claude_rust_types/domain/
message.rs

1use serde::{Deserialize, Serialize};
2use serde_json::Value;
3
4#[derive(Debug, Clone, Serialize, Deserialize)]
5#[serde(rename_all = "snake_case")]
6pub enum Role {
7    User,
8    Assistant,
9}
10
11#[derive(Debug, Clone, Serialize, Deserialize)]
12#[serde(tag = "type", rename_all = "snake_case")]
13pub enum ContentBlock {
14    Text {
15        text: String,
16    },
17    ToolUse {
18        id: String,
19        name: String,
20        input: Value,
21    },
22    ToolResult {
23        tool_use_id: String,
24        content: String,
25        #[serde(skip_serializing_if = "Option::is_none")]
26        is_error: Option<bool>,
27    },
28}
29
30#[derive(Debug, Clone, Serialize, Deserialize)]
31pub struct Message {
32    pub role: Role,
33    pub content: Vec<ContentBlock>,
34}
35
36impl Message {
37    pub fn user(text: impl Into<String>) -> Self {
38        Self {
39            role: Role::User,
40            content: vec![ContentBlock::Text { text: text.into() }],
41        }
42    }
43
44    pub fn assistant(blocks: Vec<ContentBlock>) -> Self {
45        Self {
46            role: Role::Assistant,
47            content: blocks,
48        }
49    }
50}
51
52#[derive(Debug, Clone, Default, Serialize, Deserialize)]
53pub struct Conversation {
54    pub system: Option<String>,
55    pub messages: Vec<Message>,
56}
57
58impl Conversation {
59    pub fn push(&mut self, msg: Message) {
60        self.messages.push(msg);
61    }
62}
claude_rust_types/domain/message.rs

claude_rust_types/domain/
message.rs
