Merge pull request #21 from erhant/main

fix: handle error instead of `expect`

src/api_interface/openai_api.rs

@@ -42,21 +42,20 @@ impl OpenAIExecutor {
         let messages = vec![ChatMessageBuilder::default()
             .content(ChatMessageContent::Text(prompt.to_string()))
             .build()
-            .expect("OpenAI function call message build error")];
+            .map_err(|e| OllamaError::from(format!("Could not build chat message: {:?}", e)))?];
 
         let parameters = ChatCompletionParametersBuilder::default()
             .model(self.model.clone())
             .messages(messages)
             .tools(openai_tools)
             .build()
-            .expect("Error while building tools.");
+            .map_err(|e| {
+                OllamaError::from(format!("Could not build message parameters: {:?}", e))
+            })?;
 
-        let result = self
-            .client
-            .chat()
-            .create(parameters)
-            .await
-            .expect("OpenAI Function call failed");
+        let result = self.client.chat().create(parameters).await.map_err(|e| {
+            OllamaError::from(format!("Failed to parse OpenAI API response: {:?}", e))
+        })?;
         let message = result.choices[0].message.clone();
 
         if raw_mode {

src/memory/files.rs

@@ -105,7 +105,7 @@ impl Embedder for OpenAIEmbedder {
             .input(EmbeddingInput::String(_prompt.to_string()))
             .encoding_format(EmbeddingEncodingFormat::Float)
             .build()
-            .expect("Error building OpenAI embedder");
+            .map_err(EmbeddingError::BuilderError)?;
 
         let result = client.embeddings().create(parameters).await;
 

src/program/errors.rs

@@ -1,6 +1,8 @@
 use std::error::Error;
 use std::fmt;
 
+use openai_dive::v1::resources::embedding::EmbeddingParametersBuilderError;
+
 #[allow(clippy::enum_variant_names)]
 #[derive(Debug)]
 pub enum CustomError {
@@ -25,6 +27,7 @@ pub enum EmbeddingError {
     DocumentEmbedding(String),
     QueryEmbedding(String),
     ModelDoesNotExist,
+    BuilderError(EmbeddingParametersBuilderError),
 }
 
 #[derive(Debug)]
@@ -78,6 +81,7 @@ impl fmt::Display for EmbeddingError {
             EmbeddingError::DocumentEmbedding(doc) => write!(f, "Error while generating embeddings for doc: {}", doc),
             EmbeddingError::QueryEmbedding(query) => write!(f, "Error while generating embeddings for query: {}", query),
             EmbeddingError::ModelDoesNotExist => write!(f, "Model does not exist. run ollama run hellord/mxbai-embed-large-v1:f16 to create it."),
+            EmbeddingError::BuilderError(err) => write!(f, "Embedding builder error: {}", err),
         }
     }
 }

src/tools/browserless.rs

@@ -35,8 +35,7 @@ impl Tool for Browserless {
 
     async fn run(&self, input: Value) -> Result<String, Box<dyn Error>> {
         let website = input["website"].as_str().ok_or("Website URL is required")?;
-        let browserless_token =
-            env::var("BROWSERLESS_TOKEN").expect("BROWSERLESS_TOKEN must be set");
+        let browserless_token = env::var("BROWSERLESS_TOKEN")?;
         let url = format!("http://0.0.0.0:3000/content?token={}", browserless_token);
         let payload = json!({
             "url": website

src/tools/serper.rs

@@ -254,7 +254,7 @@ impl Tool for SearchTool {
         }
 
         let client = Client::new();
-        let api_key = env::var("SERPER_API_KEY").expect("SERPER_API_KEY must be set");
+        let api_key = env::var("SERPER_API_KEY")?;
         let response = client
             .post(&url)
             .header("X-API-KEY", api_key)