mirascope 2.0.1__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/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/reference.md

@@ -1682,7 +1682,7 @@ client.organization_invitations.create(
 <dl>
 <dd>
 
-**recipient_email:** `str` — a string matching the pattern ^[^\s@]+@[^\s@]+\.[^\s@]+$
+**recipient_email:** `str` — a string matching the pattern ^[^ \t\n\r\f\v@]+@[^ \t\n\r\f\v@]+[.][^ \t\n\r\f\v@]+$
     
 </dd>
 </dl>

mirascope/llm/calls/calls.py

@@ -1,8 +1,10 @@
 """The Call module for generating responses using LLMs."""
 
-from dataclasses import dataclass
-from typing import Generic, TypeVar, overload
+from collections.abc import Callable
+from dataclasses import dataclass, field
+from typing import Any, Generic, TypeVar, overload
 
+from ..._utils import copy_function_metadata
 from ..context import Context, DepsT
 from ..formatting import FormattableT
 from ..models import Model, use_model
@@ -12,6 +14,7 @@ from ..prompts import (
     ContextPrompt,
     Prompt,
 )
+from ..prompts.prompts import BasePrompt
 from ..responses import (
     AsyncContextResponse,
     AsyncContextStreamResponse,
@@ -24,24 +27,35 @@ from ..responses import (
 )
 from ..types import P
 
-CallT = TypeVar("CallT", bound="BaseCall")
+PromptT = TypeVar("PromptT", bound=BasePrompt[Callable[..., Any]])
+CallT = TypeVar("CallT", bound="BaseCall[Any]")
 
 
-@dataclass
-class BaseCall:
+@dataclass(kw_only=True)
+class BaseCall(Generic[PromptT]):
     """Base class for all Call types with shared model functionality."""
 
     default_model: Model
     """The default model that will be used if no model is set in context."""
 
+    prompt: PromptT
+    """The underlying Prompt instance that generates messages with tools and format."""
+
+    __name__: str = field(init=False, repr=False, default="")
+    """The name of the underlying function (preserved for decorator stacking)."""
+
     @property
     def model(self) -> Model:
         """The model used for generating responses. May be overwritten via `with llm.model(...)`."""
         return use_model(self.default_model)
 
+    def __post_init__(self) -> None:
+        """Preserve standard function attributes for decorator stacking."""
+        copy_function_metadata(self, self.prompt.fn)
+
 
 @dataclass
-class Call(BaseCall, Generic[P, FormattableT]):
+class Call(BaseCall[Prompt[P, FormattableT]], Generic[P, FormattableT]):
     """A call that directly generates LLM responses without requiring a model argument.
 
     Created by decorating a `MessageTemplate` with `llm.call`. The decorated function
@@ -53,9 +67,6 @@ class Call(BaseCall, Generic[P, FormattableT]):
     The model can be overridden at runtime using `with llm.model(...)` context manager.
     """
 
-    prompt: Prompt[P, FormattableT]
-    """The underlying Prompt instance that generates messages with tools and format."""
-
     @overload
     def __call__(
         self: "Call[P, None]", *args: P.args, **kwargs: P.kwargs
@@ -104,7 +115,7 @@ class Call(BaseCall, Generic[P, FormattableT]):
 
 
 @dataclass
-class AsyncCall(BaseCall, Generic[P, FormattableT]):
+class AsyncCall(BaseCall[AsyncPrompt[P, FormattableT]], Generic[P, FormattableT]):
     """An async call that directly generates LLM responses without requiring a model argument.
 
     Created by decorating an async `MessageTemplate` with `llm.call`. The decorated async
@@ -116,9 +127,6 @@ class AsyncCall(BaseCall, Generic[P, FormattableT]):
     The model can be overridden at runtime using `with llm.model(...)` context manager.
     """
 
-    prompt: AsyncPrompt[P, FormattableT]
-    """The underlying AsyncPrompt instance that generates messages with tools and format."""
-
     @overload
     async def __call__(
         self: "AsyncCall[P, None]", *args: P.args, **kwargs: P.kwargs
@@ -169,7 +177,9 @@ class AsyncCall(BaseCall, Generic[P, FormattableT]):
 
 
 @dataclass
-class ContextCall(BaseCall, Generic[P, DepsT, FormattableT]):
+class ContextCall(
+    BaseCall[ContextPrompt[P, DepsT, FormattableT]], Generic[P, DepsT, FormattableT]
+):
     """A context-aware call that directly generates LLM responses without requiring a model argument.
 
     Created by decorating a `ContextMessageTemplate` with `llm.call`. The decorated function
@@ -182,9 +192,6 @@ class ContextCall(BaseCall, Generic[P, DepsT, FormattableT]):
     The model can be overridden at runtime using `with llm.model(...)` context manager.
     """
 
-    prompt: ContextPrompt[P, DepsT, FormattableT]
-    """The underlying ContextPrompt instance that generates messages with tools and format."""
-
     @overload
     def __call__(
         self: "ContextCall[P, DepsT, None]",
@@ -255,7 +262,10 @@ class ContextCall(BaseCall, Generic[P, DepsT, FormattableT]):
 
 
 @dataclass
-class AsyncContextCall(BaseCall, Generic[P, DepsT, FormattableT]):
+class AsyncContextCall(
+    BaseCall[AsyncContextPrompt[P, DepsT, FormattableT]],
+    Generic[P, DepsT, FormattableT],
+):
     """An async context-aware call that directly generates LLM responses without requiring a model argument.
 
     Created by decorating an async `ContextMessageTemplate` with `llm.call`. The decorated async
@@ -268,9 +278,6 @@ class AsyncContextCall(BaseCall, Generic[P, DepsT, FormattableT]):
     The model can be overridden at runtime using `with llm.model(...)` context manager.
     """
 
-    prompt: AsyncContextPrompt[P, DepsT, FormattableT]
-    """The underlying AsyncContextPrompt instance that generates messages with tools and format."""
-
     @overload
     async def __call__(
         self: "AsyncContextCall[P, DepsT, None]",

mirascope/llm/prompts/prompts.py

@@ -1,9 +1,10 @@
 """Concrete Prompt classes for generating messages with tools and formatting."""
 
-from collections.abc import Sequence
-from dataclasses import dataclass
-from typing import Generic, overload
+from collections.abc import Callable, Sequence
+from dataclasses import dataclass, field
+from typing import Any, Generic, TypeVar, overload
 
+from ..._utils import copy_function_metadata
 from ..context import Context, DepsT
 from ..formatting import Format, FormattableT, OutputParser
 from ..messages import Message, promote_to_messages
@@ -19,12 +20,7 @@ from ..responses import (
     Response,
     StreamResponse,
 )
-from ..tools import (
-    AsyncContextToolkit,
-    AsyncToolkit,
-    ContextToolkit,
-    Toolkit,
-)
+from ..tools import AsyncContextToolkit, AsyncToolkit, ContextToolkit, Toolkit
 from ..types import P
 from .protocols import (
     AsyncContextMessageTemplate,
@@ -33,9 +29,26 @@ from .protocols import (
     MessageTemplate,
 )
 
+FunctionT = TypeVar("FunctionT", bound=Callable[..., Any])
+
+
+@dataclass(kw_only=True)
+class BasePrompt(Generic[FunctionT]):
+    """Base class for all Prompt types with shared metadata functionality."""
+
+    fn: FunctionT
+    """The underlying prompt function that generates message content."""
+
+    __name__: str = field(init=False, repr=False, default="")
+    """The name of the underlying function (preserved for decorator stacking)."""
+
+    def __post_init__(self) -> None:
+        """Preserve standard function attributes for decorator stacking."""
+        copy_function_metadata(self, self.fn)
+
 
 @dataclass
-class Prompt(Generic[P, FormattableT]):
+class Prompt(BasePrompt[MessageTemplate[P]], Generic[P, FormattableT]):
     """A prompt that can be called with a model to generate a response.
 
     Created by decorating a `MessageTemplate` with `llm.prompt`. The decorated
@@ -45,9 +58,6 @@ class Prompt(Generic[P, FormattableT]):
     It can be invoked with a model: `prompt(model, *args, **kwargs)`.
     """
 
-    fn: MessageTemplate[P]
-    """The underlying prompt function that generates message content."""
-
     toolkit: Toolkit
     """The toolkit containing this prompt's tools."""
 
@@ -134,7 +144,7 @@ class Prompt(Generic[P, FormattableT]):
 
 
 @dataclass
-class AsyncPrompt(Generic[P, FormattableT]):
+class AsyncPrompt(BasePrompt[AsyncMessageTemplate[P]], Generic[P, FormattableT]):
     """An async prompt that can be called with a model to generate a response.
 
     Created by decorating an async `MessageTemplate` with `llm.prompt`. The decorated
@@ -144,9 +154,6 @@ class AsyncPrompt(Generic[P, FormattableT]):
     It can be invoked with a model: `await prompt(model, *args, **kwargs)`.
     """
 
-    fn: AsyncMessageTemplate[P]
-    """The underlying async prompt function that generates message content."""
-
     toolkit: AsyncToolkit
     """The toolkit containing this prompt's async tools."""
 
@@ -235,7 +242,9 @@ class AsyncPrompt(Generic[P, FormattableT]):
 
 
 @dataclass
-class ContextPrompt(Generic[P, DepsT, FormattableT]):
+class ContextPrompt(
+    BasePrompt[ContextMessageTemplate[P, DepsT]], Generic[P, DepsT, FormattableT]
+):
     """A context-aware prompt that can be called with a model to generate a response.
 
     Created by decorating a `ContextMessageTemplate` with `llm.prompt`. The decorated
@@ -246,9 +255,6 @@ class ContextPrompt(Generic[P, DepsT, FormattableT]):
     It can be invoked with a model: `prompt(model, ctx, *args, **kwargs)`.
     """
 
-    fn: ContextMessageTemplate[P, DepsT]
-    """The underlying context-aware prompt function that generates message content."""
-
     toolkit: ContextToolkit[DepsT]
     """The toolkit containing this prompt's context-aware tools."""
 
@@ -361,7 +367,9 @@ class ContextPrompt(Generic[P, DepsT, FormattableT]):
 
 
 @dataclass
-class AsyncContextPrompt(Generic[P, DepsT, FormattableT]):
+class AsyncContextPrompt(
+    BasePrompt[AsyncContextMessageTemplate[P, DepsT]], Generic[P, DepsT, FormattableT]
+):
     """An async context-aware prompt that can be called with a model to generate a response.
 
     Created by decorating an async `ContextMessageTemplate` with `llm.prompt`. The decorated
@@ -372,9 +380,6 @@ class AsyncContextPrompt(Generic[P, DepsT, FormattableT]):
     It can be invoked with a model: `await prompt(model, ctx, *args, **kwargs)`.
     """
 
-    fn: AsyncContextMessageTemplate[P, DepsT]
-    """The underlying async context-aware prompt function that generates message content."""
-
     toolkit: AsyncContextToolkit[DepsT]
     """The toolkit containing this prompt's async context-aware tools."""
 

mirascope/llm/tools/tool_schema.py

@@ -22,6 +22,7 @@ from docstring_parser import parse
 from pydantic import BaseModel, Field, create_model
 from pydantic.fields import FieldInfo
 
+from ..._utils import copy_function_metadata
 from ..content import ToolCall
 from ..types import Jsonable
 from .protocols import AsyncContextToolFn, AsyncToolFn, ContextToolFn, ToolFn
@@ -159,11 +160,14 @@ class ToolSchema(Generic[ToolFnT]):
 
     strict: bool | None
     """Whether the tool should use strict mode when supported by the model.
-    
-    If set to None, will use the provider's default setting (usually as strict as 
+
+    If set to None, will use the provider's default setting (usually as strict as
     possible).
     """
 
+    __name__: str
+    """The name of the underlying function (preserved for decorator stacking)."""
+
     def __hash__(self) -> int:
         if not hasattr(self, "_hash"):
             self._hash = hash(
@@ -200,6 +204,7 @@ class ToolSchema(Generic[ToolFnT]):
         self.description = description
         self.parameters = parameters
         self.strict = strict
+        copy_function_metadata(self, fn)
 
     @classmethod
     def from_function(

mirascope/ops/_internal/closure.py

@@ -738,12 +738,15 @@ class _DependencyCollector:
             # For Python 3.13+
             return definition.func  # pyright: ignore[reportFunctionMemberAccess] # pragma: no cover
 
+        # Handle objects with .fn but no __qualname__ (e.g., old-style wrappers).
+        # With copy_function_metadata() now copying __qualname__ to ToolSchema, Prompt,
+        # Call, etc., this branch is no longer reached in normal usage.
         if (
             (wrapped_function := getattr(definition, "fn", None)) is not None
             and not hasattr(definition, "__qualname__")
             and callable(wrapped_function)
         ):
-            return wrapped_function
+            return wrapped_function  # pragma: no cover
 
         return definition
 

mirascope/ops/_internal/exporters/exporters.py

@@ -20,14 +20,17 @@ from ....api._generated.traces.types import (
     TracesCreateRequestResourceSpansItemResource,
     TracesCreateRequestResourceSpansItemResourceAttributesItem,
     TracesCreateRequestResourceSpansItemResourceAttributesItemValue,
+    TracesCreateRequestResourceSpansItemResourceAttributesItemValueArrayValue,
     TracesCreateRequestResourceSpansItemScopeSpansItem,
     TracesCreateRequestResourceSpansItemScopeSpansItemScope,
     TracesCreateRequestResourceSpansItemScopeSpansItemSpansItem,
     TracesCreateRequestResourceSpansItemScopeSpansItemSpansItemAttributesItem,
     TracesCreateRequestResourceSpansItemScopeSpansItemSpansItemAttributesItemValue,
+    TracesCreateRequestResourceSpansItemScopeSpansItemSpansItemAttributesItemValueArrayValue,
     TracesCreateRequestResourceSpansItemScopeSpansItemSpansItemStatus,
 )
 from ....api.client import Mirascope
+from .utils import to_otlp_any_value
 
 logger = logging.getLogger(__name__)
 
@@ -282,17 +285,7 @@ class MirascopeOTLPExporter(SpanExporter):
     def _convert_attribute_value(
         self, value: AttributeValue
     ) -> TracesCreateRequestResourceSpansItemScopeSpansItemSpansItemAttributesItemValue:
-        """Convert OpenTelemetry AttributeValue to Mirascope API's KeyValueValue.
-
-        This conversion is necessary because the Fern-generated API client
-        expects KeyValueValue objects, not OpenTelemetry's AttributeValue types.
-
-        Args:
-            value: An OpenTelemetry AttributeValue (bool, int, float, str, or Sequence)
-
-        Returns:
-            A KeyValueValue object for the Mirascope API
-        """
+        """Convert OpenTelemetry AttributeValue to Mirascope API's KeyValueValue."""
         match value:
             case str():
                 return TracesCreateRequestResourceSpansItemScopeSpansItemSpansItemAttributesItemValue(
@@ -312,49 +305,21 @@ class MirascopeOTLPExporter(SpanExporter):
                 )
             case _:
                 return TracesCreateRequestResourceSpansItemScopeSpansItemSpansItemAttributesItemValue(
-                    string_value=str(list(value))
+                    array_value=TracesCreateRequestResourceSpansItemScopeSpansItemSpansItemAttributesItemValueArrayValue(
+                        values=[to_otlp_any_value(v) for v in value]
+                    )
                 )
 
     def _convert_event_attribute_value(
         self, value: AttributeValue
     ) -> dict[str, object]:
-        """Convert OpenTelemetry AttributeValue to OTLP event attribute value format.
-
-        This uses the OTLP JSON format with typed value wrappers (e.g., stringValue, intValue).
-
-        Args:
-            value: An OpenTelemetry AttributeValue (bool, int, float, str, or Sequence)
-
-        Returns:
-            A dict with the typed value (e.g., {"stringValue": "..."})
-        """
-        match value:
-            case str():
-                return {"stringValue": value}
-            case bool():
-                return {"boolValue": value}
-            case int():
-                return {"intValue": str(value)}
-            case float():
-                return {"doubleValue": value}
-            case _:
-                # Sequences - convert to string representation
-                return {"stringValue": str(list(value))}
+        """Convert OpenTelemetry AttributeValue to OTLP event attribute value format."""
+        return to_otlp_any_value(value)
 
     def _convert_resource_attribute_value(
         self, value: AttributeValue
     ) -> TracesCreateRequestResourceSpansItemResourceAttributesItemValue:
-        """Convert OpenTelemetry AttributeValue to Mirascope API's resource KeyValueValue.
-
-        This conversion is necessary because the Fern-generated API client
-        expects KeyValueValue objects, not OpenTelemetry's AttributeValue types.
-
-        Args:
-            value: An OpenTelemetry AttributeValue (bool, int, float, str, or Sequence)
-
-        Returns:
-            A KeyValueValue object for the Mirascope API resource attributes
-        """
+        """Convert OpenTelemetry AttributeValue to Mirascope API's resource KeyValueValue."""
         match value:
             case str():
                 return TracesCreateRequestResourceSpansItemResourceAttributesItemValue(
@@ -374,7 +339,9 @@ class MirascopeOTLPExporter(SpanExporter):
                 )
             case _:
                 return TracesCreateRequestResourceSpansItemResourceAttributesItemValue(
-                    string_value=str(list(value))
+                    array_value=TracesCreateRequestResourceSpansItemResourceAttributesItemValueArrayValue(
+                        values=[to_otlp_any_value(v) for v in value]
+                    )
                 )
 
     def shutdown(self) -> None:

mirascope/ops/_internal/exporters/utils.py

@@ -4,6 +4,43 @@ This module provides helper functions for formatting and converting
 OpenTelemetry data types for export.
 """
 
+from __future__ import annotations
+
+from collections.abc import Mapping, Sequence
+
+from opentelemetry.util.types import AttributeValue
+
+
+def to_otlp_any_value(value: AttributeValue) -> dict[str, object]:
+    """Convert AttributeValue to OTLP AnyValue (dict form).
+
+    - string/bool/int/float are converted to stringValue/boolValue/intValue/doubleValue
+    - Sequence (excluding str/bytes/Mapping) is converted to arrayValue.values
+    - Unsupported types fallback to stringValue=str(value)
+
+    Args:
+        value: An OpenTelemetry AttributeValue (bool, int, float, str, or Sequence)
+
+    Returns:
+        A dict representing OTLP AnyValue (e.g., {"stringValue": "..."})
+    """
+    match value:
+        case str():
+            return {"stringValue": value}
+        case bool():
+            return {"boolValue": value}
+        case int():
+            return {"intValue": str(value)}
+        case float():
+            return {"doubleValue": value}
+        case _ if isinstance(value, bytes | bytearray | memoryview | Mapping):
+            return {"stringValue": str(value)}
+        case _ if isinstance(value, Sequence):
+            values = [to_otlp_any_value(v) for v in value]
+            return {"arrayValue": {"values": values}}
+        case _:
+            return {"stringValue": str(value)}
+
 
 def format_trace_id(trace_id: int) -> str:
     """Format a trace ID as a 32-character hex string.

mirascope/ops/_internal/traced_calls.py

@@ -6,6 +6,7 @@ from dataclasses import dataclass, field
 from typing import Any, Generic, TypeVar
 from typing_extensions import TypeIs
 
+from ..._utils import copy_function_metadata
 from ...llm.calls import AsyncCall, AsyncContextCall, Call, ContextCall
 from ...llm.context import Context, DepsT
 from ...llm.formatting import FormattableT
@@ -35,6 +36,7 @@ from .traced_functions import (
     TracedContextFunction,
     TracedFunction,
 )
+from .utils import get_original_fn
 
 CallT = TypeVar(
     "CallT",
@@ -106,6 +108,14 @@ class _BaseTracedCall(Generic[CallT]):
     metadata: dict[str, str] = field(default_factory=dict)
     """Arbitrary key-value pairs for additional metadata."""
 
+    __name__: str = field(init=False, repr=False, default="")
+    """The name of the underlying function (preserved for decorator stacking)."""
+
+    def __post_init__(self) -> None:
+        """Preserve standard function attributes for decorator stacking."""
+        original_fn = get_original_fn(self._call.prompt.fn)
+        copy_function_metadata(self, original_fn)
+
 
 @dataclass(kw_only=True)
 class TracedCall(_BaseTracedCall[Call[P, FormattableT]]):
@@ -153,6 +163,7 @@ class TracedCall(_BaseTracedCall[Call[P, FormattableT]]):
 
     def __post_init__(self) -> None:
         """Initialize TracedFunction wrappers for call and stream methods."""
+        super().__post_init__()
         self.call = TracedFunction(
             fn=self._call.call, tags=self.tags, metadata=self.metadata
         )
@@ -213,6 +224,7 @@ class TracedAsyncCall(_BaseTracedCall[AsyncCall[P, FormattableT]]):
 
     def __post_init__(self) -> None:
         """Initialize AsyncTracedFunction wrappers for call and stream methods."""
+        super().__post_init__()
         self.call = AsyncTracedFunction(
             fn=self._call.call, tags=self.tags, metadata=self.metadata
         )
@@ -276,6 +288,7 @@ class TracedContextCall(_BaseTracedCall[ContextCall[P, DepsT, FormattableT]]):
 
     def __post_init__(self) -> None:
         """Initialize TracedContextFunction wrappers for call and stream methods."""
+        super().__post_init__()
         self.call = TracedContextFunction(
             fn=self._call.call, tags=self.tags, metadata=self.metadata
         )
@@ -344,6 +357,7 @@ class TracedAsyncContextCall(_BaseTracedCall[AsyncContextCall[P, DepsT, Formatta
 
     def __post_init__(self) -> None:
         """Initialize AsyncTracedContextFunction wrappers for call and stream methods."""
+        super().__post_init__()
         self.call = AsyncTracedContextFunction(
             fn=self._call.call, tags=self.tags, metadata=self.metadata
         )

mirascope/ops/_internal/traced_functions.py

@@ -10,6 +10,7 @@ from typing import Any, Generic, Literal, TypeVar
 
 from opentelemetry.util.types import AttributeValue
 
+from ..._utils import copy_function_metadata
 from ...api.client import get_async_client, get_sync_client
 from ...llm.context import Context, DepsT
 from ...llm.responses.root_response import RootResponse
@@ -175,11 +176,15 @@ class _BaseFunction(Generic[P, R, FunctionT], ABC):
     _is_async: bool = field(init=False)
     """Whether the wrapped function is asynchronous."""
 
+    __name__: str = field(init=False, repr=False, default="")
+    """The name of the underlying function (preserved for decorator stacking)."""
+
     def __post_init__(self) -> None:
         """Initialize additional attributes after dataclass init."""
         self._qualified_name = get_qualified_name(self.fn)
         original_fn = get_original_fn(self.fn)
         self._module_name = getattr(original_fn, "__module__", "")
+        copy_function_metadata(self, original_fn)
 
 
 @dataclass(kw_only=True)
@@ -200,7 +205,7 @@ class _BaseTracedFunction(_BaseFunction[P, R, FunctionT]):
                 "mirascope.trace.arg_values": json_dumps(arg_values),
             }
             if self.tags:
-                attributes["mirascope.trace.tags"] = self.tags
+                attributes["mirascope.trace.tags"] = list(self.tags)
             if self.metadata:
                 attributes["mirascope.trace.metadata"] = json_dumps(self.metadata)
             span.set(**attributes)
@@ -314,7 +319,7 @@ class _BaseTracedContextFunction(
                 "mirascope.trace.arg_values": json_dumps(arg_values),
             }
             if self.tags:
-                attributes["mirascope.trace.tags"] = self.tags
+                attributes["mirascope.trace.tags"] = list(self.tags)
             if self.metadata:
                 attributes["mirascope.trace.metadata"] = json_dumps(self.metadata)
             span.set(**attributes)

mirascope/ops/_internal/utils.py

@@ -33,15 +33,23 @@ def _is_call_method(fn: Callable[..., Any]) -> bool:
 
 
 def get_original_fn(fn: Callable[..., Any]) -> Callable[..., Any]:
-    """Get the original function from a Call method or return fn as-is.
+    """Get the original unwrapped function.
 
-    When fn is a bound method of a Call object (e.g., Call.call or Call.stream),
-    returns the original decorated function from prompt.fn. Otherwise returns fn.
+    Follows the __wrapped__ chain set by copy_function_metadata() to find the
+    original function. Falls back to checking for Call methods for bound methods.
     """
+    # Follow __wrapped__ chain if available (set by copy_function_metadata)
+    # In practice, get_original_fn is called with original functions or bound methods,
+    # so this loop is defensive code for edge cases like @functools.wraps chains.
+    while hasattr(fn, "__wrapped__"):
+        fn = fn.__wrapped__  # pyright: ignore[reportFunctionMemberAccess] # pragma: no cover
+
+    # Handle bound methods of Call objects (e.g., Call.call or Call.stream)
     if _is_call_method(fn):
         prompt = fn.__self__.prompt  # pyright: ignore[reportFunctionMemberAccess]
         if hasattr(prompt, "fn"):
-            return prompt.fn
+            return get_original_fn(prompt.fn)
+
     return fn
 
 

mirascope/ops/_internal/versioned_functions.py

@@ -196,7 +196,7 @@ class _BaseVersionedFunction(_BaseTracedFunction[P, R, Any]):
             if self.name:
                 span.set(**{"mirascope.version.name": self.name})
             if self.tags:
-                span.set(**{"mirascope.version.tags": self.tags})
+                span.set(**{"mirascope.version.tags": list(self.tags)})
             if self.metadata:
                 for key, value in self.metadata.items():
                     span.set(**{f"mirascope.version.meta.{key}": value})

{mirascope-2.0.1.dist-info → mirascope-2.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mirascope
-Version: 2.0.1
+Version: 2.0.2
 Summary: Every frontier LLM. One unified interface.
 Project-URL: Homepage, https://mirascope.com
 Project-URL: Documentation, https://mirascope.com/docs/mirascope/v2

{mirascope-2.0.1.dist-info → mirascope-2.0.2.dist-info}/RECORD

@@ -1,5 +1,6 @@
 mirascope/__init__.py,sha256=V3T7JlZJ4G9dA5lmq5eTtLYUajbQmTvT3-tUlv8o-fg,265
 mirascope/_stubs.py,sha256=iq5hxVYdLEZ_b2fyBNeY5HXcL50Dcw22HBaBr3ydPgU,12179
+mirascope/_utils.py,sha256=hSwsMKURXrje8Dpl2-KhEBLWJrudWKqExbemKwozRyE,1036
 mirascope/api/__init__.py,sha256=FL0OpTweJC0SUpOnZJddDzeC0JwxPDkIaL78VpeMtko,498
 mirascope/api/client.py,sha256=3JrrC2zNlEttAUXHGN1kbDpGczyI9zCNiSZ3dEyqtAI,8253
 mirascope/api/settings.py,sha256=N9mqpEpx7dupszKYK9li8vY-iDq2FFvlus0fK-gYPu0,2639
@@ -7,7 +8,7 @@ mirascope/api/_generated/README.md,sha256=7-B_qhXCHaXKX9FPG-G112hKmXM64uM6eNqmSd
 mirascope/api/_generated/__init__.py,sha256=ZIX_SjuD8PIB3gifL9sDgEt8zqfEFTv2IvjK-_GzOsw,16800
 mirascope/api/_generated/client.py,sha256=Xh_VGnnbdZaSUpWzjGWu-1lwoZ_6isz9eU-BATjo_vs,8775
 mirascope/api/_generated/environment.py,sha256=eKhvb5ZF3qzJj2OtEI7E9STAquQ1VivYLdmHRc4hFV4,262
-mirascope/api/_generated/reference.md,sha256=Zb8SXjz4ZRLdaGdWGzCldyCwykWkmyotNK2qD3XJX9A,53535
+mirascope/api/_generated/reference.md,sha256=I_GJI0SONEQYPuVAlss8lzcJpUx2Mjra9kzYREppwpg,53563
 mirascope/api/_generated/annotations/__init__.py,sha256=z1rsEyNXWcegCIeC-Y2P9TBI0fVbNs1FohqSJiqrK84,987
 mirascope/api/_generated/annotations/client.py,sha256=rikFtCmou4D4hSZqtaxFnIc7ZKMt86XztUyh4pMcXII,13818
 mirascope/api/_generated/annotations/raw_client.py,sha256=EEC2Rr5KT_WmLn9aFgQxApnxHGl3MdOOcoYAti4syhE,54205
@@ -96,8 +97,8 @@ mirascope/api/_generated/health/types/__init__.py,sha256=dYIZPgww1FuKOKJTeUWMDad
 mirascope/api/_generated/health/types/health_check_response.py,sha256=OCmNPA0BN9mfNoQKzl1aS2uFwzG60su540pZtC8gFJI,659
 mirascope/api/_generated/health/types/health_check_response_status.py,sha256=eCSX3a4cA5zETaIbLvtYkSeS3585DuJa9VnM1fw3VxE,155
 mirascope/api/_generated/organization_invitations/__init__.py,sha256=4LHOmmo5wJRLPqyiZ3jVG_VW3LK-VQVbFUfTwweMbcw,1247
-mirascope/api/_generated/organization_invitations/client.py,sha256=4vpnNlGWnIJ4L8PAkUxn5iU4dTv_qCIP-gMZ6Ap2rCk,13992
-mirascope/api/_generated/organization_invitations/raw_client.py,sha256=MQ1AopdvO0xGMuxTnd-2nS0ytalLW2IFp2D9PUysUfI,58289
+mirascope/api/_generated/organization_invitations/client.py,sha256=EHGrekNKf4uhupP9IffP20SGBKtYJotyv34Tjs3haqY,14048
+mirascope/api/_generated/organization_invitations/raw_client.py,sha256=6Or-I-vkvnx2LHC3G50oCExFAfWdRJIaSNP6OtJuP3U,58345
 mirascope/api/_generated/organization_invitations/types/__init__.py,sha256=eoOMa4pr1aks2d79z2YXDV8JsPs2giYVC-KVLxFWmos,1956
 mirascope/api/_generated/organization_invitations/types/organization_invitations_accept_response.py,sha256=pbvwLUaCkCQ5fuPKlh2OR0U5JINptYKHL4E_g9JusmA,1129
 mirascope/api/_generated/organization_invitations/types/organization_invitations_accept_response_role.py,sha256=s7pe7kJSJ_QZuvK1GNmD3v8Kwyy8GhDPrE7wokylXbw,199
@@ -265,7 +266,7 @@ mirascope/api/_generated/types/unauthorized_error_tag.py,sha256=sx-QdNmQjboMaPni
 mirascope/llm/__init__.py,sha256=1hP54tE5tO6lyrgiZHafc8JbF3LOIKrPON9UQJPo9uw,5879
 mirascope/llm/exceptions.py,sha256=hYc8A8A6unqPaB9GmvdlWIEUso4NWlR7lXELEVVyGFc,10839
 mirascope/llm/calls/__init__.py,sha256=aA-IoOlXdmhVybHTj_fGx1jDlIL-a7ed7xncO2aMIJY,293
-mirascope/llm/calls/calls.py,sha256=K4lLbAgfpvuNyYfh8UYm37onUr8QcEm4BMK_1CBr7i4,11836
+mirascope/llm/calls/calls.py,sha256=g71njIpo5hWvNCnSsDeT8c3l-IaAes01iKrwMyPIYlY,12089
 mirascope/llm/calls/decorator.py,sha256=hMJscJ1cPnQctGZfTqJV6hoCScYherCutOecIcpJfbk,9006
 mirascope/llm/content/__init__.py,sha256=4a5UaoiF8OKfXS9j7INFvOnU7bkNpacpiZ6IlLgC6Lc,1926
 mirascope/llm/content/audio.py,sha256=W6liY1PuQ0pF91eDBnNKW52dr-G7-vnd-rhlSqebx9Y,5311
@@ -297,7 +298,7 @@ mirascope/llm/models/thinking_config.py,sha256=CkFOkw-fjDlsZbHJbV68v4yRwb5o6miVx
 mirascope/llm/prompts/__init__.py,sha256=rOtSiZvw6vsJl7vCFWzKZnm3haNhnLtPRzPfdbgoH7U,727
 mirascope/llm/prompts/_utils.py,sha256=42hTqm8PuLpDOz4Mi59kaUou43qeo5R-9703l8cciSg,1018
 mirascope/llm/prompts/decorator.py,sha256=UiCA921u1qCQZPqrDRE4xERg7nzyzf1rChAEN216ojA,7179
-mirascope/llm/prompts/prompts.py,sha256=WZUERWWVIvJq6LGMl6ixSGFoAayVwdVjm16CLEUGyew,15926
+mirascope/llm/prompts/prompts.py,sha256=Yb4ijwZCokkPkBv4rWfH22qNzG8gm9ueuIjIMYrwpww,16277
 mirascope/llm/prompts/protocols.py,sha256=auopJnLAfOognCFVVnXrvb4zFj3102Ofi2iVHfublM0,2242
 mirascope/llm/providers/__init__.py,sha256=JP1cW_IFUZwkcHWP2Kc9ezihfQu7CY80ayqmkVeR9pc,1953
 mirascope/llm/providers/model_id.py,sha256=qr7RKkP_3Blgxjzw_HvQVJLM1aTzsP-jVFsycz0itI8,292
@@ -373,7 +374,7 @@ mirascope/llm/tools/__init__.py,sha256=P9NX6pP5fKScTAuR9K3fODdW17wQ-6fcArVx2klGi
 mirascope/llm/tools/_utils.py,sha256=mUi6vUEwYuwu8W9FCQcQ_9h_VE_U-Y3Y2UQ2YptPZL4,1171
 mirascope/llm/tools/decorator.py,sha256=2ZTvYH0jrft0E3BdQ2BB9Okyb72JOIqQvBkLWmWuo7w,5663
 mirascope/llm/tools/protocols.py,sha256=pUgGuh4TilRuoipeEHfTzu5cXacMnUpz0Q5YXQyAY2I,3050
-mirascope/llm/tools/tool_schema.py,sha256=2wVqbXx2B12dE1u_By9Vlbrjz0R_s8pmE6M1Av4GrAE,10291
+mirascope/llm/tools/tool_schema.py,sha256=1fwEFZpxfFyXKFrJmsj0A1W6ptaMHaJQRUc0Yz0GEZk,10473
 mirascope/llm/tools/toolkit.py,sha256=YFBEUWEeFRsujYjhsBvD7Hu3-tlvQP3sihp_c65ZVHE,5284
 mirascope/llm/tools/tools.py,sha256=CaSGvjFnj-A8KIpInc7m5VNj7wY-eW25f4fvAZeDwWQ,9844
 mirascope/llm/types/__init__.py,sha256=lqzi1FkZ-s-D9-KQzVkAHuQQ1zp6B6yM3r9UNo46teE,357
@@ -383,26 +384,26 @@ mirascope/llm/types/type_vars.py,sha256=OsAcQAZh5T_X8ZTLlP4GC1x3qgVY9rfYSnME8aMT
 mirascope/ops/__init__.py,sha256=tS2HIXbbKo9VY2LPngNaAtel9aNQblt2Tz3scyYUmyw,2617
 mirascope/ops/exceptions.py,sha256=N5AES285tTxFJ1THIeJht-_ZEQ7-7_FWoes4EBAu3B8,641
 mirascope/ops/_internal/__init__.py,sha256=yCf8bnHiQFWYsBkwBPyMqknI5apmOwlmyf105mH9EAE,173
-mirascope/ops/_internal/closure.py,sha256=lPkZLfL33OZ3yxdZhT3RXqhD0ychKWH0zevU4V-VHCY,42465
+mirascope/ops/_internal/closure.py,sha256=lZuT71ufwHd987BWb7mdk88_d0_eO18fNs9sS16zLss,42727
 mirascope/ops/_internal/configuration.py,sha256=kWgqLM6XOJ9WK_7y7R3ULQ0UE7r9dgxxMIfg4YRZ0c4,5228
 mirascope/ops/_internal/context.py,sha256=G0Jiv6XfWjeDyRrh5-Wg6I0UlqiDj17SNMhtQ6Rw3lM,2522
 mirascope/ops/_internal/propagation.py,sha256=2in7uttv2b5MBoe3Ol1mgn6N20TqhH0sErGak7hMUDY,7032
 mirascope/ops/_internal/protocols.py,sha256=72K-AIHpDsflo58cig_jO1_xdAwuqZaQBU96G4sGe_I,4229
 mirascope/ops/_internal/session.py,sha256=-7r2TbJiAKeEQ0YaJ9X_2BlumLK7x6h9wP6f8ei-5GA,4105
 mirascope/ops/_internal/spans.py,sha256=Mx4x2sSYYIhibcHKr99TTboUdGrSPV09NpftJD5kkBE,7675
-mirascope/ops/_internal/traced_calls.py,sha256=nHz2NgG9IHyhDKozKp7m252lDetdSaGn-tYhtZSO9LM,13000
-mirascope/ops/_internal/traced_functions.py,sha256=dAMx6BfBs6phc8LgdbsBZFUCN7el-sRMhK6yBYm6zVw,18884
+mirascope/ops/_internal/traced_calls.py,sha256=L0LG1SWW7DNYnvqajPoD1ptPi8IUM9JssYBqHo7rp0E,13577
+mirascope/ops/_internal/traced_functions.py,sha256=lPN8c-lzHY-BxhhbcDpKC7JQtvgnmYMDDpx7M46VP-8,19136
 mirascope/ops/_internal/tracing.py,sha256=RzHBc-yEbj3Z_Kjobh26FWDSnF6AGbRCsb0JXBSzPdc,10244
 mirascope/ops/_internal/types.py,sha256=5B1RgChTEbGkLZvdoX8XA2vMKn4bS8BczjGDElxmugI,412
-mirascope/ops/_internal/utils.py,sha256=qxr0ysyrBZP7ov9pqAirtDjftprMswwq3szbbVqjabk,4466
+mirascope/ops/_internal/utils.py,sha256=uGGKuAToxUUlkiqcmZIAICCAtmQay_fQ_ciVWF6jcIQ,4907
 mirascope/ops/_internal/versioned_calls.py,sha256=7oePDEuf98ph6yrHKJmXSagc1mGz5sdTTJgQnB2CyHM,18445
-mirascope/ops/_internal/versioned_functions.py,sha256=lX0jgOPdtaajFM8kEkkEBRbGCrLuTo65EApnYCSDo_U,13522
+mirascope/ops/_internal/versioned_functions.py,sha256=g_m-szx6NiBmwxakygNDnJTOyOQN-cVEgsUZZXLFh0w,13528
 mirascope/ops/_internal/versioning.py,sha256=MxiezWrGUPiLZxZnwwXDFTR7EMrfTHR7rmd14TuDCL8,8286
 mirascope/ops/_internal/exporters/__init__.py,sha256=3PUd75TczW_8bUImcWgddWyoRmygGf6T_jL7Jo3t_sY,608
-mirascope/ops/_internal/exporters/exporters.py,sha256=nRyDXxf0Mgy0D5sni60tdi8Hgfivj2eLO1dWwtaydRI,14932
+mirascope/ops/_internal/exporters/exporters.py,sha256=9RUKKgtKCxrhh-9H0eaYdG_LApmirmvRZjNIlhr3tGA,14043
 mirascope/ops/_internal/exporters/processors.py,sha256=hkNwA5m0e_KJoPQf_EPUaK6ItZlJ8D5TCBs-JAhGKXo,3511
 mirascope/ops/_internal/exporters/types.py,sha256=1ge7nwNQ3HUH4qVoYeHhSUp3LxohkvDSoqPw2MDLEqw,4057
-mirascope/ops/_internal/exporters/utils.py,sha256=zDZihsj4BxLZH3o2RKCofCvSUf4Nbo8Yro2dqkm0rI4,727
+mirascope/ops/_internal/exporters/utils.py,sha256=1EwxK70bKrB5fkm9y3MnXduSF-NKJqRbBDgqD1T0fnc,2026
 mirascope/ops/_internal/instrumentation/__init__.py,sha256=Fz2WUX_vFvLqHbMxkuVQAGRXyWkk9Oqug8DcA-oa4p8,180
 mirascope/ops/_internal/instrumentation/llm/__init__.py,sha256=_L6JL3sw6XbW9ICsWe4RZpkT-_BFSIBbh_n00O9h_SM,184
 mirascope/ops/_internal/instrumentation/llm/common.py,sha256=bb4rHjnQyAs55Cf_qy0OupEuzxvskpOTvRBGEJyByfY,17637
@@ -417,7 +418,7 @@ mirascope/ops/_internal/instrumentation/llm/gen_ai_types/gen_ai_input_messages.p
 mirascope/ops/_internal/instrumentation/llm/gen_ai_types/gen_ai_output_messages.py,sha256=qxLxSmwd_2EdymR4a5cDXc9DTSFGQXMhqSHGgS219Y4,957
 mirascope/ops/_internal/instrumentation/llm/gen_ai_types/gen_ai_system_instructions.py,sha256=ZkWNOh6-tG6LtFP_0Hjfdf_8BqAsIFyP3tNybeo1FnY,388
 mirascope/ops/_internal/instrumentation/llm/gen_ai_types/shared.py,sha256=JpmcF1i25EHc8KMv6nkpBwyIU46B_T8v1L71C16n080,3004
-mirascope-2.0.1.dist-info/METADATA,sha256=C9jPBWHGrM_vyZG6XA69yUmOwOz9uDb2UMq_jpC3yvI,9019
-mirascope-2.0.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-mirascope-2.0.1.dist-info/licenses/LICENSE,sha256=MxD-HQ4YHSi2YqZ3UH2MsjclZbesds6j5it4O2IWyJs,1072
-mirascope-2.0.1.dist-info/RECORD,,
+mirascope-2.0.2.dist-info/METADATA,sha256=G9TopFOv0bxICfwTZO2IY5Nk3pxz06oUC24kTOaL-yo,9019
+mirascope-2.0.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mirascope-2.0.2.dist-info/licenses/LICENSE,sha256=MxD-HQ4YHSi2YqZ3UH2MsjclZbesds6j5it4O2IWyJs,1072
+mirascope-2.0.2.dist-info/RECORD,,