objectiveai_sdk/agent/
continuation.rs

1use base64::Engine;
2use schemars::JsonSchema;
3use serde::{Deserialize, Serialize};
4
5/// Continuation state for multi-turn agent completions.
6///
7/// Returned in the final streaming chunk and in unary responses.
8/// Pass it back in the next request to continue the conversation.
9/// Serialized as base64-encoded JSON.
10#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
11#[serde(untagged)]
12#[schemars(rename = "agent.Continuation")]
13pub enum Continuation {
14    #[schemars(title = "Openrouter")]
15    Openrouter(super::openrouter::Continuation),
16    #[schemars(title = "ClaudeAgentSdk")]
17    ClaudeAgentSdk(super::claude_agent_sdk::Continuation),
18    #[schemars(title = "CodexSdk")]
19    CodexSdk(super::codex_sdk::Continuation),
20    #[schemars(title = "Mock")]
21    Mock(super::mock::Continuation),
22}
23
24impl From<super::openrouter::Continuation> for Continuation {
25    fn from(inner: super::openrouter::Continuation) -> Self {
26        Self::Openrouter(inner)
27    }
28}
29
30impl From<super::claude_agent_sdk::Continuation> for Continuation {
31    fn from(inner: super::claude_agent_sdk::Continuation) -> Self {
32        Self::ClaudeAgentSdk(inner)
33    }
34}
35
36impl From<super::codex_sdk::Continuation> for Continuation {
37    fn from(inner: super::codex_sdk::Continuation) -> Self {
38        Self::CodexSdk(inner)
39    }
40}
41
42impl From<super::mock::Continuation> for Continuation {
43    fn from(inner: super::mock::Continuation) -> Self {
44        Self::Mock(inner)
45    }
46}
47
48impl Continuation {
49    /// Returns the MCP sessions map for this continuation.
50    pub fn mcp_sessions(&self) -> &indexmap::IndexMap<String, String> {
51        match self {
52            Self::Openrouter(c) => &c.mcp_sessions,
53            Self::ClaudeAgentSdk(c) => &c.mcp_sessions,
54            Self::CodexSdk(c) => &c.mcp_sessions,
55            Self::Mock(c) => &c.mcp_sessions,
56        }
57    }
58
59    /// Returns the upstream type for this continuation.
60    pub fn upstream(&self) -> super::Upstream {
61        match self {
62            Self::Openrouter(_) => super::Upstream::Openrouter,
63            Self::ClaudeAgentSdk(_) => super::Upstream::ClaudeAgentSdk,
64            Self::CodexSdk(_) => super::Upstream::CodexSdk,
65            Self::Mock(_) => super::Upstream::Mock,
66        }
67    }
68
69    /// Serializes the continuation to a base64-encoded string.
70    pub fn to_string(&self) -> String {
71        let json = serde_json::to_string(self).unwrap();
72        base64::engine::general_purpose::STANDARD.encode(json)
73    }
74
75    /// Attempts to deserialize a continuation from a base64-encoded string.
76    pub fn try_from_string(s: &str) -> Option<Self> {
77        let json = base64::engine::general_purpose::STANDARD.decode(s).ok()?;
78        let continuation = serde_json::from_slice(&json).ok()?;
79        Some(continuation)
80    }
81}
objectiveai_sdk/agent/continuation.rs

objectiveai_sdk/agent/
continuation.rs
