228 lines
10 KiB
Python
228 lines
10 KiB
Python
"""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 data["voiceOutputEnabled"] is True
|
|
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,
|
|
"voiceOutputEnabled": False,
|
|
}
|
|
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
|
|
assert data["voiceOutputEnabled"] is False
|
|
|
|
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"
|
|
|
|
assistant_id = response.json()["id"]
|
|
runtime_resp = client.get(f"/api/assistants/{assistant_id}/runtime-config")
|
|
assert runtime_resp.status_code == 200
|
|
metadata = runtime_resp.json()["sessionStartMetadata"]
|
|
assert metadata["knowledgeBaseId"] == "non-existent-kb"
|
|
assert metadata["knowledge"]["enabled"] is True
|
|
assert metadata["knowledge"]["kbId"] == "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
|
|
|
|
def test_get_runtime_config(self, client, sample_assistant_data, sample_llm_model_data, sample_asr_model_data, sample_voice_data):
|
|
"""Test resolved runtime config endpoint for WS session.start metadata."""
|
|
llm_resp = client.post("/api/llm", json=sample_llm_model_data)
|
|
assert llm_resp.status_code == 200
|
|
|
|
asr_resp = client.post("/api/asr", json=sample_asr_model_data)
|
|
assert asr_resp.status_code == 200
|
|
|
|
voice_resp = client.post("/api/voices", json=sample_voice_data)
|
|
assert voice_resp.status_code == 200
|
|
voice_id = voice_resp.json()["id"]
|
|
|
|
sample_assistant_data.update({
|
|
"llmModelId": sample_llm_model_data["id"],
|
|
"asrModelId": sample_asr_model_data["id"],
|
|
"voice": voice_id,
|
|
"prompt": "runtime prompt",
|
|
"opener": "runtime opener",
|
|
"speed": 1.1,
|
|
})
|
|
assistant_resp = client.post("/api/assistants", json=sample_assistant_data)
|
|
assert assistant_resp.status_code == 200
|
|
assistant_id = assistant_resp.json()["id"]
|
|
|
|
runtime_resp = client.get(f"/api/assistants/{assistant_id}/runtime-config")
|
|
assert runtime_resp.status_code == 200
|
|
payload = runtime_resp.json()
|
|
|
|
assert payload["assistantId"] == assistant_id
|
|
metadata = payload["sessionStartMetadata"]
|
|
assert metadata["systemPrompt"] == "runtime prompt"
|
|
assert metadata["greeting"] == "runtime opener"
|
|
assert metadata["services"]["llm"]["model"] == sample_llm_model_data["model_name"]
|
|
assert metadata["services"]["asr"]["model"] == sample_asr_model_data["model_name"]
|
|
assert metadata["services"]["tts"]["voice"] == sample_voice_data["voice_key"]
|
|
|
|
def test_runtime_config_text_mode_when_voice_output_disabled(self, client, sample_assistant_data):
|
|
sample_assistant_data["voiceOutputEnabled"] = False
|
|
assistant_resp = client.post("/api/assistants", json=sample_assistant_data)
|
|
assert assistant_resp.status_code == 200
|
|
assistant_id = assistant_resp.json()["id"]
|
|
|
|
runtime_resp = client.get(f"/api/assistants/{assistant_id}/runtime-config")
|
|
assert runtime_resp.status_code == 200
|
|
metadata = runtime_resp.json()["sessionStartMetadata"]
|
|
assert metadata["output"]["mode"] == "text"
|
|
assert metadata["services"]["tts"]["enabled"] is False
|