Update docs and tests for DBOS v2.0 (#3004)

Author: qianl15

docs/durable_execution/dbos.md

@@ -123,7 +123,7 @@ Other than that, any agent and toolset will just work!
 
 ### Agent Run Context and Dependencies
 
-DBOS checkpoints workflow inputs/outputs and step outputs into a database using `jsonpickle`. This means you need to make sure [dependencies](../dependencies.md) object provided to [`DBOSAgent.run()`][pydantic_ai.durable_exec.dbos.DBOSAgent.run] or [`DBOSAgent.run_sync()`][pydantic_ai.durable_exec.dbos.DBOSAgent.run_sync], and tool outputs can be serialized using jsonpickle. You may also want to keep the inputs and outputs small (under \~2 MB). PostgreSQL and SQLite support up to 1 GB per field, but large objects may impact performance.
+DBOS checkpoints workflow inputs/outputs and step outputs into a database using [`pickle`](https://docs.python.org/3/library/pickle.html). This means you need to make sure [dependencies](../dependencies.md) object provided to [`DBOSAgent.run()`][pydantic_ai.durable_exec.dbos.DBOSAgent.run] or [`DBOSAgent.run_sync()`][pydantic_ai.durable_exec.dbos.DBOSAgent.run_sync], and tool outputs can be serialized using pickle. You may also want to keep the inputs and outputs small (under \~2 MB). PostgreSQL and SQLite support up to 1 GB per field, but large objects may impact performance.
 
 ### Streaming
 
@@ -153,6 +153,6 @@ You can customize DBOS's retry policy using [step configuration](#step-configura
 
 ## Observability with Logfire
 
-DBOS automatically generates OpenTelemetry spans for each workflow and step execution, and Pydantic AI emits spans for each agent run, model request, and tool invocation. You can send these spans to [Pydantic Logfire](../logfire.md) to get a full, end-to-end view of what's happening in your application.
+DBOS can be configured to generate OpenTelemetry spans for each workflow and step execution, and Pydantic AI emits spans for each agent run, model request, and tool invocation. You can send these spans to [Pydantic Logfire](../logfire.md) to get a full, end-to-end view of what's happening in your application.
 
 For more information about DBOS logging and tracing, please see the [DBOS docs](https://docs.dbos.dev/python/tutorials/logging-and-tracing) for details.

pydantic_ai_slim/pydantic_ai/mcp.py

@@ -540,7 +540,7 @@ def __repr__(self) -> str:
             f'args={self.args!r}',
         ]
         if self.id:
-            repr_args.append(f'id={self.id!r}')
+            repr_args.append(f'id={self.id!r}')  # pragma: lax no cover
         return f'{self.__class__.__name__}({", ".join(repr_args)})'
 
     def __eq__(self, value: object, /) -> bool:

tests/test_dbos.py

@@ -43,9 +43,14 @@
 from .conftest import IsDatetime, IsStr
 
 try:
+    import importlib.metadata
+
     from dbos import DBOS, DBOSConfig, SetWorkflowID
+    from packaging.version import Version
 
     from pydantic_ai.durable_exec.dbos import DBOSAgent, DBOSMCPServer, DBOSModel
+
+    dbos_version = Version(importlib.metadata.version('dbos'))
 except ImportError:  # pragma: lax no cover
     pytest.skip('DBOS is not installed', allow_module_level=True)
 
@@ -110,9 +115,10 @@ def workflow_raises(exc_type: type[Exception], exc_message: str) -> Iterator[Non
 DBOS_SQLITE_FILE = 'dbostest.sqlite'
 DBOS_CONFIG: DBOSConfig = {
     'name': 'pydantic_dbos_tests',
-    'database_url': f'sqlite:///{DBOS_SQLITE_FILE}',
     'system_database_url': f'sqlite:///{DBOS_SQLITE_FILE}',
     'run_admin_server': False,
+    # enable_otlp requires dbos>1.14
+    'enable_otlp': True,
 }
 
 
@@ -1042,9 +1048,19 @@ async def simple_event_stream_handler(
     ):
         pass
 
-    with workflow_raises(TypeError, snapshot('Serialized function should be defined at the top level of a module')):
+    with pytest.raises(Exception) as exc_info:
         await simple_dbos_agent.run('What is the capital of Mexico?', event_stream_handler=simple_event_stream_handler)
 
+    if dbos_version <= Version('1.14'):  # pragma: lax no cover
+        # Older DBOS versions used jsonpickle
+        assert str(exc_info.value) == snapshot('Serialized function should be defined at the top level of a module')
+    else:
+        # Newer DBOS versions use pickle
+        assert (
+            "local object 'test_dbos_agent_run_in_workflow_with_event_stream_handler.<locals>.simple_event_stream_handler'"
+            in str(exc_info.value)
+        )
+
 
 async def test_dbos_agent_run_in_workflow_with_model(allow_model_requests: None, dbos: DBOS):
     # A non-DBOS model is not wrapped as steps so it's not deterministic and cannot be used in a DBOS workflow.
@@ -1142,15 +1158,19 @@ async def get_model_name(ctx: RunContext[UnserializableDeps]) -> int:
 async def test_dbos_agent_with_unserializable_deps_type(allow_model_requests: None, dbos: DBOS):
     unserializable_deps_dbos_agent = DBOSAgent(unserializable_deps_agent)
     # Test this raises a serialization error because httpx.AsyncClient is not serializable.
-    with pytest.raises(
-        Exception,
-        match='object proxy must define __reduce_ex__()',
-    ):
+    with pytest.raises(Exception) as exc_info:
         async with AsyncClient() as client:
             # This will trigger the client to be unserializable
             logfire.instrument_httpx(client, capture_all=True)
             await unserializable_deps_dbos_agent.run('What is the model name?', deps=UnserializableDeps(client=client))
 
+    if dbos_version <= Version('1.14'):  # pragma: lax no cover
+        # Older DBOS versions used jsonpickle
+        assert str(exc_info.value) == snapshot('object proxy must define __reduce_ex__()')
+    else:
+        # Newer DBOS versions use pickle
+        assert str(exc_info.value) == snapshot("cannot pickle '_thread.RLock' object")
+
 
 # Test dynamic toolsets in an agent with DBOS