Skip to content

Commit 5379ef4

Browse files
committed
Address comments
1 parent 04e3fb5 commit 5379ef4

File tree

2 files changed

+1
-16
lines changed

2 files changed

+1
-16
lines changed

pydantic_ai_slim/pydantic_ai/durable_exec/dbos/_agent.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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
)

tests/test_dbos.py

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff 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-
10251015
async 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

0 commit comments

Comments
 (0)