@devangkumar/dvai 1.0.3 → 1.0.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@devangkumar/dvai",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "description": "A powerful AI assistant for your terminal",
   "type": "module",
   "main": "src/index.js",

package/src/index.js

@@ -1,5 +1,4 @@
 import { GoogleGenerativeAI } from "@google/generative-ai";
-import { Command } from "commander";
 import ora from "ora";
 import inquirer from "inquirer";
 
@@ -8,8 +7,6 @@ const HARDCODED_API_KEY = "AIzaSyDkVgrWXHa7oyPasZGqNOZ-OyA-uYIDtXc";
 const genAI = new GoogleGenerativeAI(HARDCODED_API_KEY);
 const model = genAI.getGenerativeModel({ model: "gemini-2.5-flash" });
 
-const program = new Command();
-
 async function startChat(initialHistory = []) {
     let chatHistory = initialHistory;
     while (true) {
@@ -38,18 +35,23 @@ async function startChat(initialHistory = []) {
     }
 }
 
-program
-  .name("myai")
-  .version("1.0.3")
-  .argument("[prompt]", "Prompt to send to AI")
-  .action(async (prompt) => {
-    // If no prompt, start chat immediately
-    if (!prompt) {
-      await startChat();
-      return;
+async function run() {
+    const args = process.argv.slice(2);
+    
+    // If no arguments, start chat immediately
+    if (args.length === 0) {
+        await startChat();
+        return;
+    }
+
+    // If "chat" command is used
+    if (args[0] === "chat") {
+        await startChat();
+        return;
     }
 
-    // Otherwise, handle single question
+    // Otherwise, treat arguments as a single prompt
+    const prompt = args.join(" ");
     const spinner = ora({ text: "", isSilent: true }).start();
     try {
         const result = await model.generateContent(prompt);
@@ -59,6 +61,6 @@ program
         spinner.stop();
         console.error("Error:", error.message);
     }
-  });
+}
 
-program.parse();
+run();