From 52249b0bc0a333b17f42bf86d18d8676d9498c41 Mon Sep 17 00:00:00 2001
From: EvolvedGhost <92856393+EvolvedGhost@users.noreply.github.com>
Date: Thu, 3 Aug 2023 12:15:41 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=A0=E9=99=A4Poe=E5=B7=B2=E5=BC=83?=
 =?UTF-8?q?=E7=94=A8=E7=9A=84=E6=A8=A1=E5=9E=8B=E5=B9=B6=E6=B7=BB=E5=8A=A0?=
 =?UTF-8?q?=E5=87=A0=E4=B8=AA=E6=96=B0=E7=9A=84Poe=E6=A8=A1=E5=9E=8B=20(#1?=
 =?UTF-8?q?110)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* Fit newest POE

* Fit newest POE

* Fix name

* Fit newest poe

* Update poe to newest fix some error

* Fix name error
---
 adapter/quora/poe.py | 11 +++++++----
 constants.py         |  6 ++++--
 manager/bot.py       |  6 ++++--
 requirements.txt     |  2 +-
 4 files changed, 16 insertions(+), 9 deletions(-)

diff --git a/adapter/quora/poe.py b/adapter/quora/poe.py
index 04b6659d..1f144ec0 100644
--- 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
index 868b2e84..9cbe58f3 100644
--- 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
index f04154dd..bf334714 100644
--- 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
index bd744107..605d537a 100644
--- 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