feat: add Poe Code-Llama-7b, Code-Llama-13b & Code-Llama-34b

josStorer · Aug 30, 2023 · 0f7a477 · 0f7a477
1 parent 9aa2adc
commit 0f7a477
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/config/index.mjs b/src/config/index.mjs
@@ -58,6 +58,9 @@ export const poeWebModelKeys = [
   'poeAiWeb_Llama_2_7b',
   'poeAiWeb_Llama_2_13b',
   'poeAiWeb_Llama_2_70b',
+  'poeAiWeb_Code_Llama_7b',
+  'poeAiWeb_Code_Llama_13b',
+  'poeAiWeb_Code_Llama_34b',
 ]
 
 /**
@@ -94,6 +97,9 @@ export const Models = {
   poeAiWeb_Llama_2_7b: { value: 'Llama-2-7b', desc: 'Poe AI (Web, Llama-2-7b)' },
   poeAiWeb_Llama_2_13b: { value: 'Llama-2-13b', desc: 'Poe AI (Web, Llama-2-13b)' },
   poeAiWeb_Llama_2_70b: { value: 'Llama-2-70b', desc: 'Poe AI (Web, Llama-2-70b)' },
+  poeAiWeb_Code_Llama_7b: { value: 'Code-Llama-7b', desc: 'Poe AI (Web, Code-Llama-7b)' },
+  poeAiWeb_Code_Llama_13b: { value: 'Code-Llama-13b', desc: 'Poe AI (Web, Code-Llama-13b)' },
+  poeAiWeb_Code_Llama_34b: { value: 'Code-Llama-34b', desc: 'Poe AI (Web, Code-Llama-34b)' },
   chatgptApi4_8k: { value: 'gpt-4', desc: 'ChatGPT (GPT-4-8k)' },
   chatgptApi4_8k_0613: { value: 'gpt-4', desc: 'ChatGPT (GPT-4-8k 0613)' },
   chatgptApi4_32k: { value: 'gpt-4-32k', desc: 'ChatGPT (GPT-4-32k)' },

diff --git a/src/services/clients/poe/index.mjs b/src/services/clients/poe/index.mjs
@@ -155,6 +155,12 @@ export default class PoeAiClient {
       bot = 'llama_2_13b_chat'
     } else if (bot === 'Llama-2-70b') {
       bot = 'llama_2_70b_chat'
+    } else if (bot === 'Code-Llama-7b') {
+      bot = 'code_llama_7b_instruct'
+    } else if (bot === 'Code-Llama-13b') {
+      bot = 'code_llama_13b_instruct'
+    } else if (bot === 'Code-Llama-34b') {
+      bot = 'code_llama_34b_instruct'
     }
 
     this.bot = bot