Skip to main content

objectiveai_sdk/functions/executions/response/streaming/
function_execution_task_chunk.rs

1use crate::agent::completions::response::streaming::AgentCompletionIds;
2use schemars::JsonSchema;
3use serde::{Deserialize, Serialize};
4
5#[derive(
6    Debug,
7    Clone,
8    PartialEq,
9    Serialize,
10    Deserialize,
11    JsonSchema,
12    arbitrary::Arbitrary,
13)]
14#[schemars(
15    rename = "functions.executions.response.streaming.FunctionExecutionTaskChunk"
16)]
17pub struct FunctionExecutionTaskChunk {
18    #[arbitrary(with = crate::arbitrary_util::arbitrary_u64)]
19    pub index: u64,
20    #[arbitrary(with = crate::arbitrary_util::arbitrary_u64)]
21    pub task_index: u64,
22    #[arbitrary(with = crate::arbitrary_util::arbitrary_vec_u64)]
23    pub task_path: Vec<u64>,
24    #[serde(skip_serializing_if = "Option::is_none")]
25    #[schemars(extend("omitempty" = true))]
26    #[arbitrary(with = crate::arbitrary_util::arbitrary_option_u64)]
27    pub swiss_pool_index: Option<u64>,
28    #[serde(skip_serializing_if = "Option::is_none")]
29    #[schemars(extend("omitempty" = true))]
30    #[arbitrary(with = crate::arbitrary_util::arbitrary_option_u64)]
31    pub swiss_round: Option<u64>,
32    #[serde(skip_serializing_if = "Option::is_none")]
33    #[schemars(extend("omitempty" = true))]
34    #[arbitrary(with = crate::arbitrary_util::arbitrary_option_u64)]
35    pub split_index: Option<u64>,
36    #[serde(flatten)]
37    pub inner: super::FunctionExecutionChunk,
38}
39
40impl AgentCompletionIds for FunctionExecutionTaskChunk {
41    fn agent_completion_ids(&self) -> impl Iterator<Item = &str> + Send {
42        self.inner.agent_completion_ids()
43    }
44}
45
46impl FunctionExecutionTaskChunk {
47    pub fn push(&mut self, other: &super::FunctionExecutionTaskChunk) {
48        self.inner.push(&other.inner);
49    }
50
51}