mirascope 2.0.0a6__py3-none-any.whl → 2.0.2__py3-none-any.whl

mirascope/api/_generated/tags/types/__init__.py

@@ -0,0 +1,17 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .tags_create_response import TagsCreateResponse
+from .tags_get_response import TagsGetResponse
+from .tags_list_response import TagsListResponse
+from .tags_list_response_tags_item import TagsListResponseTagsItem
+from .tags_update_response import TagsUpdateResponse
+
+__all__ = [
+    "TagsCreateResponse",
+    "TagsGetResponse",
+    "TagsListResponse",
+    "TagsListResponseTagsItem",
+    "TagsUpdateResponse",
+]

mirascope/api/_generated/tags/types/tags_create_response.py

@@ -0,0 +1,41 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ...core.serialization import FieldMetadata
+
+
+class TagsCreateResponse(UniversalBaseModel):
+    id: str
+    name: str = pydantic.Field()
+    """
+    a string at most 100 character(s) long
+    """
+
+    project_id: typing_extensions.Annotated[str, FieldMetadata(alias="projectId")]
+    organization_id: typing_extensions.Annotated[
+        str, FieldMetadata(alias="organizationId")
+    ]
+    created_by: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="createdBy")
+    ] = None
+    created_at: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="createdAt")
+    ] = None
+    updated_at: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="updatedAt")
+    ] = None
+
+    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/tags/types/tags_get_response.py

@@ -0,0 +1,41 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ...core.serialization import FieldMetadata
+
+
+class TagsGetResponse(UniversalBaseModel):
+    id: str
+    name: str = pydantic.Field()
+    """
+    a string at most 100 character(s) long
+    """
+
+    project_id: typing_extensions.Annotated[str, FieldMetadata(alias="projectId")]
+    organization_id: typing_extensions.Annotated[
+        str, FieldMetadata(alias="organizationId")
+    ]
+    created_by: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="createdBy")
+    ] = None
+    created_at: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="createdAt")
+    ] = None
+    updated_at: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="updatedAt")
+    ] = None
+
+    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/tags/types/tags_list_response.py

@@ -0,0 +1,23 @@
+# 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 .tags_list_response_tags_item import TagsListResponseTagsItem
+
+
+class TagsListResponse(UniversalBaseModel):
+    tags: typing.List[TagsListResponseTagsItem]
+    total: float
+
+    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/tags/types/tags_list_response_tags_item.py

@@ -0,0 +1,41 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ...core.serialization import FieldMetadata
+
+
+class TagsListResponseTagsItem(UniversalBaseModel):
+    id: str
+    name: str = pydantic.Field()
+    """
+    a string at most 100 character(s) long
+    """
+
+    project_id: typing_extensions.Annotated[str, FieldMetadata(alias="projectId")]
+    organization_id: typing_extensions.Annotated[
+        str, FieldMetadata(alias="organizationId")
+    ]
+    created_by: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="createdBy")
+    ] = None
+    created_at: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="createdAt")
+    ] = None
+    updated_at: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="updatedAt")
+    ] = None
+
+    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/tags/types/tags_update_response.py

@@ -0,0 +1,41 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ...core.serialization import FieldMetadata
+
+
+class TagsUpdateResponse(UniversalBaseModel):
+    id: str
+    name: str = pydantic.Field()
+    """
+    a string at most 100 character(s) long
+    """
+
+    project_id: typing_extensions.Annotated[str, FieldMetadata(alias="projectId")]
+    organization_id: typing_extensions.Annotated[
+        str, FieldMetadata(alias="organizationId")
+    ]
+    created_by: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="createdBy")
+    ] = None
+    created_at: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="createdAt")
+    ] = None
+    updated_at: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="updatedAt")
+    ] = None
+
+    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/token_cost/__init__.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .types import TokenCostCalculateRequestUsage, TokenCostCalculateResponse
+
+__all__ = ["TokenCostCalculateRequestUsage", "TokenCostCalculateResponse"]

mirascope/api/_generated/token_cost/client.py

