genai_types/
messages.rs

1use crate::models::ModelInfo;
2use crate::tool_choice::ToolChoice;
3use mcp_protocol::tool::{Tool, ToolContent};
4use serde::{Deserialize, Serialize};
5
6/// Different types of content that can be in a message
7#[derive(Serialize, Deserialize, Debug, Clone)]
8#[serde(tag = "type", rename_all = "snake_case")]
9pub enum MessageContent {
10    Text {
11        text: String,
12    },
13
14    ToolUse {
15        id: String,
16        name: String,
17        input: serde_json::Value,
18    },
19
20    ToolResult {
21        tool_use_id: String,
22        content: Vec<ToolContent>,
23        is_error: Option<bool>,
24    },
25}
26
27/// A single message in a conversation with Claude
28#[derive(Serialize, Deserialize, Debug, Clone)]
29pub struct Message {
30    /// Role of the message sender (user, assistant, system)
31    pub role: Role,
32
33    /// Content of the message as vector of MessageContent objects
34    pub content: Vec<MessageContent>,
35}
36
37/// Role of the message sender
38#[derive(Serialize, Deserialize, Debug, Clone)]
39pub enum Role {
40    #[serde(rename = "user")]
41    User,
42
43    #[serde(rename = "assistant")]
44    Assistant,
45
46    #[serde(rename = "system")]
47    System,
48}
49
50impl Message {
51    /// Create a new message with structured content
52    pub fn new_structured(role: impl Into<String>, content: Vec<MessageContent>) -> Self {
53        Self {
54            role: match role.into().as_str() {
55                "user" => Role::User,
56                "assistant" => Role::Assistant,
57                "system" => Role::System,
58                _ => panic!("Invalid role"),
59            },
60            content,
61        }
62    }
63}
64
65/// Request to generate a completion from Claude
66#[derive(Serialize, Deserialize, Debug, Clone)]
67pub struct CompletionRequest {
68    /// The model to use
69    pub model: String,
70
71    /// List of messages in the conversation
72    pub messages: Vec<Message>,
73
74    /// Maximum number of tokens to generate
75    pub max_tokens: u32,
76
77    /// Temperature parameter (0.0 to 1.0)
78    pub temperature: Option<f32>,
79
80    /// System prompt to use
81    pub system: Option<String>,
82
83    /// Tools to make available to the model
84    pub tools: Option<Vec<Tool>>,
85
86    /// Tool choice configuration
87    pub tool_choice: Option<ToolChoice>,
88
89    /// Whether to disable parallel tool use
90    pub disable_parallel_tool_use: Option<bool>,
91}
92
93/// Information about token usage
94#[derive(Serialize, Deserialize, Debug, Clone)]
95pub struct Usage {
96    pub input_tokens: u32,
97
98    pub output_tokens: u32,
99}
100
101/// Response from a completion request
102#[derive(Serialize, Deserialize, Debug, Clone)]
103pub struct CompletionResponse {
104    /// Generated content blocks
105    pub content: Vec<MessageContent>,
106
107    /// ID of the message
108    pub id: String,
109
110    /// Model used for generation
111    pub model: String,
112
113    // always "assistant"
114    pub role: Role,
115
116    pub stop_reason: StopReason,
117
118    pub stop_sequence: Option<String>,
119
120    /// Message type
121    #[serde(rename = "type")]
122    pub message_type: String,
123
124    /// Token usage information
125    pub usage: Usage,
126}
127
128/// Reason why generation stopped
129#[derive(Serialize, Deserialize, Debug, Clone)]
130pub enum StopReason {
131    /// Generation stopped because the end of a turn was reached
132    #[serde(rename = "end_turn")]
133    EndTurn,
134
135    /// Generation stopped because the maximum token limit was reached
136    #[serde(rename = "max_tokens")]
137    MaxTokens,
138
139    /// Generation stopped because a stop sequence was encountered
140    #[serde(rename = "stop_sequence")]
141    StopSequence,
142
143    /// Generation stopped because a tool was used
144    #[serde(rename = "tool_use")]
145    ToolUse,
146}
147
148/// Request format for the anthropic-proxy actor
149#[derive(Serialize, Deserialize, Debug, Clone)]
150pub enum ProxyRequest {
151    ListModels,
152
153    GenerateCompletion { request: CompletionRequest },
154}
155
156/// Response format from the anthropic-proxy actor
157#[derive(Serialize, Deserialize, Debug, Clone)]
158pub enum ProxyResponse {
159    /// List of available models
160    ListModels { models: Vec<ModelInfo> },
161
162    /// Generated completion
163    Completion { completion: CompletionResponse },
164
165    /// Error response
166    Error { error: String },
167}
genai_types/messages.rs

genai_types/
messages.rs
