diff --git a/crates/goose-server/src/routes/audio.rs b/crates/goose-server/src/routes/audio.rs index 8fa36a1945aa..6985d96d1ce0 100644 --- a/crates/goose-server/src/routes/audio.rs +++ b/crates/goose-server/src/routes/audio.rs @@ -391,7 +391,6 @@ pub fn routes(state: Arc) -> Router { mod tests { use super::*; use axum::{body::Body, http::Request}; - use serde_json::json; use tower::ServiceExt; #[tokio::test(flavor = "multi_thread")] @@ -424,7 +423,6 @@ mod tests { let state = AppState::new().await.unwrap(); let app = routes(state); - let large_data = "a".repeat(30 * 1024 * 1024); // 30MB let request = Request::builder() .uri("/audio/transcribe") .method("POST") diff --git a/crates/goose-server/src/routes/reply.rs b/crates/goose-server/src/routes/reply.rs index e68aef92837d..9f118a3b5add 100644 --- a/crates/goose-server/src/routes/reply.rs +++ b/crates/goose-server/src/routes/reply.rs @@ -557,58 +557,15 @@ pub fn routes(state: Arc) -> Router { #[cfg(test)] mod tests { use super::*; - use goose::conversation::message::Message; - use goose::{ - model::ModelConfig, - providers::{ - base::{Provider, ProviderUsage, Usage}, - errors::ProviderError, - }, - }; - - #[derive(Clone)] - struct MockProvider { - model_config: ModelConfig, - } - - #[async_trait::async_trait] - impl Provider for MockProvider { - fn metadata() -> goose::providers::base::ProviderMetadata { - goose::providers::base::ProviderMetadata::empty() - } - - async fn complete_with_model( - &self, - _model_config: &ModelConfig, - _system: &str, - _messages: &[Message], - _tools: &[rmcp::model::Tool], - ) -> anyhow::Result<(Message, ProviderUsage), ProviderError> { - Ok(( - Message::assistant().with_text("Mock response"), - ProviderUsage::new("mock".to_string(), Usage::default()), - )) - } - - fn get_model_config(&self) -> ModelConfig { - self.model_config.clone() - } - } mod integration_tests { use super::*; use axum::{body::Body, http::Request}; use goose::conversation::message::Message; - use serde_json::json; use tower::ServiceExt; #[tokio::test(flavor = "multi_thread")] async fn test_reply_endpoint() { - let mock_model_config = ModelConfig::new("test-model").unwrap(); - let mock_provider = MockProvider { - model_config: mock_model_config, - }; - let state = AppState::new().await.unwrap(); let app = routes(state); diff --git a/crates/goose/src/providers/factory.rs b/crates/goose/src/providers/factory.rs index 22a02ae15347..3796f88fd0a2 100644 --- a/crates/goose/src/providers/factory.rs +++ b/crates/goose/src/providers/factory.rs @@ -30,11 +30,6 @@ use crate::model::ModelConfig; use anyhow::Result; use once_cell::sync::Lazy; -#[cfg(test)] -use super::errors::ProviderError; -#[cfg(test)] -use rmcp::model::Tool; - const DEFAULT_LEAD_TURNS: usize = 3; const DEFAULT_FAILURE_THRESHOLD: usize = 2; const DEFAULT_FALLBACK_TURNS: usize = 2; @@ -174,63 +169,8 @@ fn create_worker_model_config(default_model: &ModelConfig) -> Result ProviderMetadata { - ProviderMetadata::new( - "mock_test", - "Mock Test Provider", - "A mock provider for testing", - "mock-model", - vec!["mock-model"], - "", - vec![], - ) - } - - fn get_model_config(&self) -> ModelConfig { - self.model_config.clone() - } - - async fn complete_with_model( - &self, - _model_config: &ModelConfig, - _system: &str, - _messages: &[Message], - _tools: &[Tool], - ) -> Result<(Message, ProviderUsage), ProviderError> { - Ok(( - Message::new( - Role::Assistant, - Utc::now().timestamp(), - vec![MessageContent::Text( - RawTextContent { - text: format!( - "Response from {} with model {}", - self.name, self.model_config.model_name - ), - meta: None, - } - .no_annotation(), - )], - ), - ProviderUsage::new(self.model_config.model_name.clone(), Usage::default()), - )) - } - } - struct EnvVarGuard { vars: Vec<(String, Option)>, } diff --git a/crates/goose/src/tool_inspection.rs b/crates/goose/src/tool_inspection.rs index a80c597e4c3d..b2ed584c2c15 100644 --- a/crates/goose/src/tool_inspection.rs +++ b/crates/goose/src/tool_inspection.rs @@ -274,30 +274,6 @@ mod tests { use mcp_core::ToolCall; use serde_json::json; - struct MockInspector { - name: &'static str, - results: Vec, - } - - #[async_trait] - impl ToolInspector for MockInspector { - fn name(&self) -> &'static str { - self.name - } - - fn as_any(&self) -> &dyn std::any::Any { - self - } - - async fn inspect( - &self, - _tool_requests: &[ToolRequest], - _messages: &[Message], - ) -> Result> { - Ok(self.results.clone()) - } - } - #[test] fn test_apply_inspection_results() { let tool_request = ToolRequest { diff --git a/crates/goose/tests/agent.rs b/crates/goose/tests/agent.rs index 99ae07ad44c2..a3ab902cdf7e 100644 --- a/crates/goose/tests/agent.rs +++ b/crates/goose/tests/agent.rs @@ -234,6 +234,16 @@ mod tests { .await } + #[tokio::test] + async fn test_agent_with_anthropic() -> Result<()> { + run_test_with_config(TestConfig { + provider_type: ProviderType::Anthropic, + model: "claude-sonnet-4", + context_window: 200_000, + }) + .await + } + #[tokio::test] async fn test_agent_with_azure() -> Result<()> { run_test_with_config(TestConfig { @@ -540,9 +550,7 @@ mod schedule_tool_tests { mod final_output_tool_tests { use super::*; use futures::stream; - use goose::agents::final_output_tool::{ - FINAL_OUTPUT_CONTINUATION_MESSAGE, FINAL_OUTPUT_TOOL_NAME, - }; + use goose::agents::final_output_tool::FINAL_OUTPUT_TOOL_NAME; use goose::conversation::Conversation; use goose::providers::base::MessageStream; use goose::recipe::Response;