Function test_generate_streaming_tool_mode

core/src/llm/structured_tests.rs:642–676

()

#[tokio::test]
async fn test_generate_streaming_tool_mode() {
    let client = MockStructuredClient::new(vec![MockStructuredClient::tool_call_response(
        "emit_result",
        serde_json::json!({"items": ["a", "b", "c"]}),
    )]);

    let req = StructuredRequest {
        prompt: "List items".to_string(),
        system: None,
        schema: serde_json::json!({
            "type": "object",
            "required": ["items"],
            "properties": {
                "items": {"type": "array", "items": {"type": "string"}}
            }
        }),
        schema_name: "result".to_string(),
        schema_description: None,
        mode: StructuredMode::Tool,
        max_repair_attempts: 0,
    };

    let partials = Arc::new(Mutex::new(Vec::new()));
    let partials_clone = Arc::clone(&partials);
    let callback: PartialObjectCallback = Box::new(move |v: &Value| {
        partials_clone.lock().unwrap().push(v.clone());
    });

    let result = generate_streaming(&client, &req, callback).await.unwrap();
    assert_eq!(result.object["items"][0], "a");
    assert_eq!(result.object["items"][2], "c");

    let p = partials.lock().unwrap();
    assert!(!p.is_empty());
}

// ========================================================================
// Edge case: extract_json_value