hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-23 09:46:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

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
parent 45d21677a0
commit 049abd698f
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)