Initial commit of integrated agent system

This commit is contained in:
cillin
2026-03-30 17:46:44 +08:00
commit 0fa413380c
337 changed files with 75268 additions and 0 deletions

View File

@@ -0,0 +1,25 @@
# -*- 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"