systemprompt-agent 0.2.2

Agent-to-Agent (A2A) protocol for systemprompt.io AI governance: streaming, JSON-RPC models, task lifecycle, .well-known discovery, and governed agent orchestration.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

use std::sync::Arc;

use crate::models::a2a::{A2aRequestParams, Task};
use crate::repository::task::TaskRepository;
use crate::services::a2a_server::handlers::state::AgentHandlerState;
use crate::services::a2a_server::processing::message::MessageProcessor;
use crate::services::a2a_server::processing::task_builder::build_canceled_task;
use systemprompt_models::RequestContext;

use super::validation::validate_message_context;

pub async fn handle_non_streaming_request(
    request: A2aRequestParams,
    state: &AgentHandlerState,
    context: &RequestContext,
) -> Result<Task, Box<dyn std::error::Error + Send + Sync>> {
    let config = state.config.read().await;
    let agent_name = config.name.clone();
    drop(config);

    match request {
        A2aRequestParams::SendMessage(params) => {
            tracing::info!("Handling SendMessage request");

            validate_message_context(&params.message, Some(context.user_id()), &state.db_pool)
                .await?;

            let message_processor =
                MessageProcessor::new(&state.db_pool, Arc::clone(&state.ai_service))?;

            message_processor
                .handle_message(params.message, &agent_name, context)
                .await
                .map_err(Into::into)
        },
        A2aRequestParams::SendStreamingMessage(params) => {
            tracing::info!("Handling SendStreamingMessage request (fallback to non-streaming)");

            validate_message_context(&params.message, Some(context.user_id()), &state.db_pool)
                .await?;

            let message_processor =
                MessageProcessor::new(&state.db_pool, Arc::clone(&state.ai_service))?;

            message_processor
                .handle_message(params.message, &agent_name, context)
                .await
                .map_err(Into::into)
        },
        A2aRequestParams::GetTask(params) => {
            tracing::info!(task_id = %params.id, "Handling GetTask request");

            let task_repo = TaskRepository::new(&state.db_pool)?;
            let task_id = systemprompt_identifiers::TaskId::new(&params.id);

            match task_repo.get_task(&task_id).await {
                Ok(Some(task)) => Ok(task),
                Ok(None) => Err(format!("Task not found: {}", params.id).into()),
                Err(e) => Err(format!("Failed to retrieve task: {e}").into()),
            }
        },
        A2aRequestParams::CancelTask(params) => {
            tracing::info!(task_id = %params.id, "Handling CancelTask request");

            let task_repo = TaskRepository::new(&state.db_pool)?;
            let task_id = params.id.clone();

            match task_repo.get_task(&task_id).await {
                Ok(Some(task)) => Ok(build_canceled_task(params.id, task.context_id)),
                Ok(None) => Err(format!("Task not found: {}", params.id).into()),
                Err(e) => Err(format!("Failed to look up task: {e}").into()),
            }
        },
        A2aRequestParams::SetTaskPushNotificationConfig(_)
        | A2aRequestParams::GetTaskPushNotificationConfig(_)
        | A2aRequestParams::ListTaskPushNotificationConfig(_)
        | A2aRequestParams::DeleteTaskPushNotificationConfig(_) => {
            Err("Push notification config requests should be handled before this point".into())
        },
        _ => {
            tracing::warn!(request = ?request, "Unsupported A2A request type");
            Err("Unsupported request type".into())
        },
    }
}