encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-03-30 10:57:20 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix: Fix unsafe_local_code_executor for import scope

Browse Source 
Merge https://github.com/google/adk-python/pull/869

How to reproduce the error:

```
from google.adk.code_executors import UnsafeLocalCodeExecutor
from google.adk.code_executors.code_execution_utils import CodeExecutionInput

result = UnsafeLocalCodeExecutor().execute_code(
    invocation_context=None,
    code_execution_input=CodeExecutionInput(
        code='''
import math

def pi():
    return math.pi

print(pi())
'''
    )
)

print(result)
```

output:
```
CodeExecutionResult(stdout='', stderr="name 'math' is not defined", output_files=[])
```
COPYBARA_INTEGRATE_REVIEW=https://github.com/google/adk-python/pull/869 from qieqieplus:main 63f557bbd3b7aa5c2801f5cc9e022d3364177308
PiperOrigin-RevId: 787145189
This commit is contained in:
qieqieplus
2025-07-25 09:44:59 -07:00
committed by Copybara-Service
parent a858d79b3a
commit 1778490e64
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/code_executors/unsafe_local_code_executor.py
+1 -2
View File
@@ -66,10 +66,9 @@ class UnsafeLocalCodeExecutor(BaseCodeExecutor):
try:
globals_ = {}
_prepare_globals(code_execution_input.code, globals_)
locals_ = {}
stdout = io.StringIO()
with redirect_stdout(stdout):
exec(code_execution_input.code, globals_, locals_)
exec(code_execution_input.code, globals_)
output = stdout.getvalue()
except Exception as e:
error = str(e)