Function collect_stream

crates/opencode-session/src/llm.rs:1432–1519

(
    mut output: StreamOutput,
    abort: tokio_util::sync::CancellationToken,
)

}

pub async fn collect_stream(
    mut output: StreamOutput,
    abort: tokio_util::sync::CancellationToken,
) -> anyhow::Result<StreamResult> {
    let mut text = String::new();
    let mut tool_calls = Vec::new();
    let mut current_tool: Option<ToolCallResult> = None;
    let mut finish_reason = None;

    while let Some(event) = output.events.recv().await {
        if abort.is_cancelled() {
            break;
        }

        match event {
            StreamEvent::Start => {}
            StreamEvent::TextStart => {}
            StreamEvent::TextDelta(delta) => {
                text.push_str(&delta);
            }
            StreamEvent::TextEnd => {}
            StreamEvent::ReasoningStart { .. } => {}
            StreamEvent::ReasoningDelta { .. } => {}
            StreamEvent::ReasoningEnd { .. } => {}
            StreamEvent::ToolInputStart { id, tool_name } => {
                current_tool = Some(ToolCallResult {
                    id,
                    name: tool_name,
                    input: serde_json::Value::Null,
                });
            }
            StreamEvent::ToolInputDelta { delta, .. } => {
                if let Some(ref mut tool) = current_tool {
                    if tool.input.is_null() {
                        tool.input = serde_json::Value::String(delta);
                    } else if let Some(s) = tool.input.as_str() {
                        let mut existing = s.to_string();
                        existing.push_str(&delta);
                        tool.input = serde_json::Value::String(existing);
                    }
                }
            }
            StreamEvent::ToolInputEnd { .. } => {}
            StreamEvent::ToolCallStart { id, name } => {
                current_tool = Some(ToolCallResult {
                    id,
                    name,
                    input: serde_json::Value::Null,
                });
            }
            StreamEvent::ToolCallDelta { input, .. } => {
                if let Some(ref mut tool) = current_tool {
                    if tool.input.is_null() {
                        tool.input = serde_json::Value::String(input);
                    } else if let Some(s) = tool.input.as_str() {
                        let mut existing = s.to_string();
                        existing.push_str(&input);
                        tool.input = serde_json::Value::String(existing);