clau-core 0.1.1

Core types and traits for Claude AI interactions
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

use serde::{Deserialize, Serialize};
use std::time::SystemTime;

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "snake_case")]
pub enum MessageType {
    Init,
    User,
    Assistant,
    Result,
    System,
    Tool,
    ToolResult,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct MessageMeta {
    pub session_id: String,
    pub timestamp: Option<SystemTime>,
    pub cost_usd: Option<f64>,
    pub duration_ms: Option<u64>,
    pub tokens_used: Option<TokenUsage>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct TokenUsage {
    pub input: u64,
    pub output: u64,
    pub total: u64,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(tag = "type", rename_all = "snake_case")]
pub enum Message {
    Init {
        #[serde(flatten)]
        meta: MessageMeta,
    },
    User {
        content: String,
        #[serde(flatten)]
        meta: MessageMeta,
    },
    Assistant {
        content: String,
        #[serde(flatten)]
        meta: MessageMeta,
    },
    Result {
        #[serde(flatten)]
        meta: MessageMeta,
        stats: ConversationStats,
    },
    System {
        content: String,
        #[serde(flatten)]
        meta: MessageMeta,
    },
    Tool {
        name: String,
        parameters: serde_json::Value,
        #[serde(flatten)]
        meta: MessageMeta,
    },
    ToolResult {
        tool_name: String,
        result: serde_json::Value,
        #[serde(flatten)]
        meta: MessageMeta,
    },
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ConversationStats {
    pub total_messages: u64,
    pub total_cost_usd: f64,
    pub total_duration_ms: u64,
    pub total_tokens: TokenUsage,
}

impl Message {
    pub fn message_type(&self) -> MessageType {
        match self {
            Message::Init { .. } => MessageType::Init,
            Message::User { .. } => MessageType::User,
            Message::Assistant { .. } => MessageType::Assistant,
            Message::Result { .. } => MessageType::Result,
            Message::System { .. } => MessageType::System,
            Message::Tool { .. } => MessageType::Tool,
            Message::ToolResult { .. } => MessageType::ToolResult,
        }
    }
    
    pub fn meta(&self) -> &MessageMeta {
        match self {
            Message::Init { meta, .. } |
            Message::User { meta, .. } |
            Message::Assistant { meta, .. } |
            Message::Result { meta, .. } |
            Message::System { meta, .. } |
            Message::Tool { meta, .. } |
            Message::ToolResult { meta, .. } => meta,
        }
    }
}