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

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

@@ -4,6 +4,12 @@
 
 from .api_keys_create_response import ApiKeysCreateResponse
 from .api_keys_get_response import ApiKeysGetResponse
+from .api_keys_list_all_for_org_response_item import ApiKeysListAllForOrgResponseItem
 from .api_keys_list_response_item import ApiKeysListResponseItem
 
-__all__ = ["ApiKeysCreateResponse", "ApiKeysGetResponse", "ApiKeysListResponseItem"]
+__all__ = [
+    "ApiKeysCreateResponse",
+    "ApiKeysGetResponse",
+    "ApiKeysListAllForOrgResponseItem",
+    "ApiKeysListResponseItem",
+]

mirascope/api/_generated/api_keys/types/api_keys_list_all_for_org_response_item.py

@@ -0,0 +1,40 @@
+# 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 ApiKeysListAllForOrgResponseItem(UniversalBaseModel):
+    id: str
+    name: str
+    key_prefix: typing_extensions.Annotated[str, FieldMetadata(alias="keyPrefix")]
+    environment_id: typing_extensions.Annotated[
+        str, FieldMetadata(alias="environmentId")
+    ]
+    owner_id: typing_extensions.Annotated[str, FieldMetadata(alias="ownerId")]
+    created_at: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="createdAt")
+    ] = None
+    last_used_at: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="lastUsedAt")
+    ] = None
+    project_id: typing_extensions.Annotated[str, FieldMetadata(alias="projectId")]
+    project_name: typing_extensions.Annotated[str, FieldMetadata(alias="projectName")]
+    environment_name: typing_extensions.Annotated[
+        str, FieldMetadata(alias="environmentName")
+    ]
+
+    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/client.py

@@ -11,8 +11,22 @@ from .environment import MirascopeEnvironment
 from .environments.client import AsyncEnvironmentsClient, EnvironmentsClient
 from .functions.client import AsyncFunctionsClient, FunctionsClient
 from .health.client import AsyncHealthClient, HealthClient
+from .organization_invitations.client import (
+    AsyncOrganizationInvitationsClient,
+    OrganizationInvitationsClient,
+)
+from .organization_memberships.client import (
+    AsyncOrganizationMembershipsClient,
+    OrganizationMembershipsClient,
+)
 from .organizations.client import AsyncOrganizationsClient, OrganizationsClient
+from .project_memberships.client import (
+    AsyncProjectMembershipsClient,
+    ProjectMembershipsClient,
+)
 from .projects.client import AsyncProjectsClient, ProjectsClient
+from .tags.client import AsyncTagsClient, TagsClient
+from .token_cost.client import AsyncTokenCostClient, TokenCostClient
 from .traces.client import AsyncTracesClient, TracesClient
 
 
@@ -81,11 +95,22 @@ class Mirascope:
         self.traces = TracesClient(client_wrapper=self._client_wrapper)
         self.docs = DocsClient(client_wrapper=self._client_wrapper)
         self.organizations = OrganizationsClient(client_wrapper=self._client_wrapper)
+        self.organization_invitations = OrganizationInvitationsClient(
+            client_wrapper=self._client_wrapper
+        )
+        self.organization_memberships = OrganizationMembershipsClient(
+            client_wrapper=self._client_wrapper
+        )
         self.projects = ProjectsClient(client_wrapper=self._client_wrapper)
+        self.project_memberships = ProjectMembershipsClient(
+            client_wrapper=self._client_wrapper
+        )
         self.environments = EnvironmentsClient(client_wrapper=self._client_wrapper)
         self.api_keys = ApiKeysClient(client_wrapper=self._client_wrapper)
         self.functions = FunctionsClient(client_wrapper=self._client_wrapper)
         self.annotations = AnnotationsClient(client_wrapper=self._client_wrapper)
+        self.tags = TagsClient(client_wrapper=self._client_wrapper)
+        self.token_cost = TokenCostClient(client_wrapper=self._client_wrapper)
 
 
 class AsyncMirascope:
@@ -155,11 +180,22 @@ class AsyncMirascope:
         self.organizations = AsyncOrganizationsClient(
             client_wrapper=self._client_wrapper
         )
+        self.organization_invitations = AsyncOrganizationInvitationsClient(
+            client_wrapper=self._client_wrapper
+        )
+        self.organization_memberships = AsyncOrganizationMembershipsClient(
+            client_wrapper=self._client_wrapper
+        )
         self.projects = AsyncProjectsClient(client_wrapper=self._client_wrapper)
