129 lines
4.6 KiB
Python
129 lines
4.6 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, Query, Body
|
|
from sqlalchemy.orm import Session
|
|
from app.core.db import get_db
|
|
from app.models import AIModel, Meeting, TranscriptTask, SummarizeTask
|
|
from app.schemas.ai_model import AIModelOut, AIModelCreate, AIModelUpdate
|
|
from typing import List, Optional
|
|
import httpx
|
|
|
|
router = APIRouter(prefix="/ai-models", tags=["ai-models"])
|
|
|
|
|
|
@router.post("/fetch-remote-models")
|
|
async def fetch_remote_models(base_url: str = Body(..., embed=True)):
|
|
"""
|
|
Fetch available models from a remote ASR server.
|
|
"""
|
|
try:
|
|
url = base_url.rstrip('/') + "/api/asrconfig"
|
|
|
|
async with httpx.AsyncClient() as client:
|
|
resp = await client.get(url, timeout=5.0)
|
|
|
|
if resp.status_code != 200:
|
|
raise HTTPException(status_code=400, detail=f"Failed to connect: {resp.status_code}")
|
|
|
|
data = resp.json()
|
|
|
|
# Check for standard response format (code 200 or 0 for success)
|
|
is_success = False
|
|
if isinstance(data, dict):
|
|
code = data.get("code")
|
|
if code == 0 or code == 200:
|
|
is_success = True
|
|
|
|
if is_success and "data" in data:
|
|
available_models = data["data"].get("available_models", [])
|
|
return {"models": available_models}
|
|
else:
|
|
raise HTTPException(status_code=400, detail="Invalid response format from ASR server")
|
|
|
|
except httpx.RequestError as e:
|
|
raise HTTPException(status_code=400, detail=f"Network error: {str(e)}")
|
|
except Exception as e:
|
|
raise HTTPException(status_code=400, detail=f"Error: {str(e)}")
|
|
|
|
|
|
@router.get("", response_model=List[AIModelOut])
|
|
def list_ai_models(
|
|
model_type: Optional[str] = Query(None, pattern="^(asr|llm)$"),
|
|
db: Session = Depends(get_db)
|
|
):
|
|
query = db.query(AIModel)
|
|
if model_type:
|
|
query = query.filter(AIModel.model_type == model_type)
|
|
return query.order_by(AIModel.created_at.desc()).all()
|
|
|
|
|
|
@router.post("", response_model=AIModelOut)
|
|
def create_ai_model(payload: AIModelCreate, db: Session = Depends(get_db)):
|
|
# Convert boolean to int for SmallInteger field
|
|
data = payload.model_dump()
|
|
if 'is_default' in data:
|
|
data['is_default'] = int(data['is_default'])
|
|
|
|
# Logic: If set as default, unset others of same type
|
|
if data.get('is_default'):
|
|
db.query(AIModel).filter(AIModel.model_type == payload.model_type).update({"is_default": 0})
|
|
|
|
item = AIModel(**data)
|
|
db.add(item)
|
|
db.commit()
|
|
db.refresh(item)
|
|
return item
|
|
|
|
|
|
@router.put("/{model_id}", response_model=AIModelOut)
|
|
def update_ai_model(model_id: int, payload: AIModelUpdate, db: Session = Depends(get_db)):
|
|
item = db.query(AIModel).filter(AIModel.model_id == model_id).first()
|
|
if not item:
|
|
raise HTTPException(status_code=404, detail="Model not found")
|
|
|
|
update_data = payload.model_dump(exclude_unset=True)
|
|
|
|
if "is_default" in update_data:
|
|
update_data["is_default"] = int(update_data["is_default"])
|
|
|
|
if update_data.get("is_default"):
|
|
# Unset others
|
|
target_type = update_data.get("model_type", item.model_type)
|
|
db.query(AIModel).filter(AIModel.model_type == target_type, AIModel.model_id != item.model_id).update({"is_default": 0})
|
|
|
|
for k, v in update_data.items():
|
|
setattr(item, k, v)
|
|
|
|
db.commit()
|
|
db.refresh(item)
|
|
return item
|
|
|
|
|
|
from sqlalchemy.exc import IntegrityError
|
|
|
|
@router.delete("/{model_id}")
|
|
def delete_ai_model(model_id: int, db: Session = Depends(get_db)):
|
|
item = db.query(AIModel).filter(AIModel.model_id == model_id).first()
|
|
if not item:
|
|
raise HTTPException(status_code=404, detail="Model not found")
|
|
|
|
if item.is_default:
|
|
raise HTTPException(status_code=400, detail="Cannot delete default model")
|
|
|
|
try:
|
|
# Manually unlink from related tables before deletion
|
|
# Set asr_model_id to NULL in Meeting
|
|
db.query(Meeting).filter(Meeting.asr_model_id == model_id).update({"asr_model_id": None})
|
|
|
|
# Set model_id to NULL in TranscriptTask
|
|
db.query(TranscriptTask).filter(TranscriptTask.model_id == model_id).update({"model_id": None})
|
|
|
|
# Set model_id to NULL in SummarizeTask
|
|
db.query(SummarizeTask).filter(SummarizeTask.model_id == model_id).update({"model_id": None})
|
|
|
|
db.delete(item)
|
|
db.commit()
|
|
except IntegrityError:
|
|
db.rollback()
|
|
raise HTTPException(status_code=400, detail="该模型已被其他资源引用，无法删除")
|
|
|
|
return {"status": "ok"}
|