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

mirascope/_utils.py

@@ -0,0 +1,34 @@
+"""Shared internal utilities for mirascope."""
+
+from typing import Any
+
+# Attributes to copy from wrapped functions (matches functools.WRAPPER_ASSIGNMENTS)
+WRAPPER_ASSIGNMENTS = (
+    "__module__",
+    "__name__",
+    "__qualname__",
+    "__annotations__",
+    "__doc__",
+)
+
+
+def copy_function_metadata(target: Any, source: Any) -> None:  # noqa: ANN401
+    """Copy standard function metadata from source to target.
+
+    Copies __module__, __name__, __qualname__, __annotations__, __doc__
+    from source to target, and sets __wrapped__ to source.
+
+    This enables decorator stacking by preserving the original function's
+    metadata on wrapper objects.
+
+    Args:
+        target: The wrapper object to copy metadata to
+        source: The original function to copy metadata from
+    """
+    for attr in WRAPPER_ASSIGNMENTS:
+        try:
+            value = getattr(source, attr)
+            object.__setattr__(target, attr, value)
+        except AttributeError:
+            pass
+    object.__setattr__(target, "__wrapped__", source)

mirascope/api/_generated/__init__.py

@@ -5,14 +5,19 @@
 from .types import (
     AlreadyExistsError,
     AlreadyExistsErrorTag,
+    BadRequestErrorBody,
+    BadRequestErrorBody_HttpApiDecodeError,
+    BadRequestErrorBody_ImmutableResourceError,
     ClickHouseError,
     DatabaseError,
     DatabaseErrorTag,
+    Date,
     HttpApiDecodeError,
     HttpApiDecodeErrorTag,
+    ImmutableResourceError,
     InternalServerErrorBody,
+    InternalServerErrorBody_ClickHouseError,
     InternalServerErrorBody_DatabaseError,
-    InternalServerErrorBody_StripeError,
     Issue,
     IssueTag,
     NotFoundErrorBody,
@@ -20,10 +25,19 @@ from .types import (
     NumberFromString,
     PermissionDeniedError,
     PermissionDeniedErrorTag,
+    PlanLimitExceededError,
+    PlanLimitExceededErrorTag,
+    PricingUnavailableError,
     PropertyKey,
     PropertyKeyKey,
     PropertyKeyKeyTag,
+    RateLimitError,
+    RateLimitErrorTag,
+    ServiceUnavailableErrorBody,
+    ServiceUnavailableErrorTag,
     StripeError,
+    SubscriptionPastDueError,
+    SubscriptionPastDueErrorTag,
     UnauthorizedErrorBody,
     UnauthorizedErrorTag,
 )
@@ -33,6 +47,9 @@ from .errors import (
     ForbiddenError,
     InternalServerError,
     NotFoundError,
+    PaymentRequiredError,
+    ServiceUnavailableError,
+    TooManyRequestsError,
     UnauthorizedError,
 )
 from . import (
@@ -42,8 +59,13 @@ from . import (
     environments,
     functions,
     health,
+    organization_invitations,
+    organization_memberships,
     organizations,
+    project_memberships,
     projects,
+    tags,
+    token_cost,
     traces,
 )
 from .annotations import (
@@ -60,11 +82,19 @@ from .annotations import (
     AnnotationsUpdateResponse,
     AnnotationsUpdateResponseLabel,
 )
-from .api_keys import ApiKeysCreateResponse, ApiKeysGetResponse, ApiKeysListResponseItem
+from .api_keys import (
+    ApiKeysCreateResponse,
+    ApiKeysGetResponse,
+    ApiKeysListAllForOrgResponseItem,
+    ApiKeysListResponseItem,
+)
 from .client import AsyncMirascope, Mirascope
 from .environment import MirascopeEnvironment
 from .environments import (
     EnvironmentsCreateResponse,
+    EnvironmentsGetAnalyticsResponse,
+    EnvironmentsGetAnalyticsResponseTopFunctionsItem,
+    EnvironmentsGetAnalyticsResponseTopModelsItem,
     EnvironmentsGetResponse,
     EnvironmentsListResponseItem,
     EnvironmentsUpdateResponse,
@@ -75,23 +105,71 @@ from .functions import (
     FunctionsCreateResponseDependenciesValue,
     FunctionsFindByHashResponse,
     FunctionsFindByHashResponseDependenciesValue,
+    FunctionsGetByEnvResponse,
+    FunctionsGetByEnvResponseDependenciesValue,
     FunctionsGetResponse,
     FunctionsGetResponseDependenciesValue,
+    FunctionsListByEnvResponse,
+    FunctionsListByEnvResponseFunctionsItem,
+    FunctionsListByEnvResponseFunctionsItemDependenciesValue,
     FunctionsListResponse,
     FunctionsListResponseFunctionsItem,
     FunctionsListResponseFunctionsItemDependenciesValue,
 )
 from .health import HealthCheckResponse, HealthCheckResponseStatus
+from .organization_invitations import (
+    OrganizationInvitationsAcceptResponse,
+    OrganizationInvitationsAcceptResponseRole,
+    OrganizationInvitationsCreateRequestRole,
+    OrganizationInvitationsCreateResponse,
+    OrganizationInvitationsCreateResponseRole,
+    OrganizationInvitationsCreateResponseStatus,
+    OrganizationInvitationsGetResponse,
+    OrganizationInvitationsGetResponseRole,
+    OrganizationInvitationsGetResponseStatus,
+    OrganizationInvitationsListResponseItem,
+    OrganizationInvitationsListResponseItemRole,
+    OrganizationInvitationsListResponseItemStatus,
+)
+from .organization_memberships import (
+    OrganizationMembershipsListResponseItem,
+    OrganizationMembershipsListResponseItemRole,
+    OrganizationMembershipsUpdateRequestRole,
+    OrganizationMembershipsUpdateResponse,
+    OrganizationMembershipsUpdateResponseRole,
+)
 from .organizations import (
+    OrganizationsCreatePaymentIntentResponse,
     OrganizationsCreateResponse,
     OrganizationsCreateResponseRole,
-    OrganizationsCreditsResponse,
     OrganizationsGetResponse,
     OrganizationsGetResponseRole,
     OrganizationsListResponseItem,
     OrganizationsListResponseItemRole,
+    OrganizationsPreviewSubscriptionChangeRequestTargetPlan,
+    OrganizationsPreviewSubscriptionChangeResponse,
+    OrganizationsPreviewSubscriptionChangeResponseValidationErrorsItem,
+    OrganizationsPreviewSubscriptionChangeResponseValidationErrorsItemResource,
+    OrganizationsRouterBalanceResponse,
+    OrganizationsSubscriptionResponse,
+    OrganizationsSubscriptionResponseCurrentPlan,
+    OrganizationsSubscriptionResponsePaymentMethod,
+    OrganizationsSubscriptionResponseScheduledChange,
+    OrganizationsSubscriptionResponseScheduledChangeTargetPlan,
     OrganizationsUpdateResponse,
     OrganizationsUpdateResponseRole,
+    OrganizationsUpdateSubscriptionRequestTargetPlan,
+    OrganizationsUpdateSubscriptionResponse,
+)
+from .project_memberships import (
+    ProjectMembershipsCreateRequestRole,
+    ProjectMembershipsCreateResponse,
+    ProjectMembershipsCreateResponseRole,
+    ProjectMembershipsListResponseItem,
+    ProjectMembershipsListResponseItemRole,
+    ProjectMembershipsUpdateRequestRole,
+    ProjectMembershipsUpdateResponse,
+    ProjectMembershipsUpdateResponseRole,
 )
 from .projects import (
     ProjectsCreateResponse,
@@ -99,6 +177,14 @@ from .projects import (
     ProjectsListResponseItem,
     ProjectsUpdateResponse,
 )
+from .tags import (
+    TagsCreateResponse,
+    TagsGetResponse,
+    TagsListResponse,
+    TagsListResponseTagsItem,
+    TagsUpdateResponse,
+)
+from .token_cost import TokenCostCalculateRequestUsage, TokenCostCalculateResponse
 from .traces import (
     TracesCreateRequestResourceSpansItem,
     TracesCreateRequestResourceSpansItemResource,
@@ -126,8 +212,18 @@ from .traces import (
     TracesGetAnalyticsSummaryResponse,
     TracesGetAnalyticsSummaryResponseTopFunctionsItem,
     TracesGetAnalyticsSummaryResponseTopModelsItem,
+    TracesGetTraceDetailByEnvResponse,
+    TracesGetTraceDetailByEnvResponseSpansItem,
     TracesGetTraceDetailResponse,
     TracesGetTraceDetailResponseSpansItem,
+    TracesListByFunctionHashResponse,
+    TracesListByFunctionHashResponseTracesItem,
+    TracesSearchByEnvRequestAttributeFiltersItem,
+    TracesSearchByEnvRequestAttributeFiltersItemOperator,
+    TracesSearchByEnvRequestSortBy,
+    TracesSearchByEnvRequestSortOrder,
+    TracesSearchByEnvResponse,
+    TracesSearchByEnvResponseSpansItem,
     TracesSearchRequestAttributeFiltersItem,
     TracesSearchRequestAttributeFiltersItemOperator,
     TracesSearchRequestSortBy,
@@ -153,14 +249,22 @@ __all__ = [
     "AnnotationsUpdateResponseLabel",
     "ApiKeysCreateResponse",
     "ApiKeysGetResponse",
+    "ApiKeysListAllForOrgResponseItem",
     "ApiKeysListResponseItem",
     "AsyncMirascope",
     "BadRequestError",
+    "BadRequestErrorBody",
+    "BadRequestErrorBody_HttpApiDecodeError",
+    "BadRequestErrorBody_ImmutableResourceError",
     "ClickHouseError",
     "ConflictError",
     "DatabaseError",
     "DatabaseErrorTag",
+    "Date",
     "EnvironmentsCreateResponse",
+    "EnvironmentsGetAnalyticsResponse",
+    "EnvironmentsGetAnalyticsResponseTopFunctionsItem",
+    "EnvironmentsGetAnalyticsResponseTopModelsItem",
     "EnvironmentsGetResponse",
     "EnvironmentsListResponseItem",
     "EnvironmentsUpdateResponse",
@@ -170,8 +274,13 @@ __all__ = [
     "FunctionsCreateResponseDependenciesValue",
     "FunctionsFindByHashResponse",
     "FunctionsFindByHashResponseDependenciesValue",
+    "FunctionsGetByEnvResponse",
+    "FunctionsGetByEnvResponseDependenciesValue",
     "FunctionsGetResponse",
     "FunctionsGetResponseDependenciesValue",
+    "FunctionsListByEnvResponse",
+    "FunctionsListByEnvResponseFunctionsItem",
+    "FunctionsListByEnvResponseFunctionsItemDependenciesValue",
     "FunctionsListResponse",
     "FunctionsListResponseFunctionsItem",
     "FunctionsListResponseFunctionsItemDependenciesValue",
@@ -179,10 +288,11 @@ __all__ = [
     "HealthCheckResponseStatus",
     "HttpApiDecodeError",
     "HttpApiDecodeErrorTag",
+    "ImmutableResourceError",
     "InternalServerError",
     "InternalServerErrorBody",
+    "InternalServerErrorBody_ClickHouseError",
     "InternalServerErrorBody_DatabaseError",
-    "InternalServerErrorBody_StripeError",
     "Issue",
     "IssueTag",
     "Mirascope",
@@ -191,17 +301,58 @@ __all__ = [
     "NotFoundErrorBody",
     "NotFoundErrorTag",
     "NumberFromString",
+    "OrganizationInvitationsAcceptResponse",
+    "OrganizationInvitationsAcceptResponseRole",
+    "OrganizationInvitationsCreateRequestRole",
+    "OrganizationInvitationsCreateResponse",
+    "OrganizationInvitationsCreateResponseRole",
+    "OrganizationInvitationsCreateResponseStatus",
+    "OrganizationInvitationsGetResponse",
+    "OrganizationInvitationsGetResponseRole",
+    "OrganizationInvitationsGetResponseStatus",
+    "OrganizationInvitationsListResponseItem",
+    "OrganizationInvitationsListResponseItemRole",
+    "OrganizationInvitationsListResponseItemStatus",
+    "OrganizationMembershipsListResponseItem",
+    "OrganizationMembershipsListResponseItemRole",
+    "OrganizationMembershipsUpdateRequestRole",
+    "OrganizationMembershipsUpdateResponse",
+    "OrganizationMembershipsUpdateResponseRole",
+    "OrganizationsCreatePaymentIntentResponse",
     "OrganizationsCreateResponse",
     "OrganizationsCreateResponseRole",
-    "OrganizationsCreditsResponse",
     "OrganizationsGetResponse",
     "OrganizationsGetResponseRole",
     "OrganizationsListResponseItem",
     "OrganizationsListResponseItemRole",
+    "OrganizationsPreviewSubscriptionChangeRequestTargetPlan",
+    "OrganizationsPreviewSubscriptionChangeResponse",
+    "OrganizationsPreviewSubscriptionChangeResponseValidationErrorsItem",
+    "OrganizationsPreviewSubscriptionChangeResponseValidationErrorsItemResource",
+    "OrganizationsRouterBalanceResponse",
+    "OrganizationsSubscriptionResponse",
+    "OrganizationsSubscriptionResponseCurrentPlan",
+    "OrganizationsSubscriptionResponsePaymentMethod",
+    "OrganizationsSubscriptionResponseScheduledChange",
+    "OrganizationsSubscriptionResponseScheduledChangeTargetPlan",
     "OrganizationsUpdateResponse",
     "OrganizationsUpdateResponseRole",
+    "OrganizationsUpdateSubscriptionRequestTargetPlan",
+    "OrganizationsUpdateSubscriptionResponse",
+    "PaymentRequiredError",
     "PermissionDeniedError",
     "PermissionDeniedErrorTag",
+    "PlanLimitExceededError",
+    "PlanLimitExceededErrorTag",
+    "PricingUnavailableError",
+    "ProjectMembershipsCreateRequestRole",
+    "ProjectMembershipsCreateResponse",
+    "ProjectMembershipsCreateResponseRole",
+    "ProjectMembershipsListResponseItem",
+    "ProjectMembershipsListResponseItemRole",
+    "ProjectMembershipsUpdateRequestRole",
+    "ProjectMembershipsUpdateResponse",
+    "ProjectMembershipsUpdateResponseRole",
     "ProjectsCreateResponse",
     "ProjectsGetResponse",
     "ProjectsListResponseItem",
@@ -209,7 +360,22 @@ __all__ = [
     "PropertyKey",
     "PropertyKeyKey",
     "PropertyKeyKeyTag",
+    "RateLimitError",
+    "RateLimitErrorTag",
+    "ServiceUnavailableError",
+    "ServiceUnavailableErrorBody",
+    "ServiceUnavailableErrorTag",
     "StripeError",
+    "SubscriptionPastDueError",
+    "SubscriptionPastDueErrorTag",
+    "TagsCreateResponse",
+    "TagsGetResponse",
+    "TagsListResponse",
+    "TagsListResponseTagsItem",
+    "TagsUpdateResponse",
+    "TokenCostCalculateRequestUsage",
+    "TokenCostCalculateResponse",
+    "TooManyRequestsError",
     "TracesCreateRequestResourceSpansItem",
     "TracesCreateRequestResourceSpansItemResource",
     "TracesCreateRequestResourceSpansItemResourceAttributesItem",
@@ -236,8 +402,18 @@ __all__ = [
     "TracesGetAnalyticsSummaryResponse",
     "TracesGetAnalyticsSummaryResponseTopFunctionsItem",
     "TracesGetAnalyticsSummaryResponseTopModelsItem",
+    "TracesGetTraceDetailByEnvResponse",
+    "TracesGetTraceDetailByEnvResponseSpansItem",
     "TracesGetTraceDetailResponse",
     "TracesGetTraceDetailResponseSpansItem",
+    "TracesListByFunctionHashResponse",
+    "TracesListByFunctionHashResponseTracesItem",
+    "TracesSearchByEnvRequestAttributeFiltersItem",
+    "TracesSearchByEnvRequestAttributeFiltersItemOperator",
+    "TracesSearchByEnvRequestSortBy",
+    "TracesSearchByEnvRequestSortOrder",
+    "TracesSearchByEnvResponse",
+    "TracesSearchByEnvResponseSpansItem",
     "TracesSearchRequestAttributeFiltersItem",
     "TracesSearchRequestAttributeFiltersItemOperator",
     "TracesSearchRequestSortBy",
@@ -253,7 +429,12 @@ __all__ = [
     "environments",
     "functions",
     "health",
+    "organization_invitations",
+    "organization_memberships",
     "organizations",
+    "project_memberships",
     "projects",
+    "tags",
+    "token_cost",
     "traces",
 ]

mirascope/api/_generated/annotations/client.py

@@ -89,6 +89,7 @@ class AnnotationsClient:
         label: typing.Optional[AnnotationsCreateRequestLabel] = OMIT,
         reasoning: typing.Optional[str] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AnnotationsCreateResponse:
         """
@@ -104,6 +105,8 @@ class AnnotationsClient:
 
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
 
+        tags : typing.Optional[typing.Sequence[str]]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -128,11 +131,14 @@ class AnnotationsClient:
             label=label,
             reasoning=reasoning,
             metadata=metadata,
+            tags=tags,
             request_options=request_options,
         )
         return _response.data
 
-    def get(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> AnnotationsGetResponse:
+    def get(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AnnotationsGetResponse:
         """
         Parameters
         ----------
@@ -165,6 +171,7 @@ class AnnotationsClient:
         label: typing.Optional[AnnotationsUpdateRequestLabel] = OMIT,
         reasoning: typing.Optional[str] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AnnotationsUpdateResponse:
         """
@@ -178,6 +185,8 @@ class AnnotationsClient:
 
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
 
+        tags : typing.Optional[typing.Sequence[str]]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -196,11 +205,18 @@ class AnnotationsClient:
         )
         """
         _response = self._raw_client.update(
-            id, label=label, reasoning=reasoning, metadata=metadata, request_options=request_options
+            id,
+            label=label,
+            reasoning=reasoning,
+            metadata=metadata,
+            tags=tags,
+            request_options=request_options,
         )
         return _response.data
 
-    def delete(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+    def delete(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
         """
         Parameters
         ----------
@@ -305,6 +321,7 @@ class AsyncAnnotationsClient:
         label: typing.Optional[AnnotationsCreateRequestLabel] = OMIT,
         reasoning: typing.Optional[str] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AnnotationsCreateResponse:
         """
@@ -320,6 +337,8 @@ class AsyncAnnotationsClient:
 
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
 
+        tags : typing.Optional[typing.Sequence[str]]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -352,11 +371,14 @@ class AsyncAnnotationsClient:
             label=label,
             reasoning=reasoning,
             metadata=metadata,
+            tags=tags,
             request_options=request_options,
         )
         return _response.data
 
-    async def get(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> AnnotationsGetResponse:
+    async def get(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AnnotationsGetResponse:
         """
         Parameters
         ----------
@@ -397,6 +419,7 @@ class AsyncAnnotationsClient:
         label: typing.Optional[AnnotationsUpdateRequestLabel] = OMIT,
         reasoning: typing.Optional[str] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AnnotationsUpdateResponse:
         """
@@ -410,6 +433,8 @@ class AsyncAnnotationsClient:
 
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
 
+        tags : typing.Optional[typing.Sequence[str]]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -436,11 +461,18 @@ class AsyncAnnotationsClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.update(
-            id, label=label, reasoning=reasoning, metadata=metadata, request_options=request_options
+            id,
+            label=label,
+            reasoning=reasoning,
+            metadata=metadata,
+            tags=tags,
+            request_options=request_options,
         )
         return _response.data
 
-    async def delete(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+    async def delete(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
         """
         Parameters
         ----------