初始化提交

backend/tests/ai_provider/config/test_models.py

@@ -0,0 +1,69 @@
+import pytest
+
+from app.ai_provider.config.models import (
+    ModelConfig,
+    ProviderConfig,
+    derive_capabilities,
+)
+
+
+class TestModelConfig:
+    def test_minimal_config(self):
+        cfg = ModelConfig(
+            name="deepseek-v4-flash",
+            provider="deepseek",
+            model="deepseek-v4-flash",
+            format="openai",
+        )
+        assert cfg.name == "deepseek-v4-flash"
+        assert cfg.format == "openai"
+        assert not cfg.supports_vision  # defaulted
+
+    def test_capability_override(self):
+        cfg = ModelConfig(
+            name="claude-sonnet",
+            provider="anthropic",
+            model="claude-sonnet-4-6",
+            format="anthropic",
+            supports_vision=True,
+            supports_tools=True,
+        )
+        assert cfg.supports_vision
+        assert cfg.supports_tools
+
+    def test_capability_auto_derive(self):
+        caps = derive_capabilities(provider="anthropic", model="claude-fable-5")
+        assert "chat" in caps
+        assert "tool_use" in caps
+        assert "vision" in caps
+        assert "reasoning" in caps
+
+
+class TestProviderConfig:
+    def test_minimal(self):
+        cfg = ProviderConfig(
+            name="deepseek",
+            format="openai",
+            base_url_env="DEEPSEEK_BASE_URL",
+            api_key_env="DEEPSEEK_API_KEY",
+        )
+        assert cfg.name == "deepseek"
+
+    def test_with_parameter_mapping(self):
+        cfg = ProviderConfig(
+            name="deepseek",
+            format="openai",
+            base_url_env="DEEPSEEK_BASE_URL",
+            api_key_env="DEEPSEEK_API_KEY",
+            parameter_mappings={
+                "max_tokens": {"provider_param": "max_tokens"},
+                "temperature": {"provider_param": "temperature", "min": 0, "max": 2},
+            },
+        )
+        mapping = cfg.parameter_mappings["temperature"]
+        assert mapping.min == 0
+        assert mapping.max == 2
+
+    def test_format_validation(self):
+        with pytest.raises(ValueError, match="format"):
+            ProviderConfig(name="bad", format="unknown", base_url_env="X", api_key_env="Y")