Update backend api

api/app/routers/assistants.py

@@ -1,6 +1,6 @@
 from fastapi import APIRouter, Depends, HTTPException
 from sqlalchemy.orm import Session
-from typing import List
+from typing import List, Optional
 import uuid
 from datetime import datetime
 
@@ -8,7 +8,7 @@ from ..db import get_db
 from ..models import Assistant, Voice, Workflow
 from ..schemas import (
     AssistantCreate, AssistantUpdate, AssistantOut,
-    VoiceOut,
+    VoiceCreate, VoiceUpdate, VoiceOut,
     WorkflowCreate, WorkflowUpdate, WorkflowOut
 )
 
@@ -16,11 +16,88 @@ router = APIRouter()
 
 
 # ============ Voices ============
-@router.get("/voices", response_model=List[VoiceOut])
-def list_voices(db: Session = Depends(get_db)):
+@router.get("/voices")
+def list_voices(
+    vendor: Optional[str] = None,
+    language: Optional[str] = None,
+    gender: Optional[str] = None,
+    page: int = 1,
+    limit: int = 50,
+    db: Session = Depends(get_db)
+):
     """获取声音库列表"""
-    voices = db.query(Voice).all()
-    return voices
+    query = db.query(Voice)
+    if vendor:
+        query = query.filter(Voice.vendor == vendor)
+    if language:
+        query = query.filter(Voice.language == language)
+    if gender:
+        query = query.filter(Voice.gender == gender)
+
+    total = query.count()
+    voices = query.order_by(Voice.created_at.desc()) \
+        .offset((page-1)*limit).limit(limit).all()
+    return {"total": total, "page": page, "limit": limit, "list": voices}
+
+
+@router.post("/voices", response_model=VoiceOut)
+def create_voice(data: VoiceCreate, db: Session = Depends(get_db)):
+    """创建声音"""
+    voice = Voice(
+        id=data.id or str(uuid.uuid4())[:8],
+        user_id=1,
+        name=data.name,
+        vendor=data.vendor,
+        gender=data.gender,
+        language=data.language,
+        description=data.description,
+        model=data.model,
+        voice_key=data.voice_key,
+        speed=data.speed,
+        gain=data.gain,
+        pitch=data.pitch,
+        enabled=data.enabled,
+    )
+    db.add(voice)
+    db.commit()
+    db.refresh(voice)
+    return voice
+
+
+@router.get("/voices/{id}", response_model=VoiceOut)
+def get_voice(id: str, db: Session = Depends(get_db)):
+    """获取单个声音详情"""
+    voice = db.query(Voice).filter(Voice.id == id).first()
+    if not voice:
+        raise HTTPException(status_code=404, detail="Voice not found")
+    return voice
+
+
+@router.put("/voices/{id}", response_model=VoiceOut)
+def update_voice(id: str, data: VoiceUpdate, db: Session = Depends(get_db)):
+    """更新声音"""
+    voice = db.query(Voice).filter(Voice.id == id).first()
+    if not voice:
+        raise HTTPException(status_code=404, detail="Voice not found")
+
+    update_data = data.model_dump(exclude_unset=True)
+    for field, value in update_data.items():
+        setattr(voice, field, value)
+
+    db.commit()
+    db.refresh(voice)
+    return voice
+
+
+@router.delete("/voices/{id}")
+def delete_voice(id: str, db: Session = Depends(get_db)):
+    """删除声音"""
+    voice = db.query(Voice).filter(Voice.id == id).first()
+    if not voice:
+        raise HTTPException(status_code=404, detail="Voice not found")
+    db.delete(voice)
+    db.commit()
+    return {"message": "Deleted successfully"}
 
 
 # ============ Assistants ============
@@ -79,11 +156,11 @@ def update_assistant(id: str, data: AssistantUpdate, db: Session = Depends(get_d
     assistant = db.query(Assistant).filter(Assistant.id == id).first()
     if not assistant:
         raise HTTPException(status_code=404, detail="Assistant not found")
-    
+
     update_data = data.model_dump(exclude_unset=True)
     for field, value in update_data.items():
         setattr(assistant, field, value)
-    
+
     assistant.updated_at = datetime.utcnow()
     db.commit()
     db.refresh(assistant)
@@ -103,10 +180,17 @@ def delete_assistant(id: str, db: Session = Depends(get_db)):
 
 # ============ Workflows ============
 @router.get("/workflows", response_model=List[WorkflowOut])
-def list_workflows(db: Session = Depends(get_db)):
+def list_workflows(
+    page: int = 1,
+    limit: int = 50,
+    db: Session = Depends(get_db)
+):
     """获取工作流列表"""
-    workflows = db.query(Workflow).all()
-    return workflows
+    query = db.query(Workflow)
+    total = query.count()
+    workflows = query.order_by(Workflow.created_at.desc()) \
+        .offset((page-1)*limit).limit(limit).all()
+    return {"total": total, "page": page, "limit": limit, "list": workflows}
 
 
 @router.post("/workflows", response_model=WorkflowOut)
@@ -129,17 +213,26 @@ def create_workflow(data: WorkflowCreate, db: Session = Depends(get_db)):
     return workflow
 
 
+@router.get("/workflows/{id}", response_model=WorkflowOut)
+def get_workflow(id: str, db: Session = Depends(get_db)):
+    """获取单个工作流"""
+    workflow = db.query(Workflow).filter(Workflow.id == id).first()
+    if not workflow:
+        raise HTTPException(status_code=404, detail="Workflow not found")
+    return workflow
+
+
 @router.put("/workflows/{id}", response_model=WorkflowOut)
 def update_workflow(id: str, data: WorkflowUpdate, db: Session = Depends(get_db)):
     """更新工作流"""
     workflow = db.query(Workflow).filter(Workflow.id == id).first()
     if not workflow:
         raise HTTPException(status_code=404, detail="Workflow not found")
-    
+
     update_data = data.model_dump(exclude_unset=True)
     for field, value in update_data.items():
         setattr(workflow, field, value)
-    
+
     workflow.updated_at = datetime.utcnow().isoformat()
     db.commit()
     db.refresh(workflow)