@devangkumar/dvai 1.0.1 → 1.0.3

package/package.json

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

package/src/index.js

@@ -1,6 +1,5 @@
 import { GoogleGenerativeAI } from "@google/generative-ai";
 import { Command } from "commander";
-import chalk from "chalk";
 import ora from "ora";
 import inquirer from "inquirer";
 
@@ -11,110 +10,55 @@ const model = genAI.getGenerativeModel({ model: "gemini-2.5-flash" });
 
 const program = new Command();
 
-program
-  .name("myai")
-  .description("A stunning AI assistant in your terminal")
-  .version("1.0.1");
+async function startChat(initialHistory = []) {
+    let chatHistory = initialHistory;
+    while (true) {
+        const { userInput } = await inquirer.prompt([{
+            type: "input",
+            name: "userInput",
+            message: ">",
+        }]);
 
-/**
- * Main AI function to get response
- */
-async function getAIResponse(prompt, chatHistory = []) {
-  const spinner = ora({
-    text: chalk.cyan("AI is thinking..."),
-    color: "cyan",
-    spinner: "dots"
-  }).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(chalk.red("\n✖ AI Error:"), error.message);
-    return null;
-  }
 }
 
-/**
- * Handle Single Question
- */
 program
-  .argument("[prompt]", "The question you want to ask AI")
+  .name("myai")
+  .version("1.0.3")
+  .argument("[prompt]", "Prompt to send to AI")
   .action(async (prompt) => {
+    // If no prompt, start chat immediately
     if (!prompt) {
-      displayWelcome();
+      await startChat();
       return;
     }
 
-    const response = await getAIResponse(prompt);
-    if (response) {
-      console.log(chalk.blue.bold("\nAI Response:"));
-      console.log(chalk.white(response));
-      console.log("\n");
-    }
-  });
-
-/**
- * Handle Interactive Chat
- */
-program
-  .command("chat")
-  .description("Start an interactive chat session")
-  .action(async () => {
-    console.clear();
-    console.log(chalk.cyan.bold("╔════════════════════════════════════════════╗"));
-    console.log(chalk.cyan.bold("║          WELCOME TO MYAI TERMINAL          ║"));
-    console.log(chalk.cyan.bold("╚════════════════════════════════════════════╝"));
-    console.log(chalk.dim("Type 'exit' or 'quit' to end the session.\n"));
-
-    let chatHistory = [];
-    
-    while (true) {
-      const { userInput } = await inquirer.prompt([
-        {
-          type: "input",
-          name: "userInput",
-          message: chalk.green("? You:"),
-          validate: (input) => input.trim() !== "" || "Please enter a prompt."
-        }
-      ]);
-
-      if (userInput.toLowerCase() === "exit" || userInput.toLowerCase() === "quit") {
-        console.log(chalk.yellow("\nGoodbye! 👋\n"));
-        break;
-      }
-
-      const response = await getAIResponse(userInput, chatHistory);
-      
-      if (response) {
-        console.log(chalk.blue.bold("\nAI:"));
-        console.log(chalk.white(response));
-        console.log("\n");
-        
-        // Update history
-        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);
     }
   });
 
-function displayWelcome() {
-    console.log(chalk.cyan.bold("\n✨ MyAI Terminal Assistant ✨"));
-    console.log(chalk.white("Usage:"));
-    console.log(chalk.dim("  myai \"your question\"  ") + chalk.white("- Ask a quick question"));
-    console.log(chalk.dim("  myai chat            ") + chalk.white("- Start interactive session"));
-    console.log(chalk.dim("  myai --help          ") + chalk.white("- Show help\n"));
-}
-
 program.parse();