90 lines
2.7 KiB
Python
90 lines
2.7 KiB
Python
|
"""
|
||
|
|
对话相关 API 路由
|
||
|
|
"""
|
||
|
|
from datetime import datetime, timezone
|
||
|
|
from typing import List, Optional
|
||
|
|
from fastapi import APIRouter, Depends, HTTPException
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
from sqlalchemy import select
|
||
|
|
import uuid
|
||
|
|
|
||
|
|
from database import get_async_db, Conversation, Segment, User
|
||
|
|
from database.models import Conversation as ConversationModel, Segment as SegmentModel
|
||
|
|
|
||
|
|
router = APIRouter(prefix="/api/conversations", tags=["conversations"])
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("")
|
||
|
|
async def create_conversation(
|
||
|
|
user_id: str,
|
||
|
|
db: AsyncSession = Depends(get_async_db)
|
||
|
|
):
|
||
|
|
"""创建新对话"""
|
||
|
|
conversation = ConversationModel(
|
||
|
|
id=str(uuid.uuid4()),
|
||
|
|
user_id=user_id,
|
||
|
|
started_at=datetime.now(timezone.utc),
|
||
|
|
status="active"
|
||
|
|
)
|
||
|
|
db.add(conversation)
|
||
|
|
await db.commit()
|
||
|
|
await db.refresh(conversation)
|
||
|
|
|
||
|
|
return {
|
||
|
|
"id": conversation.id,
|
||
|
|
"user_id": conversation.user_id,
|
||
|
|
"started_at": conversation.started_at.isoformat(),
|
||
|
|
"status": conversation.status
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/{conversation_id}")
|
||
|
|
async def get_conversation(
|
||
|
|
conversation_id: str,
|
||
|
|
db: AsyncSession = Depends(get_async_db)
|
||
|
|
):
|
||
|
|
"""获取对话详情"""
|
||
|
|
conversation = await db.get(ConversationModel, conversation_id)
|
||
|
|
if not conversation:
|
||
|
|
raise HTTPException(status_code=404, detail="Conversation not found")
|
||
|
|
|
||
|
|
return {
|
||
|
|
"id": conversation.id,
|
||
|
|
"user_id": conversation.user_id,
|
||
|
|
"started_at": conversation.started_at.isoformat(),
|
||
|
|
"ended_at": conversation.ended_at.isoformat() if conversation.ended_at else None,
|
||
|
|
"duration_seconds": conversation.duration_seconds,
|
||
|
|
"summary": conversation.summary,
|
||
|
|
"status": conversation.status,
|
||
|
|
"current_topic": conversation.current_topic,
|
||
|
|
"conversation_stage": conversation.conversation_stage
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/{conversation_id}/end")
|
||
|
|
async def end_conversation(
|
||
|
|
conversation_id: str,
|
||
|
|
db: AsyncSession = Depends(get_async_db)
|
||
|
|
):
|
||
|
|
"""结束对话"""
|
||
|
|
conversation = await db.get(ConversationModel, conversation_id)
|
||
|
|
if not conversation:
|
||
|
|
raise HTTPException(status_code=404, detail="Conversation not found")
|
||
|
|
|
||
|
|
conversation.status = "ended"
|
||
|
|
conversation.ended_at = datetime.now(timezone.utc)
|
||
|
|
|
||
|
|
if conversation.started_at:
|
||
|
|
duration = (conversation.ended_at - conversation.started_at).total_seconds()
|
||
|
|
conversation.duration_seconds = int(duration)
|
||
|
|
|
||
|
|
await db.commit()
|
||
|
|
|
||
|
|
return {
|
||
|
|
"id": conversation.id,
|
||
|
|
"status": conversation.status,
|
||
|
|
"ended_at": conversation.ended_at.isoformat(),
|
||
|
|
"duration_seconds": conversation.duration_seconds
|
||
|
|
}
|
||
|
|
|