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    /// Full slash-separated lineage of the agent this continuation
50    /// belongs to. See per-upstream struct docs for the semantic.
51    /// Empty string if the continuation was minted before this field
52    /// existed (pre-field tokens deserialize the default).
53    pub fn agent_instance_hierarchy(&self) -> &str {
54        match self {
55            Self::Openrouter(c) => c.agent_instance_hierarchy.as_str(),
56            Self::ClaudeAgentSdk(c) => c.agent_instance_hierarchy.as_str(),
57            Self::CodexSdk(c) => c.agent_instance_hierarchy.as_str(),
58            Self::Mock(c) => c.agent_instance_hierarchy.as_str(),
59        }
60    }
61
62    /// Stamps `agent_instance_hierarchy` on the outgoing continuation.
63    /// The api server calls this just before returning the wire
64    /// continuation so the next round inherits the same lineage.
65    pub fn set_agent_instance_hierarchy(&mut self, id: String) {
66        match self {
67            Self::Openrouter(c) => c.agent_instance_hierarchy = id,
68            Self::ClaudeAgentSdk(c) => c.agent_instance_hierarchy = id,
69            Self::CodexSdk(c) => c.agent_instance_hierarchy = id,
70            Self::Mock(c) => c.agent_instance_hierarchy = id,
71        }
72    }
73
74    /// Returns the upstream type for this continuation.
75    pub fn upstream(&self) -> super::Upstream {
76        match self {
77            Self::Openrouter(_) => super::Upstream::Openrouter,
78            Self::ClaudeAgentSdk(_) => super::Upstream::ClaudeAgentSdk,
79            Self::CodexSdk(_) => super::Upstream::CodexSdk,
80            Self::Mock(_) => super::Upstream::Mock,
81        }
82    }
83
84    /// Serializes the continuation to a base64-encoded string.
85    pub fn to_string(&self) -> String {
86        let json = serde_json::to_string(self).unwrap();
87        base64::engine::general_purpose::STANDARD.encode(json)
88    }
89
90    /// Attempts to deserialize a continuation from a base64-encoded string.
91    pub fn try_from_string(s: &str) -> Option<Self> {
92        let json = base64::engine::general_purpose::STANDARD.decode(s).ok()?;
93        let continuation = serde_json::from_slice(&json).ok()?;
94        Some(continuation)
95    }
96}
objectiveai_sdk/agent/continuation.rs

objectiveai_sdk/agent/
continuation.rs
