refactor: simplify error handling in MCP server code execution

Author: liuruibin

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

@@ -10,7 +10,9 @@
 import json
 import os
 import re
+import sys
 import time
+import traceback
 from functools import reduce
 from typing import List, Dict
 
@@ -143,7 +145,7 @@ def mcp_response_generator(chat_model, message_list, mcp_servers):
             except StopAsyncIteration:
                 break
     except Exception as e:
-        maxkb_logger.error(f'Exception: {e}')
+        maxkb_logger.error(f'Exception: {e}', traceback.format_exc())
     finally:
         loop.close()
 
@@ -285,9 +287,10 @@ def _handle_mcp_request(self, mcp_enable, tool_enable, mcp_source, mcp_servers,
                     code_path = f'{executor.sandbox_path}/execute/{tool_id}.py'
                     with open(code_path, 'w') as f:
                         f.write(code)
+                        os.system(f"chown sandbox:root {code_path}")
 
                     tool_config = {
-                        'command': 'python',
+                        'command': sys.executable,
                         'args': [code_path],
                         'transport': 'stdio',
                     }

apps/common/utils/tool_code.py

@@ -120,19 +120,16 @@ 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
-    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]
-    exec({dedent(code)!a})
-except Exception as e:
-    pass
+import os
+import sys
+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]
+exec({dedent(code)!a})
 """
 
     def _exec_sandbox(self, _code, _id):