mbxai 0.6.4__py3-none-any.whl → 0.6.6__py3-none-any.whl

mbxai/__init__.py

@@ -2,4 +2,4 @@
 MBX AI package.
 """
 
-__version__ = "0.6.4" 
+__version__ = "0.6.6" 

mbxai/mcp/server.py

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

mbxai/tools/client.py

@@ -44,7 +44,7 @@ class ToolClient:
         functions = [tool.to_openai_function() for tool in self._tools.values()]
         
         # Make the chat request
-        response = await self._openrouter_client.chat(
+        response = await self._openrouter_client.chat_completion(
             messages=messages,
             model=model,
             functions=functions,
@@ -89,7 +89,7 @@ class ToolClient:
             })
             
             # Get final response
-            final_response = await self._openrouter_client.chat(
+            final_response = await self._openrouter_client.chat_completion(
                 messages=messages,
                 model=model,
             )
@@ -252,7 +252,6 @@ class ToolClient:
         response_format: type[T],
         *,
         model: str | None = None,
-        stream: bool = False,
         **kwargs: Any,
     ) -> Any:
         """Chat with the model and parse the response into a Pydantic model.
@@ -261,22 +260,36 @@ class ToolClient:
             messages: The conversation messages
             response_format: The Pydantic model to parse the response into
             model: Optional model override
-            stream: Whether to stream the response
             **kwargs: Additional parameters for the chat completion
 
         Returns:
             The parsed response from the model
         """
-        # First use the chat function to handle any tool calls
+        # First use our own chat function to handle any tool calls
         response = await self.chat(
             messages=messages,
             model=model,
-            stream=stream,
             **kwargs,
         )
 
-        if stream:
-            return response
+        if not response or not response.choices:
+            raise ValueError("No response received from OpenRouter")
+            
+        choice = response.choices[0]
+        if not choice:
+            raise ValueError("Empty choice in response")
+            
+        message = choice.message
+        if not message:
+            raise ValueError("Choice missing message")
+
+        # If we still have tool calls, something went wrong
+        if message.tool_calls:
+            raise ValueError("Unexpected tool calls in final response")
+
+        # Ensure we have content to parse
+        if not message.content:
+            raise ValueError("No content in final response to parse")
 
         # Now that we have the final response, parse it into the desired format
         try:
@@ -284,8 +297,6 @@ class ToolClient:
                 messages=messages,
                 response_format=response_format,
                 model=model,
-                stream=stream,
-                tools=None,  # No tools for the final parse
                 **kwargs,
             )
             return final_response

{mbxai-0.6.4.dist-info → mbxai-0.6.6.dist-info}/METADATA

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

{mbxai-0.6.4.dist-info → mbxai-0.6.6.dist-info}/RECORD

@@ -1,18 +1,18 @@
-mbxai/__init__.py,sha256=VUrUuLr8MZYb-vl76YFDkrPwXUdeqUgt9qpYFCWD4pU,47
+mbxai/__init__.py,sha256=E7N5mx6c02zw5ZAm-EzmKVJVqQaE5_Q9uDbuHaRNipI,47
 mbxai/core.py,sha256=WMvmU9TTa7M_m-qWsUew4xH8Ul6xseCZ2iBCXJTW-Bs,196
 mbxai/mcp/__init__.py,sha256=_ek9iYdYqW5saKetj4qDci11jxesQDiHPJRpHMKkxgU,175
 mbxai/mcp/client.py,sha256=B8ZpH-uecmTCgoDw65LwwVxsFWVoX-08t5ff0hOEPXk,6011
 mbxai/mcp/example.py,sha256=oaol7AvvZnX86JWNz64KvPjab5gg1VjVN3G8eFSzuaE,2350
-mbxai/mcp/server.py,sha256=QFx8o8DqnH8MYMNu1mGoj-QWTRfwLXOQ9DaNNEZunv8,3462
+mbxai/mcp/server.py,sha256=CAO3UOAlmBX8knZCs6_Q3FiwAa-GnPXXepuXwSYN_40,3462
 mbxai/openrouter/__init__.py,sha256=Ito9Qp_B6q-RLGAQcYyTJVWwR2YAZvNqE-HIYXxhtD8,298
 mbxai/openrouter/client.py,sha256=RO5tbF42vkcjxjvC-QFB8DGA0gQLljH3KPBn3HgZV8I,13662
 mbxai/openrouter/config.py,sha256=Ia93s-auim9Sq71eunVDbn9ET5xX2zusXpV4JBdHAzs,3251
 mbxai/openrouter/models.py,sha256=b3IjjtZAjeGOf2rLsdnCD1HacjTnS8jmv_ZXorc-KJQ,2604
 mbxai/tools/__init__.py,sha256=QUFaXhDm-UKcuAtT1rbKzhBkvyRBVokcQIOf9cxIuwc,160
-mbxai/tools/client.py,sha256=lRrVoaGwcnKUkwONVYXNGG9ao4vXgX-fsniPjlXWUZo,11445
+mbxai/tools/client.py,sha256=XZsSK3-YNvMK_V-byov8-vtAaVVhC19MemFoQ3Laa3Q,11907
 mbxai/tools/example.py,sha256=1HgKK39zzUuwFbnp3f0ThyWVfA_8P28PZcTwaUw5K78,2232
 mbxai/tools/types.py,sha256=fo5t9UbsHGynhA88vD_ecgDqL8iLvt2E1h1ym43Rrgk,745
-mbxai-0.6.4.dist-info/METADATA,sha256=dz-ZLvT6BoU95BV9TcN483JJO91n9SRJpn6XVdEp2w8,4107
-mbxai-0.6.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mbxai-0.6.4.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
-mbxai-0.6.4.dist-info/RECORD,,
+mbxai-0.6.6.dist-info/METADATA,sha256=6USMdTMlJQZNzwy13gx4svOLYwOO7fNoUwdqhM3QWM8,4107
+mbxai-0.6.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mbxai-0.6.6.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
+mbxai-0.6.6.dist-info/RECORD,,