AstrBot 4.5.0__py3-none-any.whl → 4.5.2__py3-none-any.whl

astrbot/dashboard/routes/conversation.py

@@ -1,10 +1,13 @@
-import traceback
 import json
-from .route import Route, Response, RouteContext
-from astrbot.core import logger
+import traceback
+
 from quart import request
-from astrbot.core.db import BaseDatabase
+
+from astrbot.core import logger
 from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
+from astrbot.core.db import BaseDatabase
+
+from .route import Response, Route, RouteContext
 
 
 class ConversationRoute(Route):
@@ -55,12 +58,10 @@ class ConversationRoute(Route):
                 exclude_platforms.split(",") if exclude_platforms else []
             )
 
-            if page < 1:
-                page = 1
+            page = max(page, 1)
             if page_size < 1:
                 page_size = 20
-            if page_size > 100:
-                page_size = 100
+            page_size = min(page_size, 100)
 
             try:
                 (
@@ -76,8 +77,8 @@ class ConversationRoute(Route):
                     exclude_platforms=exclude_platform_list,
                 )
             except Exception as e:
-                logger.error(f"数据库查询出错: {str(e)}\n{traceback.format_exc()}")
-                return Response().error(f"数据库查询出错: {str(e)}").__dict__
+                logger.error(f"数据库查询出错: {e!s}\n{traceback.format_exc()}")
+                return Response().error(f"数据库查询出错: {e!s}").__dict__
 
             # 计算总页数
             total_pages = (
@@ -96,9 +97,9 @@ class ConversationRoute(Route):
             return Response().ok(result).__dict__
 
         except Exception as e:
-            error_msg = f"获取对话列表失败: {str(e)}\n{traceback.format_exc()}"
+            error_msg = f"获取对话列表失败: {e!s}\n{traceback.format_exc()}"
             logger.error(error_msg)
-            return Response().error(f"获取对话列表失败: {str(e)}").__dict__
+            return Response().error(f"获取对话列表失败: {e!s}").__dict__
 
     async def get_conv_detail(self):
         """获取指定对话详情（通过POST请求）"""
@@ -111,7 +112,8 @@ class ConversationRoute(Route):
                 return Response().error("缺少必要参数: user_id 和 cid").__dict__
 
             conversation = await self.conv_mgr.get_conversation(
-                unified_msg_origin=user_id, conversation_id=cid
+                unified_msg_origin=user_id,
+                conversation_id=cid,
             )
             if not conversation:
                 return Response().error("对话不存在").__dict__
@@ -127,14 +129,14 @@ class ConversationRoute(Route):
                         "history": conversation.history,
                         "created_at": conversation.created_at,
                         "updated_at": conversation.updated_at,
-                    }
+                    },
                 )
                 .__dict__
             )
 
         except Exception as e:
-            logger.error(f"获取对话详情失败: {str(e)}\n{traceback.format_exc()}")
-            return Response().error(f"获取对话详情失败: {str(e)}").__dict__
+            logger.error(f"获取对话详情失败: {e!s}\n{traceback.format_exc()}")
+            return Response().error(f"获取对话详情失败: {e!s}").__dict__
 
     async def upd_conv(self):
         """更新对话信息(标题和角色ID)"""
@@ -148,7 +150,8 @@ class ConversationRoute(Route):
             if not user_id or not cid:
                 return Response().error("缺少必要参数: user_id 和 cid").__dict__
             conversation = await self.conv_mgr.get_conversation(
-                unified_msg_origin=user_id, conversation_id=cid
+                unified_msg_origin=user_id,
+                conversation_id=cid,
             )
             if not conversation:
                 return Response().error("对话不存在").__dict__
@@ -162,8 +165,8 @@ class ConversationRoute(Route):
             return Response().ok({"message": "对话信息更新成功"}).__dict__
 
         except Exception as e:
-            logger.error(f"更新对话信息失败: {str(e)}\n{traceback.format_exc()}")
-            return Response().error(f"更新对话信息失败: {str(e)}").__dict__
+            logger.error(f"更新对话信息失败: {e!s}\n{traceback.format_exc()}")
+            return Response().error(f"更新对话信息失败: {e!s}").__dict__
 
     async def del_conv(self):
         """删除对话"""
@@ -188,17 +191,18 @@ class ConversationRoute(Route):
 
                     if not user_id or not cid:
                         failed_items.append(
-                            f"user_id:{user_id}, cid:{cid} - 缺少必要参数"
+                            f"user_id:{user_id}, cid:{cid} - 缺少必要参数",
                         )
                         continue
 
                     try:
                         await self.core_lifecycle.conversation_manager.delete_conversation(
-                            unified_msg_origin=user_id, conversation_id=cid
+                            unified_msg_origin=user_id,
+                            conversation_id=cid,
                         )
                         deleted_count += 1
                     except Exception as e:
