truffle-ai · rahulkarajgikar · Apr 3, 2025 · Mar 28, 2025 · Mar 28, 2025 · Mar 28, 2025
diff --git a/.env.example b/.env.example
@@ -1,14 +1,11 @@
 # Provider-specific API keys
-# You only need to set the API key for the provider you're using
 
 # OpenAI API key (required when using OpenAI models)
-# Must start with 'sk-' 
 OPENAI_API_KEY=your_openai_api_key_here
 
 # Anthropic API key (required when using Anthropic Claude models)
-# Must start with 'sk-ant-'
 ANTHROPIC_API_KEY=your_anthropic_api_key_here
 
-# Optional: Set the log level to debug or silly to see more verbose output
+# Set the log level to debug or silly to see more verbose output. Default is info.
 # LOG_LEVEL=debug
 # LOG_LEVEL=silly
diff --git a/app/cli.ts b/app/cli.ts
@@ -2,9 +2,10 @@ import readline from 'readline';
 import chalk from 'chalk';
 import { ClientManager } from '../src/client/manager.js';
 import { logger } from '../src/utils/logger.js';
-import { LLMCallbacks, LLMService } from '../src/ai/llm/types.js';
+import { LLMCallbacks, ILLMService } from '../src/ai/llm/types.js';
 import { AgentConfig } from '../src/config/types.js';
 import { initializeServices } from '../src/utils/service-initializer.js';
