Add Default trait to TokenUsage struct

scald · Nov 25, 2024 · 5532b50 · 5532b50
1 parent 0f7e89b
commit 5532b50
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/src/providers/mod.rs b/src/providers/mod.rs
@@ -8,21 +8,13 @@ pub use openai::OpenAIProvider;
 use serde::Serialize;
 
 #[derive(Debug, Clone, Serialize)]
+#[derive(Default)]
 pub struct TokenUsage {
     pub prompt_tokens: usize,
     pub completion_tokens: usize,
     pub total_tokens: usize,
 }
 
-impl Default for TokenUsage {
-    fn default() -> Self {
-        Self {
-            prompt_tokens: 0,
-            completion_tokens: 0,
-            total_tokens: 0,
-        }
-    }
-}
 
 #[derive(Debug, Clone, Copy)]
 pub enum AIProvider {