-                        failed_items.append(f"user_id:{user_id}, cid:{cid} - {str(e)}")
+                        failed_items.append(f"user_id:{user_id}, cid:{cid} - {e!s}")
 
                 message = f"成功删除 {deleted_count} 个对话"
                 if failed_items:
@@ -212,26 +216,26 @@ class ConversationRoute(Route):
                             "deleted_count": deleted_count,
                             "failed_count": len(failed_items),
                             "failed_items": failed_items,
-                        }
+                        },
                     )
                     .__dict__
                 )
-            else:
-                # 单个删除
-                user_id = data.get("user_id")
-                cid = data.get("cid")
+            # 单个删除
+            user_id = data.get("user_id")
+            cid = data.get("cid")
 
-                if not user_id or not cid:
-                    return Response().error("缺少必要参数: user_id 和 cid").__dict__
+            if not user_id or not cid:
+                return Response().error("缺少必要参数: user_id 和 cid").__dict__
 
-                await self.core_lifecycle.conversation_manager.delete_conversation(
-                    unified_msg_origin=user_id, conversation_id=cid
-                )
-                return Response().ok({"message": "对话删除成功"}).__dict__
+            await self.core_lifecycle.conversation_manager.delete_conversation(
+                unified_msg_origin=user_id,
+                conversation_id=cid,
+            )
+            return Response().ok({"message": "对话删除成功"}).__dict__
 
         except Exception as e:
-            logger.error(f"删除对话失败: {str(e)}\n{traceback.format_exc()}")
-            return Response().error(f"删除对话失败: {str(e)}").__dict__
+            logger.error(f"删除对话失败: {e!s}\n{traceback.format_exc()}")
+            return Response().error(f"删除对话失败: {e!s}").__dict__
 
     async def update_history(self):
         """更新对话历史内容"""
@@ -260,7 +264,8 @@ class ConversationRoute(Route):
                 )
 
             conversation = await self.conv_mgr.get_conversation(
-                unified_msg_origin=user_id, conversation_id=cid
+                unified_msg_origin=user_id,
+                conversation_id=cid,
             )
             if not conversation:
                 return Response().error("对话不存在").__dict__
@@ -268,11 +273,13 @@ class ConversationRoute(Route):
             history = json.loads(history) if isinstance(history, str) else history
 
             await self.conv_mgr.update_conversation(
-                unified_msg_origin=user_id, conversation_id=cid, history=history
+                unified_msg_origin=user_id,
+                conversation_id=cid,
+                history=history,
             )
 
             return Response().ok({"message": "对话历史更新成功"}).__dict__
 
         except Exception as e:
-            logger.error(f"更新对话历史失败: {str(e)}\n{traceback.format_exc()}")
-            return Response().error(f"更新对话历史失败: {str(e)}").__dict__
+            logger.error(f"更新对话历史失败: {e!s}\n{traceback.format_exc()}")
+            return Response().error(f"更新对话历史失败: {e!s}").__dict__

astrbot/dashboard/routes/file.py

@@ -1,8 +1,10 @@
-from .route import Route, RouteContext
-from astrbot import logger
 from quart import abort, send_file
+
+from astrbot import logger
 from astrbot.core import file_token_service
 
