diff --git a/tests/run_agent/test_run_agent.py b/tests/run_agent/test_run_agent.py index 2dc46c1d785b..7aec0e3dde25 100644 --- a/tests/run_agent/test_run_agent.py +++ b/tests/run_agent/test_run_agent.py @@ -3973,7 +3973,9 @@ class TestHandleMaxIterations: kwargs = agent.client.chat.completions.create.call_args.kwargs from agent.portal_tags import nous_portal_tags - assert kwargs["extra_body"]["tags"] == nous_portal_tags() + assert kwargs["extra_body"]["tags"] == nous_portal_tags( + session_id=agent.session_id + ) assert kwargs["extra_body"]["provider"] == { "only": ["deepseek"], "ignore": ["deepinfra"], @@ -3995,7 +3997,9 @@ class TestHandleMaxIterations: kwargs = agent.client.chat.completions.create.call_args.kwargs from agent.portal_tags import nous_portal_tags - assert kwargs["extra_body"] == {"tags": nous_portal_tags()} + assert kwargs["extra_body"] == { + "tags": nous_portal_tags(session_id=agent.session_id) + } def test_summary_drops_invalid_provider_sort(self, agent): agent.base_url = "https://openrouter.ai/api/v1"