agent: python tool: test serializability of variables
This commit is contained in:
parent
082d54db14
commit
f9afb041e2
1 changed files with 12 additions and 1 deletions
|
@ -3,6 +3,13 @@ import sys
|
||||||
import types
|
import types
|
||||||
from typing import Dict, Union
|
from typing import Dict, Union
|
||||||
|
|
||||||
|
def _is_serializable(obj) -> bool:
|
||||||
|
try:
|
||||||
|
json.dumps(obj)
|
||||||
|
return True
|
||||||
|
except Exception as e:
|
||||||
|
return False
|
||||||
|
|
||||||
def execute_python(source: str) -> Union[Dict, str]:
|
def execute_python(source: str) -> Union[Dict, str]:
|
||||||
"""
|
"""
|
||||||
Evaluate a Python program and return the globals it declared.
|
Evaluate a Python program and return the globals it declared.
|
||||||
|
@ -21,7 +28,11 @@ def execute_python(source: str) -> Union[Dict, str]:
|
||||||
results = {
|
results = {
|
||||||
k: v
|
k: v
|
||||||
for k, v in namespace.items()
|
for k, v in namespace.items()
|
||||||
if not k.startswith('_') and not isinstance(v, type) and not callable(v) and not isinstance(v, types.ModuleType)
|
if not k.startswith('_') \
|
||||||
|
and not isinstance(v, type) \
|
||||||
|
and not isinstance(v, types.ModuleType) \
|
||||||
|
and not callable(v) \
|
||||||
|
and _is_serializable(v)
|
||||||
}
|
}
|
||||||
sys.stderr.write(f"Results: {json.dumps(results, indent=2)}\n")
|
sys.stderr.write(f"Results: {json.dumps(results, indent=2)}\n")
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue