♻️ The agent import performance improvement: add existing tool check

Author: Phinease

backend/config_service.py

@@ -1,7 +1,6 @@
 import uvicorn
 import logging
 import warnings
-import asyncio
 
 from consts.const import APP_VERSION
 
@@ -12,30 +11,14 @@
 
 from apps.config_app import app
 from utils.logging_utils import configure_logging, configure_elasticsearch_logging
-from services.tool_configuration_service import initialize_tools_on_startup
+
 
 configure_logging(logging.INFO)
 configure_elasticsearch_logging()
 logger = logging.getLogger("config_service")
 
 
-async def startup_initialization():
-    """
-    Perform initialization tasks during server startup
-    """
+if __name__ == "__main__":
     logger.info("Starting server initialization...")
     logger.info(f"APP version is: {APP_VERSION}")
-    try:
-        # Initialize tools on startup - service layer handles detailed logging
-        await initialize_tools_on_startup()
-        logger.info("Server initialization completed successfully!")
-            
-    except Exception as e:
-        logger.error(f"Server initialization failed: {str(e)}")
-        # Don't raise the exception to allow server to start even if initialization fails
-        logger.warning("Server will continue to start despite initialization issues")
-
-
-if __name__ == "__main__":
-    asyncio.run(startup_initialization())
     uvicorn.run(app, host="0.0.0.0", port=5010, log_level="info")

backend/runtime_service.py

@@ -1,7 +1,6 @@
 import uvicorn
 import logging
 import warnings
-import asyncio
 
 from consts.const import APP_VERSION
 
@@ -12,31 +11,16 @@
 
 from apps.runtime_app import app
 from utils.logging_utils import configure_logging, configure_elasticsearch_logging
-from services.tool_configuration_service import initialize_tools_on_startup
+
 
 configure_logging(logging.INFO)
 configure_elasticsearch_logging()
 logger = logging.getLogger("runtime_service")
 
 
-async def startup_initialization():
-    """
-    Perform initialization tasks during server startup
-    """
+if __name__ == "__main__":
     logger.info("Starting server initialization...")
     logger.info(f"APP version is: {APP_VERSION}")
-    try:
-        # Initialize tools on startup - service layer handles detailed logging
-        await initialize_tools_on_startup()
-        logger.info("Server initialization completed successfully!")
-    except Exception as e:
-        logger.error(f"Server initialization failed: {str(e)}")
-        # Don't raise the exception to allow server to start even if initialization fails
-        logger.warning("Server will continue to start despite initialization issues")
-
-
-if __name__ == "__main__":
-    asyncio.run(startup_initialization())
     uvicorn.run(app, host="0.0.0.0", port=5014, log_level="info")
 
 

backend/services/tool_configuration_service.py

@@ -11,7 +11,7 @@
 import jsonref
 from mcpadapt.smolagents_adapter import _sanitize_function_name
 
-from consts.const import DEFAULT_USER_ID, LOCAL_MCP_SERVER, DATA_PROCESS_SERVICE
+from consts.const import LOCAL_MCP_SERVER, DATA_PROCESS_SERVICE
 from consts.exceptions import MCPConnectionError, ToolExecutionException, NotFoundException
 from consts.model import ToolInstanceInfoRequest, ToolInfo, ToolSourceEnum, ToolValidateRequest
 from database.remote_mcp_db import get_mcp_records_by_tenant, get_mcp_server_by_name_and_tenant
@@ -22,7 +22,6 @@
     update_tool_table_from_scan_tool_list,
     search_last_tool_instance_by_tool_id,
 )
-from database.user_tenant_db import get_all_tenant_ids
 from services.file_management_service import get_llm_model
 from services.vectordatabase_service import get_embedding_model, get_vector_db_core
 from services.tenant_config_service import get_selected_knowledge_list, build_knowledge_name_mapping
@@ -367,71 +366,6 @@ async def list_all_tools(tenant_id: str):
     return formatted_tools
 
 
