alexhancock/mcp-crate-cleanup (#4885)

This commit is contained in:
Alex Hancock
2025-09-29 17:00:17 -04:00
committed by GitHub
parent 2bd18a757b
commit cb0eca9966
78 changed files with 1039 additions and 1090 deletions
+14 -8
View File
@@ -554,6 +554,8 @@ mod final_output_tool_tests {
use goose::conversation::Conversation;
use goose::providers::base::MessageStream;
use goose::recipe::Response;
use rmcp::model::CallToolRequestParam;
use rmcp::object;
#[tokio::test]
async fn test_final_output_assistant_message_in_reply() -> Result<()> {
@@ -620,12 +622,13 @@ mod final_output_tool_tests {
agent.add_final_output_tool(response).await;
// Simulate a final output tool call occurring.
let tool_call = mcp_core::tool::ToolCall::new(
FINAL_OUTPUT_TOOL_NAME,
serde_json::json!({
let tool_call = CallToolRequestParam {
name: FINAL_OUTPUT_TOOL_NAME.into(),
arguments: Some(object!({
"result": "Test output"
}),
);
})),
};
let (_, result) = agent
.dispatch_tool_call(tool_call, "request_id".to_string(), None, &None)
.await;
@@ -1039,8 +1042,8 @@ mod max_turns_tests {
use goose::model::ModelConfig;
use goose::providers::base::{Provider, ProviderMetadata, ProviderUsage, Usage};
use goose::providers::errors::ProviderError;
use mcp_core::tool::ToolCall;
use rmcp::model::Tool;
use rmcp::model::{CallToolRequestParam, Tool};
use rmcp::object;
struct MockToolProvider {}
@@ -1058,7 +1061,10 @@ mod max_turns_tests {
_messages: &[Message],
_tools: &[Tool],
) -> Result<(Message, ProviderUsage), ProviderError> {
let tool_call = ToolCall::new("test_tool", serde_json::json!({"param": "value"}));
let tool_call = CallToolRequestParam {
name: "test_tool".into(),
arguments: Some(object!({"param": "value"})),
};
let message = Message::assistant().with_tool_request("call_123", Ok(tool_call));
let usage = ProviderUsage::new(