Fix/incorrect code template (#7490)
This commit is contained in:
parent
85fc0fdb51
commit
a02118d5bc
3 changed files with 6 additions and 7 deletions
|
|
@ -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:
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue