Merge pull request #46 from SecretiveShell/fix-tool-api

do not force str on tool endpoints

Author: SecretiveShell

mcp_bridge/mcpManagement/prompts.py

@@ -19,7 +19,7 @@ async def get_prompts() -> dict[str, ListPromptsResult]:
 
 
 @router.post("/{prompt_name}")
-async def get_prompt(prompt_name: str, args: dict[str, str] = {}) -> GetPromptResult:
+async def get_prompt(prompt_name: str, args: dict[str, Any] = {}) -> GetPromptResult:
     """Evaluate a prompt"""
 
     client = await ClientManager.get_client_from_prompt(prompt_name)

mcp_bridge/mcpManagement/tools.py

@@ -1,3 +1,4 @@
+from typing import Any
 from fastapi import APIRouter, HTTPException
 from mcp_bridge.mcp_clients.McpClientManager import ClientManager
 from mcp.types import ListToolsResult, CallToolResult
@@ -18,7 +19,7 @@ async def get_tools() -> dict[str, ListToolsResult]:
 
 
 @router.post("/{tool_name}/call")
-async def call_tool(tool_name: str, arguments: dict[str, str] = {}) -> CallToolResult:
+async def call_tool(tool_name: str, arguments: dict[str, Any] = {}) -> CallToolResult:
     """Call a tool"""
 
     client = await ClientManager.get_client_from_tool(tool_name)