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

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.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mirascope
-Version: 1.21.0
+Version: 1.21.1
 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.1.dist-info}/RECORD

@@ -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.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,,