23 lines
841 B
Python
23 lines
841 B
Python
"""Tests for src.llm.registry helpers."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from src.llm.registry import _default_headers_for # pyright: ignore[reportPrivateUsage]
|
|
|
|
|
|
def test_default_headers_for_openrouter_base_url() -> None:
|
|
"""OpenRouter base URLs get the app-attribution headers."""
|
|
headers = _default_headers_for("https://openrouter.ai/api/v1")
|
|
assert headers["HTTP-Referer"] == "https://honcho.dev"
|
|
assert headers["X-Openrouter-Title"] == "Honcho"
|
|
|
|
|
|
def test_default_headers_for_non_openrouter_base_url() -> None:
|
|
"""Other OpenAI-compatible providers get no extra headers."""
|
|
assert _default_headers_for("https://api.openai.com/v1") == {}
|
|
|
|
|
|
def test_default_headers_for_none_base_url() -> None:
|
|
"""A missing base URL (default OpenAI) gets no extra headers."""
|
|
assert _default_headers_for(None) == {}
|