first commit

backend/tests/test_model_adapters.py

@@ -0,0 +1,26 @@
+"""Tests for model adapters."""
+
+import pytest
+
+from app.model_adapters.base import ModelAdapter, ChatResponse
+from app.model_adapters.providers import DeepSeekAdapter
+
+
+def test_adapter_payload_and_extraction():
+    adapter = DeepSeekAdapter(api_key="test", model="deepseek-chat", base_url="https://x")
+    payload = adapter._build_payload("hello", {"temperature": 0.5, "max_tokens": 100})
+    assert payload["model"] == "deepseek-chat"
+    assert payload["messages"][0]["content"] == "hello"
+
+    response_text = adapter._extract_text(
+        {"choices": [{"message": {"content": "hi"}}]}
+    )
+    assert response_text == "hi"
+
+
+def test_adapter_token_usage():
+    adapter = DeepSeekAdapter(api_key="test", model="m", base_url="https://x")
+    usage = adapter._extract_token_usage(
+        {"usage": {"prompt_tokens": 10, "completion_tokens": 5, "total_tokens": 15}}
+    )
+    assert usage["total_tokens"] == 15