trace.ai-cli 1.1.7 → 1.1.9

package/README.markdown

@@ -1,151 +1,151 @@
-# Trace.Ai CLI
-
-A powerful AI-powered command-line tool powered by mixkey for analyzing files, folders, images, and system information. Trace.AI CLI leverages artificial intelligence to provide code analysis, project structure insights, text extraction from images, and detailed system information, making it an essential tool for developers and analysts.
-
-## Features
-
-- **File Analysis**: Analyze code files (e.g., JavaScript, Python, Java) for structure, quality, and potential improvements.
-- **Folder Structure Analysis**: Get insights into project architecture, technology stack, and file organization, including a simple file tree.
-- **Image Text Extraction**: Extract text from images (e.g., PNG, JPEG) using AI-powered OCR.
-- **System Information**: Get detailed information about your system's hardware, software, and configuration.
-- **Context Management**: Add text or file-based context to enhance AI responses.
-- **Interactive CLI**: Easy-to-use interface with commands like `/file`, `/folder`, `/image`, `/system`, and more.
-- **Extensible**: Supports a wide range of file types and languages, with customizable queries.
-
-## Installation
-
-### Prerequisites
-
-- **Node.js**: Version 14.0.0 or higher.
-- **npm**: Included with Node.js.
-
-### Install via npm
-
-Install Trace.AI CLI globally:
-
-```bash
-npm install -g trace.ai-cli
-```
-
-If you encounter permission issues, try one of the following:
-
-- **With sudo**:
-
-  ```bash
-  sudo npm install -g trace.ai-cli
-  ```
-
-- **With a user-owned directory**:
-
-  ```bash
-  mkdir -p ~/.npm-global
-  npm config set prefix '~/.npm-global'
-  echo 'export PATH=~/.npm-global/bin:$PATH' >> ~/.zshrc  # or ~/.bashrc
-  source ~/.zshrc
-  npm install -g trace.ai-cli
-  ```
-
-Verify the installation:
-
-```bash
-trace-ai --version
-```
-
-## Usage
-
-### Interactive Mode
-
-Start Trace.AI CLI in interactive mode:
-
-```bash
-trace-ai
-```
-
-This launches the interactive CLI interface where you can enter commands and queries.
-
-### Commands
-
-- **File Analysis**:
-  ```
-  /file <file_path> [query]
-  ```
-  Example: `/file src/app.js explain the main function`
-
-- **Folder Analysis**:
-  ```
-  /folder <folder_path> [query]
-  ```
-  Example: `/folder src/ what is the architecture?`
-
-- **Image Analysis**:
-  ```
-  /image <image_path> [query]
-  ```
-  Example: `/image diagram.png explain this flowchart`
-
-- **System Information**:
-  ```
-  /system [query]
-  ```
-  The system information feature supports natural language queries that automatically determine which information to return.
-  
-  Example: `/system How much RAM do I have available?`
-
-- **Add Context**:
-  ```
-  /context <text>
-  ```
-  Example: `/context This is a React project using TypeScript`
-
-- **Add File as Context**:
-  ```
-  /context-file <file_path>
-  ```
-  Example: `/context-file README.md`
-
-- **View Context**:
-  ```
-  /view-context
-  ```
-
-- **Clear Context or Screen**:
-  ```
-  /clear [type]
-  ```
-  Example: `/clear context` or `/clear screen`
-
-- **Help**:
-  ```
-  /help
-  ```
-
-- **Statistics**:
-  ```
-  /stats
-  ```
-
-- **Exit**:
-  ```
-  /exit
-  ```
-
-### Direct Queries
-
-You can also ask questions directly without using commands:
-
-```
-What is the difference between let and const in JavaScript?
-```
-
-```
-How do I implement a binary search tree in Python?
-```
-
-```
-What's the current CPU usage on my system?
-```
-
-
-## License
-
+# Trace.Ai CLI
+
+A powerful AI-powered command-line tool powered by mixkey for analyzing files, folders, images, and system information. Trace.AI CLI leverages artificial intelligence to provide code analysis, project structure insights, text extraction from images, and detailed system information, making it an essential tool for developers and analysts.
+
+## Features
+
+- **File Analysis**: Analyze code files (e.g., JavaScript, Python, Java) for structure, quality, and potential improvements.
+- **Folder Structure Analysis**: Get insights into project architecture, technology stack, and file organization, including a simple file tree.
+- **Image Text Extraction**: Extract text from images (e.g., PNG, JPEG) using AI-powered OCR.
+- **System Information**: Get detailed information about your system's hardware, software, and configuration.
+- **Context Management**: Add text or file-based context to enhance AI responses.
+- **Interactive CLI**: Easy-to-use interface with commands like `/file`, `/folder`, `/image`, `/system`, and more.
+- **Extensible**: Supports a wide range of file types and languages, with customizable queries.
+
+## Installation
+
+### Prerequisites
+
+- **Node.js**: Version 14.0.0 or higher.
+- **npm**: Included with Node.js.
+
+### Install via npm
+
+Install Trace.AI CLI globally:
+
+```bash
+npm install -g trace.ai-cli
+```
+
+If you encounter permission issues, try one of the following:
+
+- **With sudo**:
+
+  ```bash
+  sudo npm install -g trace.ai-cli
+  ```
+
+- **With a user-owned directory**:
+
+  ```bash
+  mkdir -p ~/.npm-global
+  npm config set prefix '~/.npm-global'
+  echo 'export PATH=~/.npm-global/bin:$PATH' >> ~/.zshrc  # or ~/.bashrc
+  source ~/.zshrc
+  npm install -g trace.ai-cli
+  ```
+
+Verify the installation:
+
+```bash
+trace-ai --version
+```
+
+## Usage
+
+### Interactive Mode
+
+Start Trace.AI CLI in interactive mode:
+
+```bash
+trace-ai
+```
+
+This launches the interactive CLI interface where you can enter commands and queries.
+
+### Commands
+
+- **File Analysis**:
+  ```
+  /file <file_path> [query]
+  ```
+  Example: `/file src/app.js explain the main function`
+
+- **Folder Analysis**:
+  ```
+  /folder <folder_path> [query]
+  ```
+  Example: `/folder src/ what is the architecture?`
+
+- **Image Analysis**:
+  ```
+  /image <image_path> [query]
+  ```
+  Example: `/image diagram.png explain this flowchart`
+
+- **System Information**:
+  ```
+  /system [query]
+  ```
+  The system information feature supports natural language queries that automatically determine which information to return.
+  
+  Example: `/system How much RAM do I have available?`
+
+- **Add Context**:
+  ```
+  /context <text>
+  ```
+  Example: `/context This is a React project using TypeScript`
+
+- **Add File as Context**:
+  ```
+  /context-file <file_path>
+  ```
+  Example: `/context-file README.md`
+
+- **View Context**:
+  ```
+  /view-context
+  ```
+
+- **Clear Context or Screen**:
+  ```
+  /clear [type]
+  ```
+  Example: `/clear context` or `/clear screen`
+
+- **Help**:
+  ```
+  /help
+  ```
+
+- **Statistics**:
+  ```
+  /stats
+  ```
+
+- **Exit**:
+  ```
+  /exit
+  ```
+
+### Direct Queries
+
+You can also ask questions directly without using commands:
+
+```
+What is the difference between let and const in JavaScript?
+```
+
+```
+How do I implement a binary search tree in Python?
+```
+
+```
+What's the current CPU usage on my system?
+```
+
+
+## License
+
 MIT

