add /o command

harmony-one · Apr 2, 2024 · 33e6fa8 · 33e6fa8
1 parent 1f26b13
commit 33e6fa8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/modules/llms/helpers.ts b/src/modules/llms/helpers.ts
@@ -14,8 +14,9 @@ export enum SupportedCommands {
   bardF = 'bard',
   claudeOpus = 'claude',
   opus = 'opus',
+  opusShort = 'o',
   claudeSonnet = 'claudes',
-  opusShort = 'c',
+  claudeShort = 'c',
   sonnet = 'sonnet',
   sonnetShort = 's',
   claudeHaiku = 'haiku',

diff --git a/src/modules/llms/index.ts b/src/modules/llms/index.ts
@@ -133,7 +133,7 @@ export class LlmsBot implements PayableBot {
       await this.onChat(ctx, LlmsModelsEnum.GEMINI)
       return
     }
-    if (ctx.hasCommand([SupportedCommands.claudeOpus, SupportedCommands.opus, SupportedCommands.opusShort]) || (hasClaudeOpusPrefix(ctx.message?.text ?? '') !== '')) {
+    if (ctx.hasCommand([SupportedCommands.claudeOpus, SupportedCommands.opus, SupportedCommands.opusShort, SupportedCommands.claudeShort]) || (hasClaudeOpusPrefix(ctx.message?.text ?? '') !== '')) {
       await this.onChat(ctx, LlmsModelsEnum.CLAUDE_OPUS)
       return
     }