|
|
|
@@ -12,7 +12,6 @@ from controllers.console.app.error import ( |
|
|
|
) |
|
|
|
from controllers.console.wraps import account_initialization_required, setup_required |
|
|
|
from core.errors.error import ModelCurrentlyNotSupportError, ProviderTokenNotInitError, QuotaExceededError |
|
|
|
from core.helper.code_executor.code_node_provider import CodeNodeProvider |
|
|
|
from core.helper.code_executor.javascript.javascript_code_provider import JavascriptCodeProvider |
|
|
|
from core.helper.code_executor.python3.python3_code_provider import Python3CodeProvider |
|
|
|
from core.llm_generator.llm_generator import LLMGenerator |
|
|
|
@@ -126,11 +125,13 @@ class InstructionGenerateApi(Resource): |
|
|
|
parser.add_argument("model_config", type=dict, required=True, nullable=False, location="json") |
|
|
|
parser.add_argument("ideal_output", type=str, required=False, default="", location="json") |
|
|
|
args = parser.parse_args() |
|
|
|
providers: list[type[CodeNodeProvider]] = [Python3CodeProvider, JavascriptCodeProvider] |
|
|
|
code_provider: type[CodeNodeProvider] | None = next( |
|
|
|
(p for p in providers if p.is_accept_language(args["language"])), None |
|
|
|
code_template = ( |
|
|
|
Python3CodeProvider.get_default_code() |
|
|
|
if args["language"] == "python" |
|
|
|
else (JavascriptCodeProvider.get_default_code()) |
|
|
|
if args["language"] == "javascript" |
|
|
|
else "" |
|
|
|
) |
|
|
|
code_template = code_provider.get_default_code() if code_provider else "" |
|
|
|
try: |
|
|
|
# Generate from nothing for a workflow node |
|
|
|
if (args["current"] == code_template or args["current"] == "") and args["node_id"] != "": |