llm_link/llm/types.rs
1use serde_json;
2
3/// Token usage information
4#[derive(Debug, Clone)]
5#[allow(dead_code)]
6pub struct Usage {
7 pub prompt_tokens: u32,
8 pub completion_tokens: u32,
9 pub total_tokens: u32,
10}
11
12/// LLM response
13#[derive(Debug, Clone)]
14#[allow(dead_code)]
15pub struct Response {
16 pub content: String,
17 pub model: String,
18 pub usage: Usage,
19 pub tool_calls: Option<serde_json::Value>, // Store tool_calls from LLM response
20}
21
22/// Model information
23#[derive(Debug, Clone)]
24pub struct Model {
25 pub id: String,
26}
27