pulse-cli-divakar 0.0.0-stage → 1.0.1

package/src/cli/chat/chat-with-aitool.js

@@ -0,0 +1,307 @@
+import chalk from "chalk";
+import boxen from "boxen"
+import { text,isCancel,cancel,intro,outro, box, multiselect } from "@clack/prompts";
+import yoctoSpinner from "yocto-spinner";
+import {marked} from "marked"
+import {markedTerminal} from "marked-terminal"
+import { AIService } from "../ai/google-service.js";
+import { ChatService } from "../../service/chat.service.js";
+import { getStoredToken } from "../../lib/token.js";
+import {prisma} from "../../lib/db.js"
+
+import {availableTools,getEnabledToolNames,getEnabledTools,enableTools,resetTools} from "../../config/tool.config.js"
+import { displayMessages, getUserFromToken ,updateConversationTitle,saveMessage} from "./utils/chataiutils.js";
+
+const aiService=new AIService()
+const chatService=new ChatService()
+
+async function selectTools() {
+    const toolOptions=availableTools.map(tool=>({
+        value:tool.id,
+        label:tool.description,
+        hint:tool.description
+    }))
+
+    const selectedTools=await multiselect({
+        message: chalk.cyan(
+      "Select tools to enable (Space to select, Enter to confirm):"
+    ),
+    options: toolOptions,
+    required: false,
+    })
+
+    if (isCancel(selectedTools)) {
+        cancel(chalk.yellow("Tool selection cancelled"));
+        process.exit(0);
+    }
+
+    enableTools(selectedTools);
+
+    if(selectedTools.length===0){
+        console.log(
+         chalk.yellow("\n⚠️  No tools selected. AI will work without tools.\n")
+        );
+    }
+    else{
+      const toolsBox = boxen(
+      chalk.green(
+        `✅ Enabled tools:\n${selectedTools
+          .map((id) => {
+            const tool = availableTools.find((t) => t.id === id);
+            return `  • ${tool.name}`;
+          })
+          .join("\n")}`
+      ),
+      {
+        padding: 1,
+        margin: { top: 1, bottom: 1 },
+        borderStyle: "round",
+        borderColor: "green",
+        title: "🛠️  Active Tools",
+        titleAlignment: "center",
+      }
+    );
+    console.log(toolsBox);
+    }
+
+    return selectTools.length > 0;
+}
+
+
+export async function startToolChat(conversationId=null) {
+    try {
+        intro(
+            boxen(chalk.bold.cyan("🛠️ Pulse AI - Tool Calling Mode"),{
+                padding: 1,
+                borderStyle: "double",
+                borderColor: "cyan",
+            })
+        )
+
+        const user=await getUserFromToken();
+        await selectTools();
+        const conversation=await initConversation(user.id,conversationId,"tool")
+        await chatLoop(conversation)
+
+        resetTools()
+
+        outro(chalk.green("✨Thanks for using tools"))
+        
+
+    } catch (error) {
+      const errorBox = boxen(chalk.red(`❌ Error: ${error.message}`), {
+        padding: 1,
+        margin: 1,
+        borderStyle: "round",
+        borderColor: "red",
+      });
+        console.log(errorBox);
+        resetTools();
+        process.exit(1);
+    } 
+}
+
+export async function initConversation(userId,conversationId=null,mode="tool") {
+    const spinner=yoctoSpinner({text:"Loading conversation..."}).start()
+
+    const conversation=await chatService.getOrCreateConversation(
+        userId,
+        conversationId,
+        mode
+    )
+    spinner.success("Conversation Loaded")
+
+    const enabledToolNames = getEnabledToolNames();
+    const toolsDisplay =
+    enabledToolNames.length > 0
+      ? `\n${chalk.gray("Active Tools:")} ${enabledToolNames.join(", ")}`
+      : `\n${chalk.gray("No tools enabled")}`;
+
+
+
+    const conversationInfo = boxen(
+    `${chalk.bold("Conversation")}: ${conversation.title}\n${chalk.gray("ID: " + conversation.id)}\n${chalk.gray("Mode: " + conversation.mode)}`,
+    {
+      padding: 1,
+      margin: { top: 1, bottom: 1 },
+      borderStyle: "round",
+      borderColor: "cyan",
+      title: "💬 Chat Session",
+      titleAlignment: "center",
+    }
+  );
+
+  console.log(conversationInfo)
+
+  if(conversation.messages?.length>0){
+    console.log(chalk.yellow("📜 Previous messages:\n"));
+    displayMessages(conversation.messages);
+  }
+
+   return conversation
+}
+
+
+async function chatLoop(conversation) {
+    const enabledToolNames=getEnabledToolNames()
+     const helpBox = boxen(
+    `${chalk.gray("• Type your message and press Enter")}\n${chalk.gray(
+      "• AI has access to:"
+    )} ${
+      enabledToolNames.length > 0 ? enabledToolNames.join(", ") : "No tools"
+    }\n${chalk.gray('• Type "exit" to end conversation')}\n${chalk.gray(
+      "• Press Ctrl+C to quit anytime"
+    )}`,
+    {
+      padding: 1,
+      margin: { bottom: 1 },
+      borderStyle: "round",
+      borderColor: "gray",
+      dimBorder: true,
+    }
+  );
+
+  console.log(helpBox)
+
+  while(true){
+    const userInput = await text({
+      message: chalk.blue("💬 Your message"),
+      placeholder: "Type your message...",
+      validate(value) {
+        if (!value || value.trim().length === 0) {
+          return "Message cannot be empty";
+        }
+      },
+    });
+
+     if (isCancel(userInput)) {
+      const exitBox = boxen(chalk.yellow("Chat session ended. Goodbye! 👋"), {
+        padding: 1,
+        margin: 1,
+        borderStyle: "round",
+        borderColor: "yellow",
+      });
+      console.log(exitBox);
+      process.exit(0);
+    }
+
+    if (userInput.toLowerCase() === "exit") {
+      const exitBox = boxen(chalk.yellow("Chat session ended. Goodbye! 👋"), {
+        padding: 1,
+        margin: 1,
+        borderStyle: "round",
+        borderColor: "yellow",
+      });
+      console.log(exitBox);
+      break;
+    }
+
+     const userBox = boxen(chalk.white(userInput), {
+      padding: 1,
+      margin: { left: 2, top: 1, bottom: 1 },
+      borderStyle: "round",
+      borderColor: "blue",
+      title: "👤 You",
+      titleAlignment: "left",
+    });
+    console.log(userBox);
+
+     await saveMessage(conversation.id, "user", userInput);
+
+    const messages = await chatService.getMessages(conversation.id);
+
+    const aiResponse = await getAIResponse(conversation.id);
+
+    await saveMessage(conversation.id, "assistant", aiResponse);
+
+    await updateConversationTitle(conversation.id, userInput, messages.length);
+
+  }
+}
+
+async function getAIResponse(conversationId) {
+  const spinner=yoctoSpinner({
+    text:"AI is thinking...",
+    color:"cyan",
+  }).start();
+
+  const dbMessages=await chatService.getMessages(conversationId)
+  const aiMessages=chatService.formatMessageForAI(dbMessages)
+
+  const tools=getEnabledTools();
+
+  let fullResponse=""
+  let isFirstChunk=true
+  const toolCallsDetected=[]
+
+  try {
+    const result=await aiService.sendMessage(aiMessages,(chunk)=>{
+      if(isFirstChunk){
+        spinner.stop();
+        console.log("\n");
+        const header=chalk.green.bold("🤖 Assistant")
+        console.log(header)
+        console.log(chalk.gray("-".repeat(60)))
+        isFirstChunk=false
+      }
+      fullResponse+=chunk;
+    },
+    tools,(toolCall)=>{
+      toolCallsDetected.push(toolCall)
+    }
+  );
+
+   if (toolCallsDetected.length > 0) {
+      console.log("\n");
+      const toolCallBox = boxen(
+        toolCallsDetected.map(tc => 
+          `${chalk.cyan("🔧 Tool:")} ${tc.toolName}\n${chalk.gray("Args:")} ${JSON.stringify(tc.args, null, 2)}`
+        ).join("\n\n"),
+        {
+          padding: 1,
+          margin: 1,
+          borderStyle: "round",
+          borderColor: "cyan",
+          title: "🛠️  Tool Calls",
+        }
+      );
+      console.log(toolCallBox);
+    }
+
+      if (result.toolResults && result.toolResults.length > 0) {
+      const toolResultBox = boxen(
+        result.toolResults.map(tr => 
+          `${chalk.green("✅ Tool:")} ${tr.toolName}\n${chalk.gray("Result:")} ${JSON.stringify(tr.result, null, 2).slice(0, 200)}...`
+        ).join("\n\n"),
+        {
+          padding: 1,
+          margin: 1,
+          borderStyle: "round",
+          borderColor: "green",
+          title: "📊 Tool Results",
+        }
+      );
+      console.log(toolResultBox);
+    }
+
+      // Render markdown response
+    console.log("\n");
+    const renderedMarkdown = marked.parse(fullResponse);
+    console.log(renderedMarkdown);
+    console.log(chalk.gray("─".repeat(60)));
+    console.log("\n");
+    
+    return result.content;
+  } catch (error) {
+    spinner.error("Failed to get AI response");
+    throw error;
+  }
+
+}
+
+
+
+
+
+
+

