diff --git a/run_agent.py b/run_agent.py index 7187499bb1..8db343a2a5 100644 --- a/run_agent.py +++ b/run_agent.py @@ -2399,6 +2399,7 @@ class AIAgent: base_url=aux_base_url, api_key=aux_api_key, config_context_length=getattr(self, "_aux_compression_context_length_config", None), + provider=getattr(self, "provider", ""), ) # Hard floor: the auxiliary compression model must have at least diff --git a/tests/run_agent/test_compression_feasibility.py b/tests/run_agent/test_compression_feasibility.py index 2050bee28e..7b4b7da612 100644 --- a/tests/run_agent/test_compression_feasibility.py +++ b/tests/run_agent/test_compression_feasibility.py @@ -184,6 +184,7 @@ def test_feasibility_check_passes_config_context_length(mock_get_client, mock_ct base_url="http://custom-endpoint:8080/v1", api_key="sk-custom", config_context_length=1_000_000, + provider="openrouter", ) @@ -206,6 +207,7 @@ def test_feasibility_check_ignores_invalid_context_length(mock_get_client, mock_ base_url="http://custom:8080/v1", api_key="sk-test", config_context_length=None, + provider="openrouter", ) @@ -258,6 +260,7 @@ def test_init_feasibility_check_uses_aux_context_override_from_config(): base_url="http://custom-endpoint:8080/v1", api_key="sk-custom", config_context_length=1_000_000, + provider="", )