harn-serve 0.10.124

Shared outbound workflow server core for Harn adapters
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

use super::*;

#[test]
fn noncompleted_tool_calls_do_not_emit_artifact_updates() {
    let task_id = "task-tool-pending".to_string();
    let task = TaskState {
        id: task_id.clone(),
        context_id: None,
        status: TaskStatus::Working,
        history: Vec::new(),
        artifacts: Vec::new(),
        metadata: BTreeMap::new(),
        events: Vec::new(),
        subscribers: Vec::new(),
        cancel_token: None,
    };
    let tasks: TaskStore = Arc::new(Mutex::new(HashMap::from([(task_id.clone(), task)])));
    let sink = super::super::A2aWorkerSink {
        task_id: task_id.clone(),
        tasks: tasks.clone(),
    };

    for event in [
        harn_vm::agent_events::AgentEvent::ToolCallUpdate {
            session_id: super::super::a2a_worker_session_id(&task_id),
            tool_call_id: "tc-99".into(),
            tool_name: "search_files".into(),
            status: harn_vm::agent_events::ToolCallStatus::InProgress,
            raw_output: None,
            error: None,
            duration_ms: None,
            execution_duration_ms: None,
            error_category: None,
            mutation_status: harn_vm::agent_events::ToolMutationStatus::Unknown,
            changed_paths: None,
            data: None,
            executor: None,
            parsing: None,
            raw_input: None,
            raw_input_partial: None,
            audit: None,
        },
        harn_vm::agent_events::AgentEvent::ToolCallUpdate {
            session_id: super::super::a2a_worker_session_id(&task_id),
            tool_call_id: "tc-100".into(),
            tool_name: "search_files".into(),
            status: harn_vm::agent_events::ToolCallStatus::Failed,
            raw_output: None,
            error: Some("boom".into()),
            duration_ms: Some(1),
            execution_duration_ms: Some(1),
            error_category: None,
            mutation_status: harn_vm::agent_events::ToolMutationStatus::Unknown,
            changed_paths: None,
            data: None,
            executor: None,
            parsing: None,
            raw_input: None,
            raw_input_partial: None,
            audit: None,
        },
    ] {
        sink.handle_event(&event);
    }

    let tasks = tasks.lock().expect("tasks");
    let task = tasks.get(&task_id).expect("task");
    assert!(task.artifacts.is_empty());
    assert!(!task
        .events
        .iter()
        .any(|event| event.get("kind").and_then(JsonValue::as_str) == Some("artifact-update")));
}