diff --git a/src/config/index.mjs b/src/config/index.mjs index d7fd3d2f..2a72f823 100644 --- 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 index c8bef0eb..3eb75403 100644 --- 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