Skip to content

Commit 13f976f

Browse files
fix
1 parent 234ab45 commit 13f976f

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

apps/common/utils/tool_code.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,10 @@ def init_sandbox_dir():
9797
def exec_code(self, code_str, keywords, function_name=None):
9898
_id = str(uuid.uuid7())
9999
action_function = f'({function_name !a}, locals_v.get({function_name !a}))' if function_name else 'locals_v.popitem()'
100-
set_run_user = f'os.setgid({pwd.getpwnam(_run_user).pw_gid});os.setuid({pwd.getpwnam(_run_user).pw_uid});' if _enable_sandbox else ''
100+
if pwd and _enable_sandbox:
101+
set_run_user = f'os.setgid({pwd.getpwnam(_run_user).pw_gid});os.setuid({pwd.getpwnam(_run_user).pw_uid});'
102+
else:
103+
set_run_user = ''
101104
_exec_code = f"""
102105
try:
103106
import os, sys, json
@@ -293,7 +296,10 @@ def visit_Return(self, node):
293296

294297
def generate_mcp_server_code(self, code_str, params, name, description, tool_id):
295298
code = self._generate_mcp_server_code(code_str, params, name, description, tool_id)
296-
set_run_user = f'os.setgid({pwd.getpwnam(_run_user).pw_gid});os.setuid({pwd.getpwnam(_run_user).pw_uid});' if _enable_sandbox else ''
299+
if pwd and _enable_sandbox:
300+
set_run_user = f'os.setgid({pwd.getpwnam(_run_user).pw_gid});os.setuid({pwd.getpwnam(_run_user).pw_uid});'
301+
else:
302+
set_run_user = ''
297303
return f"""
298304
import os, sys, logging
299305
logging.basicConfig(level=logging.WARNING)

0 commit comments

Comments
 (0)