删除Poe已弃用的模型并添加几个新的Poe模型 (#1110)

* Fit newest POE * Fit newest POE * Fix name * Fit newest poe * Update poe to newest fix some error * Fix name error
lss233 · Aug 3, 2023 · 52249b0 · 52249b0
1 parent 5a376b8
commit 52249b0
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 9 deletions.
diff --git a/adapter/quora/poe.py b/adapter/quora/poe.py
@@ -11,16 +11,19 @@
 
 
 class PoeBot(Enum):
-    """Poe 支持的机器人：{'capybara': 'Sage', 'beaver': 'GPT-4', 'a2_2': 'Claude+','a2': 'Claude','a2_100k': 'Claude100k', 'chinchilla': 'ChatGPT', 'hutia': 'NeevaAI',
-    'nutria': 'Dragonfly'} """
+    """Poe 支持的机器人：{'capybara': 'Assistant', 'a2': 'Claude-instant', 'beaver': 'GPT-4', 'chinchilla': 'ChatGPT',
+    'llama_2_7b_chat': 'Llama-2-7b', 'a2_100k': 'Claude-instant-100k', 'llama_2_13b_chat': 'Llama-2-13b', 'agouti': 'ChatGPT-16k', 
+    'vizcacha': 'GPT-4-32k', 'acouchy': 'Google-PaLM', 'llama_2_70b_chat':'Llama-2-70b', 'a2_2': 'Claude-2-100k'} """
     Sage = "capybara"
     GPT4 = "beaver"
+    GPT432k = "vizcacha"
     Claude2 = "a2_2"
     Claude = "a2"
     Claude100k = "a2_100k"
     ChatGPT = "chinchilla"
-    NeevaAI = "hutia"
-    Dragonfly = "nutria"
+    ChatGPT16k = "agouti"
+    Llama2 = "llama_2_70b_chat"
+    PaLM = "acouchy"
 
     @staticmethod
     def parse(bot_name: str):

diff --git a/constants.py b/constants.py
@@ -13,12 +13,14 @@ class LlmName(Enum):
     SlackClaude = "slack-claude"
     PoeSage = "poe-sage"
     PoeGPT4 = "poe-gpt4"
+    PoeGPT432k = "poe-gpt432k"
     PoeClaude2 = "poe-claude2"
     PoeClaude = "poe-claude"
     PoeClaude100k = "poe-claude100k"
     PoeChatGPT = "poe-chatgpt"
-    PoeDragonfly = "poe-dragonfly"
-    PoeNeevaAI = "poe-neevaai"
+    PoeChatGPT16k = "poe-chatgpt16k"
+    PoeLlama2 = "poe-llama2"
+    PoePaLM = "poe-palm"
     ChatGPT_Web = "chatgpt-web"
     ChatGPT_Api = "chatgpt-api"
     Bing = "bing"

diff --git a/manager/bot.py b/manager/bot.py
@@ -561,12 +561,14 @@ def bots_info(self):
         if len(self.bots['poe-web']) > 0:
             bot_info += f"* {LlmName.PoeSage.value} : POE Sage 模型\n"
             bot_info += f"* {LlmName.PoeGPT4.value} : POE ChatGPT4 模型\n"
+            bot_info += f"* {LlmName.PoeGPT432k.value} : POE ChatGPT4 32k 模型\n"
             bot_info += f"* {LlmName.PoeClaude.value} : POE Claude 模型\n"
             bot_info += f"* {LlmName.PoeClaude2.value} : POE Claude+ 模型\n"
             bot_info += f"* {LlmName.PoeClaude100k.value} : POE Claude 100k 模型\n"
             bot_info += f"* {LlmName.PoeChatGPT.value} : POE ChatGPT 模型\n"
-            bot_info += f"* {LlmName.PoeDragonfly.value} : POE Dragonfly 模型\n"
-            bot_info += f"* {LlmName.PoeNeevaAI.value} : POE NeevaAI 模型\n"
+            bot_info += f"* {LlmName.PoeChatGPT16k.value} : POE ChatGPT 16k 模型\n"
+            bot_info += f"* {LlmName.PoeLlama2.value} : POE Llama2 模型\n"
+            bot_info += f"* {LlmName.PoePaLM.value} : POE PaLM 模型\n"
         if len(self.bots['slack-accesstoken']) > 0:
             bot_info += f"* {LlmName.SlackClaude.value} : Slack Claude 模型\n"
         if len(self.bots['xinghuo-cookie']) > 0:

diff --git a/requirements.txt b/requirements.txt
@@ -30,7 +30,7 @@ tls-client
 python-dateutil~=2.8.2
 discord.py
 azure-cognitiveservices-speech
-poe-api~=0.4.17
+poe-api==0.5.2
 
 regex~=2023.6.3
 httpx~=0.24.1