Better search paths and handling of CLI providers (#5554)

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
2025-11-07 19:35:26 -08:00
parent 65b4b2bb18
commit 25dfd768e5
27 changed files with 721 additions and 538 deletions
@@ -0,0 +1,58 @@
+use crate::{conversation::message::Message, model::ModelConfig, providers::create};
+use anyhow::Result;
+use rmcp::model::ToolAnnotations;
+use rmcp::{model::Tool, object};
+
+pub async fn test_provider_configuration(
+    provider_name: &str,
+    model: &str,
+    toolshim_enabled: bool,
+    toolshim_model: Option<String>,
+) -> Result<()> {
+    let model_config = ModelConfig::new(model)?
+        .with_max_tokens(Some(50))
+        .with_toolshim(toolshim_enabled)
+        .with_toolshim_model(toolshim_model);
+
+    let provider = create(provider_name, model_config).await?;
+
+    let messages =
+        vec![Message::user().with_text("What is the weather like in San Francisco today?")];
+
+    let tools = if !toolshim_enabled {
+        vec![create_sample_weather_tool()]
+    } else {
+        vec![]
+    };
+
+    let _result = provider
+        .complete(
+            "You are an AI agent called goose. You use tools of connected extensions to solve problems.",
+            &messages,
+            &tools.into_iter().collect::<Vec<_>>()
+        )
+        .await?;
+
+    Ok(())
+}
+
+fn create_sample_weather_tool() -> Tool {
+    Tool::new(
+        "get_weather".to_string(),
+        "Get current temperature for a given location.".to_string(),
+        object!({
+            "type": "object",
+            "required": ["location"],
+            "properties": {
+                "location": {"type": "string"}
+            }
+        }),
+    )
+    .annotate(ToolAnnotations {
+        title: Some("Get weather".to_string()),
+        read_only_hint: Some(true),
+        destructive_hint: Some(false),
+        idempotent_hint: Some(false),
+        open_world_hint: Some(false),
+    })
+}