ai-agent-test 0.1.4 → 0.1.5

package/README.md

@@ -63,17 +63,17 @@ Create a JSON configuration file at `~/.ai/models.json` with an array of model c
 ]
 ```
 
-Set the environment variable in your terminal before running (Default to 0):
-
+Start CLI with selected model (default is the first model in the array):
 ```bash
-export AI_MODEL_INDEX=0  # Use the first model in models.json (default)
-npx tsx src/index.ts
-```
+# option 1
+ai 1 # to select the second model (Google Gemini)
 
-Or set it inline:
+# option 2
+export AI_MODEL_INDEX=1
+ai
 
-```bash
-AI_MODEL_INDEX=0 npx tsx src/index.ts
+# option 3
+AI_MODEL_INDEX=1 ai
 ```
 
 ### CLI Commands

package/bin/ai

@@ -1,3 +1,11 @@
 #!/usr/bin/env node
 
+// Pass command-line arguments to the main module
+import { argv } from 'process';
+
+// Set AI_MODEL_INDEX environment variable if provided as first argument
+if (argv[2]) {
+  process.env.AI_MODEL_INDEX = argv[2];
+}
+
 import('../dist/index.js');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-agent-test",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "main": "dist/index.js",
   "bin": {
     "ai": "./bin/ai"