mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-21 12:52:26 +00:00
zhangtao
7d2367e34e
本次提交包含多项优化： 1. 移除大量冗余的文件头注释与过时的from __future__导入 2. 将CRUD的list方法统一重命名为get_list保持接口一致 3. 修复前后端状态字段类型不匹配问题，将string类型status改为number 4. 修正前端文案错别字，将"代办事项"修正为标准写法 5. 更新sqlalchemy版本并调整依赖配置 6. 新增缓存工具类替代fastapi-cache2，重构缓存调用逻辑 7. 新增开源授权函生成相关工具与数据库字段支持 8. 为多个业务模块添加防重复提交loading状态 9. 修复邮件模型的外键关联缺失问题 10. 优化pdf生成工具的导入时机与文档注释
312 lines
12 KiB
Python
312 lines
12 KiB
Python
|
|
from collections.abc import AsyncGenerator
|
|
from datetime import datetime
|
|
from typing import Any
|
|
|
|
from agno.run.team import TeamRunOutput
|
|
from agno.session.team import TeamSession
|
|
from agno.team.team import Team
|
|
|
|
from app.api.v1.module_system.dept.service import DeptService
|
|
from app.common.request import PaginationService
|
|
from app.core.base_schema import AuthSchema
|
|
from app.core.exceptions import CustomException
|
|
from app.core.logger import logger
|
|
|
|
from .crud import ChatSessionCRUD
|
|
from .schema import (
|
|
ChatQuerySchema,
|
|
ChatSessionCreateSchema,
|
|
ChatSessionQueryParam,
|
|
ChatSessionUpdateSchema,
|
|
)
|
|
from .utils import AgnoFactory
|
|
|
|
|
|
async def _format_session_data(session: TeamSession, auth: AuthSchema | None = None) -> dict[str, Any]:
|
|
"""格式化会话数据，添加前端需要的字段"""
|
|
if hasattr(session, "to_dict"):
|
|
session_dict = session.to_dict()
|
|
else:
|
|
session_dict = {
|
|
"session_id": getattr(session, "session_id", ""),
|
|
"agent_id": getattr(session, "agent_id", None),
|
|
"team_id": getattr(session, "team_id", None),
|
|
"workflow_id": getattr(session, "workflow_id", None),
|
|
"user_id": getattr(session, "user_id", None),
|
|
"session_data": getattr(session, "session_data", None),
|
|
"agent_data": getattr(session, "agent_data", None),
|
|
"team_data": getattr(session, "team_data", None),
|
|
"workflow_data": getattr(session, "workflow_data", None),
|
|
"metadata": getattr(session, "metadata", None),
|
|
"runs": getattr(session, "runs", []),
|
|
"summary": getattr(session, "summary", None),
|
|
"created_at": getattr(session, "created_at", None),
|
|
"updated_at": getattr(session, "updated_at", None),
|
|
}
|
|
|
|
session_data = session_dict.get("session_data") or {}
|
|
runs = session_dict.get("runs") or []
|
|
messages = _extract_messages(runs)
|
|
|
|
# 从 session_data 中获取 session_name 作为标题
|
|
session_name = session_data.get("session_name") if session_data else None
|
|
|
|
result = {
|
|
**session_dict,
|
|
"id": session_dict.get("session_id"),
|
|
"title": session_name or session_dict.get("session_id", "")[:8] or "未命名会话",
|
|
"created_time": _unix_to_datetime(session_dict.get("created_at")),
|
|
"updated_time": _unix_to_datetime(session_dict.get("updated_at")),
|
|
"message_count": len(messages),
|
|
"messages": messages,
|
|
}
|
|
|
|
# 如果有 auth，查询部门名称
|
|
if auth and session_dict.get("team_id"):
|
|
try:
|
|
team_id = session_dict.get("team_id")
|
|
if isinstance(team_id, str):
|
|
dept_name = await DeptService(auth).detail(id=int(team_id))
|
|
result["team_name"] = dept_name.get("name")
|
|
elif isinstance(team_id, int):
|
|
dept_name = await DeptService(auth).detail(id=team_id)
|
|
result["team_name"] = dept_name.get("name")
|
|
else:
|
|
result["team_name"] = None
|
|
except Exception:
|
|
result["team_name"] = None
|
|
else:
|
|
result["team_name"] = None
|
|
|
|
# 如果 summary 是 SessionSummary 对象，提取 summary 字段
|
|
summary = session_dict.get("summary")
|
|
if summary:
|
|
if isinstance(summary, dict):
|
|
result["summary"] = summary.get("summary")
|
|
else:
|
|
result["summary"] = str(summary)
|
|
|
|
return result
|
|
|
|
|
|
def _unix_to_datetime(timestamp: int | None) -> str | None:
|
|
"""将Unix时间戳转换为日期时间字符串"""
|
|
if timestamp is None:
|
|
return None
|
|
try:
|
|
dt = datetime.fromtimestamp(timestamp)
|
|
return dt.strftime("%Y-%m-%d %H:%M:%S")
|
|
except (ValueError, TypeError, OSError):
|
|
return None
|
|
|
|
|
|
def _extract_messages(runs: list[dict[str, Any]]) -> list[dict[str, Any]]:
|
|
"""从 runs 中提取消息"""
|
|
messages = []
|
|
if not runs:
|
|
return messages
|
|
for run in runs:
|
|
if not isinstance(run, dict):
|
|
continue
|
|
run_messages = run.get("messages", [])
|
|
if run_messages and isinstance(run_messages, list):
|
|
for msg in run_messages:
|
|
if isinstance(msg, dict):
|
|
role = msg.get("role")
|
|
if role in ("user", "assistant"):
|
|
messages.append(
|
|
{
|
|
"id": msg.get("id"),
|
|
"role": role,
|
|
"content": msg.get("content", ""),
|
|
"created_at": msg.get("created_at"),
|
|
}
|
|
)
|
|
return messages
|
|
|
|
|
|
class ChatService:
|
|
"""聊天会话管理模块服务层"""
|
|
|
|
def __init__(self, auth: AuthSchema) -> None:
|
|
self.auth = auth
|
|
|
|
async def chat_query(self, query: ChatQuerySchema) -> AsyncGenerator[str, None]:
|
|
"""流式 AI 对话"""
|
|
try:
|
|
crud = ChatSessionCRUD(self.auth)
|
|
|
|
session_id = query.session_id
|
|
if not session_id:
|
|
import uuid
|
|
|
|
session_id = str(uuid.uuid4())
|
|
session: TeamSession | None = await crud.create_crud(data=ChatSessionCreateSchema(title="新对话"))
|
|
if not session:
|
|
raise CustomException(msg="创建会话失败")
|
|
session_id = session.session_id
|
|
|
|
agno_factory = AgnoFactory()
|
|
dept_id = str(self.auth.user.dept_id) if self.auth and self.auth.user and hasattr(self.auth.user, "dept_id") and self.auth.user.dept_id else "default"
|
|
agent = agno_factory.create_agent(
|
|
user_id=self.auth.user.username if self.auth and self.auth.user else "user",
|
|
dept_id=dept_id,
|
|
session_id=session_id,
|
|
db=crud.db,
|
|
)
|
|
|
|
async for chunk in agent.arun(input=query.message, stream=True):
|
|
if chunk and chunk.content:
|
|
yield chunk.content
|
|
|
|
except Exception as e:
|
|
logger.error(f"聊天查询失败: {e}")
|
|
yield f"抱歉，处理您的请求时出现错误：{str(e)}"
|
|
|
|
async def chat_non_stream(self, message: str, session_id: str | None) -> dict[str, Any]:
|
|
"""非流式 AI 对话"""
|
|
try:
|
|
crud = ChatSessionCRUD(self.auth)
|
|
|
|
if not session_id:
|
|
import uuid
|
|
|
|
session_id = str(uuid.uuid4())
|
|
session: TeamSession | None = await crud.create_crud(data=ChatSessionCreateSchema(title="新对话"))
|
|
if not session:
|
|
raise CustomException(msg="创建会话失败")
|
|
session_id = session.session_id
|
|
|
|
agno_factory = AgnoFactory()
|
|
dept_id = str(self.auth.user.dept_id) if self.auth and self.auth.user and hasattr(self.auth.user, "dept_id") and self.auth.user.dept_id else "default"
|
|
agent: Team = agno_factory.create_agent(
|
|
user_id=self.auth.user.username if self.auth and self.auth.user else "user",
|
|
dept_id=dept_id,
|
|
session_id=session_id,
|
|
db=crud.db,
|
|
)
|
|
|
|
response: TeamRunOutput = await agent.arun(input=message)
|
|
|
|
response_text = ""
|
|
action = None
|
|
|
|
if response and response.content:
|
|
response_text = response.content
|
|
import json
|
|
|
|
try:
|
|
if response_text.strip().startswith("{") and response_text.strip().endswith("}"):
|
|
action = json.loads(response_text)
|
|
elif "```json" in response_text:
|
|
json_start = response_text.find("```json") + 7
|
|
json_end = response_text.find("```", json_start)
|
|
if json_end > json_start:
|
|
json_str = response_text[json_start:json_end].strip()
|
|
action = json.loads(json_str)
|
|
except (json.JSONDecodeError, Exception):
|
|
pass
|
|
|
|
if not action:
|
|
action = self._parse_action_from_response(response_text)
|
|
|
|
return {
|
|
"response": response_text,
|
|
"session_id": session_id,
|
|
"function_calls": None,
|
|
"action": action,
|
|
}
|
|
|
|
except Exception as e:
|
|
logger.error(f"聊天查询失败: {e}")
|
|
return {
|
|
"response": f"抱歉，处理您的请求时出现错误：{str(e)}",
|
|
"session_id": session_id,
|
|
"function_calls": None,
|
|
"action": None,
|
|
}
|
|
|
|
@staticmethod
|
|
def _parse_action_from_response(response_text: str) -> dict[str, Any] | None:
|
|
"""从响应文本中解析操作建议"""
|
|
|
|
route_config = {
|
|
"用户管理": {"path": "/system/user", "name": "用户管理"},
|
|
"角色管理": {"path": "/system/role", "name": "角色管理"},
|
|
"菜单管理": {"path": "/system/menu", "name": "菜单管理"},
|
|
"部门管理": {"path": "/system/dept", "name": "部门管理"},
|
|
"字典管理": {"path": "/system/dict", "name": "字典管理"},
|
|
"系统日志": {"path": "/system/log", "name": "系统日志"},
|
|
}
|
|
|
|
navigation_keywords = ["跳转", "打开", "进入", "前往", "去", "浏览", "查看"]
|
|
has_navigation = any(keyword in response_text for keyword in navigation_keywords)
|
|
|
|
if not has_navigation:
|
|
return None
|
|
|
|
for page_name, route_info in route_config.items():
|
|
if page_name in response_text:
|
|
return {
|
|
"type": "navigate",
|
|
"path": route_info["path"],
|
|
"name": route_info["name"],
|
|
}
|
|
|
|
keyword_mapping = {
|
|
"用户": {"path": "/system/user", "name": "用户管理"},
|
|
"角色": {"path": "/system/role", "name": "角色管理"},
|
|
"菜单": {"path": "/system/menu", "name": "菜单管理"},
|
|
"部门": {"path": "/system/dept", "name": "部门管理"},
|
|
"字典": {"path": "/system/dict", "name": "字典管理"},
|
|
"日志": {"path": "/system/log", "name": "系统日志"},
|
|
}
|
|
|
|
for keyword, route_info in keyword_mapping.items():
|
|
if keyword in response_text:
|
|
return {
|
|
"type": "navigate",
|
|
"path": route_info["path"],
|
|
"name": route_info["name"],
|
|
}
|
|
|
|
return None
|
|
|
|
async def get_session(self, session_id: str) -> dict[str, Any] | None:
|
|
crud = ChatSessionCRUD(self.auth)
|
|
session: TeamSession | None = await crud.get_by_id_crud(session_id=session_id)
|
|
if session:
|
|
return await _format_session_data(session, self.auth)
|
|
return None
|
|
|
|
async def create(self, data: ChatSessionCreateSchema) -> dict[str, Any] | None:
|
|
crud = ChatSessionCRUD(self.auth)
|
|
session = await crud.create_crud(data=data)
|
|
if session:
|
|
return await _format_session_data(session, self.auth)
|
|
return None
|
|
|
|
async def page(
|
|
self,
|
|
page_no: int,
|
|
page_size: int,
|
|
search: ChatSessionQueryParam,
|
|
order_by: list[dict[str, str]] | None = None,
|
|
) -> dict[str, Any]:
|
|
crud = ChatSessionCRUD(self.auth)
|
|
sessions = await crud.list_crud()
|
|
items = [await _format_session_data(s, self.auth) for s in sessions]
|
|
return await PaginationService.paginate(
|
|
data_list=items,
|
|
page_no=page_no,
|
|
page_size=page_size,
|
|
)
|
|
|
|
async def update(self, session_id: str, data: ChatSessionUpdateSchema) -> bool:
|
|
crud = ChatSessionCRUD(self.auth)
|
|
return await crud.update_crud(session_id=session_id, data=data)
|
|
|
|
async def delete(self, session_ids: list[str]) -> None:
|
|
await ChatSessionCRUD(self.auth).delete_crud(session_ids=session_ids)
|