PyPI - mbxai - Versions diffs - 0.5.11__tar.gz → 0.5.13__tar.gz - Mend

mbxai 0.5.11tar.gz → 0.5.13tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (25) hide show

{mbxai-0.5.11 → mbxai-0.5.13}/PKG-INFO RENAMED Viewed

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

{mbxai-0.5.11 → mbxai-0.5.13}/pyproject.toml RENAMED Viewed

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 [project]
 name = "mbxai"
-version = "0.5.11"
+version = "0.5.13"
 authors = [
     { name = "MBX AI" }
 ]

{mbxai-0.5.11 → mbxai-0.5.13}/setup.py RENAMED Viewed

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 setup(
     name="mbxai",
-    version="0.5.11",
+    version="0.5.13",
     author="MBX AI",
     description="MBX AI SDK",
     long_description=open("README.md").read(),

{mbxai-0.5.11 → mbxai-0.5.13}/src/mbxai/__init__.py RENAMED Viewed

@@ -2,4 +2,4 @@
 MBX AI package.
 """
-__version__ = "0.5.11"
+__version__ = "0.5.13"

{mbxai-0.5.11 → mbxai-0.5.13}/src/mbxai/mcp/server.py RENAMED Viewed

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

{mbxai-0.5.11 → mbxai-0.5.13}/src/mbxai/tools/client.py RENAMED Viewed

@@ -48,7 +48,57 @@ class ToolClient:
             schema=schema,
         )
         self._tools[name] = tool
-        logger.debug(f"Registered tool: {name}")
+        logger.info(f"Registered tool: {name}")
+    def _truncate_content(self, content: str | None, max_length: int = 100) -> str:
+        """Truncate content for logging."""
+        if not content:
+            return "None"
+        if len(content) <= max_length:
+            return content
+        return content[:max_length] + "..."
+    def _log_messages(self, messages: list[dict[str, Any]]) -> None:
+        """Log the messages being sent to OpenRouter."""
+        logger.info("Sending messages to OpenRouter:")
+        for msg in messages:
+            role = msg.get("role", "unknown")
+            content = self._truncate_content(msg.get("content"))
+            tool_calls = msg.get("tool_calls", [])
+            tool_call_id = msg.get("tool_call_id")
+            if tool_calls:
+                tool_call_info = [
+                    f"{tc['function']['name']}(id={tc['id']})"
+                    for tc in tool_calls
+                ]
+                logger.info(f"  {role}: content='{content}', tool_calls={tool_call_info}")
+            elif tool_call_id:
+                logger.info(f"  {role}: content='{content}', tool_call_id={tool_call_id}")
+            else:
+                logger.info(f"  {role}: content='{content}'")
+        # Validate tool call responses
+        tool_call_ids = set()
+        tool_response_ids = set()
+        for msg in messages:
+            if msg.get("role") == "assistant" and "tool_calls" in msg:
+                for tc in msg["tool_calls"]:
+                    tool_call_ids.add(tc["id"])
+            elif msg.get("role") == "tool":
+                tool_response_ids.add(msg["tool_call_id"])
+        missing_responses = tool_call_ids - tool_response_ids
+        if missing_responses:
+            logger.error(f"Missing tool responses for call IDs: {missing_responses}")
+            logger.error("Message sequence:")
+            for msg in messages:
+                role = msg.get("role", "unknown")
+                if role == "assistant" and "tool_calls" in msg:
+                    logger.error(f"  Assistant message with tool calls: {[tc['id'] for tc in msg['tool_calls']]}")
+                elif role == "tool":
+                    logger.error(f"  Tool response for call ID: {msg['tool_call_id']}")
     async def chat(
         self,
@@ -72,11 +122,14 @@ class ToolClient:
         tools = [tool.to_openai_function() for tool in self._tools.values()]
         if tools:
-            logger.debug(f"Using tools: {tools}")
+            logger.info(f"Available tools: {[tool['function']['name'] for tool in tools]}")
             kwargs["tools"] = tools
             kwargs["tool_choice"] = "auto"
         while True:
+            # Log messages before sending to OpenRouter
+            self._log_messages(messages)
             # Get the model's response
             response = self._client.chat_completion(
                 messages=messages,
@@ -107,92 +160,86 @@ class ToolClient:
                     for tool_call in message.tool_calls
                 ]
             messages.append(assistant_message)
-            logger.debug(f"Added assistant message: {assistant_message}")
+            logger.info(f"Assistant message: content='{self._truncate_content(message.content)}', tool_calls={[tc.function.name for tc in message.tool_calls] if message.tool_calls else None}")
             # If there are no tool calls, we're done
             if not message.tool_calls:
                 return response
-            # Handle each tool call
+            # Handle all tool calls before getting the next model response
+            tool_responses = []
             for tool_call in message.tool_calls:
-                logger.debug(f"Processing tool call: {tool_call}")
-                logger.debug(f"Tool call ID: {tool_call.id}")
-                logger.debug(f"Tool call function: {tool_call.function}")
-                logger.debug(f"Tool call arguments: {tool_call.function.arguments}")
                 tool = self._tools.get(tool_call.function.name)
                 if not tool:
                     raise ValueError(f"Unknown tool: {tool_call.function.name}")
                 # Parse arguments if they're a string
                 arguments = tool_call.function.arguments
-                logger.debug(f"Raw arguments type: {type(arguments)}")
-                logger.debug(f"Raw arguments: {arguments}")
                 if isinstance(arguments, str):
                     try:
                         arguments = json.loads(arguments)
-                        logger.debug(f"Parsed arguments: {arguments}")
                     except json.JSONDecodeError as e:
                         logger.error(f"Failed to parse tool arguments: {e}")
                         raise ValueError(f"Invalid tool arguments format: {arguments}")
                 # Call the tool
-                logger.debug(f"Calling tool {tool.name} with arguments: {arguments}")
+                logger.info(f"Calling tool: {tool.name} with args: {self._truncate_content(json.dumps(arguments))}")
                 if inspect.iscoroutinefunction(tool.function):
                     result = await tool.function(**arguments)
                 else:
                     result = tool.function(**arguments)
-                logger.debug(f"Tool result: {result}")
                 # Convert result to JSON string if it's not already
                 if not isinstance(result, str):
                     result = json.dumps(result)
-                # Add the tool response to the messages
+                # Create the tool response
                 tool_response = {
                     "role": "tool",
                     "tool_call_id": tool_call.id,
                     "content": result,
                 }
-                messages.append(tool_response)
-                logger.debug(f"Added tool response to messages: {tool_response}")
-                # Get a new response from the model with this tool result
-                response = self._client.chat_completion(
-                    messages=messages,
-                    model=model,
-                    stream=stream,
-                    **kwargs,
-                )
-                if stream:
-                    return response
-                message = response.choices[0].message
-                # Add the assistant's message with tool calls
-                assistant_message = {
-                    "role": "assistant",
-                    "content": message.content or None,  # Ensure content is None if empty
-                }
-                if message.tool_calls:
-                    assistant_message["tool_calls"] = [
-                        {
-                            "id": tool_call.id,
-                            "type": "function",
-                            "function": {
-                                "name": tool_call.function.name,
-                                "arguments": tool_call.function.arguments,
-                            },
-                        }
-                        for tool_call in message.tool_calls
-                    ]
-                messages.append(assistant_message)
-                logger.debug(f"Added assistant message: {assistant_message}")
-                # If there are no more tool calls, we're done
-                if not message.tool_calls:
-                    return response
+                tool_responses.append(tool_response)
+                logger.info(f"Tool response for call ID {tool_call.id}: {self._truncate_content(result)}")
+            # Add all tool responses to the messages
+            messages.extend(tool_responses)
+            # Get a new response from the model with all tool results
+            response = self._client.chat_completion(
+                messages=messages,
+                model=model,
+                stream=stream,
+                **kwargs,
+            )
+            if stream:
+                return response
+            message = response.choices[0].message
+            # Add the assistant's message with tool calls
+            assistant_message = {
+                "role": "assistant",
+                "content": message.content or None,  # Ensure content is None if empty
+            }
+            if message.tool_calls:
+                assistant_message["tool_calls"] = [
+                    {
+                        "id": tool_call.id,
+                        "type": "function",
+                        "function": {
+                            "name": tool_call.function.name,
+                            "arguments": tool_call.function.arguments,
+                        },
+                    }
+                    for tool_call in message.tool_calls
+                ]
+            messages.append(assistant_message)
+            logger.info(f"Assistant message: content='{self._truncate_content(message.content)}', tool_calls={[tc.function.name for tc in message.tool_calls] if message.tool_calls else None}")
+            # If there are no more tool calls, we're done
+            if not message.tool_calls:
+                return response
     async def parse(
         self,
@@ -218,11 +265,14 @@ class ToolClient:
         tools = [tool.to_openai_function() for tool in self._tools.values()]
         if tools:
-            logger.debug(f"Using tools: {tools}")
+            logger.info(f"Available tools: {[tool['function']['name'] for tool in tools]}")
             kwargs["tools"] = tools
             kwargs["tool_choice"] = "auto"
         while True:
+            # Log messages before sending to OpenRouter
+            self._log_messages(messages)
             # Get the model's response
             response = self._client.chat_completion_parse(
                 messages=messages,
@@ -254,90 +304,84 @@ class ToolClient:
                     for tool_call in message.tool_calls
                 ]
             messages.append(assistant_message)
-            logger.debug(f"Added assistant message: {assistant_message}")
+            logger.info(f"Assistant message: content='{self._truncate_content(message.content)}', tool_calls={[tc.function.name for tc in message.tool_calls] if message.tool_calls else None}")
             # If there are no tool calls, we're done
             if not message.tool_calls:
                 return response
-            # Handle each tool call
+            # Handle all tool calls before getting the next model response
+            tool_responses = []
             for tool_call in message.tool_calls:
-                logger.debug(f"Processing tool call: {tool_call}")
-                logger.debug(f"Tool call ID: {tool_call.id}")
-                logger.debug(f"Tool call function: {tool_call.function}")
-                logger.debug(f"Tool call arguments: {tool_call.function.arguments}")
                 tool = self._tools.get(tool_call.function.name)
                 if not tool:
                     raise ValueError(f"Unknown tool: {tool_call.function.name}")
                 # Parse arguments if they're a string
                 arguments = tool_call.function.arguments
-                logger.debug(f"Raw arguments type: {type(arguments)}")
-                logger.debug(f"Raw arguments: {arguments}")
                 if isinstance(arguments, str):
                     try:
                         arguments = json.loads(arguments)
-                        logger.debug(f"Parsed arguments: {arguments}")
                     except json.JSONDecodeError as e:
                         logger.error(f"Failed to parse tool arguments: {e}")
                         raise ValueError(f"Invalid tool arguments format: {arguments}")
                 # Call the tool
-                logger.debug(f"Calling tool {tool.name} with arguments: {arguments}")
+                logger.info(f"Calling tool: {tool.name} with args: {self._truncate_content(json.dumps(arguments))}")
                 if inspect.iscoroutinefunction(tool.function):
                     result = await tool.function(**arguments)
                 else:
                     result = tool.function(**arguments)
-                logger.debug(f"Tool result: {result}")
                 # Convert result to JSON string if it's not already
                 if not isinstance(result, str):
                     result = json.dumps(result)
-                # Add the tool response to the messages
+                # Create the tool response
                 tool_response = {
                     "role": "tool",
                     "tool_call_id": tool_call.id,
                     "content": result,
                 }
-                messages.append(tool_response)
-                logger.debug(f"Added tool response to messages: {tool_response}")
-                # Get a new response from the model with this tool result
-                response = self._client.chat_completion_parse(
-                    messages=messages,
-                    response_format=response_format,
-                    model=model,
-                    stream=stream,
-                    **kwargs,
-                )
-                if stream:
-                    return response
-                message = response.choices[0].message
-                # Add the assistant's message with tool calls
-                assistant_message = {
-                    "role": "assistant",
-                    "content": message.content or None,  # Ensure content is None if empty
-                }
-                if message.tool_calls:
-                    assistant_message["tool_calls"] = [
-                        {
-                            "id": tool_call.id,
-                            "type": "function",
-                            "function": {
-                                "name": tool_call.function.name,
-                                "arguments": tool_call.function.arguments,
-                            },
-                        }
-                        for tool_call in message.tool_calls
-                    ]
-                messages.append(assistant_message)
-                logger.debug(f"Added assistant message: {assistant_message}")
-                # If there are no more tool calls, we're done
-                if not message.tool_calls:
-                    return response
+                tool_responses.append(tool_response)
+                logger.info(f"Tool response: {self._truncate_content(result)}")
+            # Add all tool responses to the messages
+            messages.extend(tool_responses)
+            # Get a new response from the model with all tool results
+            response = self._client.chat_completion_parse(
+                messages=messages,
+                response_format=response_format,
+                model=model,
+                stream=stream,
+                **kwargs,
+            )
+            if stream:
+                return response
+            message = response.choices[0].message
+            # Add the assistant's message with tool calls
+            assistant_message = {
+                "role": "assistant",
+                "content": message.content or None,  # Ensure content is None if empty
+            }
+            if message.tool_calls:
+                assistant_message["tool_calls"] = [
+                    {
+                        "id": tool_call.id,
+                        "type": "function",
+                        "function": {
+                            "name": tool_call.function.name,
+                            "arguments": tool_call.function.arguments,
+                        },
+                    }
+                    for tool_call in message.tool_calls
+                ]
+            messages.append(assistant_message)
+            logger.info(f"Assistant message: content='{self._truncate_content(message.content)}', tool_calls={[tc.function.name for tc in message.tool_calls] if message.tool_calls else None}")
+            # If there are no more tool calls, we're done
+            if not message.tool_calls:
+                return response

{mbxai-0.5.11 → mbxai-0.5.13}/uv.lock RENAMED Viewed

@@ -292,11 +292,11 @@ wheels = [
 [[package]]
 name = "httpx-sse"
-version = "0.5.11"
+version = "0.5.13"
 source = { registry = "https://pypi.org/simple" }
-sdist = { url = "https://files.pythonhosted.org/packages/4c/60/8f4281fa9bbf3c8034fd54c0e7412e66edbab6bc74c4996bd616f8d0406e/httpx-sse-0.5.11.tar.gz", hash = "sha256:1e81a3a3070ce322add1d3529ed42eb5f70817f45ed6ec915ab753f961139721", size = 12624 }
+sdist = { url = "https://files.pythonhosted.org/packages/4c/60/8f4281fa9bbf3c8034fd54c0e7412e66edbab6bc74c4996bd616f8d0406e/httpx-sse-0.5.13.tar.gz", hash = "sha256:1e81a3a3070ce322add1d3529ed42eb5f70817f45ed6ec915ab753f961139721", size = 12624 }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/e1/9b/a181f281f65d776426002f330c31849b86b31fc9d848db62e16f03ff739f/httpx_sse-0.5.11-py3-none-any.whl", hash = "sha256:f329af6eae57eaa2bdfd962b42524764af68075ea87370a2de920af5341e318f", size = 7819 },
+    { url = "https://files.pythonhosted.org/packages/e1/9b/a181f281f65d776426002f330c31849b86b31fc9d848db62e16f03ff739f/httpx_sse-0.5.13-py3-none-any.whl", hash = "sha256:f329af6eae57eaa2bdfd962b42524764af68075ea87370a2de920af5341e318f", size = 7819 },
 ]
 [[package]]
@@ -446,7 +446,7 @@ wheels = [
 [[package]]
 name = "mbxai"
-version = "0.5.11"
+version = "0.5.13"
 source = { editable = "." }
 dependencies = [
     { name = "fastapi" },
@@ -980,14 +980,14 @@ wheels = [
 [[package]]
 name = "typing-inspection"
-version = "0.5.11"
+version = "0.5.13"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "typing-extensions" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/82/5c/e6082df02e215b846b4b8c0b887a64d7d08ffaba30605502639d44c06b82/typing_inspection-0.5.11.tar.gz", hash = "sha256:9765c87de36671694a67904bf2c96e395be9c6439bb6c87b5142569dcdd65122", size = 76222 }
+sdist = { url = "https://files.pythonhosted.org/packages/82/5c/e6082df02e215b846b4b8c0b887a64d7d08ffaba30605502639d44c06b82/typing_inspection-0.5.13.tar.gz", hash = "sha256:9765c87de36671694a67904bf2c96e395be9c6439bb6c87b5142569dcdd65122", size = 76222 }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/31/08/aa4fdfb71f7de5176385bd9e90852eaf6b5d622735020ad600f2bab54385/typing_inspection-0.5.11-py3-none-any.whl", hash = "sha256:50e72559fcd2a6367a19f7a7e610e6afcb9fac940c650290eed893d61386832f", size = 14125 },
+    { url = "https://files.pythonhosted.org/packages/31/08/aa4fdfb71f7de5176385bd9e90852eaf6b5d622735020ad600f2bab54385/typing_inspection-0.5.13-py3-none-any.whl", hash = "sha256:50e72559fcd2a6367a19f7a7e610e6afcb9fac940c650290eed893d61386832f", size = 14125 },
 ]
 [[package]]