mirascope 1.18.0__tar.gz → 1.18.1__tar.gz

{mirascope-1.18.0 → mirascope-1.18.1}/PKG-INFO

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

{mirascope-1.18.0 → mirascope-1.18.1}/mirascope/core/azure/_utils/_setup_call.py

@@ -10,7 +10,6 @@ from azure.ai.inference import ChatCompletionsClient
 from azure.ai.inference.aio import ChatCompletionsClient as AsyncChatCompletionsClient
 from azure.ai.inference.models import (
     ChatCompletions,
-    ChatCompletionsResponseFormatJSON,
     ChatRequestMessage,
     StreamingChatCompletionsUpdate,
     UserMessage,
@@ -29,13 +28,27 @@ from ...base._utils import (
 from ...base.call_params import CommonCallParams
 from ...base.stream_config import StreamConfig
 from .._call_kwargs import AzureCallKwargs
-from ..call_params import AzureCallParams
+from ..call_params import AzureCallParams, ResponseFormatJSON
 from ..dynamic_config import AsyncAzureDynamicConfig, AzureDynamicConfig
 from ..tool import AzureTool, GenerateAzureStrictToolJsonSchema
 from ._convert_common_call_params import convert_common_call_params
 from ._convert_message_params import convert_message_params
 from ._get_credential import get_credential
 
+try:
+    from azure.ai.inference.models._models import (
+        ChatCompletionsResponseFormatJSON as FormatJSON,
+    )
+
+    json_object: ResponseFormatJSON | None = None
+except ImportError:  # pragma: no cover
+    # https://github.com/Azure/azure-sdk-for-python/commit/cf61ada8243d619ee6164bf1d32b3939b3c271e5#diff-0fb29d8cce6df3a014592ba8e9959409b8c471088381549807239324f5e8ddc4R15-R17
+    from azure.ai.inference.models._models import (
+        JsonSchemaFormat as FormatJSON,  # pyright: ignore [reportAttributeAccessIssue]
+    )
+
+    json_object = "json_object"  # pyright: ignore [reportAssignmentType]
+
 
 @overload
 def setup_call(
@@ -116,7 +129,7 @@ def setup_call(
     messages = convert_message_params(messages)
     if json_mode:
         if response_model and response_model.model_config.get("strict", False):
-            call_kwargs["response_format"] = ChatCompletionsResponseFormatJSON(
+            call_kwargs["response_format"] = FormatJSON(
                 {
                     "name": response_model.__name__,
                     "description": response_model.__doc__ or DEFAULT_TOOL_DOCSTRING,
@@ -127,7 +140,7 @@ def setup_call(
                 }
             )
         else:
-            call_kwargs["response_format"] = ChatCompletionsResponseFormatJSON()
+            call_kwargs["response_format"] = json_object or FormatJSON()
             json_mode_content = _utils.json_mode_content(response_model).strip()
             messages.append(UserMessage(content=json_mode_content))
     elif response_model:

{mirascope-1.18.0 → mirascope-1.18.1}/mirascope/core/azure/call_params.py

@@ -2,13 +2,26 @@
 
 from __future__ import annotations
 
-from typing import Any
+from typing import Any, Literal, TypeAlias
 
 from azure.ai.inference.models import (
     ChatCompletionsNamedToolChoice,
-    ChatCompletionsResponseFormat,
     ChatCompletionsToolChoicePreset,
 )
+
+try:
+    from azure.ai.inference.models import (
+        ChatCompletionsResponseFormatJSON,
+    )
+
+    ResponseFormatJSON: TypeAlias = ChatCompletionsResponseFormatJSON  # pyright: ignore [reportRedeclaration]
+except ImportError:  # pragma: no cover
+    from azure.ai.inference.models import (
+        JsonSchemaFormat,  # pyright: ignore [reportAttributeAccessIssue]
+    )
+
+    ResponseFormatJSON: TypeAlias = JsonSchemaFormat | Literal["json_object"]  # pyright: ignore [reportRedeclaration]
+
 from typing_extensions import NotRequired
 
 from ..base import BaseCallParams
@@ -36,7 +49,7 @@ class AzureCallParams(BaseCallParams):
     max_tokens: NotRequired[int | None]
     model_extras: NotRequired[dict[str, Any] | None]
     presence_penalty: NotRequired[float | None]
-    response_format: NotRequired[ChatCompletionsResponseFormat | None]
+    response_format: NotRequired[ResponseFormatJSON | None]
     seed: NotRequired[int | None]
     stop: NotRequired[list[str] | None]
     temperature: NotRequired[float | None]

{mirascope-1.18.0 → mirascope-1.18.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mirascope"
-version = "1.18.0"
+version = "1.18.1"
 description = "LLM abstractions that aren't obstructions"
 readme = "README.md"
 license = { file = "LICENSE" }