-async def initialize_tools_on_startup():
-    """
-    Initialize and scan all tools during server startup for all tenants
-    
-    This function scans all available tools (local, LangChain, and MCP) 
-    and updates the database with the latest tool information for all tenants.
-    """
-    
-    logger.info("Starting tool initialization on server startup...")
-    
-    try:
-        # Get all tenant IDs from the database
-        tenant_ids = get_all_tenant_ids()
-        
-        if not tenant_ids:
-            logger.warning("No tenants found in database, skipping tool initialization")
-            return
-        
-        logger.info(f"Found {len(tenant_ids)} tenants: {tenant_ids}")
-        
-        total_tools = 0
-        successful_tenants = 0
-        failed_tenants = []
-        
-        # Process each tenant
-        for tenant_id in tenant_ids:
-            try:
-                logger.info(f"Initializing tools for tenant: {tenant_id}")
-                
-                # Add timeout to prevent hanging during startup
-                try:
-                    await asyncio.wait_for(
-                        update_tool_list(tenant_id=tenant_id, user_id=DEFAULT_USER_ID),
-                        timeout=60.0  # 60 seconds timeout per tenant
-                    )
-                    
-                    # Get the count of tools for this tenant
-                    tools_info = query_all_tools(tenant_id)
-                    tenant_tool_count = len(tools_info)
-                    total_tools += tenant_tool_count
-                    successful_tenants += 1
-                    
-                    logger.info(f"Tenant {tenant_id}: {tenant_tool_count} tools initialized")
-                    
-                except asyncio.TimeoutError:
-                    logger.error(f"Tool initialization timed out for tenant {tenant_id}")
-                    failed_tenants.append(f"{tenant_id} (timeout)")
-                    
-            except Exception as e:
-                logger.error(f"Tool initialization failed for tenant {tenant_id}: {str(e)}")
-                failed_tenants.append(f"{tenant_id} (error: {str(e)})")
-        
-        # Log final results
-        logger.info("Tool initialization completed!")
-        logger.info(f"Total tools available across all tenants: {total_tools}")
-        logger.info(f"Successfully processed: {successful_tenants}/{len(tenant_ids)} tenants")
-        
-        if failed_tenants:
-            logger.warning(f"Failed tenants: {', '.join(failed_tenants)}")
-        
-    except Exception as e:
-        logger.error(f"❌ Tool initialization failed: {str(e)}")
-        raise
-
-
 def load_last_tool_config_impl(tool_id: int, tenant_id: str, user_id: str):
     """
     Load the last tool configuration for a given tool ID

doc/docs/en/user-guide/agent-development.md

@@ -19,6 +19,8 @@ If you have an existing agent configuration, you can also import it:
 > - **Import anyway**: Keep the duplicate name; the imported agent will be in an unavailable state and requires manual modification of the Agent name and variable name before it can be used
 > - **Regenerate and import**: The system will call the LLM to rename the Agent, which will consume a certain amount of model tokens and may take longer
 
+> 📌 **Important:** For agents created via import, if their tools include `knowledge_base_search` or other knowledge base search tools, these tools will only search **knowledge bases that the currently logged-in user is allowed to access in this environment**. The original knowledge base configuration in the exported agent will *not* be automatically inherited, so actual search results and answer quality may differ from what the original author observed.
+
 <div style="display: flex; justify-content: left;">
   <img src="./assets/agent-development/duplicated_import.png" style="width: 80%; height: auto;" />
 </div>

doc/docs/zh/user-guide/agent-development.md

@@ -19,6 +19,8 @@
 > - **直接导入**：保留重复名称，导入后的智能体会处于不可用状态，需手动修改 Agent 名称和变量名后才能使用
 > - **重新生成并导入**：系统将调用 LLM 对 Agent 进行重命名，会消耗一定的模型 token 数，可能耗时较长
 
+> 📌 **重要说明**：通过导入创建的智能体，如果其工具中包含 `knowledge_base_search` 等知识库检索工具，这些工具只会检索**当前登录用户在本环境中有权限访问的知识库**。导入文件中原有的知识库配置不会自动继承，因此实际检索结果和回答效果，可能与智能体原作者环境下的表现存在差异。
+
 <div style="display: flex; justify-content: left;">
   <img src="./assets/agent-development/duplicated_import.png" style="width: 80%; height: auto;" />
 </div>