mirascope 2.0.0a3__py3-none-any.whl → 2.0.0a5__py3-none-any.whl

mirascope/api/_generated/types/__init__.py

@@ -2,20 +2,36 @@
 
 # isort: skip_file
 
+from .already_exists_error import AlreadyExistsError
+from .already_exists_error_tag import AlreadyExistsErrorTag
+from .database_error import DatabaseError
+from .database_error_tag import DatabaseErrorTag
 from .http_api_decode_error import HttpApiDecodeError
 from .http_api_decode_error_tag import HttpApiDecodeErrorTag
 from .issue import Issue
 from .issue_tag import IssueTag
+from .not_found_error_body import NotFoundErrorBody
+from .not_found_error_tag import NotFoundErrorTag
+from .permission_denied_error import PermissionDeniedError
+from .permission_denied_error_tag import PermissionDeniedErrorTag
 from .property_key import PropertyKey
-from .property_key_tag import PropertyKeyTag
-from .property_key_tag_tag import PropertyKeyTagTag
+from .property_key_key import PropertyKeyKey
+from .property_key_key_tag import PropertyKeyKeyTag
 
 __all__ = [
+    "AlreadyExistsError",
+    "AlreadyExistsErrorTag",
+    "DatabaseError",
+    "DatabaseErrorTag",
     "HttpApiDecodeError",
     "HttpApiDecodeErrorTag",
     "Issue",
     "IssueTag",
+    "NotFoundErrorBody",
+    "NotFoundErrorTag",
+    "PermissionDeniedError",
+    "PermissionDeniedErrorTag",
     "PropertyKey",
-    "PropertyKeyTag",
-    "PropertyKeyTagTag",
+    "PropertyKeyKey",
+    "PropertyKeyKeyTag",
 ]

mirascope/api/_generated/types/already_exists_error.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .already_exists_error_tag import AlreadyExistsErrorTag
+
+
+class AlreadyExistsError(UniversalBaseModel):
+    message: str
+    resource: typing.Optional[str] = None
+    tag: AlreadyExistsErrorTag
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

mirascope/api/_generated/types/already_exists_error_tag.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+AlreadyExistsErrorTag = typing.Union[typing.Literal["AlreadyExistsError"], typing.Any]

mirascope/api/_generated/types/database_error.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .database_error_tag import DatabaseErrorTag
+
+
+class DatabaseError(UniversalBaseModel):
+    message: str
+    cause: typing.Optional[typing.Optional[typing.Any]] = None
+    tag: DatabaseErrorTag
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

mirascope/api/_generated/types/database_error_tag.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+DatabaseErrorTag = typing.Union[typing.Literal["DatabaseError"], typing.Any]

mirascope/api/_generated/types/http_api_decode_error.py

@@ -3,9 +3,7 @@
 import typing
 
 import pydantic
-import typing_extensions
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
-from ..core.serialization import FieldMetadata
 from .http_api_decode_error_tag import HttpApiDecodeErrorTag
 from .issue import Issue
 
@@ -17,7 +15,7 @@ class HttpApiDecodeError(UniversalBaseModel):
 
     issues: typing.List[Issue]
     message: str