+from .route import Route, RouteContext
+
 
 class FileRoute(Route):
     def __init__(

astrbot/dashboard/routes/knowledge_base.py

@@ -1,17 +1,20 @@
 """知识库管理 API 路由"""
 
-import uuid
-import aiofiles
+import asyncio
 import os
 import traceback
-import asyncio
+import uuid
+
+import aiofiles
 from quart import request
+
 from astrbot.core import logger
 from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
-from .route import Route, Response, RouteContext
-from ..utils import generate_tsne_visualization
 from astrbot.core.provider.provider import EmbeddingProvider, RerankProvider
 
+from ..utils import generate_tsne_visualization
+from .route import Response, Route, RouteContext
+
 
 class KnowledgeBaseRoute(Route):
     """知识库管理路由
@@ -108,7 +111,7 @@ class KnowledgeBaseRoute(Route):
                             "stage": "parsing",
                             "current": 0,
                             "total": 100,
-                        }
+                        },
                     )
 
                     # 创建进度回调函数
@@ -122,7 +125,7 @@ class KnowledgeBaseRoute(Route):
                                     "stage": stage,
                                     "current": current,
                                     "total": total,
-                                }
+                                },
                             )
 
                     doc = await kb_helper.upload_document(
@@ -141,7 +144,7 @@ class KnowledgeBaseRoute(Route):
                 except Exception as e:
                     logger.error(f"上传文档 {file_info['file_name']} 失败: {e}")
                     failed_docs.append(
-                        {"file_name": file_info["file_name"], "error": str(e)}
+                        {"file_name": file_info["file_name"], "error": str(e)},
                     )
 
             # 更新任务完成状态
@@ -202,7 +205,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"获取知识库列表失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"获取知识库列表失败: {str(e)}").__dict__
+            return Response().error(f"获取知识库列表失败: {e!s}").__dict__
 
     async def create_kb(self):
         """创建知识库
@@ -240,7 +243,7 @@ class KnowledgeBaseRoute(Route):
             if not embedding_provider_id:
                 return Response().error("缺少参数 embedding_provider_id").__dict__
             prv = await kb_manager.provider_manager.get_provider_by_id(
-                embedding_provider_id
+                embedding_provider_id,
             )  # type: ignore
             if not prv or not isinstance(prv, EmbeddingProvider):
                 return (
@@ -250,15 +253,15 @@ class KnowledgeBaseRoute(Route):
                 vec = await prv.get_embedding("astrbot")
                 if len(vec) != prv.get_dim():
                     raise ValueError(
-                        f"嵌入向量维度不匹配，实际是 {len(vec)}，然而配置是 {prv.get_dim()}"
+                        f"嵌入向量维度不匹配，实际是 {len(vec)}，然而配置是 {prv.get_dim()}",
                     )
             except Exception as e:
-                return Response().error(f"测试嵌入模型失败: {str(e)}").__dict__
+                return Response().error(f"测试嵌入模型失败: {e!s}").__dict__
             # pre-check rerank
             if rerank_provider_id:
                 rerank_prv: RerankProvider = (
                     await kb_manager.provider_manager.get_provider_by_id(
-                        rerank_provider_id
+                        rerank_provider_id,
                     )
                 )  # type: ignore
                 if not rerank_prv:
@@ -266,14 +269,15 @@ class KnowledgeBaseRoute(Route):
                 # 检查重排序模型可用性
                 try:
                     res = await rerank_prv.rerank(
-                        query="astrbot", documents=["astrbot knowledge base"]
+                        query="astrbot",
+                        documents=["astrbot knowledge base"],
                     )
                     if not res:
                         raise ValueError("重排序模型返回结果异常")
                 except Exception as e:
                     return (
                         Response()
-                        .error(f"测试重排序模型失败: {str(e)}，请检查控制台日志输出。")
+                        .error(f"测试重排序模型失败: {e!s}，请检查控制台日志输出。")
                         .__dict__
                     )
 
@@ -298,7 +302,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"创建知识库失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"创建知识库失败: {str(e)}").__dict__
+            return Response().error(f"创建知识库失败: {e!s}").__dict__
 
     async def get_kb(self):
         """获取知识库详情
@@ -324,7 +328,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"获取知识库详情失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"获取知识库详情失败: {str(e)}").__dict__
+            return Response().error(f"获取知识库详情失败: {e!s}").__dict__
 
     async def update_kb(self):
         """更新知识库
@@ -404,7 +408,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"更新知识库失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"更新知识库失败: {str(e)}").__dict__
+            return Response().error(f"更新知识库失败: {e!s}").__dict__
 
     async def delete_kb(self):
         """删除知识库
@@ -431,7 +435,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"删除知识库失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"删除知识库失败: {str(e)}").__dict__
+            return Response().error(f"删除知识库失败: {e!s}").__dict__
 
     async def get_kb_stats(self):
         """获取知识库统计信息
@@ -466,7 +470,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"获取知识库统计失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"获取知识库统计失败: {str(e)}").__dict__
+            return Response().error(f"获取知识库统计失败: {e!s}").__dict__
 
     # ===== 文档管理 API =====
 
@@ -508,7 +512,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"获取文档列表失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"获取文档列表失败: {str(e)}").__dict__
+            return Response().error(f"获取文档列表失败: {e!s}").__dict__
 
     async def upload_document(self):
         """上传文档
@@ -597,7 +601,7 @@ class KnowledgeBaseRoute(Route):
                             "file_name": file_name,
                             "file_content": file_content,
                             "file_type": file_type,
-                        }
+                        },
                     )
                 finally:
                     # 清理临时文件
@@ -630,7 +634,7 @@ class KnowledgeBaseRoute(Route):
                     batch_size=batch_size,
                     tasks_limit=tasks_limit,
                     max_retries=max_retries,
-                )
+                ),
             )
 
             return (
@@ -640,7 +644,7 @@ class KnowledgeBaseRoute(Route):
                         "task_id": task_id,
                         "file_count": len(files_to_upload),
                         "message": "task created, processing in background",
-                    }
+                    },
                 )
                 .__dict__
             )
@@ -650,7 +654,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"上传文档失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"上传文档失败: {str(e)}").__dict__
+            return Response().error(f"上传文档失败: {e!s}").__dict__
 
     async def get_upload_progress(self):
         """获取上传进度和结果
@@ -703,7 +707,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"获取上传进度失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"获取上传进度失败: {str(e)}").__dict__
+            return Response().error(f"获取上传进度失败: {e!s}").__dict__
 
     async def get_document(self):
         """获取文档详情
@@ -734,7 +738,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"获取文档详情失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"获取文档详情失败: {str(e)}").__dict__
+            return Response().error(f"获取文档详情失败: {e!s}").__dict__
 
     async def delete_document(self):
         """删除文档
@@ -766,7 +770,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"删除文档失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"删除文档失败: {str(e)}").__dict__
+            return Response().error(f"删除文档失败: {e!s}").__dict__
 
     async def delete_chunk(self):
         """删除文本块
@@ -801,7 +805,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"删除文本块失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"删除文本块失败: {str(e)}").__dict__
+            return Response().error(f"删除文本块失败: {e!s}").__dict__
 
     async def list_chunks(self):
         """获取块列表
@@ -827,7 +831,9 @@ class KnowledgeBaseRoute(Route):
             if not kb_helper:
                 return Response().error("知识库不存在").__dict__
             chunk_list = await kb_helper.get_chunks_by_doc_id(
-                doc_id=doc_id, offset=offset, limit=limit
+                doc_id=doc_id,
+                offset=offset,
+                limit=limit,
             )
             return (
                 Response()
@@ -837,7 +843,7 @@ class KnowledgeBaseRoute(Route):
                         "page": page,
                         "page_size": page_size,
                         "total": await kb_helper.get_chunk_count_by_doc_id(doc_id),
-                    }
+                    },
                 )
                 .__dict__
             )
