Update backend schema

api/tests/test_assistants.py

@@ -0,0 +1,168 @@
+"""Tests for Assistant API endpoints"""
+import pytest
+import uuid
+
+
+class TestAssistantAPI:
+    """Test cases for Assistant endpoints"""
+
+    def test_get_assistants_empty(self, client):
+        """Test getting assistants when database is empty"""
+        response = client.get("/api/assistants")
+        assert response.status_code == 200
+        data = response.json()
+        assert "total" in data
+        assert "list" in data
+
+    def test_create_assistant(self, client, sample_assistant_data):
+        """Test creating a new assistant"""
+        response = client.post("/api/assistants", json=sample_assistant_data)
+        assert response.status_code == 200
+        data = response.json()
+        assert data["name"] == sample_assistant_data["name"]
+        assert data["opener"] == sample_assistant_data["opener"]
+        assert data["prompt"] == sample_assistant_data["prompt"]
+        assert data["language"] == sample_assistant_data["language"]
+        assert "id" in data
+        assert data["callCount"] == 0
+
+    def test_create_assistant_minimal(self, client):
+        """Test creating an assistant with minimal required data"""
+        data = {"name": "Minimal Assistant"}
+        response = client.post("/api/assistants", json=data)
+        assert response.status_code == 200
+        assert response.json()["name"] == "Minimal Assistant"
+
+    def test_get_assistant_by_id(self, client, sample_assistant_data):
+        """Test getting a specific assistant by ID"""
+        # Create first
+        create_response = client.post("/api/assistants", json=sample_assistant_data)
+        assistant_id = create_response.json()["id"]
+
+        # Get by ID
+        response = client.get(f"/api/assistants/{assistant_id}")
+        assert response.status_code == 200
+        data = response.json()
+        assert data["id"] == assistant_id
+        assert data["name"] == sample_assistant_data["name"]
+
+    def test_get_assistant_not_found(self, client):
+        """Test getting a non-existent assistant"""
+        response = client.get("/api/assistants/non-existent-id")
+        assert response.status_code == 404
+
+    def test_update_assistant(self, client, sample_assistant_data):
+        """Test updating an assistant"""
+        # Create first
+        create_response = client.post("/api/assistants", json=sample_assistant_data)
+        assistant_id = create_response.json()["id"]
+
+        # Update
+        update_data = {
+            "name": "Updated Assistant",
+            "prompt": "You are an updated assistant.",
+            "speed": 1.5
+        }
+        response = client.put(f"/api/assistants/{assistant_id}", json=update_data)
+        assert response.status_code == 200
+        data = response.json()
+        assert data["name"] == "Updated Assistant"
+        assert data["prompt"] == "You are an updated assistant."
+        assert data["speed"] == 1.5
+
+    def test_delete_assistant(self, client, sample_assistant_data):
+        """Test deleting an assistant"""
+        # Create first
+        create_response = client.post("/api/assistants", json=sample_assistant_data)
+        assistant_id = create_response.json()["id"]
+
+        # Delete
+        response = client.delete(f"/api/assistants/{assistant_id}")
+        assert response.status_code == 200
+
+        # Verify deleted
+        get_response = client.get(f"/api/assistants/{assistant_id}")
+        assert get_response.status_code == 404
+
+    def test_list_assistants_with_pagination(self, client, sample_assistant_data):
+        """Test listing assistants with pagination"""
+        # Create multiple assistants
+        for i in range(3):
+            data = sample_assistant_data.copy()
+            data["name"] = f"Assistant {i}"
+            client.post("/api/assistants", json=data)
+
+        # Test pagination
+        response = client.get("/api/assistants?page=1&limit=2")
+        assert response.status_code == 200
+        data = response.json()
+        assert data["total"] == 3
+        assert len(data["list"]) == 2
+
+    def test_create_assistant_with_voice(self, client, sample_assistant_data, sample_voice_data):
+        """Test creating an assistant with a voice reference"""
+        # Create a voice first
+        voice_response = client.post("/api/voices", json=sample_voice_data)
+        voice_id = voice_response.json()["id"]
+
+        # Create assistant with voice
+        sample_assistant_data["voice"] = voice_id
+        response = client.post("/api/assistants", json=sample_assistant_data)
+        assert response.status_code == 200
+        assert response.json()["voice"] == voice_id
+
+    def test_create_assistant_with_knowledge_base(self, client, sample_assistant_data):
+        """Test creating an assistant with knowledge base reference"""
+        # Note: This test assumes knowledge base doesn't exist
+        sample_assistant_data["knowledgeBaseId"] = "non-existent-kb"
+        response = client.post("/api/assistants", json=sample_assistant_data)
+        assert response.status_code == 200
+        assert response.json()["knowledgeBaseId"] == "non-existent-kb"
+
+    def test_assistant_with_model_references(self, client, sample_assistant_data):
+        """Test creating assistant with model references"""
+        sample_assistant_data.update({
+            "llmModelId": "llm-001",
+            "asrModelId": "asr-001",
+            "embeddingModelId": "emb-001",
+            "rerankModelId": "rerank-001"
+        })
+        response = client.post("/api/assistants", json=sample_assistant_data)
+        assert response.status_code == 200
+        data = response.json()
+        assert data["llmModelId"] == "llm-001"
+        assert data["asrModelId"] == "asr-001"
+        assert data["embeddingModelId"] == "emb-001"
+        assert data["rerankModelId"] == "rerank-001"
+
+    def test_assistant_with_tools(self, client, sample_assistant_data):
+        """Test creating assistant with tools"""
+        sample_assistant_data["tools"] = ["weather", "calculator", "search"]
+        response = client.post("/api/assistants", json=sample_assistant_data)
+        assert response.status_code == 200
+        assert response.json()["tools"] == ["weather", "calculator", "search"]
+
+    def test_assistant_with_hotwords(self, client, sample_assistant_data):
+        """Test creating assistant with hotwords"""
+        sample_assistant_data["hotwords"] = ["hello", "help", "stop"]
+        response = client.post("/api/assistants", json=sample_assistant_data)
+        assert response.status_code == 200
+        assert response.json()["hotwords"] == ["hello", "help", "stop"]
+
+    def test_different_config_modes(self, client, sample_assistant_data):
+        """Test creating assistants with different config modes"""
+        for mode in ["platform", "dify", "fastgpt", "none"]:
+            sample_assistant_data["name"] = f"Assistant {mode}"
+            sample_assistant_data["configMode"] = mode
+            response = client.post("/api/assistants", json=sample_assistant_data)
+            assert response.status_code == 200
+            assert response.json()["configMode"] == mode
+
+    def test_different_languages(self, client, sample_assistant_data):
+        """Test creating assistants with different languages"""
+        for lang in ["zh", "en", "ja", "ko"]:
+            sample_assistant_data["name"] = f"Assistant {lang}"
+            sample_assistant_data["language"] = lang
+            response = client.post("/api/assistants", json=sample_assistant_data)
+            assert response.status_code == 200
+            assert response.json()["language"] == lang