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

mbxai/__init__.py

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

mbxai/mcp/client.py

@@ -19,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."""
@@ -125,9 +126,11 @@ class MCPClient(ToolClient):
                 # 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
-                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)}")

mbxai/mcp/server.py

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

mbxai/openrouter/client.py

@@ -7,6 +7,9 @@ from openai import OpenAI, OpenAIError
 from pydantic import BaseModel, TypeAdapter
 from .models import OpenRouterModel, OpenRouterModelRegistry
 from .config import OpenRouterConfig
+import logging
+
+logger = logging.getLogger(__name__)
 
 
 class OpenRouterError(Exception):
@@ -133,13 +136,32 @@ class OpenRouterClient:
             # Remove any incompatible parameters
             kwargs.pop("parse", None)  # Remove parse parameter if present
             
-            return self._client.chat.completions.create(
+            logger.debug(f"Making chat completion request with model {model or self.model}")
+            response = self._client.chat.completions.create(
                 model=str(model or self.model),
                 messages=messages,
                 stream=stream,
                 **kwargs,
             )
+            
+            # Log response structure for debugging
+            logger.debug(f"Response type: {type(response)}")
+            logger.debug(f"Response attributes: {dir(response)}")
+            
+            # Validate response structure
+            if not hasattr(response, 'choices') or not response.choices:
+                logger.error("Invalid response: missing choices")
+                raise OpenRouterError("Invalid response: missing choices")
+            
+            if not hasattr(response.choices[0], 'message'):
+                logger.error("Invalid response: missing message in first choice")
+                raise OpenRouterError("Invalid response: missing message in first choice")
+            
+            logger.debug(f"Response message: {response.choices[0].message}")
+            return response
+            
         except Exception as e:
+            logger.error(f"Error in chat completion: {str(e)}")
             self._handle_api_error("chat completion", e)
 
     def chat_completion_parse(

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

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

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

@@ -1,18 +1,18 @@
-mbxai/__init__.py,sha256=wVJ8x7MtN4VeN_VyyppKqGepPR_Ko096DeXHM7znB7E,47
+mbxai/__init__.py,sha256=fQrXtVudyVaDP1s-lHxflDN7wNf2J4SrZHqDeDyOthk,47
 mbxai/core.py,sha256=WMvmU9TTa7M_m-qWsUew4xH8Ul6xseCZ2iBCXJTW-Bs,196
 mbxai/mcp/__init__.py,sha256=_ek9iYdYqW5saKetj4qDci11jxesQDiHPJRpHMKkxgU,175
-mbxai/mcp/client.py,sha256=N5w_cTEF9vMwnBoGvvenYSX8YswAogJBEd312sHyrPM,5324
+mbxai/mcp/client.py,sha256=CYm_5HiB3hZUYfahzM_senilPgH2oZwSevf2up1JTF0,5506
 mbxai/mcp/example.py,sha256=oaol7AvvZnX86JWNz64KvPjab5gg1VjVN3G8eFSzuaE,2350
-mbxai/mcp/server.py,sha256=QCRIoHw0eJjrYFpnD1dFngjKHcb3v6_qWAUT2sF8-dk,3462
+mbxai/mcp/server.py,sha256=rEqGWMhaWDxutwC8Eq7djq4EFH4MH8RMdZIEAYIwquc,3462
 mbxai/openrouter/__init__.py,sha256=Ito9Qp_B6q-RLGAQcYyTJVWwR2YAZvNqE-HIYXxhtD8,298
-mbxai/openrouter/client.py,sha256=NPccB7Slou5tSc8DvXvItLdoDMgU0pcxW1lKcRKQXp0,8865
+mbxai/openrouter/client.py,sha256=05_brXaQwrx_kBJPRwh5uiCHnWbgixI3nQWmx9xare0,9916
 mbxai/openrouter/config.py,sha256=MTX_YHsFrM7JYqovJSkEF6JzVyIdajeI5Dja2CALH58,2874
 mbxai/openrouter/models.py,sha256=b3IjjtZAjeGOf2rLsdnCD1HacjTnS8jmv_ZXorc-KJQ,2604
 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.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,,
+mbxai-0.5.2.dist-info/METADATA,sha256=jB1PQS8nFzupdjxuatX7Iun3JpSubaK5LZ785zbl_iE,4107
+mbxai-0.5.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mbxai-0.5.2.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
+mbxai-0.5.2.dist-info/RECORD,,