add /i command

harmony-one · Jan 11, 2024 · 499ad36 · 499ad36
1 parent bb8833f
commit 499ad36
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/src/modules/open-ai/helpers.ts b/src/modules/open-ai/helpers.ts
@@ -16,16 +16,18 @@ export const SupportedCommands = {
   ask32: { name: 'ask32' },
   gpt: { name: 'gpt' },
   last: { name: 'last' },
-  dalle: { name: 'image' },
+  dalle: { name: 'dalle' },
+  dalleImg: { name: 'image' },
   dalleShort: { name: 'img' },
+  dalleShorter: { name: 'i' },
   genImgEn: { name: 'genImgEn' },
   on: { name: 'on' },
   off: { name: 'off' }
 }
 
 export const MAX_TRIES = 3
 
-const DALLE_PREFIX_LIST = ['i.', ', ', 'image ', 'd.', 'img ']
+const DALLE_PREFIX_LIST = ['i. ', ',', 'image ', 'd.', 'img ', 'i ']
 
 export const isMentioned = (
   ctx: OnMessageContext | OnCallBackQueryData

diff --git a/src/modules/open-ai/index.ts b/src/modules/open-ai/index.ts
@@ -101,8 +101,10 @@ export class OpenAIBot implements PayableBot {
         return 0
       }
       if (
-        ctx.hasCommand(SupportedCommands.dalle.name) ||
-        ctx.hasCommand(SupportedCommands.dalleShort.name)
+        ctx.hasCommand([SupportedCommands.dalle.name,
+          SupportedCommands.dalleImg.name,
+          SupportedCommands.dalleShort.name,
+          SupportedCommands.dalleShorter.name])
       ) {
         const imageNumber = ctx.session.openAi.imageGen.numImages
         const imageSize = ctx.session.openAi.imageGen.imgSize
@@ -227,7 +229,10 @@ export class OpenAIBot implements PayableBot {
     }
 
     if (
-      ctx.hasCommand([SupportedCommands.dalle.name, SupportedCommands.dalleShort.name]) ||
+      ctx.hasCommand([SupportedCommands.dalle.name,
+        SupportedCommands.dalleImg.name,
+        SupportedCommands.dalleShort.name,
+        SupportedCommands.dalleShorter.name]) ||
       (ctx.message?.text?.startsWith('image ') && ctx.chat?.type === 'private')
     ) {
       let prompt = (ctx.match ? ctx.match : ctx.message?.text) as string