mbxai 0.4.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

mbxai/__init__.py

@@ -2,4 +2,4 @@
 MBX AI package.
 """
 
-__version__ = "0.1.0" 
+__version__ = "0.5.1" 

mbxai/mcp/client.py

@@ -2,11 +2,13 @@
 
 from typing import Any, TypeVar, Callable
 import httpx
+import logging
 from pydantic import BaseModel, Field
 
 from ..tools import ToolClient, Tool
 from ..openrouter import OpenRouterClient
 
+logger = logging.getLogger(__name__)
 
 T = TypeVar("T", bound=BaseModel)
 
@@ -17,6 +19,7 @@ class MCPTool(Tool):
     service: str = Field(description="The service that provides the tool")
     strict: bool = Field(default=True, description="Whether the tool response is strictly validated")
     input_schema: dict[str, Any] = Field(description="The input schema for the tool")
+    function: Callable[..., Any] | None = Field(default=None, description="The function that implements the tool")
 
     def to_openai_function(self) -> dict[str, Any]:
         """Convert the tool to an OpenAI function definition."""
@@ -101,16 +104,34 @@ class MCPClient(ToolClient):
         
         # Fetch tools from the server
         response = await self._http_client.get(f"{base_url}/tools")
-        tools_data = response.json()
+        response_data = response.json()
+        
+        # Extract tools array from response
+        tools_data = response_data.get("tools", [])
+        logger.info(f"Received {len(tools_data)} tools from server {name}")
         
         # Register each tool
-        for tool_data in tools_data:
-            # Create MCPTool instance
-            tool = MCPTool(**tool_data)
-            
-            # Create the tool function
-            tool_function = self._create_tool_function(tool)
+        for idx, tool_data in enumerate(tools_data):
+            logger.debug(f"Processing tool {idx}: {tool_data}")
             
-            # Register the tool with ToolClient
-            self._tools[tool.name] = tool
-            tool.function = tool_function 
+            # Ensure tool_data is a dictionary
+            if not isinstance(tool_data, dict):
+                logger.error(f"Invalid tool data type: {type(tool_data)}. Expected dict, got {tool_data}")
+                continue
+                
+            try:
+                # Create MCPTool instance with proper dictionary unpacking
+                tool = MCPTool(**tool_data)
+                
+                # Create the tool function
+                tool_function = self._create_tool_function(tool)
+                
+                # Set the function after creation
+                tool.function = tool_function
+                
+                # Register the tool with ToolClient
+                self._tools[tool.name] = tool
+                logger.info(f"Successfully registered tool: {tool.name}")
+            except Exception as e:
+                logger.error(f"Failed to register tool: {str(e)}")
+                logger.error(f"Tool data that caused the error: {tool_data}") 

mbxai/mcp/server.py

@@ -31,7 +31,7 @@ class MCPServer:
         self.app = FastAPI(
             title=self.name,
             description=self.description,
-            version="0.4.0",
+            version="0.5.1",
         )
         
         # Initialize MCP server

{mbxai-0.4.0.dist-info → mbxai-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mbxai
-Version: 0.4.0
+Version: 0.5.1
 Summary: MBX AI SDK
 Project-URL: Homepage, https://www.mibexx.de
 Project-URL: Documentation, https://www.mibexx.de

{mbxai-0.4.0.dist-info → mbxai-0.5.1.dist-info}/RECORD

@@ -1,9 +1,9 @@
-mbxai/__init__.py,sha256=K4smKVZXrhcJYppSx5ZVow5BySoCQFVW6fgtPb9MUec,47
+mbxai/__init__.py,sha256=kyNYdkqdxssgIHfUQJw8pEuP8HOGhC7hR1d9oaAcGco,47
 mbxai/core.py,sha256=WMvmU9TTa7M_m-qWsUew4xH8Ul6xseCZ2iBCXJTW-Bs,196
 mbxai/mcp/__init__.py,sha256=_ek9iYdYqW5saKetj4qDci11jxesQDiHPJRpHMKkxgU,175
-mbxai/mcp/client.py,sha256=rVYYsWSSUWmF9lWHFeFiPOnqD5yU-N1A9ApXekfhwVg,4411
+mbxai/mcp/client.py,sha256=CYm_5HiB3hZUYfahzM_senilPgH2oZwSevf2up1JTF0,5506
 mbxai/mcp/example.py,sha256=oaol7AvvZnX86JWNz64KvPjab5gg1VjVN3G8eFSzuaE,2350
-mbxai/mcp/server.py,sha256=d_p2oLZcj6_9MbpbV5xxVCz1IMTlXQDYF9d8Sq0M4HY,3462
+mbxai/mcp/server.py,sha256=11TgdqtWs_IhaqbQtLbeRbBpYwCYVo692JvQT0kbQJo,3462
 mbxai/openrouter/__init__.py,sha256=Ito9Qp_B6q-RLGAQcYyTJVWwR2YAZvNqE-HIYXxhtD8,298
 mbxai/openrouter/client.py,sha256=NPccB7Slou5tSc8DvXvItLdoDMgU0pcxW1lKcRKQXp0,8865
 mbxai/openrouter/config.py,sha256=MTX_YHsFrM7JYqovJSkEF6JzVyIdajeI5Dja2CALH58,2874
@@ -12,7 +12,7 @@ mbxai/tools/__init__.py,sha256=QUFaXhDm-UKcuAtT1rbKzhBkvyRBVokcQIOf9cxIuwc,160
 mbxai/tools/client.py,sha256=PDkDTvpwAs8-F8CcF6wjjq2Z-VhYpjA9RXTK64Rwa3Y,5418
 mbxai/tools/example.py,sha256=1HgKK39zzUuwFbnp3f0ThyWVfA_8P28PZcTwaUw5K78,2232
 mbxai/tools/types.py,sha256=ZHnmiDXpH6wZhiZ-Tj9PiPMJaW1aiDAq5It2gpiwNp0,831
-mbxai-0.4.0.dist-info/METADATA,sha256=bgDoeH6t7D0pJ30Bif6hwf9yRJ8j2mfOnmqtbgaIlVY,4107
-mbxai-0.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mbxai-0.4.0.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
-mbxai-0.4.0.dist-info/RECORD,,
+mbxai-0.5.1.dist-info/METADATA,sha256=g61BMob_rXCUGO5GOo6yRT-0U0g0MfwRJhWst9KJ02g,4107
+mbxai-0.5.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mbxai-0.5.1.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
+mbxai-0.5.1.dist-info/RECORD,,