mbxai 1.4.0__py3-none-any.whl → 1.6.0__py3-none-any.whl

mbxai/__init__.py

@@ -2,4 +2,4 @@
 MBX AI package.
 """
 
-__version__ = "1.4.0" 
+__version__ = "1.6.0" 

mbxai/mcp/server.py

@@ -31,7 +31,7 @@ class MCPServer:
         self.app = FastAPI(
             title=self.name,
             description=self.description,
-            version="1.4.0",
+            version="1.6.0",
         )
         
         # Initialize MCP server
@@ -74,13 +74,10 @@ class MCPServer:
         tools = await self.mcp_server.list_tools()
         tool_metadata = tools[-1]
         
-        # Convert FastMCP schema to our schema format
+        # Use the raw inputSchema from FastMCP - it will be processed later by convert_to_strict_schema
+        # when the tool is converted to OpenAI function format
         inputSchema = tool_metadata.inputSchema
-        if isinstance(inputSchema, dict):
-            if "$ref" in inputSchema:
-                ref = inputSchema["$ref"].split("/")[-1]
-                inputSchema = tool_metadata.inputSchema.get("$defs", {}).get(ref, {})
-                    
+        
         # Create and store Tool instance
         self._tools[tool_metadata.name] = Tool(
             name=tool_metadata.name,

mbxai/tools/types.py

@@ -9,16 +9,142 @@ import json
 
 logger = logging.getLogger(__name__)
 
+def _resolve_ref(schema: dict[str, Any], ref: str, root_schema: dict[str, Any]) -> dict[str, Any]:
+    """Resolve a $ref to its actual schema definition.
+    
+    Args:
+        schema: The current schema object
+        ref: The reference string (e.g., "#/$defs/MyType")
+        root_schema: The root schema containing $defs
+        
+    Returns:
+        The resolved schema definition with all nested $refs also resolved
+    """
+    if ref.startswith("#/"):
+        # Remove the "#/" prefix and split the path
+        path_parts = ref[2:].split("/")
+        current = root_schema
+        
+        # Navigate through the path
+        for part in path_parts:
+            if isinstance(current, dict) and part in current:
+                current = current[part]
+            else:
+                logger.warning(f"Could not resolve $ref: {ref}")
+                return {"type": "object", "additionalProperties": False}
+        
+        if isinstance(current, dict):
+            # Recursively process the resolved schema to handle nested $refs
+            return _convert_property_to_strict(current, root_schema)
+        else:
+            return {"type": "object", "additionalProperties": False}
+    else:
+        logger.warning(f"Unsupported $ref format: {ref}")
+        return {"type": "object", "additionalProperties": False}
+
+def _convert_property_to_strict(prop: dict[str, Any], root_schema: dict[str, Any]) -> dict[str, Any]:
+    """Convert a single property to strict format, recursively handling nested objects and arrays.
+    
+    Args:
+        prop: The property definition to convert
+        root_schema: The root schema containing $defs
+        
+    Returns:
+        A strict format property definition with all $refs resolved
+    """
+    # Handle $ref resolution first - this may return a completely different schema
+    if "$ref" in prop:
+        # Get the referenced schema path
+        ref = prop["$ref"]
+        if ref.startswith("#/"):
+            path_parts = ref[2:].split("/")
+            current = root_schema
+            
+            # Navigate through the path to get the referenced schema
+            for part in path_parts:
+                if isinstance(current, dict) and part in current:
+                    current = current[part]
+                else:
+                    logger.warning(f"Could not resolve $ref: {ref}")
+                    return {"type": "object", "additionalProperties": False}
+            
+            if isinstance(current, dict):
+                # Recursively process the resolved schema
+                return _convert_property_to_strict(current, root_schema)
+            else:
+                return {"type": "object", "additionalProperties": False}
+        else:
+            logger.warning(f"Unsupported $ref format: {ref}")
+            return {"type": "object", "additionalProperties": False}
+    
+    # Get the property type, defaulting to string
+    prop_type = prop.get("type", "string")
+    
+    # Start with basic property structure
+    new_prop = {
+        "type": prop_type,
+        "description": prop.get("description", f"A {prop_type} parameter")
+    }
+    
+    # Handle specific types
+    if prop_type == "object":
+        # Objects need additionalProperties: false and properties
+        new_prop["additionalProperties"] = False
+        new_prop["properties"] = {}
+        new_prop["required"] = []
+        
+        # Process nested properties recursively
+        if "properties" in prop:
+            for nested_name, nested_prop in prop["properties"].items():
+                new_prop["properties"][nested_name] = _convert_property_to_strict(nested_prop, root_schema)
+        
+        # Copy required fields
+        if "required" in prop:
+            new_prop["required"] = prop["required"]
+            
+    elif prop_type == "array":
+        # Arrays must have items definition
+        if "items" in prop:
+            # Recursively process items schema (this could also have $refs)
+            new_prop["items"] = _convert_property_to_strict(prop["items"], root_schema)
+        else:
+            # Default items schema if missing
+            new_prop["items"] = {
+                "type": "string",
+                "description": "Array item"
+            }
+            
+    elif prop_type in ["string", "number", "integer", "boolean"]:
+        # For primitive types, copy additional constraints
+        if "enum" in prop:
+            new_prop["enum"] = prop["enum"]
+        if "minimum" in prop:
+            new_prop["minimum"] = prop["minimum"]
+        if "maximum" in prop:
+            new_prop["maximum"] = prop["maximum"]
+        if "pattern" in prop:
+            new_prop["pattern"] = prop["pattern"]
+        if "format" in prop:
+            new_prop["format"] = prop["format"]
+    
+    return new_prop
+
 def convert_to_strict_schema(schema: dict[str, Any], strict: bool = True, keep_input_wrapper: bool = False) -> dict[str, Any]:
     """Convert a schema to strict format required by OpenAI.
     
