Dark-Alex-17/llm-functions
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: improve execute js/py code and collect results (#117)

Browse Source
This commit is contained in:
sigoden
2024-10-23 17:22:11 +08:00
committed by GitHub
parent d499954dbb
commit fbeaa9cb2c
6 changed files with 14 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/run-tool.py
+2 -4
View File
@@ -72,9 +72,7 @@ def load_env(file_path):
if env_name not in os.environ:
env_value = "=".join(value_parts).strip()
if env_value.startswith('"') and env_value.endswith('"'):
env_value = env_value[1:-1]
elif env_value.startswith("'") and env_value.endswith("'"):
if (env_value.startswith('"') and env_value.endswith('"')) or (env_value.startswith("'") and env_value.endswith("'")):
env_value = env_value[1:-1]
env_vars[env_name] = env_value
@@ -110,7 +108,7 @@ def return_to_llm(value):
value_type = type(value).__name__
if value_type in ("str", "int", "float", "bool"):
writer.write(value)
writer.write(str(value))
elif value_type == "dict" or value_type == "list":
value_str = json.dumps(value, indent=2)
assert value == json.loads(value_str)