瀏覽代碼

fix: code Interpreter error handling not work (#21736)

tags/1.5.1
quicksand 4 月之前
父節點
當前提交
4198a533ad
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 6 行新增3 行删除
  1. 6
    3
      api/core/tools/builtin_tool/providers/code/tools/simple_code.py

+ 6
- 3
api/core/tools/builtin_tool/providers/code/tools/simple_code.py 查看文件

@@ -4,6 +4,7 @@ from typing import Any, Optional
from core.helper.code_executor.code_executor import CodeExecutor, CodeLanguage
from core.tools.builtin_tool.tool import BuiltinTool
from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.errors import ToolInvokeError


class SimpleCode(BuiltinTool):
@@ -25,6 +26,8 @@ class SimpleCode(BuiltinTool):
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)

yield self.create_text_message(result)
try:
result = CodeExecutor.execute_code(language, "", code)
yield self.create_text_message(result)
except Exception as e:
raise ToolInvokeError(str(e))

Loading…
取消
儲存