281 lines
8.2 KiB
Python
281 lines
8.2 KiB
Python
"""
|
||
对话 API 端点。
|
||
"""
|
||
from typing import List, Optional
|
||
from fastapi import APIRouter, Depends, HTTPException, status
|
||
from pydantic import BaseModel, Field
|
||
from sqlalchemy.orm import Session
|
||
|
||
from app.db.session import get_db
|
||
from app.services.conv_manager import ConversationManager
|
||
from app.services.model_manager import ModelManager
|
||
from app.services.kb_manager import KnowledgeBaseManager
|
||
from app.services.job_manager import AsyncJobManager
|
||
from app.config import get_settings
|
||
from app.utils.exceptions import ResourceNotFoundError
|
||
from app.utils.logger import get_logger
|
||
|
||
logger = get_logger(__name__)
|
||
router = APIRouter(prefix="/conversations", tags=["conversations"])
|
||
|
||
|
||
# Pydantic 模式
|
||
class ConversationCreate(BaseModel):
|
||
"""创建对话的模式。"""
|
||
user_id: Optional[int] = Field(None, description="User ID")
|
||
title: str = Field(default="New Conversation", description="Conversation title")
|
||
|
||
|
||
class ConversationResponse(BaseModel):
|
||
"""对话响应的模式。"""
|
||
id: int
|
||
user_id: Optional[int]
|
||
title: str
|
||
created_at: str
|
||
|
||
class Config:
|
||
from_attributes = True
|
||
|
||
|
||
class MessageResponse(BaseModel):
|
||
"""消息响应的模式。"""
|
||
id: int
|
||
conversation_id: int
|
||
role: str
|
||
content: str
|
||
msg_metadata: dict
|
||
created_at: str
|
||
|
||
class Config:
|
||
from_attributes = True
|
||
|
||
|
||
class ChatRequest(BaseModel):
|
||
"""聊天请求的模式。"""
|
||
user_input: str = Field(..., description="User's message")
|
||
kb_id: Optional[int] = Field(None, description="Knowledge base ID for RAG")
|
||
llm_name: Optional[str] = Field(None, description="LLM model name")
|
||
use_rag: bool = Field(default=True, description="Enable RAG (default True)")
|
||
|
||
|
||
class ChatResponse(BaseModel):
|
||
"""聊天响应的模式。"""
|
||
message_id: int
|
||
conversation_id: int
|
||
role: str
|
||
content: str
|
||
metadata: dict
|
||
|
||
|
||
# 依赖项
|
||
def get_conv_manager(
|
||
db: Session = Depends(get_db),
|
||
) -> ConversationManager:
|
||
"""获取 ConversationManager 实例。"""
|
||
settings = get_settings()
|
||
model_manager = ModelManager(db, settings)
|
||
job_manager = AsyncJobManager()
|
||
kb_manager = KnowledgeBaseManager(db, model_manager, job_manager, settings)
|
||
return ConversationManager(db, model_manager, kb_manager, settings)
|
||
|
||
|
||
@router.post("/", response_model=ConversationResponse, status_code=status.HTTP_201_CREATED)
|
||
def create_conversation(
|
||
conv_data: ConversationCreate,
|
||
conv_manager: ConversationManager = Depends(get_conv_manager),
|
||
):
|
||
"""
|
||
创建新对话。
|
||
|
||
Args:
|
||
conv_data: 对话创建数据
|
||
conv_manager: ConversationManager 实例
|
||
|
||
Returns:
|
||
创建的对话
|
||
"""
|
||
try:
|
||
conversation = conv_manager.create_conversation(
|
||
user_id=conv_data.user_id,
|
||
title=conv_data.title,
|
||
)
|
||
|
||
return ConversationResponse(
|
||
id=conversation.id,
|
||
user_id=conversation.user_id,
|
||
title=conversation.title or "New Conversation",
|
||
created_at=conversation.created_at.isoformat() if conversation.created_at else "",
|
||
)
|
||
|
||
except Exception as e:
|
||
logger.error("conversation_creation_failed", error=str(e))
|
||
raise HTTPException(
|
||
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||
detail=f"Failed to create conversation: {str(e)}",
|
||
)
|
||
|
||
|
||
@router.get("/{conversation_id}", response_model=ConversationResponse)
|
||
def get_conversation(
|
||
conversation_id: int,
|
||
conv_manager: ConversationManager = Depends(get_conv_manager),
|
||
):
|
||
"""
|
||
根据 ID 获取对话。
|
||
|
||
Args:
|
||
conversation_id: 对话 ID
|
||
conv_manager: ConversationManager 实例
|
||
|
||
Returns:
|
||
对话详情
|
||
|
||
Raises:
|
||
404: 如果找不到对话
|
||
"""
|
||
try:
|
||
conversation = conv_manager.get_conversation(conversation_id)
|
||
|
||
return ConversationResponse(
|
||
id=conversation.id,
|
||
user_id=conversation.user_id,
|
||
title=conversation.title or "New Conversation",
|
||
created_at=conversation.created_at.isoformat() if conversation.created_at else "",
|
||
)
|
||
|
||
except ResourceNotFoundError as e:
|
||
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=str(e))
|
||
except Exception as e:
|
||
logger.error("conversation_get_failed", conversation_id=conversation_id, error=str(e))
|
||
raise HTTPException(
|
||
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||
detail=f"Failed to get conversation: {str(e)}",
|
||
)
|
||
|
||
|
||
@router.delete("/{conversation_id}", status_code=status.HTTP_204_NO_CONTENT)
|
||
def delete_conversation(
|
||
conversation_id: int,
|
||
conv_manager: ConversationManager = Depends(get_conv_manager),
|
||
):
|
||
"""
|
||
删除对话。
|
||
|
||
Args:
|
||
conversation_id: 对话 ID
|
||
conv_manager: ConversationManager 实例
|
||
|
||
Raises:
|
||
404: 如果找不到对话
|
||
"""
|
||
try:
|
||
conv_manager.delete_conversation(conversation_id)
|
||
logger.info("conversation_deleted_via_api", conversation_id=conversation_id)
|
||
|
||
except ResourceNotFoundError as e:
|
||
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=str(e))
|
||
except Exception as e:
|
||
logger.error("conversation_delete_failed", conversation_id=conversation_id, error=str(e))
|
||
raise HTTPException(
|
||
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||
detail=f"Failed to delete conversation: {str(e)}",
|
||
)
|
||
|
||
|
||
@router.get("/{conversation_id}/messages", response_model=List[MessageResponse])
|
||
def get_messages(
|
||
conversation_id: int,
|
||
limit: int = 50,
|
||
offset: int = 0,
|
||
conv_manager: ConversationManager = Depends(get_conv_manager),
|
||
):
|
||
"""
|
||
获取对话中的消息。
|
||
|
||
Args:
|
||
conversation_id: 对话 ID
|
||
limit: 最大消息数量（默认 50）
|
||
offset: 分页偏移量（默认 0）
|
||
conv_manager: ConversationManager 实例
|
||
|
||
Returns:
|
||
消息列表
|
||
|
||
Raises:
|
||
404: 如果找不到对话
|
||
"""
|
||
try:
|
||
messages = conv_manager.get_messages(
|
||
conversation_id=conversation_id,
|
||
limit=limit,
|
||
offset=offset,
|
||
)
|
||
|
||
return [
|
||
MessageResponse(
|
||
id=msg.id,
|
||
conversation_id=msg.conversation_id,
|
||
role=msg.role,
|
||
content=msg.content,
|
||
msg_metadata=msg.msg_metadata or {},
|
||
created_at=msg.created_at.isoformat() if msg.created_at else "",
|
||
)
|
||
for msg in messages
|
||
]
|
||
|
||
except ResourceNotFoundError as e:
|
||
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=str(e))
|
||
except Exception as e:
|
||
logger.error("messages_get_failed", conversation_id=conversation_id, error=str(e))
|
||
raise HTTPException(
|
||
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||
detail=f"Failed to get messages: {str(e)}",
|
||
)
|
||
|
||
|
||
@router.post("/{conversation_id}/chat", response_model=ChatResponse)
|
||
async def chat(
|
||
conversation_id: int,
|
||
chat_data: ChatRequest,
|
||
conv_manager: ConversationManager = Depends(get_conv_manager),
|
||
):
|
||
"""
|
||
发送消息并获取助手回复。
|
||
|
||
Args:
|
||
conversation_id: 对话 ID
|
||
chat_data: 聊天请求数据
|
||
conv_manager: ConversationManager 实例
|
||
|
||
Returns:
|
||
助手的回复
|
||
|
||
Raises:
|
||
404: 如果找不到对话
|
||
"""
|
||
try:
|
||
message = await conv_manager.chat(
|
||
conversation_id=conversation_id,
|
||
user_input=chat_data.user_input,
|
||
kb_id=chat_data.kb_id,
|
||
llm_name=chat_data.llm_name,
|
||
use_rag=chat_data.use_rag,
|
||
)
|
||
|
||
return ChatResponse(
|
||
message_id=message.id,
|
||
conversation_id=message.conversation_id,
|
||
role=message.role,
|
||
content=message.content,
|
||
metadata=message.msg_metadata or {},
|
||
)
|
||
|
||
except ResourceNotFoundError as e:
|
||
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=str(e))
|
||
except Exception as e:
|
||
logger.error("chat_failed", conversation_id=conversation_id, error=str(e))
|
||
raise HTTPException(
|
||
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||
detail=f"Failed to process chat: {str(e)}",
|
||
)
|