package/src/cli/chat/utils/chataiutils.js

@@ -0,0 +1,212 @@
+import yoctoSpinner from "yocto-spinner";
+import { getStoredToken } from "../../../lib/token.js";
+import { prisma } from "../../../lib/db.js";
+import { AIService } from "../../ai/google-service.js";
+import { ChatService } from "../../../service/chat.service.js";
+import { isCancel, text } from "@clack/prompts";
+import chalk from "chalk";
+import boxen from "boxen";
+import { marked } from "marked";
+
+const aiService=new AIService()
+const chatService=new ChatService()
+
+export async function getUserFromToken() {
+    const token=await getStoredToken();
+    if(!token.access_token){
+        throw new Error("Not authenticated. Please run 'pulse login' first");
+    }
+    const spinner=yoctoSpinner({text:"Authenticating..."});
+    spinner.start()
+
+    const user=await prisma.user.findFirst({
+        where:{
+            sessions:{
+                some:{token:token.access_token},
+            }
+        }
+    })
+    if(!user){
+        spinner.error("User not found")
+        throw new Error("User not found.Please login again")
+    }
+    spinner.success(`Welcome back, ${user.name}!`)
+    return user;
+}
+
+
+export async function getAIResponse(conversationId) {
+    const spinner = yoctoSpinner({ 
+        text: "AI is thinking...", 
+        color: "cyan" 
+    }).start();
+
+    const dbMessages=await chatService.getMessages(conversationId)
+    const aiMessages=chatService.formatMessageForAI(dbMessages)
+
+    let fullResponse=""
+
+    let isFirstChunk=true;
+
+    try {
+        const result=await aiService.sendMessage(aiMessages,(chunk)=>{
+             if (isFirstChunk) {
+                spinner.stop();
+                console.log("\n");
+                const header = chalk.green.bold("🤖 Assistant:");
+                console.log(header);
+                console.log(chalk.gray("─".repeat(60)));
+                isFirstChunk = false;
+            }
+            fullResponse += chunk;
+        })
+
+        console.log("\n");
+        const renderedMarkdown = marked.parse(fullResponse);
+        console.log(renderedMarkdown);
+        console.log(chalk.gray("─".repeat(60)));
+        console.log("\n");
+
+        return result.content;
+
+    } catch (error) {
+        spinner.error("Failed to get AI response");
+        throw error;
+    }
+}
+
+export async function saveMessage(conversationId,role,content) {
+    return await chatService.addMessage(conversationId,role,content)
+}
+
+export async function updateConversationTitle(conversationId, userInput, messageCount) {
+  if (messageCount === 1) {
+    const title = userInput.slice(0, 50) + (userInput.length > 50 ? "..." : "");
+    await chatService.updateTitle(conversationId, title);
+  }
+}
+
+
+export async function initConversation(userId,conversationId=null,mode="chat") {
+    const spinner=yoctoSpinner({text:"Loading conversation..."}).start()
+
+    const conversation=await chatService.getOrCreateConversation(
+        userId,
+        conversationId,
+        mode
+    )
+
+    spinner.success("Conversation Loaded")
+
+    const conversationInfo = boxen(
+    `${chalk.bold("Conversation")}: ${conversation.title}\n${chalk.gray("ID: " + conversation.id)}\n${chalk.gray("Mode: " + conversation.mode)}`,
+    {
+      padding: 1,
+      margin: { top: 1, bottom: 1 },
+      borderStyle: "round",
+      borderColor: "cyan",
+      title: "💬 Chat Session",
+      titleAlignment: "center",
+    }
+  );
+
+  console.log(conversationInfo)
+
+  if(conversation.messages?.length>0){
+    console.log(chalk.yellow("📜 Previous messages:\n"));
+    displayMessages(conversation.messages);
+  }
+
+   return conversation
+}
+
+
+export function displayMessages(messages) {
+  messages.forEach((msg) => {
+    if (msg.role === "user") {
+      const userBox = boxen(chalk.white(msg.content), {
+        padding: 1,
+        margin: { left: 2, bottom: 1 },
+        borderStyle: "round",
+        borderColor: "blue",
+        title: "👤 You",
+        titleAlignment: "left",
+      });
+      console.log(userBox);
+    } else {
+      // Render markdown for assistant messages
+      const renderedContent = marked.parse(msg.content);
+      const assistantBox = boxen(renderedContent.trim(), {
+        padding: 1,
+        margin: { left: 2, bottom: 1 },
+        borderStyle: "round",
+        borderColor: "green",
+        title: "🤖 Assistant",
+        titleAlignment: "left",
+      });
+      console.log(assistantBox);
+    }
+  });
+}
+
+
+export async function chatLoop(conversation) {
+     const helpBox = boxen(
+    `${chalk.gray('• Type your message and press Enter')}\n${chalk.gray('• Markdown formatting is supported in responses')}\n${chalk.gray('• Type "exit" to end conversation')}\n${chalk.gray('• Press Ctrl+C to quit anytime')}`,
+    {
+      padding: 1,
+      margin: { bottom: 1 },
+      borderStyle: "round",
+      borderColor: "gray",
+      dimBorder: true,
+    }
+   );
+
+  console.log(helpBox)
+
+  while(true){
+    const userInput=await text({
+        message:chalk.blue("Your message"),
+        placeholder:"Type your message...",
+        validate(value){
+            if(!value || value.trim().length===0){
+                return "Message cannot be empty"
+            }
+        }
+    })
+
+    if(isCancel(userInput)){
+        const exitBox = boxen(chalk.yellow("Chat session ended. Goodbye! 👋"), {
+            padding: 1,
+            margin: 1,
+            borderStyle: "round",
+            borderColor: "yellow",
+        });
+        console.log(exitBox);
+        process.exit(0);
+    }
+
+    if(userInput.toLowerCase()==="exit"){
+        const exitBox = boxen(chalk.yellow("Chat session ended. Goodbye! 👋"), {
+        padding: 1,
+        margin: 1,
+        borderStyle: "round",
+        borderColor: "yellow",
+      });
+      console.log(exitBox);
+      break;
+    }
+
+    await saveMessage(conversation.id , "user" , userInput);
+
+    const messages = await chatService.getMessages(conversation.id)
+
+    const aiResponse = await getAIResponse(conversation.id)
+
+    await saveMessage(conversation.id , "assistant" , aiResponse)
+
+    await updateConversationTitle(conversation.id , userInput , messages.length)
+
+  }
+
+}