@@ -846,7 +852,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"获取块列表失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"获取块列表失败: {str(e)}").__dict__
+            return Response().error(f"获取块列表失败: {e!s}").__dict__
 
     # ===== 检索 API =====
 
@@ -893,7 +899,9 @@ class KnowledgeBaseRoute(Route):
             if debug:
                 try:
                     img_base64 = await generate_tsne_visualization(
-                        query, kb_names, kb_manager
+                        query,
+                        kb_names,
+                        kb_manager,
                     )
                     if img_base64:
                         response_data["visualization"] = img_base64
@@ -909,7 +917,7 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"检索失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"检索失败: {str(e)}").__dict__
+            return Response().error(f"检索失败: {e!s}").__dict__
 
     # ===== 会话知识库配置 API =====
 
@@ -945,7 +953,7 @@ class KnowledgeBaseRoute(Route):
 
         except Exception as e:
             logger.error(f"[KB配置] 获取配置时出错: {e}", exc_info=True)
-            return Response().error(f"获取会话知识库配置失败: {str(e)}").__dict__
+            return Response().error(f"获取会话知识库配置失败: {e!s}").__dict__
 
     async def set_session_kb_config(self):
         """设置会话的知识库配置
@@ -1024,13 +1032,12 @@ class KnowledgeBaseRoute(Route):
                     )
                     .__dict__
                 )
-            else:
-                logger.error("[KB配置] 配置保存失败，验证不匹配")
-                return Response().error("配置保存失败").__dict__
+            logger.error("[KB配置] 配置保存失败，验证不匹配")
+            return Response().error("配置保存失败").__dict__
 
         except Exception as e:
             logger.error(f"[KB配置] 设置配置时出错: {e}", exc_info=True)
-            return Response().error(f"设置会话知识库配置失败: {str(e)}").__dict__
+            return Response().error(f"设置会话知识库配置失败: {e!s}").__dict__
 
     async def delete_session_kb_config(self):
         """删除会话的知识库配置
@@ -1062,4 +1069,4 @@ class KnowledgeBaseRoute(Route):
         except Exception as e:
             logger.error(f"删除会话知识库配置失败: {e}")
             logger.error(traceback.format_exc())
