rustic-ai 0.2.0

A Rust-native agent framework with tool calling, streaming, and multi-provider support for OpenAI, Anthropic, Gemini, and Grok
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

#[path = "support/scripted_model.rs"]
mod scripted_support;

use std::sync::Arc;

use rustic_ai::{
    Agent, ModelResponse, ModelResponsePart, RunInput, TextPart, UsageLimits, UserContent,
};
use scripted_support::ScriptedModel;

#[tokio::test]
async fn scripted_model_returns_queued_response() {
    let response = ModelResponse {
        parts: vec![ModelResponsePart::Text(TextPart {
            content: "ok".to_string(),
        })],
        usage: None,
        model_name: Some("scripted".to_string()),
        finish_reason: Some("stop".to_string()),
    };

    let model = Arc::new(ScriptedModel::new("scripted", Vec::new()));
    model.push_response(response).await;
    let agent = Agent::new(model);
    let input = RunInput::new(
        vec![UserContent::Text("hello".to_string())],
        vec![],
        (),
        UsageLimits::default(),
    );

    let result = agent.run(input).await.expect("run succeeds");
    assert_eq!(result.output, "ok");
}