test: isolate context-length metadata stub

This commit is contained in:
b3nw committed 2026-06-12 16:18:50 +00:00
1 parent b2eda41b49
commit 8ee8523340
1 file changed
+14 -2
@@ -21,6 +21,8 @@ config-override args again.
"""
from pathlib import Path
import sys
import types
REPO = Path(__file__).resolve().parent.parent
@@ -322,7 +324,17 @@ def test_routes_session_model_resolver_passes_custom_provider_api_key(monkeypatc
"""
from api import config as cfg_mod
from api import routes
import agent.model_metadata as metadata
# Some unrelated route tests install ``sys.modules["agent"]`` as a plain
# module stub at collection time. Make this regression test own a package-
# shaped temporary ``agent.model_metadata`` import target so the production
# helper's local import is order-independent.
fake_agent = types.ModuleType("agent")
fake_agent.__path__ = []
metadata = types.ModuleType("agent.model_metadata")
fake_agent.model_metadata = metadata # type: ignore[attr-defined]
monkeypatch.setitem(sys.modules, "agent", fake_agent)
monkeypatch.setitem(sys.modules, "agent.model_metadata", metadata)
seen = {}
@@ -345,7 +357,7 @@ def test_routes_session_model_resolver_passes_custom_provider_api_key(monkeypatc
seen.update(model=model, base_url=base_url, kwargs=kwargs)
return 500_000 if kwargs.get("api_key") == "sk-test-route" else 256_000
monkeypatch.setattr(metadata, "get_model_context_length", fake_get_model_context_length)
monkeypatch.setattr(metadata, "get_model_context_length", fake_get_model_context_length, raising=False)
assert routes._resolve_context_length_for_session_model(
"custom-model-id",