orbital-cli-agent 1.0.0

package/src/cli/chat/chat-with-ai-tool.js

@@ -0,0 +1,371 @@
+import chalk from "chalk";
+import boxen from "boxen";
+import { text, isCancel, cancel, intro, outro, 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 { getStoredToken } from "../commands/auth/login.js";
+import * as apiClient from "../api/api-client.js";
+import { 
+  availableTools, 
+  getEnabledTools, 
+  enableTools, 
+  getEnabledToolNames,
+  resetTools 
+} from "../../config/tool.config.js";
+
+// Configure marked for terminal
+marked.use(
+  markedTerminal({
+    code: chalk.cyan,
+    blockquote: chalk.gray.italic,
+    heading: chalk.green.bold,
+    firstHeading: chalk.magenta.underline.bold,
+    hr: chalk.reset,
+    listitem: chalk.reset,
+    list: chalk.reset,
+    paragraph: chalk.reset,
+    strong: chalk.bold,
+    em: chalk.italic,
+    codespan: chalk.yellow.bgBlack,
+    del: chalk.dim.gray.strikethrough,
+    link: chalk.blue.underline,
+    href: chalk.blue.underline,
+  })
+);
+
+const aiService = new AIService();
+
+async function getUserFromToken() {
+  const token = await getStoredToken();
+  
+  if (!token?.access_token) {
+    throw new Error("Not authenticated. Please run 'orbit login' first.");
+  }
+
+  const spinner = yoctoSpinner({ text: "Authenticating..." }).start();
+
+  const user = await apiClient.getUserFromApi(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, token };
+}
+
+async function selectTools() {
+  const toolOptions = availableTools.map(tool => ({
+    value: tool.id,
+    label: tool.name,
+    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);
+  }
+
+  // Enable selected tools
+  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 selectedTools.length > 0;
+}
+
+async function initConversation(token, userId, conversationId = null, mode = "tool") {
+  const spinner = yoctoSpinner({ text: "Loading conversation..." }).start();
+  
+  const conversation = await apiClient.getOrCreateConversation(
+    token,
+    conversationId,
+    mode
+  );
+  
+  spinner.success("Conversation loaded");
+  
+  // Get enabled tool names for display
+  const enabledToolNames = getEnabledToolNames();
+  const toolsDisplay = enabledToolNames.length > 0 
+    ? `\n${chalk.gray("Active Tools:")} ${enabledToolNames.join(", ")}`
+    : `\n${chalk.gray("No tools enabled")}`;
+  
+  // Display conversation info in a box
+  const conversationInfo = boxen(
+    `${chalk.bold("Conversation")}: ${conversation.title}\n${chalk.gray("ID: " + conversation.id)}\n${chalk.gray("Mode: " + conversation.mode)}${toolsDisplay}`,
+    {
+      padding: 1,
+      margin: { top: 1, bottom: 1 },
+      borderStyle: "round",
+      borderColor: "cyan",
+      title: "💬 Tool Calling Session",
+      titleAlignment: "center",
+    }
+  );
+  
+  console.log(conversationInfo);
+  
+  // Display existing messages if any
+  if (conversation.messages?.length > 0) {
+    console.log(chalk.yellow("📜 Previous messages:\n"));
+    displayMessages(conversation.messages);
+  }
+  
+  return conversation;
+}
+
+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 if (msg.role === "assistant") {
+      const renderedContent = marked.parse(msg.content);
+      const assistantBox = boxen(renderedContent.trim(), {
+        padding: 1,
+        margin: { left: 2, bottom: 1 },
+        borderStyle: "round",
+        borderColor: "green",
+        title: "🤖 Assistant (with tools)",
+        titleAlignment: "left",
+      });
+      console.log(assistantBox);
+    }
+  });
+}
+
+async function saveMessage(token, conversationId, role, content) {
+  return await apiClient.addMessage(token, conversationId, role, content);
+}
+
+async function getAIResponse(token, conversationId) {
+  const spinner = yoctoSpinner({ 
+    text: "AI is thinking...", 
+    color: "cyan" 
+  }).start();
+
+  const dbMessages = await apiClient.getMessages(token, conversationId);
+  const aiMessages = dbMessages.map((msg) => ({
+    role: msg.role,
+    content: typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content),
+  }));
+
+  const tools = getEnabledTools();
+  
+  let fullResponse = "";
+  let isFirstChunk = true;
+  const toolCallsDetected = [];
+  
+  try {
+    // IMPORTANT: Pass tools in the streamText config
+    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);
+      }
+    );
+    
+    // Display tool calls if any
+    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);
+    }
+
+    // Display tool results if any
+    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;
+  }
+}
+
+
+async function updateConversationTitle(token, conversationId, userId, userInput, messageCount) {
+  if (messageCount === 1) {
+    const title = userInput.slice(0, 50) + (userInput.length > 50 ? "..." : "");
+    await apiClient.updateTitle(token, conversationId, title);
+  }
+}
+
+async function chatLoop(token, 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(token, conversation.id, "user", userInput);
+    const messages = await apiClient.getMessages(token, conversation.id);
+    const aiResponse = await getAIResponse(token, conversation.id);
+    await saveMessage(token, conversation.id, "assistant", aiResponse);
+    await updateConversationTitle(token, conversation.id, conversation.userId, userInput, messages.length);
+  }
+}
+
+export async function startToolChat(conversationId = null) {
+  try {
+    intro(
+      boxen(chalk.bold.cyan("🛠️  Orbit AI - Tool Calling Mode"), {
+        padding: 1,
+        borderStyle: "double",
+        borderColor: "cyan",
+      })
+    );
+
+    const { user, token } = await getUserFromToken();
+    
+    // Select tools
+    await selectTools();
+    
+    const conversation = await initConversation(token.access_token, user.id, conversationId, "tool");
+    await chatLoop(token.access_token, conversation);
+    
+    // Reset tools on exit
+    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);
+  }
+}

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