+    This function handles:
+    - Resolving all $ref and $defs to inline definitions
+    - Adding required 'items' property for arrays
+    - Ensuring all objects have 'additionalProperties: false'
+    - Recursively processing nested schemas
+    
     Args:
         schema: The input schema to validate and convert
         strict: Whether to enforce strict validation with additionalProperties: false
         keep_input_wrapper: Whether to keep the input wrapper (for MCP tools)
         
     Returns:
-        A schema in strict format
+        A schema in strict format with all references resolved
     """
     if not schema:
         return {"type": "object", "properties": {}, "required": [], "additionalProperties": False}
@@ -31,14 +157,16 @@ def convert_to_strict_schema(schema: dict[str, Any], strict: bool = True, keep_i
         "additionalProperties": False  # Always enforce additionalProperties: false for OpenRouter
     }
 
+    # Store the root schema for $ref resolution
+    root_schema = schema
+
     # Handle input wrapper
     if "properties" in schema and "input" in schema["properties"]:
         inputSchema = schema["properties"]["input"]
         
         # If input has a $ref, resolve it
         if "$ref" in inputSchema:
-            ref = inputSchema["$ref"].split("/")[-1]
-            inputSchema = schema.get("$defs", {}).get(ref, {})
+            inputSchema = _resolve_ref(inputSchema, inputSchema["$ref"], root_schema)
         
         if keep_input_wrapper:
             # Create the input property schema
@@ -49,20 +177,10 @@ def convert_to_strict_schema(schema: dict[str, Any], strict: bool = True, keep_i
                 "additionalProperties": False  # Always enforce additionalProperties: false for OpenRouter
             }
             
-            # Copy over input properties
+            # Process input properties
             if "properties" in inputSchema:
                 for prop_name, prop in inputSchema["properties"].items():
-                    # Create a new property object with only allowed fields
-                    new_prop = {
-                        "type": prop.get("type", "string"),
-                        "description": prop.get("description", f"The {prop_name} parameter")
-                    }
-                    
-                    # If the property is an object, ensure it has additionalProperties: false
-                    if new_prop["type"] == "object":
-                        new_prop["additionalProperties"] = False
-                    
-                    input_prop_schema["properties"][prop_name] = new_prop
+                    input_prop_schema["properties"][prop_name] = _convert_property_to_strict(prop, root_schema)
             
             # Copy over required fields for input schema
             if "required" in inputSchema:
@@ -78,17 +196,7 @@ def convert_to_strict_schema(schema: dict[str, Any], strict: bool = True, keep_i
             # If not keeping input wrapper, use input schema directly
             if "properties" in inputSchema:
                 for prop_name, prop in inputSchema["properties"].items():
-                    # Create a new property object with only allowed fields
-                    new_prop = {
-                        "type": prop.get("type", "string"),
-                        "description": prop.get("description", f"The {prop_name} parameter")
-                    }
-                    
-                    # If the property is an object, ensure it has additionalProperties: false
-                    if new_prop["type"] == "object":
-                        new_prop["additionalProperties"] = False
-                    
-                    strict_schema["properties"][prop_name] = new_prop
+                    strict_schema["properties"][prop_name] = _convert_property_to_strict(prop, root_schema)
             
             # Copy over required fields
             if "required" in inputSchema:
@@ -97,17 +205,7 @@ def convert_to_strict_schema(schema: dict[str, Any], strict: bool = True, keep_i
         # If no input wrapper, use the schema as is
         if "properties" in schema:
             for prop_name, prop in schema["properties"].items():
-                # Create a new property object with only allowed fields
-                new_prop = {
-                    "type": prop.get("type", "string"),
-                    "description": prop.get("description", f"The {prop_name} parameter")
-                }
-                
-                # If the property is an object, ensure it has additionalProperties: false
-                if new_prop["type"] == "object":
-                    new_prop["additionalProperties"] = False
-                
-                strict_schema["properties"][prop_name] = new_prop
+                strict_schema["properties"][prop_name] = _convert_property_to_strict(prop, root_schema)
         
         # Copy over required fields
         if "required" in schema:

{mbxai-1.4.0.dist-info → mbxai-1.6.0.dist-info}/METADATA

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

{mbxai-1.4.0.dist-info → mbxai-1.6.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-mbxai/__init__.py,sha256=E7p4_3-qHNUH7NcRh8unQZwHeS5pRX-O8zL-n0jaR2M,47
+mbxai/__init__.py,sha256=k9FHAi4YDTkuiAHKU5ktijzvf4s2I3EFN1s7jqJPA0o,47
 mbxai/core.py,sha256=WMvmU9TTa7M_m-qWsUew4xH8Ul6xseCZ2iBCXJTW-Bs,196
 mbxai/examples/openrouter_example.py,sha256=-grXHKMmFLoh-yUIEMc31n8Gg1S7uSazBWCIOWxgbyQ,1317
 mbxai/examples/parse_example.py,sha256=eCKMJoOl6qwo8sDP6Trc6ncgjPlgTqi5tPE2kB5_P0k,3821
@@ -12,7 +12,7 @@ mbxai/examples/mcp/mcp_server_example.py,sha256=nFfg22Jnc6HMW_ezLO3So1xwDdx2_rIt
 mbxai/mcp/__init__.py,sha256=_ek9iYdYqW5saKetj4qDci11jxesQDiHPJRpHMKkxgU,175
 mbxai/mcp/client.py,sha256=QRzId6o4_WRWVv3rtm8cfZZGaoY_UlaOO-oqNjY-tmw,5219
 mbxai/mcp/example.py,sha256=oaol7AvvZnX86JWNz64KvPjab5gg1VjVN3G8eFSzuaE,2350
-mbxai/mcp/server.py,sha256=V-o8BugXj9Jj_yUIGepH0fCw8lrxYSpVRpI5CFfAvnk,3454
+mbxai/mcp/server.py,sha256=tSc05RxhXpRdXSpu-noJmK0V48uru0v1JpcUr0P5fmA,3332
 mbxai/openrouter/__init__.py,sha256=Ito9Qp_B6q-RLGAQcYyTJVWwR2YAZvNqE-HIYXxhtD8,298
 mbxai/openrouter/client.py,sha256=3LD6WDJ8wjo_nefH5d1NJCsrWPvBc_KBf2NsItUoSt8,18302
 mbxai/openrouter/config.py,sha256=Ia93s-auim9Sq71eunVDbn9ET5xX2zusXpV4JBdHAzs,3251
@@ -21,8 +21,8 @@ mbxai/openrouter/schema.py,sha256=H_77ZrA9zmbX155bWpCJj1jehUyJPS0QybEW1IVAoe0,54
 mbxai/tools/__init__.py,sha256=ogxrHvgJ7OR62Lmd5x9Eh5d2C0jqWyQis7Zy3yKpZ78,218
 mbxai/tools/client.py,sha256=j6yB2hYxvWbaQ5SqN1Fs_YFdPtwettdcMoXcdeV-520,14930
 mbxai/tools/example.py,sha256=1HgKK39zzUuwFbnp3f0ThyWVfA_8P28PZcTwaUw5K78,2232
-mbxai/tools/types.py,sha256=7gNIJBjzr9i4DT50OGLMjn3-6yBXqlK-kIz_RWcqywo,5875
-mbxai-1.4.0.dist-info/METADATA,sha256=EWpGN5OMQTKiNodh1zxL-tU1Ud3U3-Q20Sz-Ep7YUuA,4147
-mbxai-1.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mbxai-1.4.0.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
-mbxai-1.4.0.dist-info/RECORD,,
+mbxai/tools/types.py,sha256=OFfM7scDGTm4FOcJA2ecj-fxL1MEBkqPsT3hqCL1Jto,9505
+mbxai-1.6.0.dist-info/METADATA,sha256=8xzN7T2h5j8XyFMaq2CpXi3zMoe7tyM1RvjzYCpJz6U,4147
+mbxai-1.6.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mbxai-1.6.0.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
+mbxai-1.6.0.dist-info/RECORD,,