@react-frameui/loki-ai 1.0.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 LOKI Contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,158 @@
+# 🧠 LOKI - Local Omni Knowledge Interface
+
+[![npm version](https://badge.fury.io/js/%40iamnishant51%2Floki-ai.svg)](https://www.npmjs.com/package/@iamnishant51/loki-ai)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+**LOKI** is a powerful, privacy-first AI assistant that runs entirely on your local machine. No cloud, no subscriptions, just pure local AI power.
+
+## ✨ Features
+
+- 🔒 **100% Local** - Your data never leaves your machine
+- 💬 **WhatsApp Integration** - Chat with LOKI via WhatsApp
+- 🛠️ **Built-in Tools** - File system, math, time, and more
+- 🧠 **Memory** - Remembers your conversations
+- 🎨 **Beautiful TUI** - Intuitive terminal interface
+- ⚡ **LangChain Powered** - Use multiple LLM providers (Ollama, Groq, OpenAI)
+- 🔌 **VS Code Extension** - Available on the marketplace
+
+## 🚀 Quick Start
+
+```bash
+# Install globally
+npm install -g @iamnishant51/loki-ai
+
+# Run LOKI
+loki
+
+# Or use npx (no install needed)
+npx @iamnishant51/loki-ai
+```
+
+## 📋 Prerequisites
+
+- **Node.js** 18+
+- **Ollama** (for local LLM) - [Install here](https://ollama.ai)
+
+## 🎯 Usage
+
+### Interactive TUI Mode
+```bash
+loki
+```
+
+### Chat Mode
+```bash
+loki chat
+```
+
+### WhatsApp Integration
+```bash
+loki server
+```
+Then scan the QR code with WhatsApp!
+
+### Direct Questions
+```bash
+loki ask "What time is it?"
+```
+
+## 🛠️ Available Tools
+
+LOKI comes with built-in tools:
+
+- **📁 File System** - List files, read directories
+- **🧮 Math** - Calculate expressions
+- **⏰ Time** - Get current time
+- **💾 Memory** - Semantic and conversation memory
+- **🔍 RAG** - Index repositories for context-aware responses
+
+## 🔧 Configuration
+
+Create a `.env` file:
+
+```env
+# LLM Provider (ollama, groq, openai)
+LLM_PROVIDER=ollama
+
+# Ollama settings
+OLLAMA_BASE_URL=http://localhost:11434
+OLLAMA_MODEL=llama3.2
+
+# Optional: Groq API
+GROQ_API_KEY=your_groq_key
+
+# Optional: OpenAI API
+OPENAI_API_KEY=your_openai_key
+```
+
+## 📱 WhatsApp Integration
+
+1. Run `loki server`
+2. Scan QR code with WhatsApp
+3. Message yourself or share your number
+4. LOKI responds automatically!
+
+## 💻 VS Code Extension
+
+Install from the [VS Code Marketplace](https://marketplace.visualstudio.com/) - Search for "LOKI AI"
+
+Features:
+- Explain code selections
+- In-editor AI assistance
+- Context-aware suggestions
+
+## 🏗️ Development
+
+```bash
+# Clone repository
+git clone https://github.com/IamNishant51/LOKI.git
+cd LOKI
+
+# Install dependencies
+npm install
+
+# Run in dev mode
+npm run dev
+
+# Build
+npm run build
+```
+
+## 📦 Project Structure
+
+```
+loki-ai/
+├── src/
+│   ├── agents/         # AI agent configurations
+│   ├── cli/            # CLI & TUI interface
+│   ├── core/           # Core logic & agent runner
+│   ├── tools/          # Built-in tools
+│   ├── memory/         # Memory management
+│   ├── interfaces/     # WhatsApp, etc.
+│   └── index.ts        # Entry point
+├── vscode-extension/   # VS Code extension
+└── dist/               # Compiled output
+```
+
+## 🤝 Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+## 📄 License
+
+MIT © Nishant
+
+## 🔗 Links
+
+- [GitHub](https://github.com/IamNishant51/LOKI)
+- [NPM Package](https://www.npmjs.com/package/@iamnishant51/loki-ai)
+- [VS Code Extension](https://marketplace.visualstudio.com/)
+- [Documentation](#) *(coming soon)*
+
+## ⭐ Show Your Support
+
+If you like LOKI, please give it a ⭐ on [GitHub](https://github.com/IamNishant51/LOKI)!
+
+---
+
+Made with ❤️ by [Nishant](https://github.com/IamNishant51)

package/dist/agents/devAgent.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DEV_AGENT_SYSTEM_PROMPT = void 0;
+/**
+ * System prompt for the developer assistant agent.
+ * Designed to be helpful, concise, and coding-focused.
+ */
+exports.DEV_AGENT_SYSTEM_PROMPT = `
+You are a senior software engineer assistant.
+Your goal is to help with coding tasks, debugging, and architecture design.
+Keep your answers concise, practical, and code-centric.
+Avoid unnecessary fluff. Focus on modern best practices.
+If you provide code explanation, keep it brief and relevant.
+`.trim();

package/dist/agents/index.js

@@ -0,0 +1,50 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AGENTS = void 0;
+exports.getSystemPrompt = getSystemPrompt;
+exports.AGENTS = {
+    dev: `
+You are LOKI (Local Omni Knowledge Interface), a helpful AI assistant.
+
+Your Capabilities:
+- Answer questions and have conversations
+- Use tools to check time, do math, list files/folders, and more
+- Remember context from our conversation
+
+Rules:
+1. When using tools, execute them and present results in NATURAL LANGUAGE
+2. NEVER show raw JSON or code blocks to the user unless they specifically ask for code
+3. Format file/folder listings as a nice readable list
+4. Be concise and professional - keep responses under 3 sentences when possible
+5. Use minimal emojis - 1 at most per response, or none
+6. If a tool fails, explain politely what went wrong
+7. IMPORTANT: Use the paths from SYSTEM CONTEXT section (e.g., Desktop Path, Home Directory)
+8. Don't over-explain - be direct and helpful
+
+Example - If user asks "what folders are on my desktop":
+- Check SYSTEM CONTEXT for Desktop Path
+- Use the list_files tool with that exact path
+- Then respond like: "Here are the folders on your Desktop: Projects, Documents, Downloads"
+  `.trim(),
+    explain: `
+You are LOKI. Your task is to EXPLAIN code or concepts simply.
+Target Audience: a developer who needs clarity.
+Style: Educational, clear, using analogies if helpful.
+Rules:
+- Break down complex logic.
+- Do not just read the code aloud; explain the *intent*.
+  `.trim(),
+    refactor: `
+You are LOKI. Your task is to REFACTOR code.
+Goal: Improve readability, performance, and structure.
+Style: Strict and critical but constructive.
+Rules:
+- Maintain original functionality.
+- Add comments where logic is complex.
+- Remove dead code.
+- Apply clean code principles (SOLID, DRY).
+  `.trim()
+};
+function getSystemPrompt(agentName = 'dev') {
+    return exports.AGENTS[agentName] || exports.AGENTS['dev'];
+}

package/dist/agents/orchestrator.js

@@ -0,0 +1,76 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runRefactorWorkflow = runRefactorWorkflow;
+const providerFactory_1 = require("../llm/providerFactory");
+const projectContext_1 = require("../context/projectContext");
+// 1. Planner Agent
+const PLANNER_PROMPT = `
+You are the PLANNER Agent.
+Analyze the request and project context.
+Create a step-by-step plan to achieve the goal.
+Do NOT write code. Just plan.
+Output format:
+- Step 1: [Action]
+- Step 2: [Action]
+...
+`;
+// 2. Analyzer Agent
+const ANALYZER_PROMPT = `
+You are the ANALYZER Agent.
+Read the plan and relevant files.
+Explain the current state and what needs to change.
+Focus on impact and risks.
+`;
+// 3. Refactor Agent
+const REFACTOR_PROMPT = `
+You are the REFACTOR Agent.
+Generate the actual code changes based on the plan and analysis.
+Show clear Before/After blocks or unified diffs.
+Do NOT apply changes. Just propose them.
+`;
+async function runRefactorWorkflow(instruction) {
+    const provider = (0, providerFactory_1.getProvider)('ollama'); // Default for internal thought loops
+    const context = (0, projectContext_1.getProjectContext)();
+    // --- Step 1: Plan ---
+    const planPrompt = `
+${PLANNER_PROMPT}
+Project: ${context.name}
+User Goal: ${instruction}
+Plan:`;
+    console.log('[Orchestrator] Planning...');
+    const plan = await provider.generate(planPrompt);
+    // --- Step 2: Analyze ---
+    const analysisPrompt = `
+${ANALYZER_PROMPT}
+Plan:
+${plan}
+User Goal: ${instruction}
+Analysis:`;
+    console.log('[Orchestrator] Analyzing...');
+    const analysis = await provider.generate(analysisPrompt);
+    // --- Step 3: Propose Refactor ---
+    const refactorPrompt = `
+${REFACTOR_PROMPT}
+Plan:
+${plan}
+Analysis:
+${analysis}
+Proposal:`;
+    console.log('[Orchestrator] Generating Proposal...');
+    const proposal = await provider.generate(refactorPrompt);
+    // Provide full report
+    return {
+        success: true,
+        summary: "Refactor Proposal Generated",
+        details: `
+=== PLAN ===
+${plan.trim()}
+
+=== ANALYSIS ===
+${analysis.trim()}
+
+=== PROPOSAL ===
+${proposal.trim()}
+        `
+    };
+}

package/dist/cli/chat.js

@@ -0,0 +1,142 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.startChatSession = startChatSession;
+exports.chatCommand = chatCommand;
+const readline_1 = __importDefault(require("readline"));
+const chalk_1 = __importDefault(require("chalk"));
+const agentRunner_1 = require("../core/agentRunner");
+const spinner_1 = require("../utils/spinner");
+const LOKI_PREFIX = chalk_1.default.green.bold('LOKI │');
+const USER_PREFIX = chalk_1.default.blue.bold('USER │');
+const ERROR_PREFIX = chalk_1.default.red.bold('ERR  │');
+const BANNER = chalk_1.default.green.bold(`
+██╗      ██████╗ ██╗  ██╗██╗
+██║     ██╔═══██╗██║ ██╔╝██║
+██║     ██║   ██║█████╔╝ ██║
+██║     ██║   ██║██╔═██╗ ██║
+███████╗╚██████╔╝██║  ██╗██║
+╚══════╝ ╚═════╝ ╚═╝  ╚═╝╚═╝
+`);
+// Handle cancellation
+let abortController = null;
+let spinner = null;
+async function activeChat(message, options) {
+    const contextPaths = [];
+    if (options.file)
+        contextPaths.push(options.file);
+    if (options.dir)
+        contextPaths.push(options.dir);
+    const shouldStream = options.stream !== false;
+    abortController = new AbortController();
+    try {
+        process.stdout.write(`\n${LOKI_PREFIX} `);
+        // Start spinner
+        spinner = new spinner_1.Spinner();
+        spinner.start();
+        const response = await (0, agentRunner_1.agentRunner)(message, {
+            provider: options.provider,
+            agent: options.agent,
+            useMemory: options.memory,
+            fileContextPaths: contextPaths,
+            stream: shouldStream,
+            onToken: (token) => {
+                // Stop spinner on first token
+                if (spinner) {
+                    spinner.stop();
+                    spinner = null;
+                }
+                process.stdout.write(token);
+            },
+            signal: abortController.signal
+        });
+        // Ensure spinner is stopped if logic completed without streaming or buffered
+        if (spinner) {
+            spinner.stop();
+            spinner = null;
+        }
+        if (!shouldStream) {
+            console.log(response);
+        }
+        console.log('\n');
+    }
+    catch (error) {
+        if (spinner) {
+            spinner.stop();
+            spinner = null;
+        }
+        if (error.message.includes('cancelled') || error.message.includes('Aborted')) {
+            console.log(chalk_1.default.gray(' [Cancelled]\n'));
+        }
+        else {
+            console.error(`\n${ERROR_PREFIX} ${error.message}\n`);
+        }
+    }
+    finally {
+        abortController = null;
+    }
+}
+/**
+ * Interactive Chat Session.
+ * Returns a promise that resolves when the user types 'exit' or 'back'.
+ */
+async function startChatSession(options) {
+    // Note: We don't clear screen here to preserve context if coming from menu, 
+    // or let the clear happen if desired.
+    // Actually, for a "sub-screen", a clear is nice, but maybe we just print a header.
+    process.stdout.write('\x1Bc');
+    console.log(chalk_1.default.green.bold('LOKI CHAT MODE'));
+    console.log(chalk_1.default.gray('Type "exit" or "back" to return to main menu.\n'));
+    const rl = readline_1.default.createInterface({
+        input: process.stdin,
+        output: process.stdout,
+        prompt: `${USER_PREFIX} `
+    });
+    return new Promise((resolve) => {
+        rl.prompt();
+        rl.on('SIGINT', () => {
+            if (abortController) {
+                abortController.abort();
+            }
+            else {
+                // Return to menu on Ctrl+C (twice?) or just ask? 
+                // For simplicity, let's treat idle Ctrl+C as "Back"
+                console.log(chalk_1.default.gray('\nReturning to menu...'));
+                rl.close();
+                resolve();
+            }
+        });
+        rl.on('line', async (line) => {
+            const input = line.trim();
+            if (input.toLowerCase() === 'exit' || input.toLowerCase() === 'back') {
+                rl.close();
+                resolve();
+                return;
+            }
+            if (input) {
+                rl.pause();
+                await activeChat(input, options);
+                rl.resume();
+            }
+            rl.prompt();
+        });
+    });
+}
+async function chatCommand(message, options) {
+    const opts = {
+        provider: options.provider,
+        agent: options.agent,
+        memory: options.memory,
+        file: options.file,
+        dir: options.dir,
+        stream: options.stream
+    };
+    if (message) {
+        await activeChat(message, opts);
+    }
+    else {
+        await startChatSession(opts);
+    }
+}

package/dist/cli/doctor.js

@@ -0,0 +1,67 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.doctorCommand = doctorCommand;
+const fs_1 = __importDefault(require("fs"));
+const config_1 = require("../utils/config");
+const providerFactory_1 = require("../llm/providerFactory");
+const chalk_1 = __importDefault(require("chalk"));
+async function doctorCommand() {
+    console.log(`\nLOKI DOCTOR - SYSTEM HEALTH CHECK\n`);
+    let allGood = true;
+    // Helper for clean output
+    const check = (label, status, info) => {
+        const mark = status ? chalk_1.default.green('[OK]  ') : chalk_1.default.red('[FAIL]');
+        console.log(`${mark} ${label.padEnd(20)} : ${info}`);
+        if (!status)
+            allGood = false;
+    };
+    const warn = (label, info) => {
+        console.log(`${chalk_1.default.yellow('[WARN]')} ${label.padEnd(20)} : ${info}`);
+    };
+    // 1. Check Config Dir
+    try {
+        if (fs_1.default.existsSync(config_1.CONFIG.CONFIG_DIR)) {
+            try {
+                fs_1.default.accessSync(config_1.CONFIG.CONFIG_DIR, fs_1.default.constants.W_OK);
+                check('Config Directory', true, config_1.CONFIG.CONFIG_DIR);
+            }
+            catch {
+                check('Config Directory', false, `Not Writable (${config_1.CONFIG.CONFIG_DIR})`);
+            }
+        }
+        else {
+            warn('Config Directory', 'Missing (Will be created on first run)');
+        }
+    }
+    catch (e) {
+        check('Config Directory', false, 'Check Failed');
+    }
+    // 2. Check Ollama
+    const ollama = (0, providerFactory_1.getProvider)('ollama');
+    const ollamaHealth = await ollama.checkHealth();
+    if (ollamaHealth) {
+        check('Ollama Connection', true, config_1.CONFIG.OLLAMA_HOST);
+    }
+    else {
+        check('Ollama Connection', false, 'Not running or unreachable');
+        console.log(chalk_1.default.gray(`      -> Run 'ollama serve' to start`));
+    }
+    // 3. Check Groq (Optional)
+    if (config_1.CONFIG.GROQ_API_KEY) {
+        check('Groq API Key', true, 'Present');
+    }
+    else {
+        // Just info, not warn/fail since it's optional
+        console.log(`${chalk_1.default.blue('[INFO]')} ${'Groq API Key'.padEnd(20)} : Not Set (Optional)`);
+    }
+    console.log(`\n----------------------------------------`);
+    if (allGood) {
+        console.log(chalk_1.default.green(`System ready.`));
+    }
+    else {
+        console.log(chalk_1.default.red(`System has issues. Please fix above errors.`));
+    }
+}

package/dist/cli/tui.js

@@ -0,0 +1,97 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.startLoop = startLoop;
+const readline_1 = __importDefault(require("readline"));
+const chalk_1 = __importDefault(require("chalk"));
+const chat_1 = require("./chat");
+const doctor_1 = require("./doctor");
+const refactorEngine_1 = require("../refactor/refactorEngine");
+const indexer_1 = require("../rag/indexer");
+const BANNER = chalk_1.default.green.bold(`
+██╗      ██████╗ ██╗  ██╗██╗
+██║     ██╔═══██╗██║ ██╔╝██║
+██║     ██║   ██║█████╔╝ ██║
+██║     ██║   ██║██╔═██╗ ██║
+███████╗╚██████╔╝██║  ██╗██║
+╚══════╝ ╚═════╝ ╚═╝  ╚═╝╚═╝
+`);
+// Simple prompt helper
+function ask(question) {
+    const rl = readline_1.default.createInterface({
+        input: process.stdin,
+        output: process.stdout,
+    });
+    return new Promise(resolve => {
+        rl.question(question, (answer) => {
+            rl.close();
+            resolve(answer);
+        });
+    });
+}
+async function waitForKey() {
+    console.log(chalk_1.default.gray('\nPress any key to continue...'));
+    process.stdin.setRawMode(true);
+    return new Promise(resolve => {
+        process.stdin.once('data', () => {
+            process.stdin.setRawMode(false);
+            resolve();
+        });
+    });
+}
+async function startLoop() {
+    while (true) {
+        process.stdout.write('\x1Bc'); // Clear screen
+        console.log(BANNER);
+        console.log(chalk_1.default.gray('  Local Omni Knowledge Interface\n'));
+        console.log(chalk_1.default.white.bold('  MAIN MENU'));
+        console.log(chalk_1.default.cyan('  [1]') + ' Chat with LOKI');
+        console.log(chalk_1.default.cyan('  [2]') + ' Agent Workflow (Refactor/Plan)');
+        console.log(chalk_1.default.cyan('  [3]') + ' Index Repository (RAG)');
+        console.log(chalk_1.default.cyan('  [4]') + ' WhatsApp Link (QR)');
+        console.log(chalk_1.default.cyan('  [5]') + ' System Doctor');
+        console.log(chalk_1.default.cyan('  [6]') + ' Exit');
+        console.log();
+        const answer = await ask(chalk_1.default.green.bold('  SELECT ▸ '));
+        switch (answer.trim()) {
+            case '1':
+                await (0, chat_1.startChatSession)({});
+                break;
+            case '2':
+                process.stdout.write('\x1Bc');
+                console.log(chalk_1.default.green.bold('LOKI AGENT WORKFLOW'));
+                console.log(chalk_1.default.gray('Example: "Refactor gitTool to use async/await"\n'));
+                const instruction = await ask(chalk_1.default.cyan('Describe task: '));
+                if (instruction.trim()) {
+                    await (0, refactorEngine_1.engineCommand)(instruction);
+                }
+                await waitForKey();
+                break;
+            case '3':
+                process.stdout.write('\x1Bc');
+                console.log(chalk_1.default.green.bold('LOKI RAG INDEXER'));
+                await (0, indexer_1.indexRepository)();
+                await waitForKey();
+                break;
+            case '4':
+                process.stdout.write('\x1Bc');
+                const { startWhatsAppClient } = require('../interfaces/whatsapp/client');
+                await startWhatsAppClient();
+                await waitForKey();
+                break;
+            case '5':
+                process.stdout.write('\x1Bc');
+                await (0, doctor_1.doctorCommand)();
+                await waitForKey();
+                break;
+            case '6':
+            case 'exit':
+                console.log('Goodbye.');
+                process.exit(0);
+            default:
+                break;
+        }
+    }
+}