ralph-agent-loop 0.4.0

A Rust CLI for managing AI agent loops with a structured JSON task queue
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

//! Gemini-style stream event extraction.
//!
//! Responsibilities:
//! - Render `tool_use` / `tool_result` events keyed by `tool_name`.
//! - Render assistant `message` payloads that surface as plain strings.
//!
//! Does not handle:
//! - Claude `message_end` semantics.
//! - Kimi tool-call arrays.

use crate::outpututil;
use serde_json::Value as JsonValue;

use super::super::stream_tool_details::format_tool_details;
use super::common;

pub(super) fn collect_lines(json: &JsonValue, lines: &mut Vec<String>) {
    let Some(event_type) = json.get("type").and_then(|t| t.as_str()) else {
        return;
    };

    match event_type {
        "tool_use" => {
            if let Some(tool) = json.get("tool_name").and_then(|t| t.as_str()) {
                let details = json.get("parameters").and_then(format_tool_details);
                lines.push(outpututil::format_tool_call(tool, details.as_deref()));
            }
        }
        "tool_result" => {
            if let Some(tool) = json.get("tool_name").and_then(|t| t.as_str()) {
                let status = json
                    .get("status")
                    .and_then(|s| s.as_str())
                    .unwrap_or("completed");
                lines.push(outpututil::format_tool_call(
                    tool,
                    Some(&format!("({status})")),
                ));
            }
        }
        "message" => {
            if json.get("role").and_then(|r| r.as_str()) == Some("assistant") {
                common::push_text(lines, json.get("content").and_then(|c| c.as_str()));
            }
        }
        _ => {}
    }
}