Add DSPy integration to logfire (#1625)

bdsaglam · alexmojaki · web-flow · commit 7a3e157f830b · 2026-01-16T09:56:30.000Z
Co-authored-by: Alex Hall &lt;alex.mojaki@gmail.com&gt;
diff --git a/docs/integrations/llms/dspy.md b/docs/integrations/llms/dspy.md
@@ -0,0 +1,54 @@
+---
+title: "Pydantic Logfire Integrations: DSPy"
+description: "Instrument DSPy with Pydantic Logfire using OpenInference for end-to-end LLM workflow tracing."
+integration: logfire
+---
+# DSPy
+
+**Logfire** supports instrumenting [DSPy](https://dspy.ai/) with the
+[`logfire.instrument_dspy()`][logfire.Logfire.instrument_dspy] method.
+
+## Installation
+
+Install `logfire` with the `dspy` extra and the DSPy package:
+
+{{ install_logfire(extras=['dspy']) }}
+
+```bash
+pip install dspy-ai
+```
+
+## Usage
+
+```python hl_lines="6"
+import dspy
+
+import logfire
+
+logfire.configure()
+logfire.instrument_dspy()
+
+lm = dspy.LM("openai/gpt-5-mini")
+dspy.configure(lm=lm)
+
+class ExtractInfo(dspy.Signature):
+    """Extract structured information from text."""
+
+    text: str = dspy.InputField()
+    title: str = dspy.OutputField()
+    headings: list[str] = dspy.OutputField()
+    entities: list[dict[str, str]] = dspy.OutputField(desc="a list of entities and their metadata")
+
+module = dspy.Predict(ExtractInfo)
+
+text = "Apple Inc. announced its latest iPhone 14 today." \
+    "The CEO, Tim Cook, highlighted its new features in a press release."
+response = module(text=text)
+
+print(response.title)
+print(response.headings)
+print(response.entities)
+```
+
+[`logfire.instrument_dspy()`][logfire.Logfire.instrument_dspy] uses the `DSPyInstrumentor().instrument()` method of
+the [`openinference-instrumentation-dspy`](https://pypi.org/project/openinference-instrumentation-dspy/) package.
diff --git a/logfire-api/logfire_api/__init__.py b/logfire-api/logfire_api/__init__.py
@@ -191,6 +191,8 @@ def instrument_google_genai(self, *args, **kwargs) -> None: ...
 
             def instrument_litellm(self, *args, **kwargs) -> None: ...
 
+            def instrument_dspy(self, *args, **kwargs) -> None: ...
+
             def instrument_aiohttp_client(self, *args, **kwargs) -> None: ...
 
             def instrument_aiohttp_server(self, *args, **kwargs) -> None: ...
@@ -229,6 +231,7 @@ def shutdown(self, *args, **kwargs) -> None: ...
         instrument_anthropic = DEFAULT_LOGFIRE_INSTANCE.instrument_anthropic
         instrument_google_genai = DEFAULT_LOGFIRE_INSTANCE.instrument_google_genai
         instrument_litellm = DEFAULT_LOGFIRE_INSTANCE.instrument_litellm
+        instrument_dspy = DEFAULT_LOGFIRE_INSTANCE.instrument_dspy
         instrument_asyncpg = DEFAULT_LOGFIRE_INSTANCE.instrument_asyncpg
         instrument_print = DEFAULT_LOGFIRE_INSTANCE.instrument_print
         instrument_celery = DEFAULT_LOGFIRE_INSTANCE.instrument_celery
diff --git a/logfire-api/logfire_api/__init__.pyi b/logfire-api/logfire_api/__init__.pyi
@@ -15,7 +15,7 @@ from logfire.propagate import attach_context as attach_context, get_context as g
 from logfire.sampling import SamplingOptions as SamplingOptions
 from typing import Any
 
-__all__ = ['Logfire', 'LogfireSpan', 'LevelName', 'AdvancedOptions', 'ConsoleOptions', 'CodeSource', 'PydanticPlugin', 'configure', 'span', 'instrument', 'log', 'trace', 'debug', 'notice', 'info', 'warn', 'warning', 'error', 'exception', 'fatal', 'force_flush', 'log_slow_async_callbacks', 'install_auto_tracing', 'instrument_asgi', 'instrument_wsgi', 'instrument_pydantic', 'instrument_pydantic_ai', 'instrument_fastapi', 'instrument_openai', 'instrument_openai_agents', 'instrument_anthropic', 'instrument_google_genai', 'instrument_litellm', 'instrument_print', 'instrument_asyncpg', 'instrument_httpx', 'instrument_celery', 'instrument_requests', 'instrument_psycopg', 'instrument_django', 'instrument_flask', 'instrument_starlette', 'instrument_aiohttp_client', 'instrument_aiohttp_server', 'instrument_sqlalchemy', 'instrument_sqlite3', 'instrument_aws_lambda', 'instrument_redis', 'instrument_pymongo', 'instrument_mysql', 'instrument_surrealdb', 'instrument_system_metrics', 'instrument_mcp', 'AutoTraceModule', 'with_tags', 'with_settings', 'suppress_scopes', 'shutdown', 'no_auto_trace', 'ScrubMatch', 'ScrubbingOptions', 'VERSION', 'add_non_user_code_prefix', 'suppress_instrumentation', 'StructlogProcessor', 'LogfireLoggingHandler', 'loguru_handler', 'SamplingOptions', 'MetricsOptions', 'logfire_info', 'get_baggage', 'set_baggage', 'get_context', 'attach_context']
+__all__ = ['Logfire', 'LogfireSpan', 'LevelName', 'AdvancedOptions', 'ConsoleOptions', 'CodeSource', 'PydanticPlugin', 'configure', 'span', 'instrument', 'log', 'trace', 'debug', 'notice', 'info', 'warn', 'warning', 'error', 'exception', 'fatal', 'force_flush', 'log_slow_async_callbacks', 'install_auto_tracing', 'instrument_asgi', 'instrument_wsgi', 'instrument_pydantic', 'instrument_pydantic_ai', 'instrument_fastapi', 'instrument_openai', 'instrument_openai_agents', 'instrument_anthropic', 'instrument_google_genai', 'instrument_litellm', 'instrument_dspy', 'instrument_print', 'instrument_asyncpg', 'instrument_httpx', 'instrument_celery', 'instrument_requests', 'instrument_psycopg', 'instrument_django', 'instrument_flask', 'instrument_starlette', 'instrument_aiohttp_client', 'instrument_aiohttp_server', 'instrument_sqlalchemy', 'instrument_sqlite3', 'instrument_aws_lambda', 'instrument_redis', 'instrument_pymongo', 'instrument_mysql', 'instrument_surrealdb', 'instrument_system_metrics', 'instrument_mcp', 'AutoTraceModule', 'with_tags', 'with_settings', 'suppress_scopes', 'shutdown', 'no_auto_trace', 'ScrubMatch', 'ScrubbingOptions', 'VERSION', 'add_non_user_code_prefix', 'suppress_instrumentation', 'StructlogProcessor', 'LogfireLoggingHandler', 'loguru_handler', 'SamplingOptions', 'MetricsOptions', 'logfire_info', 'get_baggage', 'set_baggage', 'get_context', 'attach_context']
 
 DEFAULT_LOGFIRE_INSTANCE = Logfire()
 span = DEFAULT_LOGFIRE_INSTANCE.span
@@ -33,6 +33,7 @@ instrument_openai_agents = DEFAULT_LOGFIRE_INSTANCE.instrument_openai_agents
 instrument_anthropic = DEFAULT_LOGFIRE_INSTANCE.instrument_anthropic
 instrument_google_genai = DEFAULT_LOGFIRE_INSTANCE.instrument_google_genai
 instrument_litellm = DEFAULT_LOGFIRE_INSTANCE.instrument_litellm
+instrument_dspy = DEFAULT_LOGFIRE_INSTANCE.instrument_dspy
 instrument_print = DEFAULT_LOGFIRE_INSTANCE.instrument_print
 instrument_asyncpg = DEFAULT_LOGFIRE_INSTANCE.instrument_asyncpg
 instrument_httpx = DEFAULT_LOGFIRE_INSTANCE.instrument_httpx
diff --git a/logfire/__init__.py b/logfire/__init__.py
@@ -37,6 +37,7 @@
 instrument_anthropic = DEFAULT_LOGFIRE_INSTANCE.instrument_anthropic
 instrument_google_genai = DEFAULT_LOGFIRE_INSTANCE.instrument_google_genai
 instrument_litellm = DEFAULT_LOGFIRE_INSTANCE.instrument_litellm
+instrument_dspy = DEFAULT_LOGFIRE_INSTANCE.instrument_dspy
 instrument_print = DEFAULT_LOGFIRE_INSTANCE.instrument_print
 instrument_asyncpg = DEFAULT_LOGFIRE_INSTANCE.instrument_asyncpg
 instrument_httpx = DEFAULT_LOGFIRE_INSTANCE.instrument_httpx
@@ -132,6 +133,7 @@ def loguru_handler() -> Any:
     'instrument_anthropic',
     'instrument_google_genai',
     'instrument_litellm',
+    'instrument_dspy',
     'instrument_print',
     'instrument_asyncpg',
     'instrument_httpx',
diff --git a/logfire/_internal/integrations/dspy.py b/logfire/_internal/integrations/dspy.py
@@ -0,0 +1,20 @@
+from typing import Any
+
+import logfire
+
+try:
+    from openinference.instrumentation.dspy import DSPyInstrumentor
+except ImportError:
+    raise RuntimeError(
+        'The `logfire.instrument_dspy()` method '
+        'requires the `openinference-instrumentation-dspy` package.\n'
+        'You can install this with:\n'
+        "    pip install 'logfire[dspy]'"
+    )
+
+
+def instrument_dspy(logfire_instance: logfire.Logfire, **kwargs: Any):
+    DSPyInstrumentor().instrument(
+        tracer_provider=logfire_instance.config.get_tracer_provider(),
+        **kwargs,
+    )
diff --git a/logfire/_internal/main.py b/logfire/_internal/main.py
@@ -1375,6 +1375,18 @@ def instrument_litellm(self, **kwargs: Any):
         self._warn_if_not_initialized_for_instrumentation()
         instrument_litellm(self, **kwargs)
 
+    def instrument_dspy(self, **kwargs: Any):
+        """Instrument [DSPy](https://dspy.ai/).
+
+        Uses the `DSPyInstrumentor().instrument()` method of the
+        [`openinference-instrumentation-dspy`](https://pypi.org/project/openinference-instrumentation-dspy/)
+        package, to which it passes `**kwargs`.
+        """
+        from .integrations.dspy import instrument_dspy
+
+        self._warn_if_not_initialized_for_instrumentation()
+        instrument_dspy(self, **kwargs)
+
     def instrument_print(self) -> AbstractContextManager[None]:
         """Instrument the built-in `print` function so that calls to it are logged.
 
diff --git a/mkdocs.yml b/mkdocs.yml
@@ -120,6 +120,7 @@ nav:
           - Anthropic: integrations/llms/anthropic.md
           - LangChain: integrations/llms/langchain.md
           - LiteLLM: integrations/llms/litellm.md
+          - DSPy: integrations/llms/dspy.md
           - MCP: integrations/llms/mcp.md
           - Claude Agent SDK: integrations/llms/claude-agent-sdk.md
           - LLamaIndex: integrations/llms/llamaindex.md
diff --git a/pyproject.toml b/pyproject.toml
@@ -81,6 +81,7 @@ sqlite3 = ["opentelemetry-instrumentation-sqlite3 >= 0.42b0"]
 aws-lambda = ["opentelemetry-instrumentation-aws-lambda >= 0.42b0"]
 google-genai = ["opentelemetry-instrumentation-google-genai >= 0.4b0"]
 litellm = ["openinference-instrumentation-litellm >= 0"]
+dspy = ["openinference-instrumentation-dspy >= 0"]
 
 [project.urls]
 Homepage = "https://logfire.pydantic.dev/"
@@ -183,6 +184,8 @@ dev = [
     "google-genai >= 0",
     "openinference-instrumentation-litellm >= 0",
     "litellm != 1.80.9",
+    "openinference-instrumentation-dspy >= 0; python_version >= '3.10'",
+    "dspy >= 3.1.0; python_version >= '3.10'",
     "pip >= 0",
     "surrealdb >= 0",
 ]
diff --git a/tests/otel_integrations/cassettes/test_dspy/test_dspy_instrumentation.yaml b/tests/otel_integrations/cassettes/test_dspy/test_dspy_instrumentation.yaml
diff --git a/tests/otel_integrations/test_dspy.py b/tests/otel_integrations/test_dspy.py
diff --git a/tests/test_logfire_api.py b/tests/test_logfire_api.py
diff --git a/uv.lock b/uv.lock
