rust-agent 0.0.5

Next Generation AI Agent Framework for Web3.
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

// Message type definitions
use serde::Deserialize;
use std::collections::HashMap;
use serde_json::Value;

// Message content structure
#[derive(Clone, Debug)]
pub struct ChatMessageContent {
    pub content: String,
    pub name: Option<String>,
    // OpenAI API tool_call_id parameter
    pub additional_kwargs: HashMap<String, Value>,
}

// Simplified message type system (aligned with langchain-core)
#[derive(Clone, Debug)]
pub enum ChatMessage {
    System(ChatMessageContent),
    Human(ChatMessageContent),
    AIMessage(ChatMessageContent),
    ToolMessage(ChatMessageContent),
}

// Token usage statistics
#[derive(Deserialize)]
pub struct TokenUsage {
    pub prompt_tokens: usize,
    pub completion_tokens: usize,
    pub total_tokens: usize,
}