package/src/cli/commands/ai/wakeup.js

@@ -0,0 +1,85 @@
+import chalk from "chalk"
+import { Command } from "commander"
+import yoctoSpinner from "yocto-spinner"
+import { getStoredToken } from "../../../lib/token.js"
+import { select } from "@clack/prompts"
+import { prisma } from "../../../lib/db.js"
+
+
+
+const wakeUpAction=async ()=>{
+    const token=await getStoredToken()
+    if(!token?.access_token){
+        console.log(chalk.red("Not Authenticated.Please login"))
+        return;
+    }
+
+    const spinner=yoctoSpinner({text:"Fetching user information..."})
+    spinner.start()
+
+    const user=await prisma.user.findFirst({
+        where:{
+        sessions:{
+            some:{
+            token:token.access_token,
+            }
+        }
+        },
+        select:{
+        id:true,
+        name:true,
+        email:true,
+        image:true
+        }
+    })
+
+    if(!user){
+        console.log(chalk.red("User not Found."))
+        return;
+    }
+
+    spinner.stop()
+    console.log(chalk.green(`Welcome back, ${user.name}!\n`))
+
+    const choice=await select({
+        message:"Select an Option:",
+        options:[
+            {
+                value:"chat",
+                label:"Chat",
+                hint:"Simple chat with AI"
+            },
+            {
+                value:"tool",
+                label:"Tool Calling",
+                hint:"Chat with tools.(Google Search,Code Execution)"
+            },
+            {
+                value:"agent",
+                label:"Agentic Mode",
+                hint:"Advanced AI agent."
+            }
+        ]
+    })
+
+    switch(choice){
+        case "chat":
+              const { startChat } = await import("../../chat/chat-with-ai.js");
+            await startChat("chat");
+            break;
+        case "tool":
+             const { startToolChat } = await import("../../chat/chat-with-aitool.js");
+            
+            await startToolChat();
+            break;
+        case "agent":
+            const { startAgentChat } = await import("../../chat/chat-with-aiAgent.js");
+            await startAgentChat();
+            break;
+    }
+}
+
+
+export const wakeUp=new Command("wakeup")
+.description("Wake up the AI.")
+.action(wakeUpAction)