mirascope 1.21.0__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/llm/_call.py

@@ -50,24 +50,36 @@ _ResultT = TypeVar("_ResultT")
 def _get_local_provider_call(
     provider: LocalProvider,
     client: Any | None,  # noqa: ANN401
+    is_async: bool,
 ) -> tuple[Callable, Any | None]:
     if provider == "ollama":
         from ..core.openai import openai_call
 
         if client:
             return openai_call, client
-        from openai import OpenAI
+        if is_async:
+            from openai import AsyncOpenAI
 
-        client = OpenAI(api_key="ollama", base_url="http://localhost:11434/v1")
+            client = AsyncOpenAI(api_key="ollama", base_url="http://localhost:11434/v1")
+        else:
+            from openai import OpenAI
+
+            client = OpenAI(api_key="ollama", base_url="http://localhost:11434/v1")
         return openai_call, client
     else:  # provider == "vllm"
         from ..core.openai import openai_call
 
         if client:
             return openai_call, client
-        from openai import OpenAI
 
-        client = OpenAI(api_key="ollama", base_url="http://localhost:8000/v1")
+        if is_async:
+            from openai import AsyncOpenAI
+
+            client = AsyncOpenAI(api_key="ollama", base_url="http://localhost:8000/v1")
+        else:
+            from openai import OpenAI
+
+            client = OpenAI(api_key="ollama", base_url="http://localhost:8000/v1")
         return openai_call, client
 
 
@@ -245,7 +257,9 @@ def _call(
 
                 if effective_provider in get_args(LocalProvider):
                     provider_call, effective_client = _get_local_provider_call(
-                        cast(LocalProvider, effective_provider), effective_client
+                        cast(LocalProvider, effective_provider),
+                        effective_client,
+                        True,
                     )
                     effective_call_args["client"] = effective_client
                 else:
@@ -293,7 +307,9 @@ def _call(
 
                 if effective_provider in get_args(LocalProvider):
                     provider_call, effective_client = _get_local_provider_call(
-                        cast(LocalProvider, effective_provider), effective_client
+                        cast(LocalProvider, effective_provider),
+                        effective_client,
+                        False,
                     )
                     effective_call_args["client"] = effective_client
                 else:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mirascope
-Version: 1.21.0
+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.0.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
@@ -332,7 +332,7 @@ mirascope/integrations/otel/_utils.py,sha256=SCVb3MpcpqLpCpumJEbEdINceNdusnyT6iu
 mirascope/integrations/otel/_with_hyperdx.py,sha256=f17uxXQk5zZPtyj6zwPwJz5i7atsnUPOoq1LqT8JO0E,1637
 mirascope/integrations/otel/_with_otel.py,sha256=tbjd6BEbcSfnsm5CWHBoHwbRNrHt6-t4or-SYGQSD-w,1659
 mirascope/llm/__init__.py,sha256=zo7QD39zQ6HIU_N2yHrzd6bbcT9YmlikZ84JN_r5k4E,397
-mirascope/llm/_call.py,sha256=Xv9Ok0gZjLETQUOavYOk4rapnhOqfpNGw0mxlDf-nV4,12585
+mirascope/llm/_call.py,sha256=XyMMZzg_DUW1pZgVcOP0QxyVDCYW8s7Xj_XMDwCzb_w,13068
 mirascope/llm/_context.py,sha256=Q4M-XRVsAvDW2LPW2XNtCm5HRU36BLFJYuahfGMfTGI,12332
 mirascope/llm/_override.py,sha256=m4MdOhM-aJRIGP7NBJhscq3ISNct6FBPn3jjmryFo_Q,112292
 mirascope/llm/_protocols.py,sha256=HXspRAC0PduGqbh2BM0CGe5iVj7CC3ZKMPAYvFvbDNQ,16406
@@ -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.0.dist-info/METADATA,sha256=3RrfjlPJIf5q2D1pf1jx9Pgrp8DBwnQYCNuGDbQ_ciE,8730
-mirascope-1.21.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mirascope-1.21.0.dist-info/licenses/LICENSE,sha256=LAs5Q8mdawTsVdONpDGukwsoc4KEUBmmonDEL39b23Y,1072
-mirascope-1.21.0.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,,