mirascope 2.0.1__py3-none-any.whl → 2.1.0__py3-none-any.whl

mirascope/_stubs.py

@@ -1,7 +1,7 @@
 """Utilities for stubbing modules with missing optional dependencies."""
 
 import sys
-from collections.abc import MutableSequence
+from collections.abc import MutableSequence, Sequence
 from importlib.abc import Loader, MetaPathFinder
 from importlib.machinery import ModuleSpec
 from types import ModuleType
@@ -33,6 +33,9 @@ EXTRA_IMPORTS: dict[str, list[str]] = {
         "opentelemetry.propagators.jaeger",
         "libcst",
         "packaging",
+        "opentelemetry.instrumentation.openai_v2",
+        "opentelemetry.instrumentation.anthropic",
+        "opentelemetry.instrumentation.google_genai",
     ],
     "mlx": ["mlx_lm"],
 }
@@ -314,44 +317,62 @@ class _StubModule(ModuleType):
 
 def stub_module_if_missing(
     module_path: str,
-    package_name: str,
+    package_name: str | Sequence[str],
 ) -> bool:
-    """Check if all packages for an extra are installed; if not, install a stub module.
+    """Check if all packages for one or more extras are installed; if not, stub a module.
 
     This must be called BEFORE importing from the module.
 
     Args:
         module_path: Full module path to stub (e.g., 'mirascope.ops._internal.tracing')
-        package_name: The extra name (e.g., 'ops'). Must exist in EXTRA_IMPORTS.
+        package_name: The extra name (e.g., "ops") or a sequence of extra names.
+            If multiple extras are provided, the module is available only when
+            all of the extras have all imports available.
+            An empty sequence is a no-op and returns True (vacuously satisfied).
 
     Returns:
