youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-03-03 14:56:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

add get default node config

Browse Source
This commit is contained in:
takatost
2024-03-04 13:21:24 +08:00
parent 75559bcbf9
commit df809ff435
33 changed files with 314 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
api/core/workflow/nodes/code/__init__.py Normal file
View File

64
api/core/workflow/nodes/code/code_node.py Normal file
View File

@ -0,0 +1,64 @@
from typing import Optional
from core.workflow.nodes.base_node import BaseNode
class CodeNode(BaseNode):
@classmethod
def get_default_config(cls, filters: Optional[dict] = None) -> dict:
"""
Get default config of node.
:param filters: filter by node config parameters.
:return:
"""
if filters and filters.get("code_language") == "javascript":
return {
"type": "code",
"config": {
"variables": [
{
"variable": "arg1",
"value_selector": []
},
{
"variable": "arg2",
"value_selector": []
}
],
"code_language": "javascript",
"code": "async function main(arg1, arg2) {\n return new Promise((resolve, reject) => {"
"\n if (true) {\n resolve({\n \"result\": arg1 + arg2"
"\n });\n } else {\n reject(\"e\");\n }\n });\n}",
"outputs": [
{
"variable": "result",
"variable_type": "number"
}
]
}
}
return {
"type": "code",
"config": {
"variables": [
{
"variable": "arg1",
"value_selector": []
},
{
"variable": "arg2",
"value_selector": []
}
],
"code_language": "python3",
"code": "def main(\n arg1: int,\n arg2: int,\n) -> int:\n return {\n \"result\": arg1 "
"+ arg2\n }",
"outputs": [
{
"variable": "result",
"variable_type": "number"
}
]
}
}