fixing dbos test

Author: adtyavrdhn

tests/test_dbos.py

@@ -1401,6 +1401,7 @@ async def hitl_main_loop(prompt: str) -> AgentRunResult[str | DeferredToolReques
                     )
                 ],
                 instructions='Just call tools without asking for confirmation.',
+                run_id=IsStr(),
             ),
             ModelResponse(
                 parts=[
@@ -1431,6 +1432,7 @@ async def hitl_main_loop(prompt: str) -> AgentRunResult[str | DeferredToolReques
                 provider_details={'finish_reason': 'tool_calls'},
                 provider_response_id=IsStr(),
                 finish_reason='tool_call',
+                run_id=IsStr(),
             ),
             ModelRequest(
                 parts=[
@@ -1448,6 +1450,7 @@ async def hitl_main_loop(prompt: str) -> AgentRunResult[str | DeferredToolReques
                     ),
                 ],
                 instructions='Just call tools without asking for confirmation.',
+                run_id=IsStr(),
             ),
             ModelResponse(
                 parts=[
@@ -1469,6 +1472,7 @@ async def hitl_main_loop(prompt: str) -> AgentRunResult[str | DeferredToolReques
                 provider_details={'finish_reason': 'stop'},
                 provider_response_id=IsStr(),
                 finish_reason='stop',
+                run_id=IsStr(),
             ),
         ]
     )
@@ -1528,6 +1532,7 @@ def hitl_main_loop_sync(prompt: str) -> AgentRunResult[str | DeferredToolRequest
                     )
                 ],
                 instructions='Just call tools without asking for confirmation.',
+                run_id=IsStr(),
             ),
             ModelResponse(
                 parts=[
@@ -1558,6 +1563,7 @@ def hitl_main_loop_sync(prompt: str) -> AgentRunResult[str | DeferredToolRequest
                 provider_details={'finish_reason': 'tool_calls'},
                 provider_response_id=IsStr(),
                 finish_reason='tool_call',
+                run_id=IsStr(),
             ),
             ModelRequest(
                 parts=[
@@ -1575,6 +1581,7 @@ def hitl_main_loop_sync(prompt: str) -> AgentRunResult[str | DeferredToolRequest
                     ),
                 ],
                 instructions='Just call tools without asking for confirmation.',
+                run_id=IsStr(),
             ),
             ModelResponse(
                 parts=[
@@ -1596,6 +1603,7 @@ def hitl_main_loop_sync(prompt: str) -> AgentRunResult[str | DeferredToolRequest
                 provider_details={'finish_reason': 'stop'},
                 provider_response_id=IsStr(),
                 finish_reason='stop',
+                run_id=IsStr(),
             ),
         ]
     )
@@ -1629,7 +1637,8 @@ async def test_dbos_agent_with_model_retry(allow_model_requests: None, dbos: DBO
                         content='What is the weather in CDMX?',
                         timestamp=IsDatetime(),
                     )
-                ]
+                ],
+                run_id=IsStr(),
             ),
             ModelResponse(
                 parts=[
@@ -1655,6 +1664,7 @@ async def test_dbos_agent_with_model_retry(allow_model_requests: None, dbos: DBO
                 provider_details={'finish_reason': 'tool_calls'},
                 provider_response_id=IsStr(),
                 finish_reason='tool_call',
+                run_id=IsStr(),
             ),
             ModelRequest(
                 parts=[
@@ -1664,7 +1674,8 @@ async def test_dbos_agent_with_model_retry(allow_model_requests: None, dbos: DBO
                         tool_call_id=IsStr(),
                         timestamp=IsDatetime(),
                     )
-                ]
+                ],
+                run_id=IsStr(),
             ),
             ModelResponse(
                 parts=[
@@ -1690,6 +1701,7 @@ async def test_dbos_agent_with_model_retry(allow_model_requests: None, dbos: DBO
                 provider_details={'finish_reason': 'tool_calls'},
                 provider_response_id=IsStr(),
                 finish_reason='tool_call',
+                run_id=IsStr(),
             ),
             ModelRequest(
                 parts=[
@@ -1699,7 +1711,8 @@ async def test_dbos_agent_with_model_retry(allow_model_requests: None, dbos: DBO
                         tool_call_id=IsStr(),
                         timestamp=IsDatetime(),
                     )
-                ]
+                ],
+                run_id=IsStr(),
             ),
             ModelResponse(
                 parts=[TextPart(content='The weather in Mexico City is currently sunny.')],
@@ -1719,6 +1732,7 @@ async def test_dbos_agent_with_model_retry(allow_model_requests: None, dbos: DBO
                 provider_details={'finish_reason': 'stop'},
                 provider_response_id=IsStr(),
                 finish_reason='stop',
+                run_id=IsStr(),
             ),
         ]
     )