mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-20 20:39:55 +00:00
zhangtao
e03692467b
- 移除langchain相关依赖，引入agno作为大模型开发框架 - 重构聊天会话和消息存储，使用agno的TeamSession管理 - 合并chat_message和chat_session模块为统一的chat模块 - 更新前端API和类型定义，适配新的后端接口 - 优化数据库日志字段类型，根据数据库类型自动选择 - 重构AI助手组件，使用新的聊天API - 清理不再使用的模型、控制器和服务代码
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
from dataclasses import dataclass
|
|
from typing import Any
|
|
|
|
from fastapi import Query
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from app.core.validator import DateTimeStr
|
|
|
|
|
|
class ChatQuerySchema(BaseModel):
|
|
"""WebSocket聊天查询模型"""
|
|
message: str = Field(..., description="消息内容")
|
|
session_id: str | None = Field(None, description="会话ID")
|
|
files: list[dict[str, Any]] | None = Field(None, description="文件信息")
|
|
|
|
|
|
class ChatSessionCreateSchema(BaseModel):
|
|
"""创建会话模型"""
|
|
title: str = Field(..., description="会话标题")
|
|
|
|
|
|
class ChatSessionUpdateSchema(BaseModel):
|
|
"""更新会话模型"""
|
|
title: str = Field(..., description="会话标题")
|
|
|
|
|
|
class ChatSessionMessageSchema(BaseModel):
|
|
"""会话消息模型"""
|
|
id: str = Field(..., description="消息ID")
|
|
role: str = Field(..., description="消息角色")
|
|
content: str = Field(..., description="消息内容")
|
|
created_at: int | None = Field(None, description="创建时间(Unix时间戳)")
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
@dataclass
|
|
class ChatSessionQueryParam:
|
|
"""会话查询参数"""
|
|
def __init__(
|
|
self,
|
|
title: str | None = Query(None, description="会话标题"),
|
|
created_at: list[DateTimeStr] | None = Query(
|
|
None,
|
|
description="创建时间范围",
|
|
examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"],
|
|
),
|
|
updated_at: list[DateTimeStr] | None = Query(
|
|
None,
|
|
description="更新时间范围",
|
|
examples=["2025-01-01 00:00:00", "2025-12-31 23:59:59"],
|
|
),
|
|
) -> None:
|
|
self.title = title
|
|
self.created_at = created_at
|
|
self.updated_at = updated_at
|
|
|
|
|
|
class AiChatRequestSchema(BaseModel):
|
|
"""AI 对话请求模型（非流式）"""
|
|
message: str = Field(..., description="用户消息内容")
|
|
session_id: str | None = Field(None, description="会话ID，不传则创建新会话")
|
|
|
|
|
|
class AiChatResponseSchema(BaseModel):
|
|
"""AI 对话响应模型（非流式）"""
|
|
response: str = Field(..., description="AI 回复内容")
|
|
session_id: str = Field(..., description="会话ID")
|
|
function_calls: list[dict[str, Any]] | None = Field(None, description="函数调用信息")
|
|
action: dict[str, Any] | None = Field(None, description="建议执行的操作")
|