diff --git a/src/agents/common/context.py b/src/agents/common/context.py index b7b5c91f..2f61dcd8 100644 --- a/src/agents/common/context.py +++ b/src/agents/common/context.py @@ -93,7 +93,8 @@ class BaseContext: metadata={ "name": "Skills", "options": [], - "description": "可选技能列表(由超级管理员维护)。运行时仅挂载并只读暴露选中的 skills。技能依赖的工具和 MCP 服务器也会被自动挂载。", + "description": "可选技能列表(由超级管理员维护)。运行时仅挂载并只读暴露选中的 " + "skills。技能依赖的工具和 MCP 服务器也会被自动挂载。", "type": "list", }, ) diff --git a/src/agents/common/middlewares/skills_middleware.py b/src/agents/common/middlewares/skills_middleware.py index 455f00db..74fcab65 100644 --- a/src/agents/common/middlewares/skills_middleware.py +++ b/src/agents/common/middlewares/skills_middleware.py @@ -20,7 +20,6 @@ from src.services.skill_service import _normalize_string_list, is_valid_skill_sl from src.storage.postgres.manager import pg_manager from src.utils.logging_config import logger - # ============================================================================= # 类型定义 # ============================================================================= diff --git a/src/agents/common/toolkits/buildin/tools.py b/src/agents/common/toolkits/buildin/tools.py index aa5187ea..ed23665a 100644 --- a/src/agents/common/toolkits/buildin/tools.py +++ b/src/agents/common/toolkits/buildin/tools.py @@ -6,8 +6,7 @@ from typing import Annotated, Any import requests from src import config, graph_base -from src.agents.common.toolkits.registry import tool -from src.agents.common.toolkits.registry import ToolExtraMetadata, _all_tool_instances, _extra_registry +from src.agents.common.toolkits.registry import ToolExtraMetadata, _all_tool_instances, _extra_registry, tool from src.storage.minio import aupload_file_to_minio from src.utils import logger diff --git a/src/agents/common/toolkits/mysql/tools.py b/src/agents/common/toolkits/mysql/tools.py index a8e21125..1574088a 100644 --- a/src/agents/common/toolkits/mysql/tools.py +++ b/src/agents/common/toolkits/mysql/tools.py @@ -46,7 +46,6 @@ def get_connection_manager() -> MySQLConnectionManager: return _connection_manager - @tool( category="mysql", tags=["数据库", "查询"], diff --git a/src/knowledge/base.py b/src/knowledge/base.py index 849d284c..027fc7bd 100644 --- a/src/knowledge/base.py +++ b/src/knowledge/base.py @@ -477,9 +477,9 @@ class KnowledgeBase(ABC): 操作结果 """ if db_id in self.databases_meta: + from src.knowledge.utils.kb_utils import parse_minio_url from src.repositories.knowledge_base_repository import KnowledgeBaseRepository from src.storage.minio import get_minio_client - from src.knowledge.utils.kb_utils import parse_minio_url minio_client = get_minio_client() diff --git a/test/test_kb_minio_cleanup.py b/test/test_kb_minio_cleanup.py index 2f260635..4bdec036 100644 --- a/test/test_kb_minio_cleanup.py +++ b/test/test_kb_minio_cleanup.py @@ -49,20 +49,19 @@ async def test_delete_knowledge_base_cleanup(): # 上传到 kb-documents (原始文件) test_object_name = f"{db_id}/test_file_{db_id[:8]}.txt" - upload_result = minio_client.upload_file( + minio_client.upload_file( bucket_name="kb-documents", object_name=test_object_name, data=test_file_content, content_type="text/plain", ) - test_file_url = upload_result.url print(f" 文件上传到 kb-documents: {test_object_name}") # 上传到 kb-parsed (解析后的 markdown) test_file_id = f"file_test_{db_id[:8]}" parsed_object_name = f"{db_id}/{test_file_id}/parsed.md" parsed_content = b"# Test Parsed Content\n\nThis is parsed markdown." - upload_result_parsed = minio_client.upload_file( + minio_client.upload_file( bucket_name="kb-parsed", object_name=parsed_object_name, data=parsed_content,