Add logging

Author: EricLBuehler

mistralrs-core/src/engine/search_request.rs

@@ -356,14 +356,31 @@ async fn do_custom_tool(
     }
 
     let result = if let Some(cb) = this.tool_callbacks.get(&tool_calls.function.name) {
-        cb(&tool_calls.function).unwrap_or_else(|e| format!("ERROR: {e}"))
+        tracing::info!("Called tool `{}`.", tool_calls.function.name);
+        cb(&tool_calls.function).unwrap_or_else(|e| {
+            tracing::error!(
+                "Error when calling tool `{}`: {e}",
+                tool_calls.function.name
+            );
+            format!("ERROR: {e}")
+        })
     } else if let Some(callback_with_tool) = this
         .tool_callbacks_with_tools
         .get(&tool_calls.function.name)
     {
-        (callback_with_tool.callback)(&tool_calls.function)
-            .unwrap_or_else(|e| format!("ERROR: {e}"))
+        tracing::info!("Called tool `{}`.", tool_calls.function.name);
+        (callback_with_tool.callback)(&tool_calls.function).unwrap_or_else(|e| {
+            tracing::error!(
+                "Error when calling tool `{}`: {e}",
+                tool_calls.function.name
+            );
+            format!("ERROR: {e}")
+        })
     } else {
+        tracing::error!(
+            "Attempted to call tool `{}`, but it doesn't exist.",
+            tool_calls.function.name
+        );
         format!("ERROR: no tool callback for {}", tool_calls.function.name)
     };
 

mistralrs/examples/mcp_client/main.rs

@@ -119,7 +119,7 @@ async fn main() -> Result<()> {
 
     // Build the model with MCP client configuration
     // The MCP client will automatically connect to configured servers and discover available tools
-    let model = TextModelBuilder::new("../hf_models/qwen3_4b".to_string())
+    let model = TextModelBuilder::new("Qwen/Qwen3-4B".to_string())
         .with_isq(IsqType::Q8_0) // Use 8-bit quantization for efficiency
         .with_logging()
         .with_paged_attn(|| {