@@ -0,0 +1,160 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+from .raw_client import AsyncRawTokenCostClient, RawTokenCostClient
+from .types.token_cost_calculate_request_usage import TokenCostCalculateRequestUsage
+from .types.token_cost_calculate_response import TokenCostCalculateResponse
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class TokenCostClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawTokenCostClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawTokenCostClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawTokenCostClient
+        """
+        return self._raw_client
+
+    def calculate(
+        self,
+        *,
+        provider: str,
+        model: str,
+        usage: TokenCostCalculateRequestUsage,
+        via_router: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> TokenCostCalculateResponse:
+        """
+        Parameters
+        ----------
+        provider : str
+            a non empty string
+
+        model : str
+            a non empty string
+
+        usage : TokenCostCalculateRequestUsage
+
+        via_router : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        TokenCostCalculateResponse
+            Success
+
+        Examples
+        --------
+        from mirascope.api._generated import Mirascope
+        from mirascope.api._generated.token_cost import TokenCostCalculateRequestUsage
+
+        client = Mirascope()
+        client.token_cost.calculate(
+            provider="provider",
+            model="model",
+            usage=TokenCostCalculateRequestUsage(
+                input_tokens=1.1,
+                output_tokens=1.1,
+            ),
+        )
+        """
+        _response = self._raw_client.calculate(
+            provider=provider,
+            model=model,
+            usage=usage,
+            via_router=via_router,
+            request_options=request_options,
+        )
+        return _response.data
+
+
+class AsyncTokenCostClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawTokenCostClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawTokenCostClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawTokenCostClient
+        """
+        return self._raw_client
+
+    async def calculate(
+        self,
+        *,
+        provider: str,
+        model: str,
+        usage: TokenCostCalculateRequestUsage,
+        via_router: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> TokenCostCalculateResponse:
+        """
+        Parameters
+        ----------
+        provider : str
+            a non empty string
+
+        model : str
+            a non empty string
+
+        usage : TokenCostCalculateRequestUsage
+
+        via_router : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        TokenCostCalculateResponse
+            Success
+
+        Examples
+        --------
+        import asyncio
+
+        from mirascope.api._generated import AsyncMirascope
+        from mirascope.api._generated.token_cost import TokenCostCalculateRequestUsage
+
+        client = AsyncMirascope()
+
+
+        async def main() -> None:
+            await client.token_cost.calculate(
+                provider="provider",
+                model="model",
+                usage=TokenCostCalculateRequestUsage(
+                    input_tokens=1.1,
+                    output_tokens=1.1,
+                ),
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.calculate(
+            provider=provider,
+            model=model,
+            usage=usage,
+            via_router=via_router,
+            request_options=request_options,
+        )
+        return _response.data

mirascope/api/_generated/token_cost/raw_client.py

@@ -0,0 +1,264 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.http_response import AsyncHttpResponse, HttpResponse
+from ..core.pydantic_utilities import parse_obj_as
+from ..core.request_options import RequestOptions
+from ..core.serialization import convert_and_respect_annotation_metadata
+from ..errors.bad_request_error import BadRequestError
+from ..errors.service_unavailable_error import ServiceUnavailableError
+from ..errors.too_many_requests_error import TooManyRequestsError
+from ..errors.unauthorized_error import UnauthorizedError
+from ..types.rate_limit_error import RateLimitError
+from ..types.unauthorized_error_body import UnauthorizedErrorBody
+from .types.token_cost_calculate_request_usage import TokenCostCalculateRequestUsage
+from .types.token_cost_calculate_response import TokenCostCalculateResponse
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class RawTokenCostClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def calculate(
+        self,
+        *,
+        provider: str,
+        model: str,
+        usage: TokenCostCalculateRequestUsage,
+        via_router: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[TokenCostCalculateResponse]:
+        """
+        Parameters
+        ----------
+        provider : str
+            a non empty string
+
+        model : str
+            a non empty string
+
+        usage : TokenCostCalculateRequestUsage
+
+        via_router : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[TokenCostCalculateResponse]
+            Success
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "token-cost",
+            method="POST",
+            json={
+                "provider": provider,
+                "model": model,
+                "usage": convert_and_respect_annotation_metadata(
+                    object_=usage,
+                    annotation=TokenCostCalculateRequestUsage,
+                    direction="write",
+                ),
+                "viaRouter": via_router,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    TokenCostCalculateResponse,
+                    parse_obj_as(
+                        type_=TokenCostCalculateResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        UnauthorizedErrorBody,
+                        parse_obj_as(
+                            type_=UnauthorizedErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(
+                status_code=_response.status_code,
+                headers=dict(_response.headers),
+                body=_response.text,
+            )
+        raise ApiError(
+            status_code=_response.status_code,
+            headers=dict(_response.headers),
+            body=_response_json,
+        )
+
+
+class AsyncRawTokenCostClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def calculate(
+        self,
+        *,
+        provider: str,
+        model: str,
+        usage: TokenCostCalculateRequestUsage,
+        via_router: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[TokenCostCalculateResponse]:
+        """
+        Parameters
+        ----------
+        provider : str
+            a non empty string
+
+        model : str
+            a non empty string
+
+        usage : TokenCostCalculateRequestUsage
+
+        via_router : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[TokenCostCalculateResponse]
+            Success
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "token-cost",
+            method="POST",
+            json={
+                "provider": provider,
+                "model": model,
+                "usage": convert_and_respect_annotation_metadata(
+                    object_=usage,
+                    annotation=TokenCostCalculateRequestUsage,
+                    direction="write",
+                ),
+                "viaRouter": via_router,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    TokenCostCalculateResponse,
+                    parse_obj_as(
+                        type_=TokenCostCalculateResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        UnauthorizedErrorBody,
+                        parse_obj_as(
+                            type_=UnauthorizedErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(
+                status_code=_response.status_code,
+                headers=dict(_response.headers),
+                body=_response.text,
+            )
+        raise ApiError(
+            status_code=_response.status_code,
+            headers=dict(_response.headers),
+            body=_response_json,
+        )

mirascope/api/_generated/token_cost/types/__init__.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .token_cost_calculate_request_usage import TokenCostCalculateRequestUsage
+from .token_cost_calculate_response import TokenCostCalculateResponse
+
+__all__ = ["TokenCostCalculateRequestUsage", "TokenCostCalculateResponse"]