类别调整

2024-07-31 13:12:11 +08:00
parent f8ad94de21
commit f809afe279
1 changed files with 2 additions and 1 deletions
@@ -37,7 +37,8 @@ class MoonshotChatBaseNode:
 class MoonshotSingleChatNode(MoonshotChatBaseNode):
    FUNCTION = "generate_single_response"
    RETURN_NAMES = ("response",)
-
+    CATEGORY = "🌙DW/moonshotChat"
+    
    def generate_single_response(self, prompt, model, temperature, max_tokens, system_message=""):
        if not self.client:
            return ("Error: MOONSHOT_API_KEY not set or invalid. Please check your api_key.ini file.",)