+import boxen from 'boxen';
 
 /**
  * Start AI-powered CLI with unified configuration
@@ -34,11 +35,15 @@ export async function initializeAiCli(
  */
 export async function runAiCli(
     clientManager: ClientManager,
-    llmService: LLMService
+    llmService: ILLMService
 ) {
     // Get model and provider info directly from the LLM service
-    const { provider, model } = llmService.getConfig();
-    logger.info(`Using ${provider} model: ${model}`, null, 'yellow');
+    // const { provider, model } = llmService.getConfig();
+    logger.info(
+        `Using model config:${JSON.stringify(llmService.getConfig(), null, 2)}`,
+        null,
+        'yellow'
+    );
 
     logger.debug(`Log level: ${logger.getLevel()}`);
     logger.info(`Connected servers: ${clientManager.getClients().size}`, null, 'green');
@@ -52,18 +57,18 @@ export async function runAiCli(
         // Get available tools from all connected servers
         logger.info('Loading available tools...');
 
-        // Get all tools from the manager
+        // Get all tools from the LLM service
         const tools = await clientManager.getAllTools();
 
-        logger.debug(`Received tools: ${tools.map((t) => t.name)}`);
-
-        // Update system context with available tools
+        // Update the system context with the available tools
+        logger.info('Updating system context...');
         llmService.updateSystemContext(tools);
 
         logger.info(
-            `Loaded ${tools.length} tools from ${clientManager.getClients().size} tool providers\n`
+            `Loaded ${Object.keys(tools).length} tools from ${clientManager.getClients().size} MCP servers\n`
         );
         logger.info('AI Agent initialized successfully!', null, 'green');
+
         // Create readline interface
         const rl = readline.createInterface({
             input: process.stdin,
@@ -103,8 +108,37 @@ export async function runAiCli(
                 }
 
                 try {
+                    let accumulatedResponse = '';
+                    let currentLines = 0;
                     // Create callbacks for progress indication (without spinner)
                     const callbacks: LLMCallbacks = {
+                        onChunk: (text: string) => {
+                            // Append the new chunk to the accumulated response
+                            accumulatedResponse += text;
+
+                            // Generate the new box
+                            const box = boxen(chalk.white(accumulatedResponse), {
+                                padding: 1,
+                                borderColor: 'yellow',
+                                title: '🤖 AI Response',
+                                titleAlignment: 'center',
+                            });
+                            const newLines = box.split('\n').length;
+
+                            // Move cursor up to the start of the previous box (if it exists)
+                            if (currentLines > 0) {
+                                process.stdout.write(`\x1b[${currentLines}A`); // Move up currentLines
+                            }
+
+                            // Print the new box (this overwrites the old one)
+                            process.stdout.write(box);
+
+                            // Update the line count
+                            currentLines = newLines;
+
+                            // Move cursor to the end of the box to allow logs below
+                            process.stdout.write('\n');
+                        },
                         onThinking: () => {
                             logger.info('AI thinking...');
                         },

diff --git a/app/index.ts b/app/index.ts
@@ -6,7 +6,7 @@ import dotenv from 'dotenv';
 import { logger } from '../src/utils/logger.js';
 import { initializeAiCli } from './cli.js';
 import { loadConfigFile } from '../src/config/loader.js';
-
+import { AgentConfig } from '../src/config/types.js';
 // Load environment variables
 dotenv.config();
 
@@ -44,7 +44,6 @@ program.parse();
 const options = program.opts();
 const configFile = options.configFile;
 const connectionMode = options.strict ? 'strict' : ('lenient' as 'strict' | 'lenient');
-const verbose = options.verbose !== false;
 
 // Platform-independent path handling
 const normalizedConfigPath = path.normalize(configFile);
@@ -75,37 +74,23 @@ logger.info('');
 async function startAiClient() {
     try {
         // Load the agent configuration
-        const config = await loadConfigFile(normalizedConfigPath);
-
-        // Validate MCP servers section exists
-        if (!config.mcpServers || Object.keys(config.mcpServers).length === 0) {
-            logger.error('Error: No MCP server configurations found in the provided file');
-            process.exit(1);
-        }
-        logger.info(
-            `Found ${Object.keys(config.mcpServers).length} server configurations in ${normalizedConfigPath}`,
-            null,
-            'green'
-        );
+        const config: AgentConfig = await loadConfigFile(normalizedConfigPath);
 
-        // Validate LLM section exists, use defaults if not
-        if (!config.llm) {
-            logger.info('No LLM configuration found, using defaults', null, 'yellow');
-            config.llm = {
-                provider: 'openai',
-                model: 'gpt-4o-mini',
-                apiKey: 'env:OPENAI_API_KEY',
-            };
-        }
+        validateAgentConfig(config);
 
         logger.info('===============================================');
         logger.info('Starting AI-powered MCP client...', null, 'cyanBright');
         logger.info('===============================================\n');
 
         await initializeAiCli(config, connectionMode);
     } catch (error) {
-        logger.error('Error: Failed to load configuration from file');
-        logger.error(error);
+        logger.error(
+            `Error: Failed to initialize AI CLI from config file${normalizedConfigPath}: ${JSON.stringify(
+                error,
+                null,
+                2
+            )}`
+        );
         process.exit(1);
     }
 }
@@ -116,3 +101,30 @@ startAiClient().catch((error) => {
     logger.error(error);
     process.exit(1);
 });
+
+function validateAgentConfig(config: AgentConfig): void {
+    logger.info('Validating agent config', null, 'cyanBright');
+    if (!config.mcpServers || Object.keys(config.mcpServers).length === 0) {
+        throw new Error('No MCP server configurations provided');
+    }
+
+    // Validate LLM section exists, use defaults if not
+    if (!config.llm) {
+        logger.info('No LLM configuration found, using defaults', null, 'yellow');
+        config.llm = {
+            provider: 'openai',
+            model: 'gpt-4o-mini',
+            apiKey: 'env:OPENAI_API_KEY',
+        };
+    }
+
+    if (!config.llm.provider || !config.llm.model) {
+        throw new Error('LLM configuration must specify provider and model');
+    }
+
+    logger.info(
+        `Found ${Object.keys(config.mcpServers).length} server configurations`,
+        null,
+        'green'
+    );
+}
diff --git a/configuration/mcp.json b/configuration/mcp.json
@@ -1,14 +1,17 @@
 {
     "mcpServers": {
         "desktopCommander": {
+            "type": "stdio",
             "command": "npx",
             "args": ["-y", "@wonderwhy-er/desktop-commander"]
         },
         "puppeteer": {
+            "type": "stdio",
             "command": "node",
             "args": ["--loader", "ts-node/esm", "src/servers/puppeteerServer.ts"]
         },
         "dockerServer": {
+            "type": "stdio",
             "command": "node",
             "args": ["--loader", "ts-node/esm", "src/servers/dockerServer.ts"]
         }