-        True if all packages for the extra are available, False if stubbed.
+        True if all packages for all extras are available (including empty sequence),
+        False if stubbed.
 
     Raises:
         KeyError: If package_name is not found in EXTRA_IMPORTS mapping.
     """
-    if package_name not in EXTRA_IMPORTS:
+    package_names: tuple[str, ...]
+    if isinstance(package_name, str):
+        package_names = (package_name,)
+    else:
+        package_names = tuple(package_name)
+
+    unknown = [name for name in package_names if name not in EXTRA_IMPORTS]
+    if unknown:
+        unknown_names = ", ".join(unknown)
         raise KeyError(
-            f"Unknown extra '{package_name}'. "
+            f"Unknown extra '{unknown_names}'. "
             f"Available extras: {', '.join(EXTRA_IMPORTS.keys())}"
         )
 
-    imports_to_check = EXTRA_IMPORTS[package_name]
+    def _extra_available(extra_name: str) -> bool:
+        imports_to_check = EXTRA_IMPORTS[extra_name]
+        for import_name in imports_to_check:
+            try:
+                __import__(import_name)
+            except ImportError:
+                return False
+        return True
 
-    # Check if ALL imports for this extra are available
-    all_available = True
-    for import_name in imports_to_check:
-        try:
-            __import__(import_name)
-        except ImportError:
-            all_available = False
+    # Find the first missing extra (short-circuit to avoid unnecessary imports)
+    first_missing_extra: str | None = None
+    for extra_name in package_names:
+        if not _extra_available(extra_name):
+            first_missing_extra = extra_name
             break
 
-    if all_available:
+    # If no extras are missing, all packages are available
+    if first_missing_extra is None:
         return True
 
-    # At least one import is missing, so stub the module
-    sys.modules[module_path] = _StubModule(module_path, package_name)
+    # Use the first missing extra for the error message
+    sys.modules[module_path] = _StubModule(module_path, first_missing_extra)
 
     # Register with the finder to handle nested imports
     _finder.register_stub(module_path)

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

@@ -165,6 +165,8 @@ from .project_memberships import (
     ProjectMembershipsCreateRequestRole,
     ProjectMembershipsCreateResponse,
     ProjectMembershipsCreateResponseRole,
+    ProjectMembershipsGetResponse,
+    ProjectMembershipsGetResponseRole,
     ProjectMembershipsListResponseItem,
     ProjectMembershipsListResponseItemRole,
     ProjectMembershipsUpdateRequestRole,
@@ -348,6 +350,8 @@ __all__ = [
     "ProjectMembershipsCreateRequestRole",
     "ProjectMembershipsCreateResponse",
     "ProjectMembershipsCreateResponseRole",
+    "ProjectMembershipsGetResponse",
+    "ProjectMembershipsGetResponseRole",
     "ProjectMembershipsListResponseItem",
     "ProjectMembershipsListResponseItemRole",
     "ProjectMembershipsUpdateRequestRole",

mirascope/api/_generated/organization_invitations/client.py

@@ -92,7 +92,7 @@ class OrganizationInvitationsClient:
         organization_id : str
 
         recipient_email : str
-            a string matching the pattern ^[^\s@]+@[^\s@]+\.[^\s@]+$
+            a string matching the pattern ^[^ \t\n\r\f\v@]+@[^ \t\n\r\f\v@]+[.][^ \t\n\r\f\v@]+$
 
         role : OrganizationInvitationsCreateRequestRole
 
@@ -335,7 +335,7 @@ class AsyncOrganizationInvitationsClient:
         organization_id : str
 
         recipient_email : str
-            a string matching the pattern ^[^\s@]+@[^\s@]+\.[^\s@]+$
+            a string matching the pattern ^[^ \t\n\r\f\v@]+@[^ \t\n\r\f\v@]+[.][^ \t\n\r\f\v@]+$
 
         role : OrganizationInvitationsCreateRequestRole
 

mirascope/api/_generated/organization_invitations/raw_client.py

@@ -172,7 +172,7 @@ class RawOrganizationInvitationsClient:
         organization_id : str
 
         recipient_email : str
-            a string matching the pattern ^[^\s@]+@[^\s@]+\.[^\s@]+$
+            a string matching the pattern ^[^ \t\n\r\f\v@]+@[^ \t\n\r\f\v@]+[.][^ \t\n\r\f\v@]+$
 
         role : OrganizationInvitationsCreateRequestRole
 
@@ -911,7 +911,7 @@ class AsyncRawOrganizationInvitationsClient:
         organization_id : str
 
         recipient_email : str
-            a string matching the pattern ^[^\s@]+@[^\s@]+\.[^\s@]+$
+            a string matching the pattern ^[^ \t\n\r\f\v@]+@[^ \t\n\r\f\v@]+[.][^ \t\n\r\f\v@]+$
 
         role : OrganizationInvitationsCreateRequestRole
 

mirascope/api/_generated/project_memberships/__init__.py

@@ -6,6 +6,8 @@ from .types import (
     ProjectMembershipsCreateRequestRole,
     ProjectMembershipsCreateResponse,
     ProjectMembershipsCreateResponseRole,
+    ProjectMembershipsGetResponse,
+    ProjectMembershipsGetResponseRole,
     ProjectMembershipsListResponseItem,
     ProjectMembershipsListResponseItemRole,
     ProjectMembershipsUpdateRequestRole,
@@ -17,6 +19,8 @@ __all__ = [
     "ProjectMembershipsCreateRequestRole",
     "ProjectMembershipsCreateResponse",
     "ProjectMembershipsCreateResponseRole",
+    "ProjectMembershipsGetResponse",
+    "ProjectMembershipsGetResponseRole",
     "ProjectMembershipsListResponseItem",
     "ProjectMembershipsListResponseItemRole",
     "ProjectMembershipsUpdateRequestRole",

mirascope/api/_generated/project_memberships/client.py

@@ -9,6 +9,7 @@ from .types.project_memberships_create_request_role import (
     ProjectMembershipsCreateRequestRole,
 )
 from .types.project_memberships_create_response import ProjectMembershipsCreateResponse
+from .types.project_memberships_get_response import ProjectMembershipsGetResponse
 from .types.project_memberships_list_response_item import (
     ProjectMembershipsListResponseItem,
 )
@@ -122,6 +123,47 @@ class ProjectMembershipsClient:
         )
         return _response.data
 
+    def get(
+        self,
+        organization_id: str,
+        project_id: str,
+        member_id: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ProjectMembershipsGetResponse:
+        """
+        Parameters
+        ----------
+        organization_id : str
+
+        project_id : str
+
+        member_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ProjectMembershipsGetResponse
+            Success
+
+        Examples
+        --------
+        from mirascope.api._generated import Mirascope
+
+        client = Mirascope()
+        client.project_memberships.get(
+            organization_id="organizationId",
+            project_id="projectId",
+            member_id="memberId",
+        )
+        """
+        _response = self._raw_client.get(
+            organization_id, project_id, member_id, request_options=request_options
+        )
+        return _response.data
+
     def delete(
         self,
         organization_id: str,
@@ -331,6 +373,55 @@ class AsyncProjectMembershipsClient:
         )
         return _response.data
 
+    async def get(
+        self,
+        organization_id: str,
+        project_id: str,
+        member_id: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ProjectMembershipsGetResponse:
+        """
+        Parameters
+        ----------
+        organization_id : str
+
+        project_id : str
+
+        member_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ProjectMembershipsGetResponse
+            Success
+
+        Examples
+        --------
+        import asyncio
+
+        from mirascope.api._generated import AsyncMirascope
+
+        client = AsyncMirascope()
+
+
+        async def main() -> None:
+            await client.project_memberships.get(
+                organization_id="organizationId",
+                project_id="projectId",
+                member_id="memberId",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get(
+            organization_id, project_id, member_id, request_options=request_options
+        )
+        return _response.data
+
     async def delete(
         self,
         organization_id: str,

mirascope/api/_generated/project_memberships/raw_client.py

@@ -23,6 +23,7 @@ from .types.project_memberships_create_request_role import (
     ProjectMembershipsCreateRequestRole,
 )
 from .types.project_memberships_create_response import ProjectMembershipsCreateResponse
+from .types.project_memberships_get_response import ProjectMembershipsGetResponse
 from .types.project_memberships_list_response_item import (
     ProjectMembershipsListResponseItem,
 )
@@ -296,6 +297,125 @@ class RawProjectMembershipsClient:
             body=_response_json,
         )
 
+    def get(
+        self,
+        organization_id: str,
+        project_id: str,
+        member_id: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[ProjectMembershipsGetResponse]:
+        """
+        Parameters
+        ----------
+        organization_id : str
+
+        project_id : str
+
+        member_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[ProjectMembershipsGetResponse]
+            Success
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(organization_id)}/projects/{jsonable_encoder(project_id)}/members/{jsonable_encoder(member_id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ProjectMembershipsGetResponse,
+                    parse_obj_as(
+                        type_=ProjectMembershipsGetResponse,  # 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 == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        PermissionDeniedError,
+                        parse_obj_as(
+                            type_=PermissionDeniedError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        NotFoundErrorBody,
+                        parse_obj_as(
+                            type_=NotFoundErrorBody,  # 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 == 500:
+                raise InternalServerError(
+                    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 == 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,
+        )
+
     def delete(
         self,
         organization_id: str,
@@ -798,6 +918,125 @@ class AsyncRawProjectMembershipsClient:
             body=_response_json,
         )
 
+    async def get(
+        self,
+        organization_id: str,
+        project_id: str,
+        member_id: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[ProjectMembershipsGetResponse]:
+        """
+        Parameters
+        ----------
+        organization_id : str
+
+        project_id : str
+
+        member_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[ProjectMembershipsGetResponse]
+            Success
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(organization_id)}/projects/{jsonable_encoder(project_id)}/members/{jsonable_encoder(member_id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ProjectMembershipsGetResponse,
+                    parse_obj_as(
+                        type_=ProjectMembershipsGetResponse,  # 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 == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        PermissionDeniedError,
+                        parse_obj_as(
+                            type_=PermissionDeniedError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        NotFoundErrorBody,
+                        parse_obj_as(
+                            type_=NotFoundErrorBody,  # 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 == 500:
+                raise InternalServerError(
+                    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 == 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,
+        )
+
     async def delete(
         self,
         organization_id: str,

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

@@ -7,6 +7,8 @@ from .project_memberships_create_response import ProjectMembershipsCreateRespons
 from .project_memberships_create_response_role import (
     ProjectMembershipsCreateResponseRole,
 )
+from .project_memberships_get_response import ProjectMembershipsGetResponse
+from .project_memberships_get_response_role import ProjectMembershipsGetResponseRole
 from .project_memberships_list_response_item import ProjectMembershipsListResponseItem
 from .project_memberships_list_response_item_role import (
     ProjectMembershipsListResponseItemRole,
@@ -21,6 +23,8 @@ __all__ = [
     "ProjectMembershipsCreateRequestRole",
     "ProjectMembershipsCreateResponse",
     "ProjectMembershipsCreateResponseRole",
+    "ProjectMembershipsGetResponse",
+    "ProjectMembershipsGetResponseRole",
     "ProjectMembershipsListResponseItem",
     "ProjectMembershipsListResponseItemRole",
     "ProjectMembershipsUpdateRequestRole",

mirascope/api/_generated/project_memberships/types/project_memberships_get_response.py

@@ -0,0 +1,33 @@
+# 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
+from ...types.date import Date
+from .project_memberships_get_response_role import ProjectMembershipsGetResponseRole
+
+
+class ProjectMembershipsGetResponse(UniversalBaseModel):
+    member_id: typing_extensions.Annotated[str, FieldMetadata(alias="memberId")]
+    organization_id: typing_extensions.Annotated[
+        str, FieldMetadata(alias="organizationId")
+    ]
+    project_id: typing_extensions.Annotated[str, FieldMetadata(alias="projectId")]
+    role: ProjectMembershipsGetResponseRole
+    created_at: typing_extensions.Annotated[
+        typing.Optional[Date], FieldMetadata(alias="createdAt")
+    ] = 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/project_memberships/types/project_memberships_get_response_role.py

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