mirascope 1.21.1__py3-none-any.whl → 1.21.2__py3-none-any.whl

mirascope/core/google/tool.py

@@ -70,41 +70,50 @@ class GoogleTool(BaseTool):
             fn["parameters"] = model_schema
 
         if "parameters" in fn:
-            # Resolve $defs and $ref
+            # Define a function to handle both ref resolution and examples conversion
+            def resolve_refs_and_fix_examples(
+                schema: dict[str, Any], defs: dict[str, Any] | None = None
+            ) -> dict[str, Any]:
+                """Recursively resolve $ref references and fix examples/example fields."""
+                # If this is a reference, resolve it
+                if "$ref" in schema:
+                    ref = schema["$ref"]
+                    if ref.startswith("#/$defs/") and defs:
+                        ref_key = ref.replace("#/$defs/", "")
+                        if ref_key in defs:
+                            # Merge the definition with the current schema (excluding $ref)
+                            resolved = {
+                                **{k: v for k, v in schema.items() if k != "$ref"},
+                                **resolve_refs_and_fix_examples(defs[ref_key], defs),
+                            }
+                            return resolved
+
+                # Handle examples -> example conversion
+                result = {}
+                for key, value in schema.items():
+                    # Convert "examples" to "example" for Google Schema
+                    if key == "examples":
+                        result["example"] = value
+                    elif isinstance(value, dict):
+                        result[key] = resolve_refs_and_fix_examples(value, defs)
+                    elif isinstance(value, list):
+                        result[key] = [
+                            resolve_refs_and_fix_examples(item, defs)
+                            if isinstance(item, dict)
+                            else item
+                            for item in value
+                        ]
+                    else:
+                        result[key] = value
+                return result
+
+            # Extract $defs if they exist
+            defs = {}
             if "$defs" in fn["parameters"]:
                 defs = fn["parameters"].pop("$defs")
 
-                def resolve_refs(schema: dict[str, Any]) -> dict[str, Any]:
-                    """Recursively resolve $ref references using the $defs dictionary."""
-                    # If this is a reference, resolve it
-                    if "$ref" in schema:
-                        ref = schema["$ref"]
-                        if ref.startswith("#/$defs/"):
-                            ref_key = ref.replace("#/$defs/", "")
-                            if ref_key in defs:
-                                # Merge the definition with the current schema (excluding $ref)
-                                resolved = {
-                                    **{k: v for k, v in schema.items() if k != "$ref"},
-                                    **resolve_refs(defs[ref_key]),
-                                }
-                                return resolved
-
-                    # Process all other keys recursively
-                    result = {}
-                    for key, value in schema.items():
-                        if isinstance(value, dict):
-                            result[key] = resolve_refs(value)
-                        elif isinstance(value, list):
-                            result[key] = [
-                                resolve_refs(item) if isinstance(item, dict) else item
-                                for item in value
-                            ]
-                        else:
-                            result[key] = value
-                    return result
-
-                # Resolve all references in the parameters
-                fn["parameters"] = resolve_refs(fn["parameters"])
+            # Resolve all references in the parameters and fix examples
+            fn["parameters"] = resolve_refs_and_fix_examples(fn["parameters"], defs)
 
             def handle_enum_schema(prop_schema: dict[str, Any]) -> dict[str, Any]:
                 if "enum" in prop_schema:
@@ -112,6 +121,7 @@ class GoogleTool(BaseTool):
                 return prop_schema
 
             # Process properties after resolving references
+            # We're already handling examples -> example conversion recursively above
             fn["parameters"]["properties"] = {
                 prop: {
                     key: value
@@ -120,6 +130,7 @@ class GoogleTool(BaseTool):
                 }
                 for prop, prop_schema in fn["parameters"]["properties"].items()
             }
+
         return Tool(function_declarations=[FunctionDeclaration(**fn)])
 
     @classmethod

{mirascope-1.21.1.dist-info → mirascope-1.21.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mirascope
-Version: 1.21.1
+Version: 1.21.2
 Summary: LLM abstractions that aren't obstructions
 Project-URL: Homepage, https://mirascope.com
 Project-URL: Documentation, https://mirascope.com/WELCOME

{mirascope-1.21.1.dist-info → mirascope-1.21.2.dist-info}/RECORD

@@ -217,7 +217,7 @@ mirascope/core/google/call_response.py,sha256=8yn1cRtqxhnjjzLltfWMzLbaS1Dg7gmS4I
 mirascope/core/google/call_response_chunk.py,sha256=oORYaGjOBv4U1IVhsR_ZS-Vr-B58I8G0-9d6Kcj9RM4,3427
 mirascope/core/google/dynamic_config.py,sha256=O6j8F0fLVFuuNwURneu5OpPuu_bMEtbDEFHhJXRT6V0,857
 mirascope/core/google/stream.py,sha256=bTxB8OUrKXxzmcX0C7_-LqtBfaAAazA5HjKZGSxxtLw,4466
-mirascope/core/google/tool.py,sha256=HPe7CZEn1nPFkvuigpDiTdU9y9cQGp6R68FChjETMAU,4623
+mirascope/core/google/tool.py,sha256=61a9Ejdxz41pwaab9VE2yvP_J1Aebua3BeRPJ_GJSnE,5138
 mirascope/core/google/_utils/__init__.py,sha256=vL0hx6WKW5lqpUcFTFCFGvmwtR-pts0JzWgCXhaUVrI,388
 mirascope/core/google/_utils/_convert_common_call_params.py,sha256=KA-z6uvRtdD4WydC0eXd3dzQuSh4x4WKNR8PAqFNUVY,1065
 mirascope/core/google/_utils/_convert_finish_reason_to_common_finish_reasons.py,sha256=ig4tb7Zanz-tyZpvc9Ncd47a2FNTOS7-wl1PYBq-4cY,879
@@ -368,7 +368,7 @@ mirascope/v0/base/ops_utils.py,sha256=1Qq-VIwgHBaYutiZsS2MUQ4OgPC3APyywI5bTiTAmA
 mirascope/v0/base/prompts.py,sha256=FM2Yz98cSnDceYogiwPrp4BALf3_F3d4fIOCGAkd-SE,1298
 mirascope/v0/base/types.py,sha256=ZfatJoX0Yl0e3jhv0D_MhiSVHLYUeJsdN3um3iE10zY,352
 mirascope/v0/base/utils.py,sha256=XREPENRQTu8gpMhHU8RC8qH_am3FfGUvY-dJ6x8i-mw,681
-mirascope-1.21.1.dist-info/METADATA,sha256=G2jXSfZomkT1FvuLahmGaE83CA4RMPdhwIujlADIbc8,8730
-mirascope-1.21.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mirascope-1.21.1.dist-info/licenses/LICENSE,sha256=LAs5Q8mdawTsVdONpDGukwsoc4KEUBmmonDEL39b23Y,1072
-mirascope-1.21.1.dist-info/RECORD,,
+mirascope-1.21.2.dist-info/METADATA,sha256=1dvAilky26Nim8aNFd5hQEypMwSS9MfUs0grmn__v9o,8730
+mirascope-1.21.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mirascope-1.21.2.dist-info/licenses/LICENSE,sha256=LAs5Q8mdawTsVdONpDGukwsoc4KEUBmmonDEL39b23Y,1072
+mirascope-1.21.2.dist-info/RECORD,,