Skip to content

Commit e116ada

Browse files
committed
Fix tests
1 parent 9b6a08b commit e116ada

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/contrib/openai_agents/test_openai.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -431,6 +431,7 @@ async def test_tool_workflow(client: Client, use_local_model: bool):
431431
get_weather_object,
432432
get_weather_country,
433433
get_weather_context,
434+
ActivityWeatherService().get_weather_method,
434435
],
435436
) as worker:
436437
workflow_handle = await client.start_workflow(
@@ -535,7 +536,7 @@ async def test_nexus_tool_workflow(
535536
model_params=ModelActivityParameters(
536537
start_to_close_timeout=timedelta(seconds=30)
537538
),
538-
model_provider=TestModelProvider(TestResearchModel())
539+
model_provider=TestModelProvider(TestNexusWeatherModel())
539540
if use_local_model
540541
else None,
541542
)

0 commit comments

Comments
 (0)