fix: add general command

Author: gagik

snippets/ai/index.ts

@@ -45,7 +45,7 @@ class AI {
           try {
             this.ai = getAiSdkProvider(
               models[this.config.get('provider') as keyof typeof models](
-                event.value as string,
+                event.value == 'default' ? undefined : event.value as string,
               ),
               this.cliContext,
               this.config,
@@ -100,6 +100,11 @@ class AI {
     await this.ai.shell(prompt);
   }
 
+  @aiCommand()
+  async general(prompt: string) {
+    await this.ai.general(prompt);
+  }
+
   @aiCommand()
   async data(prompt: string) {
     await this.ai.data(prompt);

snippets/ai/providers/ai-provider.ts

@@ -86,7 +86,7 @@ export abstract class AiProvider {
     const commands = [
       {
         cmd: 'ai.ask',
-        desc: 'ask questions',
+        desc: 'ask MongoDB questions',
         example: 'ai.ask how do I run queries in mongosh?',
       },
       {
@@ -111,9 +111,14 @@ export abstract class AiProvider {
       },
       {
         cmd: 'ai.shell',
-        desc: 'generate general mongosh commands',
+        desc: 'generate administrative mongosh commands',
         example: 'ai.shell get sharding info',
       },
+      {
+        cmd: 'ai.general',
+        desc: 'use your model for general questions',
+        example: 'ai.general what is the meaning of life?',
+      },
       {
         cmd: 'ai.config',
         desc: 'configure the AI commands',
@@ -209,7 +214,7 @@ export abstract class AiProvider {
 
     this.thinking.stop();
     this.setInput(
-      this.cleanUpResponse(response, { expectedOutput: 'mongoshCommand' }),
+      this.formatResponse(response, { expectedOutput: 'mongoshCommand' }),
     );
   }
 
@@ -230,10 +235,24 @@ export abstract class AiProvider {
 
     this.thinking.stop();
     this.setInput(
-      this.cleanUpResponse(response, { expectedOutput: 'mongoshCommand' }),
+      this.formatResponse(response, { expectedOutput: 'mongoshCommand' }),
     );
   }
 
+  async general(prompt: string): Promise<void> {
+    const signal = AbortSignal.timeout(30_000);
+    this.thinking.start(signal);
+
+    const response = await this.getResponse(prompt, {
+      systemPrompt: 'Give brief answers without any formatting or markdown.',
+      signal,
+      expectedOutput: 'text',
+    });
+
+    this.thinking.stop();
+    this.respond(this.formatResponse(response, { expectedOutput: 'text' }));
+  }
+
   async data(prompt: string): Promise<void> {
     await this.ensureCollectionName(prompt);
     const signal = AbortSignal.timeout(30_000);
@@ -252,7 +271,7 @@ export abstract class AiProvider {
 
     this.thinking.stop();
     this.setInput(
-      this.cleanUpResponse(response, { expectedOutput: 'mongoshCommand' }),
+      this.formatResponse(response, { expectedOutput: 'mongoshCommand' }),
     );
   }
 
@@ -274,11 +293,11 @@ export abstract class AiProvider {
 
     this.thinking.stop();
     this.setInput(
-      this.cleanUpResponse(response, { expectedOutput: 'mongoshCommand' }),
+      this.formatResponse(response, { expectedOutput: 'mongoshCommand' }),
     );
   }
 
-  cleanUpResponse(
+  formatResponse(
     text: string,
     {
       expectedOutput,
@@ -322,7 +341,7 @@ export abstract class AiProvider {
     });
 
     this.thinking.stop();
-    this.respond(this.cleanUpResponse(response, { expectedOutput: 'text' }));
+    this.respond(this.formatResponse(response, { expectedOutput: 'text' }));
   }
 
   public clear() {