@devangkumar/dvai 1.0.2 → 1.0.3

package/package.json

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

package/src/index.js

@@ -10,83 +10,54 @@ const model = genAI.getGenerativeModel({ model: "gemini-2.5-flash" });
 
 const program = new Command();
 
-program
-  .name("myai")
-  .description("AI Assistant")
-  .version("1.0.2");
+async function startChat(initialHistory = []) {
+    let chatHistory = initialHistory;
+    while (true) {
+        const { userInput } = await inquirer.prompt([{
+            type: "input",
+            name: "userInput",
+            message: ">",
+        }]);
 
-async function getAIResponse(prompt, chatHistory = []) {
-  const spinner = ora({
-    text: "",
-    spinner: "dots",
-    isSilent: true 
-  }).start();
+        const cmd = userInput.toLowerCase();
+        if (cmd === "exit" || cmd === "quit" || cmd === "bye") break;
 
-  try {
-    if (chatHistory.length > 0) {
-        const chat = model.startChat({
-            history: chatHistory,
-        });
-        const result = await chat.sendMessage(prompt);
-        spinner.stop();
-        return result.response.text();
-    } else {
-        const result = await model.generateContent(prompt);
-        spinner.stop();
-        return result.response.text();
+        const spinner = ora({ text: "", isSilent: true }).start();
+        try {
+            const chat = model.startChat({ history: chatHistory });
+            const result = await chat.sendMessage(userInput);
+            spinner.stop();
+            const response = result.response.text();
+            process.stdout.write(response + "\n\n");
+            chatHistory.push({ role: "user", parts: [{ text: userInput }] });
+            chatHistory.push({ role: "model", parts: [{ text: response }] });
+        } catch (error) {
+            spinner.stop();
+            console.error("Error:", error.message);
+        }
     }
-  } catch (error) {
-    spinner.stop();
-    console.error("Error:", error.message);
-    return null;
-  }
 }
 
-/**
- * Handle Single Question
- */
 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;
     }
 
-    const response = await getAIResponse(prompt);
-    if (response) {
-      process.stdout.write(response + "\n");
-    }
-  });
-
-/**
- * Handle Interactive Chat
- */
-program
-  .command("chat")
-  .description("Start chat session")
-  .action(async () => {
-    let chatHistory = [];
-    
-    while (true) {
-      const { userInput } = await inquirer.prompt([
-        {
-          type: "input",
-          name: "userInput",
-          message: ">",
-        }
-      ]);
-
-      const cmd = userInput.toLowerCase();
-      if (cmd === "exit" || cmd === "quit" || cmd === "bye") break;
-
-      const response = await getAIResponse(userInput, chatHistory);
-      
-      if (response) {
-        process.stdout.write(response + "\n\n");
-        
-        chatHistory.push({ role: "user", parts: [{ text: userInput }] });
-        chatHistory.push({ role: "model", parts: [{ text: response }] });
-      }
+    // Otherwise, handle single question
+    const spinner = ora({ text: "", isSilent: true }).start();
+    try {
+        const result = await model.generateContent(prompt);
+        spinner.stop();
+        process.stdout.write(result.response.text() + "\n");
+    } catch (error) {
+        spinner.stop();
+        console.error("Error:", error.message);
     }
   });