@@ -0,0 +1,267 @@
+import chalk from "chalk";
+import boxen from "boxen";
+import { text, isCancel, cancel, intro, outro } 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 { getStoredToken } from "../commands/auth/login.js";
+import * as apiClient from "../api/api-client.js";
+
+// Configure marked to use terminal renderer
+marked.use(
+  markedTerminal({
+    // Styling options for terminal output
+    code: chalk.cyan,
+    blockquote: chalk.gray.italic,
+    heading: chalk.green.bold,
+    firstHeading: chalk.magenta.underline.bold,
+    hr: chalk.reset,
+    listitem: chalk.reset,
+    list: chalk.reset,
+    paragraph: chalk.reset,
+    strong: chalk.bold,
+    em: chalk.italic,
+    codespan: chalk.yellow.bgBlack,
+    del: chalk.dim.gray.strikethrough,
+    link: chalk.blue.underline,
+    href: chalk.blue.underline,
+  })
+);
+
+// Initialize service
+const aiService = new AIService();
+
+async function getUserFromToken() {
+  const token = await getStoredToken();
+  
+  if (!token?.access_token) {
+    throw new Error("Not authenticated. Please run 'orbit login' first.");
+  }
+
+  const spinner = yoctoSpinner({ text: "Authenticating..." }).start();
+
+  const user = await apiClient.getUserFromApi(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, token };
+}
+
+async function initConversation(token, userId, conversationId = null, mode = "chat") {
+  const spinner = yoctoSpinner({ text: "Loading conversation..." }).start();
+  
+  const conversation = await apiClient.getOrCreateConversation(
+    token,
+    conversationId,
+    mode
+  );
+  
+  spinner.success("Conversation loaded");
+  
+  // Display conversation info in a box
+  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);
+  
+  // Display existing messages if any
+  if (conversation.messages?.length > 0) {
+    console.log(chalk.yellow("📜 Previous messages:\n"));
+    displayMessages(conversation.messages);
+  }
+  
+  return conversation;
+}
+
+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);
+    }
+  });
+}
+
+async function saveMessage(token, conversationId, role, content) {
+  return await apiClient.addMessage(token, conversationId, role, content);
+}
+
+async function getAIResponse(token, conversationId) {
+  const spinner = yoctoSpinner({ 
+    text: "AI is thinking...", 
+    color: "cyan" 
+  }).start();
+
+  const dbMessages = await apiClient.getMessages(token, conversationId);
+  const aiMessages = dbMessages.map((msg) => ({
+    role: msg.role,
+    content: typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content),
+  }));
+  
+  let fullResponse = "";
+  let isFirstChunk = true;
+  
+  try {
+    const result = await aiService.sendMessage(aiMessages, (chunk) => {
+      // Stop spinner on first chunk and show header
+      if (isFirstChunk) {
+        spinner.stop();
+        console.log("\n");
+        const header = chalk.green.bold("🤖 Assistant:");
+        console.log(header);
+        console.log(chalk.gray("─".repeat(60)));
+        isFirstChunk = false;
+      }
+      process.stdout.write(chunk);
+      fullResponse += chunk;
+    });
+    
+    // Now render the complete 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;
+  }
+}
+
+async function updateConversationTitle(token, conversationId, userId, userInput, messageCount) {
+  if (messageCount === 1) {
+    const title = userInput.slice(0, 50) + (userInput.length > 50 ? "..." : "");
+    await apiClient.updateTitle(token, conversationId, title);
+  }
+}
+
+async function chatLoop(token, 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";
+        }
+      },
+    });
+
+    // Handle cancellation (Ctrl+C)
+    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);
+    }
+
+    // Handle exit command
+    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;
+    }
+
+    // Save user message
+    await saveMessage(token, conversation.id, "user", userInput);
+
+    // Get messages count before AI response
+    const messages = await apiClient.getMessages(token, conversation.id);
+    
+    // Get AI response with streaming and markdown rendering
+    const aiResponse = await getAIResponse(token, conversation.id);
+
+    // Save AI response
+    await saveMessage(token, conversation.id, "assistant", aiResponse);
+
+    // Update title if first exchange
+    await updateConversationTitle(token, conversation.id, conversation.userId, userInput, messages.length);
+  }
+}
+
+// Main entry point
+export async function startChat(mode = "chat", conversationId = null) {
+  try {
+    // Display intro banner
+    intro(
+      boxen(chalk.bold.cyan("🚀 Orbit AI Chat"), {
+        padding: 1,
+        borderStyle: "double",
+        borderColor: "cyan",
+      })
+    );
+
+    const { user, token } = await getUserFromToken();
+    const conversation = await initConversation(token.access_token, user.id, conversationId, mode);
+    await chatLoop(token.access_token, conversation);
+    
+    // Display outro
+    outro(chalk.green("✨ Thanks for chatting!"));
+  } catch (error) {
+    const errorBox = boxen(chalk.red(`❌ Error: ${error.message}`), {
+      padding: 1,
+      margin: 1,
+      borderStyle: "round",
+      borderColor: "red",
+    });
+    console.log(errorBox);
+    process.exit(1);
+  }
+}

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

