加入豆包AI ：doubao-1-5-lite-32k-250115

2025-02-28 16:33:27 +08:00
parent 24502ed18c
commit 06718629e7
7 changed files with 165 additions and 7 deletions
--- a/constants.py
+++ b/constants.py
@@ -11,12 +11,14 @@ class ChatType(IntEnum):
    BardAssistant = 5  # Google Bard
    ZhiPu = 6  # ZhiPu
    CLAUDE = 7  # CLAUDE
+    DOUBAO = 8  # doubao

    @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,
-                         ChatType.BardAssistant.value, ChatType.ZhiPu.value, ChatType.CLAUDE.value]:
+                         ChatType.BardAssistant.value, ChatType.ZhiPu.value, ChatType.CLAUDE.value,
+                         ChatType.DOUBAO.value]:
            return True
        return False