-            return Response().error(f"删除会话知识库配置失败: {str(e)}").__dict__
+            return Response().error(f"删除会话知识库配置失败: {e!s}").__dict__

astrbot/dashboard/routes/log.py

@@ -1,8 +1,11 @@
 import asyncio
 import json
+
 from quart import make_response
-from astrbot.core import logger, LogBroker
-from .route import Route, RouteContext, Response
+
+from astrbot.core import LogBroker, logger
+
+from .route import Response, Route, RouteContext
 
 
 class LogRoute(Route):
@@ -11,7 +14,9 @@ class LogRoute(Route):
         self.log_broker = log_broker
         self.app.add_url_rule("/api/live-log", view_func=self.log, methods=["GET"])
         self.app.add_url_rule(
-            "/api/log-history", view_func=self.log_history, methods=["GET"]
+            "/api/log-history",
+            view_func=self.log_history,
+            methods=["GET"],
         )
 
     async def log(self):
@@ -55,7 +60,7 @@ class LogRoute(Route):
                 .ok(
                     data={
                         "logs": logs,
-                    }
+                    },
                 )
                 .__dict__
             )

astrbot/dashboard/routes/persona.py

@@ -1,9 +1,12 @@
 import traceback
-from .route import Route, Response, RouteContext
-from astrbot.core import logger
+
 from quart import request
-from astrbot.core.db import BaseDatabase
+
+from astrbot.core import logger
 from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
+from astrbot.core.db import BaseDatabase
+
+from .route import Response, Route, RouteContext
 
 
 class PersonaRoute(Route):
@@ -46,13 +49,13 @@ class PersonaRoute(Route):
                             else None,
                         }
                         for persona in personas
-                    ]
+                    ],
                 )
                 .__dict__
             )
         except Exception as e:
-            logger.error(f"获取人格列表失败: {str(e)}\n{traceback.format_exc()}")
-            return Response().error(f"获取人格列表失败: {str(e)}").__dict__
+            logger.error(f"获取人格列表失败: {e!s}\n{traceback.format_exc()}")
+            return Response().error(f"获取人格列表失败: {e!s}").__dict__
 
     async def get_persona_detail(self):
         """获取指定人格的详细信息"""
@@ -81,13 +84,13 @@ class PersonaRoute(Route):
                         "updated_at": persona.updated_at.isoformat()
                         if persona.updated_at
                         else None,
-                    }
+                    },
                 )
                 .__dict__
             )
         except Exception as e:
-            logger.error(f"获取人格详情失败: {str(e)}\n{traceback.format_exc()}")
-            return Response().error(f"获取人格详情失败: {str(e)}").__dict__
+            logger.error(f"获取人格详情失败: {e!s}\n{traceback.format_exc()}")
+            return Response().error(f"获取人格详情失败: {e!s}").__dict__
 
     async def create_persona(self):
         """创建新人格"""
@@ -136,15 +139,15 @@ class PersonaRoute(Route):
                             if persona.updated_at
                             else None,
                         },
-                    }
+                    },
                 )
                 .__dict__
             )
         except ValueError as e:
             return Response().error(str(e)).__dict__
         except Exception as e:
-            logger.error(f"创建人格失败: {str(e)}\n{traceback.format_exc()}")
-            return Response().error(f"创建人格失败: {str(e)}").__dict__
+            logger.error(f"创建人格失败: {e!s}\n{traceback.format_exc()}")
+            return Response().error(f"创建人格失败: {e!s}").__dict__
 
     async def update_persona(self):
         """更新人格信息"""
@@ -177,8 +180,8 @@ class PersonaRoute(Route):
         except ValueError as e:
             return Response().error(str(e)).__dict__
         except Exception as e:
-            logger.error(f"更新人格失败: {str(e)}\n{traceback.format_exc()}")
-            return Response().error(f"更新人格失败: {str(e)}").__dict__
+            logger.error(f"更新人格失败: {e!s}\n{traceback.format_exc()}")
+            return Response().error(f"更新人格失败: {e!s}").__dict__
 
     async def delete_persona(self):
         """删除人格"""
@@ -195,5 +198,5 @@ class PersonaRoute(Route):
         except ValueError as e:
             return Response().error(str(e)).__dict__
         except Exception as e:
-            logger.error(f"删除人格失败: {str(e)}\n{traceback.format_exc()}")
-            return Response().error(f"删除人格失败: {str(e)}").__dict__
+            logger.error(f"删除人格失败: {e!s}\n{traceback.format_exc()}")
+            return Response().error(f"删除人格失败: {e!s}").__dict__