Skip to content

Commit ae7de7d

Browse files
fix: treat default template of code as empty (langgenius#24106)
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
1 parent ef5decc commit ae7de7d

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

api/controllers/console/app/generator.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,9 @@
1212
)
1313
from controllers.console.wraps import account_initialization_required, setup_required
1414
from core.errors.error import ModelCurrentlyNotSupportError, ProviderTokenNotInitError, QuotaExceededError
15+
from core.helper.code_executor.code_node_provider import CodeNodeProvider
16+
from core.helper.code_executor.javascript.javascript_code_provider import JavascriptCodeProvider
17+
from core.helper.code_executor.python3.python3_code_provider import Python3CodeProvider
1518
from core.llm_generator.llm_generator import LLMGenerator
1619
from core.model_runtime.errors.invoke import InvokeError
1720
from libs.login import login_required
@@ -123,9 +126,14 @@ def post(self):
123126
parser.add_argument("model_config", type=dict, required=True, nullable=False, location="json")
124127
parser.add_argument("ideal_output", type=str, required=False, default="", location="json")
125128
args = parser.parse_args()
126-
129+
providers: list[type[CodeNodeProvider]] = [Python3CodeProvider, JavascriptCodeProvider]
130+
code_provider: type[CodeNodeProvider] | None = next(
131+
(p for p in providers if p.is_accept_language(args["language"])), None
132+
)
133+
code_template = code_provider.get_default_code() if code_provider else ""
127134
try:
128-
if args["current"] == "" and args["node_id"] != "": # Generate from nothing for a workflow node
135+
# Generate from nothing for a workflow node
136+
if (args["current"] == code_template or args["current"] == "") and args["node_id"] != "":
129137
from models import App, db
130138
from services.workflow_service import WorkflowService
131139

0 commit comments

Comments
 (0)