package/cli/traceAI.js

@@ -7,6 +7,7 @@ const { analyzeFile } = require('../services/fileAnalyzer');
 const { analyzeFolderStructure } = require('../services/folderAnalyzer');
 const { extractTextFromImage } = require('../services/imageService');
 const { processWithAI } = require('../services/aiService');
+const { markdownToAnsi } = require('../utils/markdown');
 
 class TraceAI {
     constructor() {
@@ -18,6 +19,7 @@ class TraceAI {
         this.contexts = [];
         this.sessionStartTime = new Date();
         this.queryCount = 0;
+        this.mode = 2; // 1: Fast, 2: Balanced (default), 3: Think
     }
 
     // Enhanced startup with animated loading
@@ -146,6 +148,12 @@ class TraceAI {
                 example: '/system cpu',
                 icon: '💻'
             },
+            {
+                cmd: '/mode <1|2|3>',
+                desc: 'Set AI mode: 1=Fast, 2=Balanced, 3=Think',
+                example: '/mode 2',
+                icon: '⚙️'
+            },
             {
                 cmd: '/exit',
                 desc: 'Exit the application',
@@ -177,11 +185,13 @@ class TraceAI {
     displayStatusBar() {
         const uptime = Math.floor((new Date() - this.sessionStartTime) / 1000);
         const contextCount = this.contexts.length;
+        const modeLabel = this.mode === 1 ? 'Fast' : this.mode === 2 ? 'Balanced' : 'Think';
         
         console.log(chalk.gray('\n┌─ SESSION INFO ') + chalk.gray('─'.repeat(34)));
         console.log(chalk.gray('│ ') + chalk.white('Uptime: ') + chalk.green(`${uptime}s`) + 
                    chalk.gray(' │ ') + chalk.white('Contexts: ') + chalk.cyan(contextCount) + 
-                   chalk.gray(' │ ') + chalk.white('Queries: ') + chalk.blueBright(this.queryCount));
+                   chalk.gray(' │ ') + chalk.white('Queries: ') + chalk.blueBright(this.queryCount) +
+                   chalk.gray(' │ ') + chalk.white('Mode: ') + chalk.magenta(modeLabel));
         console.log(chalk.gray('└─') + chalk.gray('─'.repeat(48)));
     }
 
@@ -220,6 +230,7 @@ class TraceAI {
             '/help': () => this.displayHelp(),
             '/stats': () => this.displayStats(),
             '/system': () => this.handleSystemCommand(trimmedInput),
+            '/mode': () => this.handleModeCommand(trimmedInput),
             '/exit': () => this.close()
         };
 
@@ -275,7 +286,7 @@ class TraceAI {
         await this.executeWithSpinner(
             `Analyzing image: ${path.basename(imagePath)}`,
             async () => {
-                const result = await extractTextFromImage(path.resolve(imagePath), query);
+                const result = await extractTextFromImage(path.resolve(imagePath), query, this.mode);
                 this.displayResult('Image Analysis', result, imagePath);
             }
         );
@@ -399,7 +410,7 @@ class TraceAI {
         await this.executeWithSpinner(
             'Retrieving system information',
             async () => {
-                const result = await processWithAI(query ? `Get system information about ${query}` : 'Get system information');
+                const result = await processWithAI(query ? `Get system information about ${query}` : 'Get system information', '', this.mode);
                 this.displayResult('System Information', result);
             }
         );
@@ -410,12 +421,25 @@ class TraceAI {
             'Processing your query',
             async () => {
                 const context = this.contexts.map(ctx => ctx.content).join('\n\n');
-                const result = await processWithAI(input, context);
+                const result = await processWithAI(input, context, this.mode);
                 this.displayResult('Trace.Ai Response', result.text || result);
             }
         );
     }
 
+    handleModeCommand(input) {
+        const arg = input.substring('/mode'.length).trim();
+        const value = Number(arg);
+        if (![1, 2, 3].includes(value)) {
+            console.log(chalk.cyan('Usage: /mode <1|2|3>  (1=Fast, 2=Balanced, 3=Think)'));
+            return;
+        }
+        this.mode = value;
+        const label = this.mode === 1 ? 'Fast' : this.mode === 2 ? 'Balanced' : 'Think';
+        console.log(chalk.green(`✓ Mode set to ${label} (${this.mode})`));
+        this.displayStatusBar();
+    }
+
     // Utility methods
     parseCommand(input) {
         return input.match(/(?:[^\s"]+|"[^"]*")+/g) || [];
@@ -463,7 +487,8 @@ class TraceAI {
             console.log(chalk.gray(`📍 Source: ${filePath}`));
         }
         console.log(chalk.blueBright('═'.repeat(60)));
-        console.log(chalk.white(content));
+        const rendered = markdownToAnsi(String(content || ''));
+        console.log(rendered);
         console.log(chalk.blueBright('═'.repeat(60)));
         console.log();
     }

package/index.js

@@ -5,6 +5,7 @@ const { analyzeFile } = require('./services/fileAnalyzer');
 const { analyzeFolderStructure } = require('./services/folderAnalyzer');
 const { extractTextFromImage } = require('./services/imageService');
 const { processWithAI } = require('./services/aiService');
+const { markdownToAnsi } = require('./utils/markdown');
 
 // Handle CLI arguments
 async function handleCliArgs() {
@@ -27,7 +28,8 @@ async function handleCliArgs() {
             const result = await analyzeFile(filePath, query);
             console.log('\n📄 File Analysis:');
             console.log('=' .repeat(50));
-            console.log(result.text || result);
+            console.log(markdownToAnsi(String(result.text || result || '')));
+
             console.log('=' .repeat(50));
         } catch (error) {
             console.error('❌ Error:', error.message);
@@ -42,7 +44,8 @@ async function handleCliArgs() {
             const result = await analyzeFolderStructure(folderPath, query);
             console.log('\n📁 Folder Analysis:');
             console.log('=' .repeat(50));
-            console.log(result.text);
+            console.log(markdownToAnsi(String(result.text || '')));
+
             console.log('=' .repeat(50));
         } catch (error) {
             console.error('❌ Error:', error.message);
@@ -56,7 +59,8 @@ async function handleCliArgs() {
             const result = await extractTextFromImage(imagePath, question);
             console.log('\n🖼️ Image Analysis:');
             console.log('=' .repeat(50));
-            console.log(result);
+            console.log(markdownToAnsi(String(result || '')));
+
             console.log('=' .repeat(50));
         } catch (error) {
             console.error('❌ Error:', error.message);
@@ -74,7 +78,7 @@ async function handleCliArgs() {
             const result = await processWithAI(query);
             console.log('\nTrace.Ai Response:');
             console.log('=' .repeat(50));
-            console.log(result);
+            console.log(markdownToAnsi(String(result || '')));
             console.log('=' .repeat(50));
         } catch (error) {
             console.error('❌ Error:', error.message);

package/package.json

@@ -1,45 +1,45 @@
-{
-    "name": "trace.ai-cli",
-    "version": "1.1.7",
-    "description": "A powerful AI-powered CLI tool",
-    "main": "index.js",
-    "bin": {
-        "trace-ai": "index.js"
-    },
-    "scripts": {
-        "start": "node index.js",
-        "test": "echo \"Error: no test specified\" && exit 1"
-    },
-    "keywords": [
-        "ai",
-        "cli",
-        "code-analysis",
-        "file-analysis",
-        "code-review",
-        "bug-detection",
-        "text-processing",
-        "image-text-extraction",
-        "ocr",
-        "artificial-intelligence",
-        "command-line",
-        "tool",
-        "developer-tools",
-        "code-quality",
-        "project-analysis"
-    ],
-    "author": "Dukeindustries7",
-    "license": "MIT",
-    "dependencies": {
-        "chalk": "^4.1.2",
-        "cli-spinners": "^2.9.2",
-        "figlet": "^1.7.0",
-        "node-fetch": "^2.6.7",
-        "ora": "^5.4.1",
-        "os-utils": "^0.0.14",
-        "systeminformation": "^5.27.1",
-        "trace.ai-cli": "^1.1.6"
-    },
-    "engines": {
-        "node": ">=14.0.0"
-    }
-}
+{
+    "name": "trace.ai-cli",
+    "version": "1.1.9",
+    "description": "A powerful AI-powered CLI tool",
+    "main": "index.js",
+    "bin": {
+        "trace-ai": "index.js"
+    },
+    "scripts": {
+        "start": "node index.js",
+        "test": "echo \"Error: no test specified\" && exit 1"
+    },
+    "keywords": [
+        "ai",
+        "cli",
+        "code-analysis",
+        "file-analysis",
+        "code-review",
+        "bug-detection",
+        "text-processing",
+        "image-text-extraction",
+        "ocr",
+        "artificial-intelligence",
+        "command-line",
+        "tool",
+        "developer-tools",
+        "code-quality",
+        "project-analysis"
+    ],
+    "author": "Dukeindustries7",
+    "license": "MIT",
+    "dependencies": {
+        "chalk": "^4.1.2",
+        "cli-spinners": "^2.9.2",
+        "figlet": "^1.7.0",
+        "node-fetch": "^2.6.7",
+        "ora": "^5.4.1",
+        "os-utils": "^0.0.14",
+        "systeminformation": "^5.27.1",
+        "trace.ai-cli": "^1.1.8"
+    },
+    "engines": {
+        "node": ">=14.0.0"
+    }
+}

package/services/aiService.js

@@ -2,6 +2,20 @@ const fetch = require('node-fetch');
 const { encryptData, decryptData } = require('../utils/encryption');
 const { getSystemInfo, formatBytes } = require('./systemInfoService');
 
+// Model selection helpers
+function getTextModels(mode) {
+    switch (Number(mode)) {
+        case 1: // Fast
+            return ['g1'];
+        case 2: // Balanced
+            return ['sct', 'g1', 'gptoss20'];
+        case 3: // Think
+            return ['kimi', 'mvrk', 'gma3', 'dsv3', 'sct', 'gptoss120', 'gptoss20', 'll70b', 'qw3', 'nlm3'];
+        default:
+            return ['sct', 'g1', 'gptoss20'];
+    }
+}
+
 /**
  * Use AI to determine what system information is being requested
  * @param {string} prompt - The user's prompt
@@ -40,7 +54,7 @@ Examples:
 Only respond with the category names, nothing else. Be precise and focus on what the user actually needs.`;
         
         // Use multiple AI models to analyze the query
-        const models = ['kimi', 'mvrk', 'gma3'];
+        const models = getTextModels(2); // Balanced by default
         
         const modelRequests = models.map(async (model) => {
             try {
@@ -250,7 +264,7 @@ Make the response helpful and easy to understand for the user.`;
                 method: 'POST',
                 headers: { 'Content-Type': 'application/json' },
                 body: encryptData({
-                    a: 'gma3', // Use a good model for formatting
+                    a: getTextModels(2)[0], // Use the first Balanced text model for formatting
                     q: formattingPrompt,
                     r: [],
                     i: [],
@@ -518,7 +532,7 @@ function formatSystemInfoManually(sysInfo, prompt) {
         response += `\n`;
     }
     
-    response += `*Query processed for user: m0v0dga_walmart*`;
+    response ;
     return response;
 }
 
@@ -558,7 +572,7 @@ function getHelpMessage(prompt) {
            `*User: m0v0dga_walmart*`;
 }
 
-async function processWithAI(prompt, context = '') {
+async function processWithAI(prompt, context = '', mode = 2) {
     try {
         
         // Enhanced system query detection
@@ -575,7 +589,6 @@ async function processWithAI(prompt, context = '') {
         
         if (isSystemQuery) {
             try {
-                console.log(`[${new Date().toISOString()}] Processing system information query...`);
                 
                 // First get basic system info to provide context
                 const basicInfo = await getSystemInfo('basic');
@@ -589,7 +602,6 @@ async function processWithAI(prompt, context = '') {
                 // Format the response using AI
                 const formattedResponse = await formatSystemInfoResponse(systemInfo, prompt);
                 
-                console.log(`[${new Date().toISOString()}] System info query completed successfully`);
                 return formattedResponse;
             } catch (sysError) {
                 console.error(`[${new Date().toISOString()}] ❌ System info error:`, sysError.message);
@@ -599,8 +611,7 @@ async function processWithAI(prompt, context = '') {
         }
         
         // Regular AI processing for non-system queries
-        console.log(`[${new Date().toISOString()}] Processing query...`);
-        const models = ['kimi', 'mvrk', 'gma3', 'dsv3', 'qw32b', 'ms24b', 'll70b', 'qw3', 'nlm3'];
+        const models = getTextModels(mode); // Use selected mode
         
         const modelRequests = models.map(async (model) => {
             try {
@@ -632,6 +643,11 @@ async function processWithAI(prompt, context = '') {
             return 'I apologize, but I was unable to process your request at this time. Please try again.';
         }
 
+        // Fast mode: return the first model response without aggregation
+        if (Number(mode) === 1) {
+            return responseTexts[0];
+        }
+
         const finalResponse = await fetch('https://traceai.dukeindustries7.workers.dev/', {
             method: 'POST',
             headers: { 'Content-Type': 'application/json' },
@@ -647,7 +663,6 @@ async function processWithAI(prompt, context = '') {
         const encryptedResult = await finalResponse.text();
         const decryptedResult = decryptData(encryptedResult);
         
-        console.log(`[${new Date().toISOString()}] Query completed successfully`);
         return decryptedResult.text || 'No response generated';
     } catch (error) {
         console.error(`[${new Date().toISOString()}] ❌ Processing error:`, error.message);