agentsdk 0.9.0

A lean, OpenAI-compatible AI SDK for Rust.
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

use o3gen_openai::types;

pub use o3gen_openai::types::{
    ChatCompletionRequestMessage as Message, ChatCompletionRole as Role, ToolCall,
    ToolCallFunction as ToolFunction,
};

pub type Messages = Vec<Message>;

pub fn system(content: impl Into<String>) -> Message {
    Message::SystemMessage(types::ChatCompletionRequestSystemMessage {
        content: Some(content.into()),
        name: None,
        role: types::ChatCompletionRequestSystemMessageRole::System,
    })
}

pub fn user(content: impl Into<String>) -> Message {
    Message::UserMessage(types::ChatCompletionRequestUserMessage {
        content: Some(types::ChatCompletionRequestUserMessageContent::String(
            content.into(),
        )),
        name: None,
        role: types::ChatCompletionRequestUserMessageRole::User,
    })
}

pub fn assistant(content: impl Into<String>) -> Message {
    Message::AssistantMessage(types::ChatCompletionRequestAssistantMessage {
        content: Some(content.into()),
        name: None,
        tool_calls: None,
        role: types::ChatCompletionRequestAssistantMessageRole::Assistant,
        function_call: None,
    })
}

pub fn tool(content: impl Into<String>, tool_call_id: impl Into<String>) -> Message {
    Message::ToolMessage(types::ChatCompletionRequestToolMessage {
        content: Some(content.into()),
        tool_call_id: tool_call_id.into(),
        role: types::ChatCompletionRequestToolMessageRole::Tool,
    })
}

pub fn assistant_tool_call(
    name: impl Into<String>,
    call_id: impl Into<String>,
    arguments: &serde_json::Value,
) -> Message {
    Message::AssistantMessage(types::ChatCompletionRequestAssistantMessage {
        content: None,
        name: None,
        tool_calls: Some(vec![ToolCall {
            id: call_id.into(),
            r#type: types::ToolCallType::Function,
            function: ToolFunction {
                name: name.into(),
                arguments: arguments.to_string(),
            },
        }]),
        role: types::ChatCompletionRequestAssistantMessageRole::Assistant,
        function_call: None,
    })
}