26 lines
867 B
Python
26 lines
867 B
Python
# -*- coding: utf-8 -*-
|
|
"""Tests for normalized env config helpers."""
|
|
|
|
from backend.config.env_config import (
|
|
canonicalize_model_provider,
|
|
get_agent_model_config,
|
|
)
|
|
|
|
|
|
def test_canonicalize_model_provider_aliases():
|
|
assert canonicalize_model_provider("claude") == "ANTHROPIC"
|
|
assert canonicalize_model_provider("openai_compatible") == "OPENAI"
|
|
assert canonicalize_model_provider("google") == "GEMINI"
|
|
|
|
|
|
def test_get_agent_model_config_fallback(monkeypatch):
|
|
monkeypatch.delenv("AGENT_RISK_MANAGER_MODEL_NAME", raising=False)
|
|
monkeypatch.delenv("AGENT_RISK_MANAGER_MODEL_PROVIDER", raising=False)
|
|
monkeypatch.setenv("MODEL_NAME", "gpt-4o-mini")
|
|
monkeypatch.setenv("MODEL_PROVIDER", "openai")
|
|
|
|
config = get_agent_model_config("risk_manager")
|
|
|
|
assert config.model_name == "gpt-4o-mini"
|
|
assert config.provider == "OPENAI"
|