fix(providers): send user agent to routermint endpoints

This commit is contained in:
Reginaldas 2026-04-23 10:32:40 +03:00 committed by Teknium
parent 5fdba79eb4
commit 3e10f339fd
4 changed files with 44 additions and 3 deletions

View file

@ -47,6 +47,24 @@ def test_ai_gateway_base_url_applies_attribution_headers(mock_openai):
assert headers["User-Agent"].startswith("HermesAgent/")
@patch("run_agent.OpenAI")
def test_routermint_base_url_applies_user_agent_header(mock_openai):
mock_openai.return_value = MagicMock()
agent = AIAgent(
api_key="test-key",
base_url="https://api.routermint.com/v1",
model="test/model",
quiet_mode=True,
skip_context_files=True,
skip_memory=True,
)
agent._apply_client_headers_for_base_url("https://api.routermint.com/v1")
headers = agent._client_kwargs["default_headers"]
assert headers["User-Agent"].startswith("HermesAgent/")
@patch("run_agent.OpenAI")
def test_unknown_base_url_clears_default_headers(mock_openai):
mock_openai.return_value = MagicMock()