+        self.project_memberships = AsyncProjectMembershipsClient(
+            client_wrapper=self._client_wrapper
+        )
         self.environments = AsyncEnvironmentsClient(client_wrapper=self._client_wrapper)
         self.api_keys = AsyncApiKeysClient(client_wrapper=self._client_wrapper)
         self.functions = AsyncFunctionsClient(client_wrapper=self._client_wrapper)
         self.annotations = AsyncAnnotationsClient(client_wrapper=self._client_wrapper)
+        self.tags = AsyncTagsClient(client_wrapper=self._client_wrapper)
+        self.token_cost = AsyncTokenCostClient(client_wrapper=self._client_wrapper)
 
 
 def _get_base_url(

mirascope/api/_generated/docs/raw_client.py

@@ -9,7 +9,9 @@ from ..core.http_response import AsyncHttpResponse, HttpResponse
 from ..core.pydantic_utilities import parse_obj_as
 from ..core.request_options import RequestOptions
 from ..errors.bad_request_error import BadRequestError
-from ..types.http_api_decode_error import HttpApiDecodeError
+from ..errors.service_unavailable_error import ServiceUnavailableError
+from ..errors.too_many_requests_error import TooManyRequestsError
+from ..types.rate_limit_error import RateLimitError
 
 
 class RawDocsClient:
@@ -51,17 +53,47 @@ class RawDocsClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        HttpApiDecodeError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=HttpApiDecodeError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # 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)
+            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 AsyncRawDocsClient:
@@ -103,14 +135,44 @@ class AsyncRawDocsClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        HttpApiDecodeError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=HttpApiDecodeError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # 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)
+            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/environment.py

@@ -4,6 +4,6 @@ import enum
 
 
 class MirascopeEnvironment(enum.Enum):
-    PRODUCTION = "https://v2.mirascope.com/api/v0"
-    STAGING = "https://staging.mirascope.com/api/v0"
-    LOCAL = "http://localhost:3000/api/v0"
+    PRODUCTION = "https://mirascope.com/api/v2"
+    STAGING = "https://staging.mirascope.com/api/v2"
+    LOCAL = "http://localhost:3000/api/v2"

mirascope/api/_generated/environments/__init__.py

@@ -4,6 +4,9 @@
 
 from .types import (
     EnvironmentsCreateResponse,
+    EnvironmentsGetAnalyticsResponse,
+    EnvironmentsGetAnalyticsResponseTopFunctionsItem,
+    EnvironmentsGetAnalyticsResponseTopModelsItem,
     EnvironmentsGetResponse,
     EnvironmentsListResponseItem,
     EnvironmentsUpdateResponse,
@@ -11,6 +14,9 @@ from .types import (
 
 __all__ = [
     "EnvironmentsCreateResponse",
+    "EnvironmentsGetAnalyticsResponse",
+    "EnvironmentsGetAnalyticsResponseTopFunctionsItem",
+    "EnvironmentsGetAnalyticsResponseTopModelsItem",
     "EnvironmentsGetResponse",
     "EnvironmentsListResponseItem",
     "EnvironmentsUpdateResponse",

mirascope/api/_generated/environments/client.py

@@ -6,6 +6,7 @@ from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
 from .raw_client import AsyncRawEnvironmentsClient, RawEnvironmentsClient
 from .types.environments_create_response import EnvironmentsCreateResponse
+from .types.environments_get_analytics_response import EnvironmentsGetAnalyticsResponse
 from .types.environments_get_response import EnvironmentsGetResponse
 from .types.environments_list_response_item import EnvironmentsListResponseItem
 from .types.environments_update_response import EnvironmentsUpdateResponse
@@ -30,7 +31,11 @@ class EnvironmentsClient:
         return self._raw_client
 
     def list(
-        self, organization_id: str, project_id: str, *, request_options: typing.Optional[RequestOptions] = None
+        self,
+        organization_id: str,
+        project_id: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[EnvironmentsListResponseItem]:
         """
         Parameters
@@ -57,7 +62,9 @@ class EnvironmentsClient:
             project_id="projectId",
         )
         """
-        _response = self._raw_client.list(organization_id, project_id, request_options=request_options)
+        _response = self._raw_client.list(
+            organization_id, project_id, request_options=request_options
+        )
         return _response.data
 
     def create(
@@ -103,7 +110,11 @@ class EnvironmentsClient:
         )
         """
         _response = self._raw_client.create(
-            organization_id, project_id, name=name, slug=slug, request_options=request_options
+            organization_id,
+            project_id,
+            name=name,
+            slug=slug,
+            request_options=request_options,
         )
         return _response.data
 
@@ -143,7 +154,9 @@ class EnvironmentsClient:
             environment_id="environmentId",
         )
         """
-        _response = self._raw_client.get(organization_id, project_id, environment_id, request_options=request_options)
+        _response = self._raw_client.get(
+            organization_id, project_id, environment_id, request_options=request_options
+        )
         return _response.data
 
     def update(
@@ -191,7 +204,12 @@ class EnvironmentsClient:
         )
         """
         _response = self._raw_client.update(
-            organization_id, project_id, environment_id, name=name, slug=slug, request_options=request_options
+            organization_id,
+            project_id,
+            environment_id,
+            name=name,
+            slug=slug,
+            request_options=request_options,
         )
         return _response.data
 
@@ -235,6 +253,60 @@ class EnvironmentsClient:
         )
         return _response.data
 
+    def getanalytics(
+        self,
+        organization_id: str,
+        project_id: str,
+        environment_id: str,
+        *,
+        start_time: str,
+        end_time: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> EnvironmentsGetAnalyticsResponse:
+        """
+        Parameters
+        ----------
+        organization_id : str
+
+        project_id : str
+
+        environment_id : str
+
+        start_time : str
+
+        end_time : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        EnvironmentsGetAnalyticsResponse
+            Success
+
+        Examples
+        --------
+        from mirascope.api._generated import Mirascope
+
+        client = Mirascope()
+        client.environments.getanalytics(
+            organization_id="organizationId",
+            project_id="projectId",
+            environment_id="environmentId",
+            start_time="startTime",
+            end_time="endTime",
+        )
+        """
+        _response = self._raw_client.getanalytics(
+            organization_id,
+            project_id,
+            environment_id,
+            start_time=start_time,
+            end_time=end_time,
+            request_options=request_options,
+        )
+        return _response.data
+
 
 class AsyncEnvironmentsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -252,7 +324,11 @@ class AsyncEnvironmentsClient:
         return self._raw_client
 
     async def list(
-        self, organization_id: str, project_id: str, *, request_options: typing.Optional[RequestOptions] = None
+        self,
+        organization_id: str,
+        project_id: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[EnvironmentsListResponseItem]:
         """
         Parameters
@@ -287,7 +363,9 @@ class AsyncEnvironmentsClient:
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.list(organization_id, project_id, request_options=request_options)
+        _response = await self._raw_client.list(
+            organization_id, project_id, request_options=request_options
+        )
         return _response.data
 
     async def create(
@@ -341,7 +419,11 @@ class AsyncEnvironmentsClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.create(
-            organization_id, project_id, name=name, slug=slug, request_options=request_options
+            organization_id,
+            project_id,
+            name=name,
+            slug=slug,
+            request_options=request_options,
         )
         return _response.data
 
@@ -447,7 +529,12 @@ class AsyncEnvironmentsClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.update(
-            organization_id, project_id, environment_id, name=name, slug=slug, request_options=request_options
+            organization_id,
+            project_id,
+            environment_id,
+            name=name,
+            slug=slug,
+            request_options=request_options,
         )
         return _response.data
 
@@ -498,3 +585,65 @@ class AsyncEnvironmentsClient:
             organization_id, project_id, environment_id, request_options=request_options
         )
         return _response.data
+
+    async def getanalytics(
+        self,
+        organization_id: str,
+        project_id: str,
+        environment_id: str,
+        *,
+        start_time: str,
+        end_time: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> EnvironmentsGetAnalyticsResponse:
+        """
+        Parameters
+        ----------
+        organization_id : str
+
+        project_id : str
+
+        environment_id : str
+
+        start_time : str
+
+        end_time : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        EnvironmentsGetAnalyticsResponse
+            Success
+
+        Examples
+        --------
+        import asyncio
+
+        from mirascope.api._generated import AsyncMirascope
+
+        client = AsyncMirascope()
+
+
+        async def main() -> None:
+            await client.environments.getanalytics(
+                organization_id="organizationId",
+                project_id="projectId",
+                environment_id="environmentId",
+                start_time="startTime",
+                end_time="endTime",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.getanalytics(
+            organization_id,
+            project_id,
+            environment_id,
+            start_time=start_time,
+            end_time=end_time,
+            request_options=request_options,
+        )
+        return _response.data