feat: model name log, args help hint and the proxy of key does not participate in the judgment.

2023-11-17 11:13:25 +08:00
parent aaacbf29ea
commit e247e7832d
7 changed files with 50 additions and 23 deletions
@@ -8,3 +8,15 @@ class ChatType(IntEnum):
    CHATGPT = 2  # ChatGPT
    XINGHUO_WEB = 3  # 讯飞星火
    ChatGLM = 4  # ChatGLM
+
+    @staticmethod
+    def is_in_chat_types(chat_type: int) -> bool:
+        if chat_type in [ChatType.TIGER_BOT.value, ChatType.CHATGPT.value,
+                         ChatType.XINGHUO_WEB.value, ChatType.ChatGLM.value]:
+            return True
+        return False
+
+    @staticmethod
+    def help_hint():
+        return str({member.value: member.name for member in ChatType}).replace('{', '').replace('}', '')
+