jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

improve: test CodeExecutor with code templates and extract CodeLanguage enum (#4098)

Browse source
This commit is contained in:
Bowen Liang 2024-05-07 12:37:18 +08:00 • committed by GitHub
commit 049abd698f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 79 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

6
api/core/tools/provider/builtin/code/tools/simple_code.py
View file

@ -1,6 +1,6 @@
from typing import Any
from core.helper.code_executor.code_executor import CodeExecutor
from core.helper.code_executor.code_executor import CodeExecutor, CodeLanguage
from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.tool.builtin_tool import BuiltinTool
@ -11,10 +11,10 @@ class SimpleCode(BuiltinTool):
invoke simple code
"""
language = tool_parameters.get('language', 'python3')
language = tool_parameters.get('language', CodeLanguage.PYTHON3)
code = tool_parameters.get('code', '')
if language not in ['python3', 'javascript']:
if language not in [CodeLanguage.PYTHON3, CodeLanguage.JAVASCRIPT]:
raise ValueError(f'Only python3 and javascript are supported, not {language}')
result = CodeExecutor.execute_code(language, '', code)