From f809afe2795fc6fbd7a13f79e007457bbb7aaebc Mon Sep 17 00:00:00 2001
From: yunxiang wei <121092793+yiwangsimple@users.noreply.github.com>
Date: Wed, 31 Jul 2024 13:12:11 +0800
Subject: [PATCH] =?UTF-8?q?=20=E7=B1=BB=E5=88=AB=E8=B0=83=E6=95=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 nodes/moonshot_chat_nodes.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/nodes/moonshot_chat_nodes.py b/nodes/moonshot_chat_nodes.py
index dca0649..71b688b 100644
--- a/nodes/moonshot_chat_nodes.py
+++ b/nodes/moonshot_chat_nodes.py
@@ -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.",)