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
35
36
37
38
39
40
41
42
43

use std::sync::Arc;

use async_trait::async_trait;
use rustic_ai::{
    Model, ModelError, ModelMessage, ModelRequestParameters, ModelResponse, RunContext, RunUsage,
};

struct DummyModel;

#[async_trait]
impl Model for DummyModel {
    fn name(&self) -> &str {
        "dummy"
    }

    async fn request(
        &self,
        _messages: &[ModelMessage],
        _settings: Option<&rustic_ai::ModelSettings>,
        _params: &ModelRequestParameters,
    ) -> Result<ModelResponse, ModelError> {
        Err(ModelError::Unsupported("dummy".to_string()))
    }
}

#[test]
fn run_context_for_tool_call_sets_fields() {
    let ctx = RunContext {
        run_id: "run".to_string(),
        deps: Arc::new(()),
        model: Arc::new(DummyModel),
        usage: RunUsage::default(),
        prompt: None,
        messages: Arc::new(Vec::new()),
        tool_call_id: None,
        tool_name: None,
    };

    let child = ctx.for_tool_call("call-1".to_string(), "tool".to_string());
    assert_eq!(child.run_id, "run");
    assert_eq!(child.tool_call_id.as_deref(), Some("call-1"));
    assert_eq!(child.tool_name.as_deref(), Some("tool"));
}