diff --git a/src/modules/open-ai/helpers.ts b/src/modules/open-ai/helpers.ts index dc32975..a873124 100644 --- a/src/modules/open-ai/helpers.ts +++ b/src/modules/open-ai/helpers.ts @@ -16,8 +16,10 @@ 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' } @@ -25,7 +27,7 @@ export const SupportedCommands = { 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 index 3642eac..809006a 100644 --- 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