File tree Expand file tree Collapse file tree 2 files changed +1
-16
lines changed
pydantic_ai_slim/pydantic_ai/durable_exec/dbos Expand file tree Collapse file tree 2 files changed +1
-16
lines changed Original file line number Diff line number Diff line change @@ -640,7 +640,7 @@ async def main():
640640 Returns:
641641 The result of the run.
642642 """
643- if model is not None and not isinstance (model , ( DBOSModel ) ):
643+ if model is not None and not isinstance (model , DBOSModel ):
644644 raise UserError (
645645 'Non-DBOS model cannot be set at agent run time inside a DBOS workflow, it must be set at agent creation time.'
646646 )
Original file line number Diff line number Diff line change @@ -1012,16 +1012,6 @@ async def run_with_deps():
10121012 assert output == snapshot ('The capital of Mexico is Mexico City.' )
10131013
10141014
1015- async def test_dbos_agent_sync_tool_activity_disabled ():
1016- # Not a valid test for DBOS
1017- pass
1018-
1019-
1020- async def test_dbos_agent_mcp_server_activity_disabled ():
1021- # Not a valid test for DBOS
1022- pass
1023-
1024-
10251015async def test_dbos_model_stream_direct (allow_model_requests : None , dbos : DBOS ):
10261016 @DBOS .workflow ()
10271017 async def run_model_stream ():
@@ -1065,11 +1055,6 @@ async def test_dbos_agent_with_unserializable_deps_type(allow_model_requests: No
10651055 await unserializable_deps_dbos_agent .run ('What is the model name?' , deps = UnserializableDeps (client = client ))
10661056
10671057
1068- async def test_logfire_plugin ():
1069- # Not a valid test for DBOS, as we don't need the LogfirePlugin.
1070- pass
1071-
1072-
10731058# Test dynamic toolsets in an agent with DBOS
10741059
10751060
You can’t perform that action at this time.
0 commit comments