feat: refactor MCP code generation and update cleanup logic

Author: liuruibin

apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py

@@ -281,7 +281,7 @@ def _handle_mcp_request(self, mcp_enable, tool_enable, mcp_source, mcp_servers,
                 for tool_id in tool_ids:
                     tool = QuerySet(Tool).filter(id=tool_id).first()
                     executor = ToolExecutor()
-                    code = executor.get_exec_code(tool.code)
+                    code = executor.generate_mcp_server_code(tool.code)
                     code_path = f'{executor.sandbox_path}/execute/{tool_id}.py'
                     with open(code_path, 'w') as f:
                         f.write(code)
@@ -340,8 +340,8 @@ def get_details(self, index: int, **kwargs):
             # 清理工具代码文件，延时删除，避免文件被占用
             for tool_id in self.context.get('tool_ids'):
                 code_path = f'{executor.sandbox_path}/execute/{tool_id}.py'
-                if os.path.exists(code_path):
-                    os.remove(code_path)
+                # if os.path.exists(code_path):
+                #     os.remove(code_path)
         return {
             'name': self.node.properties.get('stepName'),
             "index": index,

apps/common/utils/tool_code.py

@@ -116,27 +116,21 @@ def _generate_mcp_server_code(self, _code):
 
         return "\n".join(code_parts)
 
-    def get_exec_code(self, code_str):
+    def generate_mcp_server_code(self, code_str):
         python_paths = CONFIG.get_sandbox_python_package_paths().split(',')
         code = self._generate_mcp_server_code(code_str)
         return f"""
 try:
     import os
     import sys
-    import pickle
     path_to_exclude = ['/opt/py3/lib/python3.11/site-packages', '/opt/maxkb-app/apps']
     sys.path = [p for p in sys.path if p not in path_to_exclude]
     sys.path += {python_paths}
     env = dict(os.environ)
     for key in list(env.keys()):
         if key in os.environ and (key.startswith('MAXKB') or key.startswith('POSTGRES') or key.startswith('PG') or key.startswith('REDIS') or key == 'PATH'):
             del os.environ[key]
-    locals_v={'{}'}
-    globals_v=globals()
-    exec({dedent(code)!a}, globals_v, locals_v)
-    f_name, f = locals_v.popitem()
-    for local in locals_v:
-        globals_v[local] = locals_v[local]
+    exec({dedent(code)!a})
 except Exception as e:
     pass
 """