diff --git a/tests/test_issue1897_profile_switch_agent_cache.py b/tests/test_issue1897_profile_switch_agent_cache.py index 97190f66..7c6782f7 100644 --- a/tests/test_issue1897_profile_switch_agent_cache.py +++ b/tests/test_issue1897_profile_switch_agent_cache.py @@ -156,7 +156,7 @@ def test_same_session_profile_switch_rebuilds_agent_under_new_soul_home(tmp_path monkeypatch.setattr( streaming, "resolve_model_provider", - lambda _model: ("test-model", "test-provider", None), + lambda _model, **_kw: ("test-model", "test-provider", None), ) monkeypatch.setattr(streaming, "_maybe_schedule_title_refresh", lambda *args, **kwargs: None) monkeypatch.setattr(profiles, "get_hermes_home_for_profile", home_for_profile) diff --git a/tests/test_issue2965_streaming_sessiondb_profile_home.py b/tests/test_issue2965_streaming_sessiondb_profile_home.py index 7988792c..520eca3e 100644 --- a/tests/test_issue2965_streaming_sessiondb_profile_home.py +++ b/tests/test_issue2965_streaming_sessiondb_profile_home.py @@ -129,7 +129,7 @@ def test_streaming_sessiondb_uses_session_profile_state_db(tmp_path, monkeypatch monkeypatch.setattr( streaming, "resolve_model_provider", - lambda _model: ("test-model", "test-provider", None), + lambda _model, **_kw: ("test-model", "test-provider", None), ) monkeypatch.setattr(streaming, "_maybe_schedule_title_refresh", lambda *args, **kwargs: None) monkeypatch.setattr(profiles, "get_hermes_home_for_profile", lambda _profile: profile_home)