Merge pull request #354 from harmony-one/voice-command

fix prefix command
harmony-one · Jan 29, 2024 · d14408b · d14408b
2 parents 50d9f79 + b2116de
commit d14408b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/src/bot.ts b/src/bot.ts
@@ -402,7 +402,6 @@ const onMessage = async (ctx: OnMessageContext): Promise<void> => {
       }
       for (const config of Object.values(PayableBots)) {
         const bot = config.bot
-
         if (!bot.isSupportedEvent(ctx)) {
           continue
         }

diff --git a/src/modules/llms/helpers.ts b/src/modules/llms/helpers.ts
@@ -20,7 +20,7 @@ export enum SupportedCommands {
 }
 
 export const MAX_TRIES = 3
-const LLAMA_PREFIX_LIST = ['*']
+const LLAMA_PREFIX_LIST = ['* ']
 const BARD_PREFIX_LIST = ['b.', 'B.']
 
 export const isMentioned = (

diff --git a/src/modules/open-ai/helpers.ts b/src/modules/open-ai/helpers.ts
@@ -27,9 +27,9 @@ export enum SupportedCommands {
 
 export const MAX_TRIES = 3
 
-export const DALLE_PREFIX_LIST = ['i. ', ',', 'image ', 'd.', 'img ']
-export const CHAT_GPT_PREFIX_LIST = ['a.', '.']
-export const NEW_PREFIX_LIST = ['n.', '..']
+export const DALLE_PREFIX_LIST = ['i.', ', ', 'image ', 'd.', 'img ']
+export const CHAT_GPT_PREFIX_LIST = ['a.', '. ']
+export const NEW_PREFIX_LIST = ['n.', '.. ']
 
 export const isMentioned = (
   ctx: OnMessageContext | OnCallBackQueryData