fix tests

scald · Nov 25, 2024 · e7ae05d · e7ae05d
1 parent 5532b50
commit e7ae05d
Show file tree

Hide file tree

Showing 3 changed files with 54 additions and 433 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -21,7 +21,7 @@
 //!     
 //!     // Process an image
 //!     let image_data = std::fs::read("image.jpg").unwrap();
-//!     let analysis = processor.process(&image_data).await.unwrap();
+//!     let (analysis, token_usage) = processor.process(&image_data).await.unwrap();
 //!     println!("Analysis: {}", analysis);
 //! }
 //! ```
@@ -36,4 +36,4 @@ pub mod utils;
 pub use errors::ProcessorError;
 pub use processor::ImageProcessor;
 pub use prompts::{ImagePrompt, PromptFormat};
-pub use providers::AIProvider;
+pub use providers::{AIProvider, TokenUsage};
diff --git a/src/processor.rs b/src/processor.rs
@@ -34,6 +34,12 @@ impl ImageProcessor {
     ///     Some("moondream".to_string()),
     ///     None
     /// );
+    /// 
+    /// # async fn example() {
+    /// let image_data = std::fs::read("image.jpg").unwrap();
+    /// let (analysis, token_usage) = processor.process(&image_data).await.unwrap();
+    /// println!("Analysis: {}", analysis);
+    /// # }
     /// ```
     pub fn new(provider: AIProvider, model: Option<String>, format: Option<PromptFormat>) -> Self {
         let provider: Box<dyn Provider> = match provider {