mbxai 0.3.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

mbxai/__init__.py

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

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)
 
@@ -101,16 +103,32 @@ 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)
+                
+                # Register the tool with ToolClient
+                self._tools[tool.name] = tool
+                tool.function = tool_function
+                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.3.0",
+            version="0.5.0",
         )
         
         # Initialize MCP server
@@ -80,14 +80,7 @@ class MCPServer:
             if "$ref" in input_schema:
                 ref = input_schema["$ref"].split("/")[-1]
                 input_schema = tool_metadata.inputSchema.get("$defs", {}).get(ref, {})
-            
-            # Handle Pydantic model input
-            if "properties" in input_schema and "input" in input_schema["properties"]:
-                input_schema = input_schema["properties"]["input"]
-                if "$ref" in input_schema:
-                    ref = input_schema["$ref"].split("/")[-1]
-                    input_schema = tool_metadata.inputSchema.get("$defs", {}).get(ref, {})
-        
+                    
         # Create and store Tool instance
         self._tools[tool_metadata.name] = Tool(
             name=tool_metadata.name,

{mbxai-0.3.0.dist-info → mbxai-0.5.0.dist-info}/METADATA

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

{mbxai-0.3.0.dist-info → mbxai-0.5.0.dist-info}/RECORD

@@ -1,9 +1,9 @@
-mbxai/__init__.py,sha256=K4smKVZXrhcJYppSx5ZVow5BySoCQFVW6fgtPb9MUec,47
+mbxai/__init__.py,sha256=wVJ8x7MtN4VeN_VyyppKqGepPR_Ko096DeXHM7znB7E,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=N5w_cTEF9vMwnBoGvvenYSX8YswAogJBEd312sHyrPM,5324
 mbxai/mcp/example.py,sha256=oaol7AvvZnX86JWNz64KvPjab5gg1VjVN3G8eFSzuaE,2350
-mbxai/mcp/server.py,sha256=UZaZAnwLNgUWJPUfwz0o2OOZOtW8MiGjH8PSKrHimmU,3855
+mbxai/mcp/server.py,sha256=QCRIoHw0eJjrYFpnD1dFngjKHcb3v6_qWAUT2sF8-dk,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.3.0.dist-info/METADATA,sha256=iIE7s6BDWC1hza4F1O2LGekdpAj8c3wvHRHVlr9aP1I,4107
-mbxai-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mbxai-0.3.0.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
-mbxai-0.3.0.dist-info/RECORD,,
+mbxai-0.5.0.dist-info/METADATA,sha256=TtjY0FTk9MG12PhtlNggSDNa5VIvp1nSOGa9oLf6pKc,4107
+mbxai-0.5.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mbxai-0.5.0.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
+mbxai-0.5.0.dist-info/RECORD,,