-    tag: typing_extensions.Annotated[HttpApiDecodeErrorTag, FieldMetadata(alias="_tag")]
+    tag: HttpApiDecodeErrorTag
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(

mirascope/api/_generated/types/issue.py

@@ -3,9 +3,7 @@
 import typing
 
 import pydantic
-import typing_extensions
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
-from ..core.serialization import FieldMetadata
 from .issue_tag import IssueTag
 from .property_key import PropertyKey
 
@@ -15,9 +13,7 @@ class Issue(UniversalBaseModel):
     Represents an error encountered while parsing a value to match the schema
     """
 
-    tag: typing_extensions.Annotated[IssueTag, FieldMetadata(alias="_tag")] = (
-        pydantic.Field()
-    )
+    tag: IssueTag = pydantic.Field()
     """
     The tag identifying the type of parse issue
     """

mirascope/api/_generated/types/not_found_error_body.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .not_found_error_tag import NotFoundErrorTag
+
+
+class NotFoundErrorBody(UniversalBaseModel):
+    message: str
+    resource: typing.Optional[str] = None
+    tag: NotFoundErrorTag
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

mirascope/api/_generated/types/not_found_error_tag.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+NotFoundErrorTag = typing.Union[typing.Literal["NotFoundError"], typing.Any]

mirascope/api/_generated/types/permission_denied_error.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .permission_denied_error_tag import PermissionDeniedErrorTag
+
+
+class PermissionDeniedError(UniversalBaseModel):
+    message: str
+    resource: typing.Optional[str] = None
+    tag: PermissionDeniedErrorTag
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

mirascope/api/_generated/types/permission_denied_error_tag.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+PermissionDeniedErrorTag = typing.Union[
+    typing.Literal["PermissionDeniedError"], typing.Any
+]

mirascope/api/_generated/types/property_key.py

@@ -2,6 +2,6 @@
 
 import typing
 
-from .property_key_tag import PropertyKeyTag
+from .property_key_key import PropertyKeyKey
 
-PropertyKey = typing.Union[str, float, PropertyKeyTag]
+PropertyKey = typing.Union[str, float, PropertyKeyKey]

mirascope/api/_generated/types/{property_key_tag.py → property_key_key.py}

@@ -3,18 +3,16 @@
 import typing
 
 import pydantic
-import typing_extensions
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
-from ..core.serialization import FieldMetadata
-from .property_key_tag_tag import PropertyKeyTagTag
+from .property_key_key_tag import PropertyKeyKeyTag
 
 
-class PropertyKeyTag(UniversalBaseModel):
+class PropertyKeyKey(UniversalBaseModel):
     """
     an object to be decoded into a globally shared symbol
     """
 
-    tag: typing_extensions.Annotated[PropertyKeyTagTag, FieldMetadata(alias="_tag")]
+    tag: PropertyKeyKeyTag
     key: str
 
     if IS_PYDANTIC_V2:

mirascope/api/_generated/types/{property_key_tag_tag.py → property_key_key_tag.py}

@@ -2,4 +2,4 @@
 
 import typing
 
-PropertyKeyTagTag = typing.Union[typing.Literal["symbol"], typing.Any]
+PropertyKeyKeyTag = typing.Union[typing.Literal["symbol"], typing.Any]

mirascope/llm/__init__.py

@@ -92,8 +92,8 @@ from .providers import (
     Params,
     Provider,
     ProviderId,
-    load_provider,
     register_provider,
+    reset_provider_registry,
 )
 from .responses import (
     AsyncChunkIterator,
@@ -117,6 +117,8 @@ from .responses import (
     TextStream,
     ThoughtStream,
     ToolCallStream,
+    Usage,
+    UsageDeltaChunk,
 )
 from .tools import (
     AsyncContextTool,
@@ -219,6 +221,8 @@ __all__ = [
     "ToolOutput",
     "Toolkit",
     "URLImageSource",
+    "Usage",
+    "UsageDeltaChunk",
     "UserContent",
     "UserContentPart",
     "UserMessage",
@@ -228,7 +232,6 @@ __all__ = [
     "exceptions",
     "format",
     "formatting",
-    "load_provider",
     "mcp",
     "messages",
     "model",
@@ -238,6 +241,7 @@ __all__ = [
     "prompts",
     "providers",
     "register_provider",
+    "reset_provider_registry",
     "responses",
     "tool",
     "tools",

mirascope/llm/exceptions.py

@@ -11,6 +11,7 @@ class MirascopeLLMError(Exception):
     """Base exception for all Mirascope LLM errors."""
 
     original_exception: Exception | None
+    provider: "ProviderId | None"
 
 
 class APIError(MirascopeLLMError):
@@ -18,6 +19,10 @@ class APIError(MirascopeLLMError):
 
     status_code: int | None
 
+    def __init__(self, message: str, status_code: int | None = None) -> None:
+        super().__init__(message)
+        self.status_code = status_code
+
 
 class ConnectionError(MirascopeLLMError):
     """Raised when unable to connect to the API (network issues, timeouts)."""
@@ -26,18 +31,30 @@ class ConnectionError(MirascopeLLMError):
 class AuthenticationError(APIError):
     """Raised for authentication failures (401, invalid API keys)."""
 
+    def __init__(self, message: str, status_code: int | None = None) -> None:
+        super().__init__(message, status_code=status_code or 401)
+
 
 class PermissionError(APIError):
     """Raised for permission/authorization failures (403)."""
 
+    def __init__(self, message: str, status_code: int | None = None) -> None:
+        super().__init__(message, status_code=status_code or 403)
+
 
 class BadRequestError(APIError):
     """Raised for malformed requests (400, 422)."""
 
+    def __init__(self, message: str, status_code: int | None = None) -> None:
+        super().__init__(message, status_code=status_code or 400)
+
 
 class NotFoundError(APIError):
     """Raised when requested resource is not found (404)."""
 
+    def __init__(self, message: str, status_code: int | None = None) -> None:
+        super().__init__(message, status_code=status_code or 404)
+
 
 class ToolNotFoundError(MirascopeLLMError):
     """Raised if a tool_call cannot be converted to any corresponding tool."""
@@ -96,15 +113,26 @@ class FormattingModeNotSupportedError(FeatureNotSupportedError):
 class RateLimitError(APIError):
     """Raised when rate limits are exceeded (429)."""
 
+    def __init__(self, message: str, status_code: int | None = None) -> None:
+        super().__init__(message, status_code=status_code or 429)
+
 
 class ServerError(APIError):
     """Raised for server-side errors (500+)."""
 
+    def __init__(self, message: str, status_code: int | None = None) -> None:
+        super().__init__(message, status_code=status_code or 500)
+
 
 class TimeoutError(MirascopeLLMError):
     """Raised when requests timeout or deadline exceeded."""
 
 
+# This wraps the APIResponseValidationErrors that OpenAI and Anthropic both return.
+class ResponseValidationError(MirascopeLLMError):
+    """Raised when API response fails validation."""
+
+
 class NoRegisteredProviderError(MirascopeLLMError):
     """Raised when no provider is registered for a given model_id."""
 

mirascope/llm/providers/__init__.py

@@ -6,33 +6,41 @@ from .anthropic import (
 )
 from .base import BaseProvider, Params, Provider
 from .google import GoogleModelId, GoogleProvider
-from .load_provider import load, load_provider
 from .mlx import MLXModelId, MLXProvider
 from .model_id import ModelId
+from .ollama import OllamaProvider
 from .openai import (
     OpenAIModelId,
     OpenAIProvider,
 )
+from .openai.completions import BaseOpenAICompletionsProvider
 from .provider_id import KNOWN_PROVIDER_IDS, ProviderId
-from .provider_registry import get_provider_for_model, register_provider
+from .provider_registry import (
+    get_provider_for_model,
+    register_provider,
+    reset_provider_registry,
+)
+from .together import TogetherProvider
 
 __all__ = [
     "KNOWN_PROVIDER_IDS",
     "AnthropicModelId",
     "AnthropicProvider",
+    "BaseOpenAICompletionsProvider",
     "BaseProvider",
     "GoogleModelId",
     "GoogleProvider",
     "MLXModelId",
     "MLXProvider",
     "ModelId",
+    "OllamaProvider",
     "OpenAIModelId",
     "OpenAIProvider",
     "Params",
     "Provider",
     "ProviderId",
+    "TogetherProvider",
     "get_provider_for_model",
-    "load",
-    "load_provider",
     "register_provider",
+    "reset_provider_registry",
 ]

mirascope/llm/providers/anthropic/__init__.py

@@ -3,22 +3,27 @@
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
+    from .beta_provider import AnthropicBetaProvider
     from .model_id import AnthropicModelId
     from .provider import AnthropicProvider
 else:
     try:
+        from .beta_provider import AnthropicBetaProvider
         from .model_id import AnthropicModelId
         from .provider import AnthropicProvider
     except ImportError:  # pragma: no cover
         from .._missing_import_stubs import (
-            create_import_error_stub,
             create_provider_stub,
         )
 
+        AnthropicBetaProvider = create_provider_stub(
+            "anthropic", "AnthropicBetaProvider"
+        )
         AnthropicProvider = create_provider_stub("anthropic", "AnthropicProvider")
         AnthropicModelId = str
 
 __all__ = [
+    "AnthropicBetaProvider",
     "AnthropicModelId",
     "AnthropicProvider",
 ]

mirascope/llm/providers/anthropic/_utils/__init__.py

@@ -1,13 +1,25 @@
-from .decode import (
-    decode_async_stream,
-    decode_response,
-    decode_stream,
+"""Shared Anthropic utilities."""
+
+from .decode import decode_async_stream, decode_response, decode_stream
+from .encode import (
+    DEFAULT_FORMAT_MODE,
+    DEFAULT_MAX_TOKENS,
+    AnthropicImageMimeType,
+    encode_image_mime_type,
+    encode_request,
+    process_params,
 )
-from .encode import encode_request
+from .errors import ANTHROPIC_ERROR_MAP
 
 __all__ = [
+    "ANTHROPIC_ERROR_MAP",
+    "DEFAULT_FORMAT_MODE",
+    "DEFAULT_MAX_TOKENS",
+    "AnthropicImageMimeType",
     "decode_async_stream",
     "decode_response",
     "decode_stream",
+    "encode_image_mime_type",
     "encode_request",
+    "process_params",
 ]