Fix/incorrect code template (#7490)

This commit is contained in:
Yeuoly
2024-08-21 15:31:13 +08:00
committed by GitHub
parent 85fc0fdb51
commit a02118d5bc
3 changed files with 6 additions and 7 deletions

View File

@@ -4,7 +4,7 @@ from enum import Enum
from threading import Lock
from typing import Literal, Optional
from httpx import get, post
from httpx import Timeout, get, post
from pydantic import BaseModel
from yarl import URL
@@ -21,7 +21,7 @@ logger = logging.getLogger(__name__)
CODE_EXECUTION_ENDPOINT = dify_config.CODE_EXECUTION_ENDPOINT
CODE_EXECUTION_API_KEY = dify_config.CODE_EXECUTION_API_KEY
CODE_EXECUTION_TIMEOUT = (10, 60)
CODE_EXECUTION_TIMEOUT = Timeout(connect=10, write=10, read=60, pool=None)
class CodeExecutionException(Exception):
pass
@@ -116,7 +116,7 @@ class CodeExecutor:
if response.data.error:
raise CodeExecutionException(response.data.error)
return response.data.stdout
return response.data.stdout or ''
@classmethod
def execute_workflow_code_template(cls, language: CodeLanguage, code: str, inputs: dict, dependencies: Optional[list[CodeDependency]] = None) -> dict:

View File

@@ -13,7 +13,7 @@ class Python3CodeProvider(CodeNodeProvider):
def get_default_code(cls) -> str:
return dedent(
"""
def main(arg1: int, arg2: int) -> dict:
def main(arg1: str, arg2: str) -> dict:
return {
"result": arg1 + arg2,
}