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

mirascope/ops/_internal/instrumentation/llm/common.py

@@ -17,10 +17,11 @@ from opentelemetry.semconv.attributes import error_attributes as ErrorAttributes
 from opentelemetry.trace import SpanKind, Status, StatusCode
 
 from .....llm import (
-    AnyToolFn,
+    AnyTools,
     AnyToolSchema,
     BaseToolkit,
     Format,
+    FormatSpec,
     FormattableT,
     Jsonable,
     Message,
@@ -28,9 +29,9 @@ from .....llm import (
     ModelId,
     Params,
     ProviderId,
+    ProviderTool,
     RootResponse,
     ToolkitT,
-    ToolSchema,
 )
 from ...configuration import get_tracer
 from ...utils import json_dumps
@@ -45,6 +46,7 @@ from .serialize import (
     serialize_mirascope_cost,
     serialize_mirascope_messages,
     serialize_mirascope_usage,
+    serialize_tools,
 )
 
 logger = logging.getLogger(__name__)
@@ -64,10 +66,7 @@ else:
     Tracer = None
 
 
-ToolsParam: TypeAlias = (
-    Sequence[ToolSchema[AnyToolFn]] | BaseToolkit[AnyToolSchema] | None
-)
-FormatParam: TypeAlias = Format[FormattableT] | None
+FormatParam: TypeAlias = FormatSpec[FormattableT] | None
 ParamsDict: TypeAlias = Mapping[str, str | int | float | bool | Sequence[str] | None]
 SpanAttributes: TypeAlias = Mapping[str, AttributeValue]
 AttributeSetter: TypeAlias = Callable[[str, AttributeValue], None]
@@ -180,53 +179,13 @@ def _assign_request_message_attributes(
     )
 
 
-def _collect_tool_schemas(
-    tools: Sequence[ToolSchema[AnyToolFn]] | BaseToolkit[AnyToolSchema],
-) -> list[ToolSchema[AnyToolFn]]:
-    """Collect ToolSchema instances from a tools parameter."""
-    iterable = list(tools.tools) if isinstance(tools, BaseToolkit) else list(tools)
-    schemas: list[ToolSchema[AnyToolFn]] = []
-    for tool in iterable:
-        if isinstance(tool, ToolSchema):
-            schemas.append(tool)
-    return schemas
-
-
-def _serialize_tool_definitions(
-    tools: ToolsParam,
-    format: FormatParam = None,
-) -> str | None:
-    """Serialize tool definitions to JSON for span attributes."""
-    if tools is None:
-        tool_schemas: list[ToolSchema[AnyToolFn]] = []
-    else:
-        tool_schemas = _collect_tool_schemas(tools)
-
-    if isinstance(format, Format) and format.mode == "tool":
-        tool_schemas.append(format.create_tool_schema())
-
-    if not tool_schemas:
-        return None
-    definitions: list[dict[str, str | int | bool | dict[str, str | int | bool]]] = []
-    for tool in tool_schemas:
-        tool_def: dict[str, str | int | bool | dict[str, str | int | bool]] = {
-            "name": tool.name,
-            "description": tool.description,
-            "parameters": tool.parameters.model_dump(by_alias=True, mode="json"),
-        }
-        if tool.strict is not None:
-            tool_def["strict"] = tool.strict
-        definitions.append(tool_def)
-    return json_dumps(definitions)
-
-
 def _build_request_attributes(
     *,
     operation: str,
     provider: ProviderId,
     model_id: ModelId,
     messages: Sequence[Message],
-    tools: ToolsParam,
+    tools: AnyTools | None,
     format: FormatParam,
     params: ParamsDict,
 ) -> dict[str, AttributeValue]:
@@ -244,7 +203,18 @@ def _build_request_attributes(
         messages=messages,
     )
 
-    tool_payload = _serialize_tool_definitions(tools, format=format)
+    tool_schemas: list[AnyToolSchema | ProviderTool] = []
+    if tools is None:
+        tool_schemas = []
+    elif isinstance(tools, BaseToolkit):
+        tool_schemas = list(tools.tools)
+    else:
+        tool_schemas = list(tools)
+
+    if isinstance(format, Format) and format.mode == "tool":
+        tool_schemas.append(format.create_tool_schema())
+
+    tool_payload = serialize_tools(tool_schemas)
     if tool_payload:
         # The incubating semconv module does not yet expose a constant for this key.
         attrs["gen_ai.tool.definitions"] = tool_payload
@@ -477,7 +447,7 @@ def start_model_span(
     model: Model,
     *,
     messages: Sequence[Message],
-    tools: ToolsParam,
+    tools: AnyTools | None,
     format: FormatParam,
     activate: bool = True,
 ) -> Iterator[SpanContext]:

mirascope/ops/_internal/instrumentation/llm/model.py

@@ -19,35 +19,30 @@ from opentelemetry import trace as otel_trace
 from .....llm import (
     AsyncContextResponse,
     AsyncContextStreamResponse,
-    AsyncContextTool,
-    AsyncContextToolkit,
+    AsyncContextTools,
     AsyncResponse,
     AsyncStreamResponse,
-    AsyncTool,
-    AsyncToolkit,
+    AsyncTools,
     Context,
     ContextResponse,
     ContextStreamResponse,
-    ContextTool,
-    ContextToolkit,
+    ContextTools,
     DepsT,
+    FormatSpec,
     FormattableT,
     Message,
     Model,
-    OutputParser,
     Response,
     RootResponse,
     StreamResponse,
     StreamResponseChunk,
-    Tool,
-    Toolkit,
+    Tools,
     UserContent,
 )
 from .....llm.messages import promote_to_messages, user
 from .common import (
     FormatParam,
     SpanContext,
-    ToolsParam,
     attach_response,
     attach_response_async,
     record_dropped_params,
@@ -108,7 +103,7 @@ def _instrumented_model_call(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[Tool] | Toolkit | None = None,
+    tools: Tools | None = None,
     format: None = None,
 ) -> Response: ...
 
@@ -118,8 +113,8 @@ def _instrumented_model_call(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[Tool] | Toolkit | None = None,
-    format: type[FormattableT] | FormatParam,
+    tools: Tools | None = None,
+    format: FormatSpec[FormattableT],
 ) -> Response[FormattableT]: ...
 
 
@@ -128,8 +123,8 @@ def _instrumented_model_call(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[Tool] | Toolkit | None = None,
-    format: type[FormattableT] | FormatParam | OutputParser[FormattableT] | None = None,
+    tools: Tools | None = None,
+    format: FormatSpec[FormattableT] | None = None,
 ) -> Response | Response[FormattableT]: ...
 
 
@@ -138,7 +133,7 @@ def _instrumented_model_call(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[Tool] | Toolkit | None = None,
+    tools: Tools | None = None,
     format: FormatParam = None,
 ) -> Response | Response[FormattableT]:
     """Returns a GenAI-instrumented result of `Model.call`."""
@@ -193,7 +188,7 @@ async def _instrumented_model_call_async(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[AsyncTool] | AsyncToolkit | None = None,
+    tools: AsyncTools | None = None,
     format: None = None,
 ) -> AsyncResponse: ...
 
@@ -203,8 +198,8 @@ async def _instrumented_model_call_async(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[AsyncTool] | AsyncToolkit | None = None,
-    format: type[FormattableT] | FormatParam,
+    tools: AsyncTools | None = None,
+    format: FormatSpec[FormattableT],
 ) -> AsyncResponse[FormattableT]: ...
 
 
@@ -213,8 +208,8 @@ async def _instrumented_model_call_async(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[AsyncTool] | AsyncToolkit | None = None,
-    format: type[FormattableT] | FormatParam | OutputParser[FormattableT] | None = None,
+    tools: AsyncTools | None = None,
+    format: FormatSpec[FormattableT] | None = None,
 ) -> AsyncResponse | AsyncResponse[FormattableT]: ...
 
 
@@ -223,7 +218,7 @@ async def _instrumented_model_call_async(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[AsyncTool] | AsyncToolkit | None = None,
+    tools: AsyncTools | None = None,
     format: FormatParam = None,
 ) -> AsyncResponse | AsyncResponse[FormattableT]:
     """Returns a GenAI-instrumented result of `Model.call_async`."""
@@ -280,7 +275,7 @@ def _instrumented_model_context_call(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[Tool | ContextTool[DepsT]] | ContextToolkit[DepsT] | None = None,
+    tools: ContextTools[DepsT] | None = None,
     format: None = None,
 ) -> ContextResponse[DepsT, None]: ...
 
@@ -291,8 +286,8 @@ def _instrumented_model_context_call(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[Tool | ContextTool[DepsT]] | ContextToolkit[DepsT] | None = None,
-    format: type[FormattableT] | FormatParam,
+    tools: ContextTools[DepsT] | None = None,
+    format: FormatSpec[FormattableT],
 ) -> ContextResponse[DepsT, FormattableT]: ...
 
 
@@ -302,8 +297,8 @@ def _instrumented_model_context_call(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[Tool | ContextTool[DepsT]] | ContextToolkit[DepsT] | None = None,
-    format: type[FormattableT] | FormatParam | OutputParser[FormattableT] | None = None,
+    tools: ContextTools[DepsT] | None = None,
+    format: FormatSpec[FormattableT] | None = None,
 ) -> ContextResponse[DepsT, None] | ContextResponse[DepsT, FormattableT]: ...
 
 
@@ -313,7 +308,7 @@ def _instrumented_model_context_call(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[Tool | ContextTool[DepsT]] | ContextToolkit[DepsT] | None = None,
+    tools: ContextTools[DepsT] | None = None,
     format: FormatParam = None,
 ) -> ContextResponse[DepsT, None] | ContextResponse[DepsT, FormattableT]:
     """Returns a GenAI-instrumented result of `Model.context_call`."""
@@ -371,9 +366,7 @@ async def _instrumented_model_context_call_async(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[AsyncTool | AsyncContextTool[DepsT]]
-    | AsyncContextToolkit[DepsT]
-    | None = None,
+    tools: AsyncContextTools[DepsT] | None = None,
     format: None = None,
 ) -> AsyncContextResponse[DepsT, None]: ...
 
@@ -384,10 +377,8 @@ async def _instrumented_model_context_call_async(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[AsyncTool | AsyncContextTool[DepsT]]
-    | AsyncContextToolkit[DepsT]
-    | None = None,
-    format: type[FormattableT] | FormatParam,
+    tools: AsyncContextTools[DepsT] | None = None,
+    format: FormatSpec[FormattableT],
 ) -> AsyncContextResponse[DepsT, FormattableT]: ...
 
 
@@ -397,10 +388,8 @@ async def _instrumented_model_context_call_async(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[AsyncTool | AsyncContextTool[DepsT]]
-    | AsyncContextToolkit[DepsT]
-    | None = None,
-    format: type[FormattableT] | FormatParam | OutputParser[FormattableT] | None = None,
+    tools: AsyncContextTools[DepsT] | None = None,
+    format: FormatSpec[FormattableT] | None = None,
 ) -> AsyncContextResponse[DepsT, None] | AsyncContextResponse[DepsT, FormattableT]: ...
 
 
@@ -410,9 +399,7 @@ async def _instrumented_model_context_call_async(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[AsyncTool | AsyncContextTool[DepsT]]
-    | AsyncContextToolkit[DepsT]
-    | None = None,
+    tools: AsyncContextTools[DepsT] | None = None,
     format: FormatParam = None,
 ) -> AsyncContextResponse[DepsT, None] | AsyncContextResponse[DepsT, FormattableT]:
     """Returns a GenAI-instrumented result of `Model.context_call_async`."""
@@ -567,7 +554,7 @@ def _instrumented_model_stream(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[Tool] | Toolkit | None = None,
+    tools: Tools | None = None,
     format: None = None,
 ) -> StreamResponse: ...
 
@@ -577,8 +564,8 @@ def _instrumented_model_stream(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[Tool] | Toolkit | None = None,
-    format: type[FormattableT] | FormatParam,
+    tools: Tools | None = None,
+    format: FormatSpec[FormattableT],
 ) -> StreamResponse[FormattableT]: ...
 
 
@@ -587,8 +574,8 @@ def _instrumented_model_stream(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[Tool] | Toolkit | None = None,
-    format: type[FormattableT] | FormatParam | OutputParser[FormattableT] | None = None,
+    tools: Tools | None = None,
+    format: FormatSpec[FormattableT] | None = None,
 ) -> StreamResponse | StreamResponse[FormattableT]: ...
 
 
@@ -597,7 +584,7 @@ def _instrumented_model_stream(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[Tool] | Toolkit | None = None,
+    tools: Tools | None = None,
     format: FormatParam = None,
 ) -> StreamResponse | StreamResponse[FormattableT]:
     """Returns a GenAI-instrumented result of `Model.stream`."""
@@ -676,7 +663,7 @@ async def _instrumented_model_stream_async(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[AsyncTool] | AsyncToolkit | None = None,
+    tools: AsyncTools | None = None,
     format: None = None,
 ) -> AsyncStreamResponse: ...
 
@@ -686,8 +673,8 @@ async def _instrumented_model_stream_async(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[AsyncTool] | AsyncToolkit | None = None,
-    format: type[FormattableT] | FormatParam,
+    tools: AsyncTools | None = None,
+    format: FormatSpec[FormattableT],
 ) -> AsyncStreamResponse[FormattableT]: ...
 
 
@@ -696,8 +683,8 @@ async def _instrumented_model_stream_async(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[AsyncTool] | AsyncToolkit | None = None,
-    format: type[FormattableT] | FormatParam | OutputParser[FormattableT] | None = None,
+    tools: AsyncTools | None = None,
+    format: FormatSpec[FormattableT] | None = None,
 ) -> AsyncStreamResponse | AsyncStreamResponse[FormattableT]: ...
 
 
@@ -706,7 +693,7 @@ async def _instrumented_model_stream_async(
     self: Model,
     content: UserContent | Sequence[Message],
     *,
-    tools: Sequence[AsyncTool] | AsyncToolkit | None = None,
+    tools: AsyncTools | None = None,
     format: FormatParam = None,
 ) -> AsyncStreamResponse | AsyncStreamResponse[FormattableT]:
     """Returns a GenAI-instrumented result of `Model.stream_async`."""
@@ -788,7 +775,7 @@ def _instrumented_model_context_stream(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[Tool | ContextTool[DepsT]] | ContextToolkit[DepsT] | None = None,
+    tools: ContextTools[DepsT] | None = None,
     format: None = None,
 ) -> ContextStreamResponse[DepsT, None]: ...
 
@@ -799,8 +786,8 @@ def _instrumented_model_context_stream(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[Tool | ContextTool[DepsT]] | ContextToolkit[DepsT] | None = None,
-    format: type[FormattableT] | FormatParam,
+    tools: ContextTools[DepsT] | None = None,
+    format: FormatSpec[FormattableT],
 ) -> ContextStreamResponse[DepsT, FormattableT]: ...
 
 
@@ -810,8 +797,8 @@ def _instrumented_model_context_stream(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[Tool | ContextTool[DepsT]] | ContextToolkit[DepsT] | None = None,
-    format: type[FormattableT] | FormatParam | OutputParser[FormattableT] | None = None,
+    tools: ContextTools[DepsT] | None = None,
+    format: FormatSpec[FormattableT] | None = None,
 ) -> (
     ContextStreamResponse[DepsT, None] | ContextStreamResponse[DepsT, FormattableT]
 ): ...
@@ -823,7 +810,7 @@ def _instrumented_model_context_stream(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[Tool | ContextTool[DepsT]] | ContextToolkit[DepsT] | None = None,
+    tools: ContextTools[DepsT] | None = None,
     format: FormatParam = None,
 ) -> ContextStreamResponse[DepsT, None] | ContextStreamResponse[DepsT, FormattableT]:
     """Returns a GenAI-instrumented result of `Model.context_stream`."""
@@ -905,9 +892,7 @@ async def _instrumented_model_context_stream_async(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[AsyncTool | AsyncContextTool[DepsT]]
-    | AsyncContextToolkit[DepsT]
-    | None = None,
+    tools: AsyncContextTools[DepsT] | None = None,
     format: None = None,
 ) -> AsyncContextStreamResponse[DepsT, None]: ...
 
@@ -918,10 +903,8 @@ async def _instrumented_model_context_stream_async(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[AsyncTool | AsyncContextTool[DepsT]]
-    | AsyncContextToolkit[DepsT]
-    | None = None,
-    format: type[FormattableT] | FormatParam,
+    tools: AsyncContextTools[DepsT] | None = None,
+    format: FormatSpec[FormattableT],
 ) -> AsyncContextStreamResponse[DepsT, FormattableT]: ...
 
 
@@ -931,10 +914,8 @@ async def _instrumented_model_context_stream_async(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[AsyncTool | AsyncContextTool[DepsT]]
-    | AsyncContextToolkit[DepsT]
-    | None = None,
-    format: type[FormattableT] | FormatParam | OutputParser[FormattableT] | None = None,
+    tools: AsyncContextTools[DepsT] | None = None,
+    format: FormatSpec[FormattableT] | None = None,
 ) -> (
     AsyncContextStreamResponse[DepsT, None]
     | AsyncContextStreamResponse[DepsT, FormattableT]
@@ -947,9 +928,7 @@ async def _instrumented_model_context_stream_async(
     content: UserContent | Sequence[Message],
     *,
     ctx: Context[DepsT],
-    tools: Sequence[AsyncTool | AsyncContextTool[DepsT]]
-    | AsyncContextToolkit[DepsT]
-    | None = None,
+    tools: AsyncContextTools[DepsT] | None = None,
     format: FormatParam = None,
 ) -> (
     AsyncContextStreamResponse[DepsT, None]
@@ -1067,7 +1046,7 @@ def _instrumented_model_resume(
     with start_model_span(
         self,
         messages=messages,
-        tools=cast(ToolsParam, response.toolkit),
+        tools=response.toolkit,
         format=cast(FormatParam, response.format),
     ) as span_ctx:
         result = _ORIGINAL_MODEL_RESUME(
@@ -1147,7 +1126,7 @@ async def _instrumented_model_resume_async(
     with start_model_span(
         self,
         messages=messages,
-        tools=cast(ToolsParam, response.toolkit),
+        tools=response.toolkit,
         format=cast(FormatParam, response.format),
         activate=True,
     ) as span_ctx:
@@ -1232,7 +1211,7 @@ def _instrumented_model_context_resume(
     with start_model_span(
         self,
         messages=messages,
-        tools=cast(ToolsParam, response.toolkit),
+        tools=response.toolkit,
         format=cast(FormatParam, response.format),
         activate=True,
     ) as span_ctx:
@@ -1320,7 +1299,7 @@ async def _instrumented_model_context_resume_async(
     with start_model_span(
         self,
         messages=messages,
-        tools=cast(ToolsParam, response.toolkit),
+        tools=response.toolkit,
         format=cast(FormatParam, response.format),
         activate=True,
     ) as span_ctx:
@@ -1402,7 +1381,7 @@ def _instrumented_model_resume_stream(
     span_cm = start_model_span(
         self,
         messages=messages,
-        tools=cast(ToolsParam, response.toolkit),
+        tools=response.toolkit,
         format=cast(FormatParam, response.format),
         activate=False,
     )
@@ -1505,7 +1484,7 @@ async def _instrumented_model_resume_stream_async(
     span_cm = start_model_span(
         self,
         messages=messages,
-        tools=cast(ToolsParam, response.toolkit),
+        tools=response.toolkit,
         format=cast(FormatParam, response.format),
         activate=False,
     )
@@ -1618,7 +1597,7 @@ def _instrumented_model_context_resume_stream(
     span_cm = start_model_span(
         self,
         messages=messages,
-        tools=cast(ToolsParam, response.toolkit),
+        tools=response.toolkit,
         format=cast(FormatParam, response.format),
         activate=False,
     )
@@ -1735,7 +1714,7 @@ async def _instrumented_model_context_resume_stream_async(
     span_cm = start_model_span(
         self,
         messages=messages,
-        tools=cast(ToolsParam, response.toolkit),
+        tools=response.toolkit,
         format=cast(FormatParam, response.format),
         activate=False,
     )

mirascope/ops/_internal/instrumentation/llm/serialize.py

@@ -4,30 +4,33 @@ from __future__ import annotations
 
 import logging
 from collections.abc import Sequence
-from typing import TYPE_CHECKING, Any, Protocol
+from typing import Any, Protocol
 
-from .....llm.content import (
+from opentelemetry.util.types import AttributeValue
+
+from .....llm import (
+    AnyToolSchema,
+    AssistantMessage,
     Audio,
     Base64ImageSource,
     Document,
     Image,
+    Jsonable,
+    Message,
+    ProviderTool,
+    RootResponse,
+    SystemMessage,
     Text,
     Thought,
     ToolCall,
     ToolOutput,
+    Usage,
+    UserMessage,
 )
 from .....llm.content.document import Base64DocumentSource, TextDocumentSource
-from .....llm.messages import AssistantMessage, Message, SystemMessage, UserMessage
-from .....llm.responses.usage import Usage
 from ...utils import json_dumps
 from .cost import calculate_cost_async, calculate_cost_sync
 
-if TYPE_CHECKING:
-    from opentelemetry.util.types import AttributeValue
-
-    from .....llm.responses.root_response import RootResponse
-    from .....llm.types import Jsonable
-
 logger = logging.getLogger(__name__)
 
 
@@ -41,7 +44,7 @@ class SpanProtocol(Protocol):
 
 def _serialize_content_part(
     part: Text | ToolCall | Thought | Image | Audio | Document | ToolOutput[Jsonable],
-) -> dict[str, Any]:
+) -> dict[str, Jsonable]:
     """Serialize a single content part to a dict matching the Mirascope dataclass structure."""
     if isinstance(part, Text):
         return {"type": "text", "text": part.text}
@@ -116,7 +119,7 @@ def _serialize_content_part(
     return {"type": "unknown"}  # pragma: no cover
 
 
-def _serialize_message(message: Message) -> dict[str, Any]:
+def _serialize_message(message: Message) -> dict[str, Jsonable]:
     """Serialize a Message to a dict matching the Mirascope dataclass structure."""
     if isinstance(message, SystemMessage):
         return {
@@ -166,6 +169,27 @@ def serialize_mirascope_usage(usage: Usage | None) -> AttributeValue | None:
     )
 
 
+def _serialize_tool(tool: AnyToolSchema | ProviderTool) -> dict[str, Jsonable]:
+    if isinstance(tool, ProviderTool):
+        return {"name": tool.name, "type": "extension"}
+    result: dict[str, Jsonable] = {
+        "name": tool.name,
+        "description": tool.description,
+        "type": "function",
+        "parameters": tool.parameters.model_dump(by_alias=True, mode="json"),
+    }
+    if tool.strict is not None:
+        result["strict"] = tool.strict
+    return result
+
+
+def serialize_tools(tools: Sequence[AnyToolSchema | ProviderTool]) -> str | None:
+    """Serialize a sequence of Mirascope tools"""
+    if not tools:
+        return None
+    return json_dumps([_serialize_tool(t) for t in tools])
+
+
 def serialize_mirascope_cost(
     input_cost: float,
     output_cost: float,

mirascope/ops/_internal/instrumentation/providers/__init__.py

@@ -0,0 +1,29 @@
+"""Provider instrumentation modules."""
+
+from .anthropic import (
+    instrument_anthropic,
+    is_anthropic_instrumented,
+    uninstrument_anthropic,
+)
+from .google_genai import (
+    instrument_google_genai,
+    is_google_genai_instrumented,
+    uninstrument_google_genai,
+)
+from .openai import (
+    instrument_openai,
+    is_openai_instrumented,
+    uninstrument_openai,
+)
+
+__all__ = [
+    "instrument_anthropic",
+    "instrument_google_genai",
+    "instrument_openai",
+    "is_anthropic_instrumented",
+    "is_google_genai_instrumented",
+    "is_openai_instrumented",
+    "uninstrument_anthropic",
+    "uninstrument_google_genai",
+    "uninstrument_openai",
+]