starweaver-runtime 0.1.1

Agent-loop graph and runtime executor primitives for Starweaver
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

//! Agent loop helper functions.

use std::collections::BTreeSet;

use starweaver_model::ModelRequestParameters;

use crate::run::AgentRunState;

pub(super) fn record_tool_control_flow(
    state: &mut AgentRunState,
    tool_return: &starweaver_model::ToolReturnPart,
) {
    let control_flow = tool_return_control_flow(tool_return);
    match control_flow {
        Some("approval_required") => state
            .pending_approval_tool_returns
            .push(tool_return.clone()),
        Some("call_deferred") => state.deferred_tool_returns.push(tool_return.clone()),
        _ => {}
    }
}

pub(super) fn tool_return_control_flow(
    tool_return: &starweaver_model::ToolReturnPart,
) -> Option<&str> {
    tool_return
        .metadata
        .get("control_flow")
        .and_then(serde_json::Value::as_str)
}

pub(super) const fn has_pending_tool_control_flow(state: &AgentRunState) -> bool {
    !state.pending_approval_tool_returns.is_empty() || !state.deferred_tool_returns.is_empty()
}

pub(super) fn is_tool_retry_return(tool_return: &starweaver_model::ToolReturnPart) -> bool {
    matches!(
        tool_return
            .metadata
            .get("error_kind")
            .and_then(serde_json::Value::as_str),
        Some("model_retry" | "invalid_arguments")
    ) || matches!(
        tool_return
            .content
            .get("kind")
            .and_then(serde_json::Value::as_str),
        Some("model_retry" | "invalid_arguments")
    )
}

pub(super) fn mark_tool_retry_return(
    tool_return: &mut starweaver_model::ToolReturnPart,
    retry: usize,
    max_retries: usize,
) {
    tool_return
        .metadata
        .insert("retry".to_string(), serde_json::json!(retry));
    tool_return
        .metadata
        .insert("max_retries".to_string(), serde_json::json!(max_retries));
}

pub(super) fn merge_request_params(
    current: &ModelRequestParameters,
    overlay: &ModelRequestParameters,
) -> ModelRequestParameters {
    let mut merged = current.clone();
    let mut names = merged
        .tools
        .iter()
        .map(|tool| tool.name.clone())
        .collect::<BTreeSet<_>>();
    for tool in &overlay.tools {
        if names.insert(tool.name.clone()) {
            merged.tools.push(tool.clone());
        }
    }
    let mut native_names = merged
        .native_tools
        .iter()
        .map(|tool| tool.tool_type.clone())
        .collect::<BTreeSet<_>>();
    for tool in &overlay.native_tools {
        if native_names.insert(tool.tool_type.clone()) {
            merged.native_tools.push(tool.clone());
        }
    }
    if overlay.output_schema.is_some() {
        merged.output_schema.clone_from(&overlay.output_schema);
    }
    if overlay.output_mode.is_some() {
        merged.output_mode = overlay.output_mode;
    }
    if !overlay.instructions.is_empty() {
        merged.instructions.extend(overlay.instructions.clone());
    }
    if overlay.thinking.is_some() {
        merged.thinking.clone_from(&overlay.thinking);
    }
    if overlay.allow_text_output.is_some() {
        merged.allow_text_output = overlay.allow_text_output;
    }
    if overlay.allow_image_output.is_some() {
        merged.allow_image_output = overlay.allow_image_output;
    }
    merged.http.headers.extend(overlay.http.headers.clone());
    merged
        .http
        .extra_body
        .extend(overlay.http.extra_body.clone());
    if overlay.http.endpoint_url.is_some() {
        merged
            .http
            .endpoint_url
            .clone_from(&overlay.http.endpoint_url);
    }
    if overlay.http.timeout_ms.is_some() {
        merged.http.timeout_ms = overlay.http.timeout_ms;
    }
    merged.http.metadata.extend(overlay.http.metadata.clone());
    merged.extra_body.extend(overlay.extra_body.clone());
    merged.metadata.extend(overlay.metadata.clone());
    merged
}