diff --git a/node.py b/node.py index 596ec55..7703239 100644 --- a/node.py +++ b/node.py @@ -3,6 +3,9 @@ from dashscope import Generation class DashscopeLLMLoader: + def __init__(self): + pass + @classmethod def INPUT_TYPES(cls): return { @@ -51,10 +54,12 @@ class DashscopeLLMLoader: } } - CATEGORY = "dashscope" FUNCTION = "select_model" + OUTPUT_NODE = True RETURN_TYPES = ("String",) + CATEGORY = "dashscope" + def select_model(self, model_type, qwen_max, qwen_plus, qwen_turbo, qwen_long): if model_type == "qwen_max": model_version = qwen_max