@@ -0,0 +1,61 @@
+import chalk from "chalk";
+import { Command } from "commander";
+import { getStoredToken } from "../auth/login.js";
+import { getUserFromApi } from "../../api/api-client.js";
+import { select, isCancel } from "@clack/prompts";
+import yoctoSpinner from "yocto-spinner";
+import { startChat } from "../../chat/chat-with-ai.js";
+import { startToolChat } from "../../chat/chat-with-ai-tool.js";
+import { startAgentChat } from "../../chat/chat-with-ai-agent.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....."}).start();
+  const user = await getUserFromApi(token.access_token);
+  spinner.stop();
+
+  if(!user){
+    console.log("User not found");
+    return;
+  }
+
+  while (true) {
+    const choice = await select({
+      message: chalk.cyan(`Welcome Back ${user.name}! How can I help you today?`),
+      options: [
+        { value: "chat", label: "AI Chat", hint: "Start a conversations with AI" },
+        { value: "tool", label: "Tool calling", hint: "Chat with tools enabled" },
+        { value: "agentic", label: "Agentic Mode", hint: "Build complete apps with AI" },
+        { value: "exit", label: "Exit", hint: "Exit the application" }
+      ]
+    });
+
+    if (isCancel(choice) || choice === "exit") {
+      console.log(chalk.green("\nGoodbye! See you next time. 👋\n"));
+      break;
+    }
+
+    switch (choice) {
+      case "chat":
+        await startChat("chat");
+        break;
+      case "tool":
+        await startToolChat();
+        break;
+      case "agentic":
+        await startAgentChat();
+        break;
+      default:
+        console.log(chalk.red("Invalid option. Please try again."));
+    }
+  }
+}
+
+export const wakeup = new Command("wakeup")
+  .description("Wake up the AI agent")
+  .action(wakeUpAction);