systemprompt-agent 0.15.0

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

use systemprompt_identifiers::MessageId;
use systemprompt_models::{AgUiEventBuilder, AgUiMessageRole};

use crate::services::a2a_server::streaming::webhook_client::WebhookContext;

pub(in crate::services::a2a_server::streaming) struct TextStreamState {
    message_started: bool,
    webhook_context: Option<WebhookContext>,
}

impl TextStreamState {
    pub(in crate::services::a2a_server::streaming) const fn new() -> Self {
        Self {
            message_started: false,
            webhook_context: None,
        }
    }

    pub(in crate::services::a2a_server::streaming) fn with_webhook_context(
        mut self,
        context: WebhookContext,
    ) -> Self {
        self.webhook_context = Some(context);
        self
    }

    pub(in crate::services::a2a_server::streaming) async fn handle_text(
        &mut self,
        text: String,
        message_id: &MessageId,
    ) {
        let Some(ref webhook_context) = self.webhook_context else {
            return;
        };

        let message_id_str = message_id.as_str();

        if !self.message_started {
            let start_event =
                AgUiEventBuilder::text_message_start(message_id_str, AgUiMessageRole::Assistant);
            if let Err(e) = webhook_context.broadcast_agui(start_event).await {
                tracing::error!(error = %e, "Failed to broadcast TEXT_MESSAGE_START");
            }
            self.message_started = true;
        }

        let content_event = AgUiEventBuilder::text_message_content(message_id_str, &text);
        if let Err(e) = webhook_context.broadcast_agui(content_event).await {
            tracing::error!(error = %e, "Failed to broadcast TEXT_MESSAGE_CONTENT");
        }
    }

    pub(in crate::services::a2a_server::streaming) async fn finalize(
        &self,
        message_id: &MessageId,
    ) {
        if self.message_started {
            if let Some(ref webhook_context) = self.webhook_context {
                let end_event = AgUiEventBuilder::text_message_end(message_id.as_str());
                if let Err(e) = webhook_context.broadcast_agui(end_event).await {
                    tracing::error!(error = %e, "Failed to broadcast TEXT_MESSAGE_END");
                }
            }
        }
    }
}

impl Default for TextStreamState {
    fn default() -> Self {
        Self::new()
    }
}