abu_base/common.rs
1use serde::{Deserialize, Serialize};
2
3#[derive(Debug, Clone, Deserialize, Serialize)]
4pub struct Usage {
5 /// Number of tokens in the generated completion.
6 pub completion_tokens: usize,
7 /// Number of tokens in the prompt.
8 pub prompt_tokens: usize,
9 /// Total number of tokens used in the request (prompt + completion).
10 pub total_tokens: usize,
11}
12
13#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
14#[serde(rename_all = "lowercase")]
15pub enum Role {
16 System,
17 User,
18 Assistant,
19 Tool,
20}