fix: README and index

Author: gagik

index.bson.br

@@ -1,52 +1,32 @@
-# LLM Command Snippet
+# AI Command Snippets
 
-This snippet adds an `llm` command to mongosh that provides helpful suggestions or recommendations for MongoDB-related tasks. The query results are generated using Groq API by default, with an option to use other models via Ollama.
+> [!CAUTION]
+> This is an experimental, early-stage snippet that is not meant for production use. 
 
-## Prerequisites
+This snippet adds a suite of commands accessible with the `ai` command. This includes:
 
-```
-export GROQ_API_KEY=gsk_XXXXXXX
-mongosh
-```
-
-## Usage
-
-After installing the snippet, you can use the `llm` command in your MongoDB shell like this:
-
-```javascript
-llm("very briefly, just the command, do not use markdown: in mongosh how to get the collections names of current db?");
-```
-
-This will output a possible solution to your query, such as `db.getCollectionNames()`.
-
-```javascript
-llm("very briefly, just the command, do not use markdown: in mongosh replace all documents of a collection with property {'set':'llm102'} with the new value {'set':'llm101'} in current db?")
-```
-
-This will output a possible solution to your query, such as `db.collection.updateMany({ set: 'llm102' }, { $set: { set: 'llm101' } })`.
-
-You can also specify a different model to use with an optional parameter:
-```javascript 
-llm("Your query here", { model: "phi3.5" });
-```
-This will use the specified model (in this case, 'phi3.5') via Ollama instead of the default Groq API.
-
-## Models
-
-By default, the `llm` command uses the Groq API with the 'llama-3.1-70b-versatile' model. You can use other models by specifying them in the optional parameter:
-
-- Groq API (default): No need to specify, just use `llm("Your query")`.
-- Ollama models: Specify the model name, e.g., `llm("Your query", { model: "gemini2" })` or `llm("Your query", { model: "phi3.5" })`.
-
-Note: When using Ollama models, make sure you have Ollama running locally on the default port (11434).
+|  |  |  |
+|---------|-------------|---------|
+| `ai.ask` | Ask questions about MongoDB | `ai.ask how do I run queries in mongosh?` |
+| `ai.data` | Generate data-related mongosh commands | `ai.data insert some sample user info` |
+| `ai.query` | Generate a MongoDB query | `ai.query find documents where name = "Ada"` |
+| `ai.aggregate` | Generate a MongoDB aggregation | `ai.aggregate find documents where name = "Ada"` |
+| `ai.collection` | Set the active collection | `ai.collection("users")` |
+| `ai.shell` | Generate general mongosh commands | `ai.shell get sharding info` |
+| `ai.general` | Ask general questions to your model | `ai.general what is the meaning of life?`
+| `ai.config` | Configure the AI commands | `ai.config.set("provider", "ollama")` |
 
+This currently supports 5 different AI providers: `docs, openai | mistral | atlas | ollama` and any model they support. For cloud providers, you can specify the API key with `MONGOSH_AI_API_KEY`.
 
 ## Installation
 
 You can install this snippet using the `snippet` command in mongosh:
 
 ```javascript
-snippet install llm-command
+config.set('snippetIndexSourceURLs',
+'https://github.com/gagik/mongosh-snippets/raw/refs/heads/ai/index.bson.br;'
++ config.get('snippetIndexSourceURLs') )
+snippet install ai
 ```
 
 ## Troubleshooting

snippets/ai/README.md

@@ -49,6 +49,7 @@ export class LoadingAnimation {
 }
 
 export interface CliContext {
+  ai: unknown;
   db: {
     _mongo: {
       _instanceState: {

snippets/ai/globals.d.ts

@@ -7,6 +7,7 @@ import { Config, ConfigSchema } from './config';
 import { CliContext, wrapAllFunctions } from './helpers';
 import chalk from 'chalk';
 
+
 class AI {
   private readonly replConfig: {
     set: (key: string, value: any) => Promise<void>;
@@ -161,6 +162,5 @@ class AI {
   }
 }
 
-module.exports = (globalThis: any) => {
-  globalThis.ai = new AI(globalThis);
-};
+
+(globalThis as unknown as CliContext).ai = new AI(globalThis as unknown as CliContext);

snippets/ai/helpers.ts

@@ -8,7 +8,7 @@
     "esModuleInterop": true,
     "skipLibCheck": true,
     "forceConsistentCasingInFileNames": true,
-    "types": ["./globals.d.ts", "node"],
+    "types": ["node"],
   },
   "include": ["index.ts"],
   "exclude": ["node_modules", "dist"]