diff --git a/packages/qllm-cli/package.json b/packages/qllm-cli/package.json
index e50e6da..9d54953 100644
--- a/packages/qllm-cli/package.json
+++ b/packages/qllm-cli/package.json
@@ -43,7 +43,9 @@
   "license": "Apache-2.0",
   "devDependencies": {
     "@types/copy-paste": "^1.1.33",
+
     "@types/jest": "^29.5.12",
+    "@types/mime-types": "^2.1.4",
     "@types/node": "^22.5.0",
     "@types/prompts": "^2.4.9",
     "@types/screenshot-desktop": "^1.12.3",
@@ -58,18 +60,19 @@
     "typescript": "^5.5.4"
   },
   "dependencies": {
-    "@types/node": "^22.5.0",
     "@npmcli/fs": "^3.1.1",
+    "@types/node": "^22.5.0",
     "cli-table3": "^0.6.5",
     "commander": "^12.1.0",
     "console-table-printer": "^2.12.1",
     "copy-paste": "^1.5.3",
+    "gradient-string": "^2.0.2",
     "jimp": "^0.22.12",
     "kleur": "^4.1.5",
     "lru-cache": "^11.0.0",
     "nanospinner": "^1.1.0",
     "prompts": "^2.4.2",
-    "qllm-lib": "1.8.5",
+    "qllm-lib": "^1.8.6",
     "readline": "^1.3.0",
     "screenshot-desktop": "^1.15.0",
     "table": "^6.8.2",
diff --git a/packages/qllm-cli/src/chat/chat-config.ts b/packages/qllm-cli/src/chat/chat-config.ts
index eee4348..9cc1051 100644
--- a/packages/qllm-cli/src/chat/chat-config.ts
+++ b/packages/qllm-cli/src/chat/chat-config.ts
@@ -1,4 +1,3 @@
-// packages/qllm-cli/src/chat/chat-config.ts
 import fs from "fs/promises";
 import path from "path";
 import os from "os";
@@ -63,7 +62,9 @@ export class ChatConfig {
     key: K,
     value: ChatConfigType[K]
   ): void {
-    this.config[key] = value;
+    const partialConfig = { [key]: value };
+    const validatedConfig = ChatConfigSchema.partial().parse(partialConfig);
+    this.config[key] = validatedConfig[key] as ChatConfigType[K];
   }
 
   public getProvider(): string | undefined {
@@ -71,7 +72,7 @@ export class ChatConfig {
   }
 
   public setProvider(provider: string): void {
-    this.config.provider = provider;
+    this.set("provider", provider);
   }
 
   public getModel(): string | undefined {
@@ -79,7 +80,7 @@ export class ChatConfig {
   }
 
   public setModel(model: string): void {
-    this.config.model = model;
+    this.set("model", model);
   }
 
   public getTemperature(): number | undefined {
@@ -87,7 +88,7 @@ export class ChatConfig {
   }
 
   public setTemperature(temperature: number): void {
-    this.config.temperature = temperature;
+    this.set("temperature", temperature);
   }
 
   public getMaxTokens(): number | undefined {
@@ -95,7 +96,7 @@ export class ChatConfig {
   }
 
   public setMaxTokens(maxTokens: number): void {
-    this.config.maxTokens = maxTokens;
+    this.set("maxTokens", maxTokens);
   }
 
   public getTopP(): number | undefined {
@@ -103,7 +104,7 @@ export class ChatConfig {
   }
 
   public setTopP(topP: number): void {
-    this.config.topP = topP;
+    this.set("topP", topP);
   }
 
   public getFrequencyPenalty(): number | undefined {
@@ -111,7 +112,7 @@ export class ChatConfig {
   }
 
   public setFrequencyPenalty(frequencyPenalty: number): void {
-    this.config.frequencyPenalty = frequencyPenalty;
+    this.set("frequencyPenalty", frequencyPenalty);
   }
 
   public getPresencePenalty(): number | undefined {
@@ -119,7 +120,7 @@ export class ChatConfig {
   }
 
   public setPresencePenalty(presencePenalty: number): void {
-    this.config.presencePenalty = presencePenalty;
+    this.set("presencePenalty", presencePenalty);
   }
 
   public getStopSequence(): string[] | undefined {
@@ -127,7 +128,7 @@ export class ChatConfig {
   }
 
   public setStopSequence(stopSequence: string[]): void {
-    this.config.stopSequence = stopSequence;
+    this.set("stopSequence", stopSequence);
   }
 
   public getCurrentConversationId(): string | undefined {
@@ -135,7 +136,7 @@ export class ChatConfig {
   }
 
   public setCurrentConversationId(conversationId: string | undefined): void {
-    this.config.currentConversationId = conversationId;
+    this.set("currentConversationId", conversationId);
   }
 
   public async initialize(): Promise<void> {
@@ -147,4 +148,4 @@ export class ChatConfig {
   }
 }
 
-export const chatConfig = ChatConfig.getInstance();
+export const chatConfig = ChatConfig.getInstance();
\ No newline at end of file
diff --git a/packages/qllm-cli/src/chat/chat.ts b/packages/qllm-cli/src/chat/chat.ts
index 27e99db..08e4a74 100644
--- a/packages/qllm-cli/src/chat/chat.ts
+++ b/packages/qllm-cli/src/chat/chat.ts
@@ -4,9 +4,9 @@ import { createConversationManager, getLLMProvider } from "qllm-lib";
 import { ChatConfig } from "./chat-config";
 import { MessageHandler } from "./message-handler";
 import { CommandProcessor } from "./command-processor";
-import { IOManager } from "./io-manager";
+import { IOManager } from "../utils/io-manager";
 import { ConfigManager } from "./config-manager";
-import { output } from "../utils/output";
+import { ioManager} from "../utils/io-manager";
 import ImageManager from "./image-manager";
 
 export class Chat {
@@ -40,11 +40,11 @@ export class Chat {
       await this.config.initialize();
       this.configManager.setProvider(this.providerName);
       this.configManager.setModel(this.modelName);
-      output.success(
+      ioManager.displaySuccess(
         `Chat initialized with ${this.providerName} provider and ${this.modelName} model.`
       );
     } catch (error) {
-      output.error(`Failed to initialize chat: ${(error as Error).message}`);
+     ioManager.displayError(`Failed to initialize chat: ${(error as Error).message}`);
       process.exit(1);
     }
   }
@@ -53,10 +53,10 @@ export class Chat {
     await this.initialize();
     const conversation = await this.conversationManager.createConversation();
     this.conversationId = conversation.id;
-    output.info(
+    ioManager.displayInfo(
       "Chat session started. Type your messages or use special commands."
     );
-    output.info("Type /help for available commands.");
+    ioManager.displayInfo("Type /help for available commands.");
     this.promptUser();
   }
 
@@ -86,7 +86,7 @@ export class Chat {
     try {
       const [command, ...args] = input.trim().split(/\s+/);
       if (!command) {
-        output.error("No command provided");
+       ioManager.displayError("No command provided");
         return;
       }
       const cleanCommand = command.substring(1).toLowerCase();
@@ -100,7 +100,7 @@ export class Chat {
       };
       await this.commandProcessor.processCommand(cleanCommand, args, context);
     } catch (error) {
-      output.error(
+     ioManager.displayError(
         "Error processing special command: " +
           (error instanceof Error ? error.message : String(error))
       );
@@ -112,7 +112,7 @@ export class Chat {
     images: string[]
   ): Promise<void> {
     if (!this.conversationId) {
-      output.error("No active conversation. Please start a chat first.");
+     ioManager.displayError("No active conversation. Please start a chat first.");
       return;
     }
     const currentProviderName = this.configManager.getProvider();
diff --git a/packages/qllm-cli/src/chat/command-processor.ts b/packages/qllm-cli/src/chat/command-processor.ts
index f945718..b8a6532 100644
--- a/packages/qllm-cli/src/chat/command-processor.ts
+++ b/packages/qllm-cli/src/chat/command-processor.ts
@@ -2,7 +2,7 @@
 import { ConversationManager } from "qllm-lib";
 import { ChatConfig } from "./chat-config";
 import { ConfigManager } from "./config-manager";
-import { IOManager } from "./io-manager";
+import { IOManager } from "../utils/io-manager";
 
 import ImageManager from "./image-manager";
 import { showHelp } from "./commands/show-help";
diff --git a/packages/qllm-cli/src/chat/commands/display-conversation.ts b/packages/qllm-cli/src/chat/commands/display-conversation.ts
index 1756f77..a221c14 100644
--- a/packages/qllm-cli/src/chat/commands/display-conversation.ts
+++ b/packages/qllm-cli/src/chat/commands/display-conversation.ts
@@ -1,5 +1,5 @@
 import { CommandContext } from "../command-processor";
-import { IOManager } from "../io-manager";
+import { IOManager } from "../../utils/io-manager";
 import { ConversationMessage, ChatMessageContent, LLMOptions } from "qllm-lib";
 
 const MESSAGES_PER_PAGE = 5;
diff --git a/packages/qllm-cli/src/chat/config-manager.ts b/packages/qllm-cli/src/chat/config-manager.ts
index e926514..103e5ab 100644
--- a/packages/qllm-cli/src/chat/config-manager.ts
+++ b/packages/qllm-cli/src/chat/config-manager.ts
@@ -1,7 +1,7 @@
 // packages/qllm-cli/src/chat/config-manager.ts
 import { ChatConfig } from "./chat-config";
 import { getLLMProvider } from "qllm-lib";
-import { output } from "../utils/output";
+import { ioManager } from "../utils/io-manager";
 import { DEFAULT_PROVIDER, DEFAULT_MODEL } from "../constants";
 
 export class ConfigManager {
@@ -11,9 +11,11 @@ export class ConfigManager {
     try {
       await getLLMProvider(providerName);
       this.config.setProvider(providerName);
-      output.success(`Provider set to: ${providerName}`);
+      ioManager.displaySuccess(`Provider set to: ${providerName}`);
     } catch (error) {
-      output.error(`Failed to set provider: ${(error as Error).message}`);
+      ioManager.displayError(
+        `Failed to set provider: ${(error as Error).message}`
+      );
     }
   }
 
@@ -27,7 +29,7 @@ export class ConfigManager {
 
   setModel(modelName: string): void {
     this.config.setModel(modelName);
-    output.success(`Model set to: ${modelName}`);
+    ioManager.displaySuccess(`Model set to: ${modelName}`);
   }
 
   getModel(): string {
@@ -56,11 +58,12 @@ export class ConfigManager {
         this.config.setStopSequence(value.split(","));
         break;
       default:
-        output.error(`Unknown option: ${option}`);
+        ioManager.displayError(`Unknown option: ${option}`);
         this.showValidOptions();
         return;
     }
-    output.success(`Option ${evalOption} set to: ${value}`);
+
+    ioManager.displaySuccess(`Option ${evalOption} set to: ${value}`);
   }
 
   getAllSettings(): Record<string, any> {
@@ -77,7 +80,7 @@ export class ConfigManager {
   }
 
   private showValidOptions(): void {
-    output.info("Valid options are:");
+    ioManager.displayInfo("Valid options are:");
     [
       "temperature",
       "max_tokens",
@@ -85,7 +88,7 @@ export class ConfigManager {
       "frequency_penalty",
       "presence_penalty",
       "stop_sequence",
-    ].forEach((opt) => output.info(`- ${opt}`));
+    ].forEach((opt) => ioManager.displayInfo(`- ${opt}`));
   }
 
   async initialize(): Promise<void> {
diff --git a/packages/qllm-cli/src/chat/image-manager.ts b/packages/qllm-cli/src/chat/image-manager.ts
index 6198d3c..a3777d0 100644
--- a/packages/qllm-cli/src/chat/image-manager.ts
+++ b/packages/qllm-cli/src/chat/image-manager.ts
@@ -1,7 +1,7 @@
 // packages/qllm-cli/src/chat/image-manager.ts
 
 import { utils } from "./utils";
-import { output } from "../utils/output";
+import { ioManager } from "../utils/io-manager";
 
 export class ImageManager {
   private images: Set<string>;
@@ -17,18 +17,18 @@ export class ImageManager {
   addImage(image: string): void {
     if (utils.isValidUrl(image) || utils.isImageFile(image)) {
       this.images.add(image);
-      output.success(`Image added: ${utils.truncateString(image, 50)}`);
+      ioManager.displaySuccess(`Image added: ${utils.truncateString(image, 50)}`);
     } else {
-      output.error("Invalid image URL or file path");
+      ioManager.displayError("Invalid image URL or file path");
     }
   }
 
   removeImage(image: string): boolean {
     const removed = this.images.delete(image);
     if (removed) {
-      output.success(`Image removed: ${utils.truncateString(image, 50)}`);
+      ioManager.displaySuccess(`Image removed: ${utils.truncateString(image, 50)}`);
     } else {
-      output.warn(`Image not found: ${utils.truncateString(image, 50)}`);
+      ioManager.displayWarning(`Image not found: ${utils.truncateString(image, 50)}`);
     }
     return removed;
   }
@@ -48,20 +48,20 @@ export class ImageManager {
   clearImages(showOutput: boolean = true): void {
     const count = this.images.size;
     this.images.clear();
-    output.success(
+    ioManager.displaySuccess(
       `Cleared ${count} image${count !== 1 ? "s" : ""} from the buffer`
     );
   }
 
   displayImages(): void {
     if (this.images.size === 0) {
-      output.info("No images in the buffer");
+      ioManager.displayInfo("No images in the buffer");
       return;
     }
 
-    output.info(`Images in the buffer (${this.images.size}):`);
+    ioManager.displayInfo(`Images in the buffer (${this.images.size}):`);
     this.getImages().forEach((image, index) => {
-      output.info(`${index + 1}. ${utils.truncateString(image, 70)}`);
+      ioManager.displayInfo(`${index + 1}. ${utils.truncateString(image, 70)}`);
     });
   }
 }
diff --git a/packages/qllm-cli/src/chat/io-manager.ts b/packages/qllm-cli/src/chat/io-manager.ts
deleted file mode 100644
index 262c865..0000000
--- a/packages/qllm-cli/src/chat/io-manager.ts
+++ /dev/null
@@ -1,278 +0,0 @@
-// packages/qllm-cli/src/chat/io-manager.ts
-import readline from "readline";
-import kleur from "kleur";
-import { getBorderCharacters, table } from "table";
-import { createSpinner } from "nanospinner";
-import { Table } from "console-table-printer";
-import prompts from 'prompts';
-
-interface Spinner {
-  start: () => void;
-  stop: () => void;
-  success: (options: { text: string }) => void;
-  error: (options: { text: string }) => void;
-}
-
-export class IOManager {
-  private rl: readline.Interface;
-  private currentSpinner: Spinner | null = null;
-
-  constructor() {
-    this.rl = readline.createInterface({
-      input: process.stdin,
-      output: process.stdout,
-    });
-  }
-
-  async getUserInput(prompt: string): Promise<string> {
-    const response = await prompts({
-      type: 'text',
-      name: 'userInput',
-      message: this.colorize(prompt, "green"),
-    });
-    return response.userInput;
-  }
-
-  async getAsyncUserInput(prompt: string): Promise<string> {
-    return this.getUserInput(prompt);
-  }
-
-  displayUserMessage(message: string): void {
-    this.displayInfo(this.colorize(`You: ${message}`, "green"));
-  }
-
-  displayAssistantMessage(message: string): void {
-    this.displayInfo(this.colorize(`Assistant: ${message}`, "blue"));
-  }
-
-  displaySystemMessage(message: string): void {
-    this.displayInfo(this.colorize(`System: ${message}`, "yellow"));
-  }
-
-  displayError(message: string): void {
-    console.error(this.colorize(`✖ ${message}`, "red"));
-  }
-
-  displaySuccess(message: string): void {
-    console.log(this.colorize(`✔ ${message}`, "green"));
-  }
-
-  displayWarning(message: string): void {
-    console.warn(this.colorize(`⚠ ${message}`, "yellow"));
-  }
-
-  displayInfo(message: string): void {
-    console.log(message);
-  }
-
-  displayTable(headers: string[], data: string[][]): void {
-    const tableData = [headers.map((h) => this.colorize(h, "cyan")), ...data];
-    console.log(
-      table(tableData, {
-        border: getBorderCharacters("norc"),
-        columnDefault: {
-          paddingLeft: 0,
-          paddingRight: 1,
-        },
-        drawHorizontalLine: () => false,
-      })
-    );
-  }
-
-  displayList(items: string[]): void {
-    items.forEach((item) => this.displayInfo(`• ${item}`));
-  }
-
-  clearLine(): void {
-    process.stdout.write("\r\x1b[K");
-  }
-
-  newLine(): void {
-    console.log();
-  }
-
-  close(): void {
-    this.rl.close();
-  }
-
-  createSpinner(message: string): Spinner {
-    const spinner = createSpinner(message);
-    this.currentSpinner = spinner;
-    return spinner;
-  }
-
-  stopSpinner(): void {
-    if (this.currentSpinner) {
-      this.currentSpinner.stop();
-      this.currentSpinner = null;
-    }
-  }
-
-  write(text: string): void {
-    process.stdout.write(text);
-  }
-
-  displayConversationList(
-    conversations: { id: string; createdAt: Date }[]
-  ): void {
-    this.displayInfo("Conversations:");
-    conversations.forEach((conv, index) => {
-      this.displayInfo(
-        `${index + 1}. ID: ${
-          conv.id
-        }, Created: ${conv.createdAt.toLocaleString()}`
-      );
-    });
-  }
-
-  displayConversationDetails(
-    id: string,
-    messages: { role: string; content: string }[]
-  ): void {
-    this.displayInfo(`Conversation ${id}:`);
-    messages.forEach((msg, index) => {
-      const roleColor = msg.role === "user" ? "green" : "blue";
-      this.displayInfo(
-        `${index + 1}. ${this.colorize(msg.role, roleColor)}: ${msg.content}`
-      );
-    });
-  }
-
-  async promptForConversationSelection(): Promise<string> {
-    const response = await prompts({
-      type: 'text',
-      name: 'selection',
-      message: 'Enter conversation number to select (or \'c\' to cancel): ',
-    });
-    return response.selection;
-  }
-
-  async promptForConversationDeletion(): Promise<string> {
-    const response = await prompts({
-      type: 'text',
-      name: 'selection',
-      message: 'Enter conversation number to delete (or \'c\' to cancel): ',
-    });
-    return response.selection;
-  }
-
-  async confirmAction(message: string): Promise<boolean> {
-    const response = await prompts({
-      type: 'confirm',
-      name: 'confirmed',
-      message: message,
-      initial: false,
-    });
-    return response.confirmed;
-  }
-
-  colorize(text: string, color: string): string {
-    switch (color) {
-      case "green":
-        return kleur.green(text);
-      case "blue":
-        return kleur.blue(text);
-      case "yellow":
-        return kleur.yellow(text);
-      case "red":
-        return kleur.red(text);
-      case "cyan":
-        return kleur.cyan(text);
-      default:
-        return text;
-    }
-  }
-
-  displayGroupedInfo(title: string, items: string[]): void {
-    this.displayInfo(this.colorize(`\n${title}:`, "yellow"));
-    items.forEach((item) => this.displayInfo(`  ${item}`));
-  }
-
-  displayTitle(title: string): void {
-    console.log(this.colorize(title, "cyan"));
-  }
-
-  displayCodeBlock(code: string, language?: string): void {
-    const formattedCode = language ? this.colorize(code, "cyan") : code;
-    console.log(this.colorize("```" + (language || ""), "gray"));
-    console.log(formattedCode);
-    console.log(this.colorize("```", "gray"));
-  }
-
-  clear(): void {
-    console.clear();
-  }
-
-  displaySectionHeader(header: string): void {
-    console.log(kleur.bold().yellow(`\n${header}`));
-  }
-
-  displayModelTable(models: { id: string; description: string }[]): void {
-    const p = new Table({
-      columns: [
-        { name: "id", alignment: "left", color: "cyan" },
-        { name: "description", alignment: "left", color: "white" },
-      ],
-      sort: (row1, row2) => row1.id.localeCompare(row2.id),
-    });
-
-    models.forEach((model) => {
-      p.addRow({ id: model.id, description: model.description || "N/A" });
-    });
-
-    p.printTable();
-  }
-
-  displayConfigOptions(options: Array<{ name: string; value: any }>): void {
-    this.displaySectionHeader("Current Configuration");
-
-    const longestNameLength = Math.max(
-      ...options.map((opt) => opt.name.length)
-    );
-
-    options.forEach(({ name, value }) => {
-      const paddedName = name.padEnd(longestNameLength);
-      const formattedValue = value !== undefined ? value.toString() : "Not set";
-      const coloredValue =
-        value !== undefined
-          ? this.colorize(formattedValue, "green")
-          : this.colorize(formattedValue, "yellow");
-      this.displayInfo(`${this.colorize(paddedName, "cyan")}: ${coloredValue}`);
-    });
-
-    this.newLine();
-    this.displayInfo(
-      this.colorize("Use 'set <option> <value>' to change a setting", "dim")
-    );
-    this.displayInfo(this.colorize("Example: set provider openai", "dim"));
-  }
-
-  displayProviderList(providers: string[]): void {
-    this.displaySectionHeader("Available Providers");
-
-    if (providers.length === 0) {
-      this.displayInfo(this.colorize("No providers available.", "yellow"));
-    } else {
-      providers.forEach((provider, index) => {
-        this.displayInfo(
-          `${this.colorize(`${index + 1}.`, "cyan")} ${this.colorize(
-            provider,
-            "green"
-          )}`
-        );
-      });
-    }
-
-    this.newLine();
-    this.displayInfo(
-      this.colorize(
-        "To set a provider, use: set provider <provider_name>",
-        "dim"
-      )
-    );
-  }
-
-  displayGroupHeader(header: string): void {
-    this.displayInfo(this.colorize(`\n${header}:`, "magenta"));
-  }
-}
\ No newline at end of file
diff --git a/packages/qllm-cli/src/chat/message-handler.ts b/packages/qllm-cli/src/chat/message-handler.ts
index f7747fc..e435b00 100644
--- a/packages/qllm-cli/src/chat/message-handler.ts
+++ b/packages/qllm-cli/src/chat/message-handler.ts
@@ -10,7 +10,7 @@ import {
   TextContent,
 } from "qllm-lib";
 import { createSpinner } from "nanospinner";
-import { output } from "../utils/output";
+import { ioManager } from "../utils/io-manager";
 import { DEFAULT_PROVIDER, DEFAULT_MODEL } from "../constants";
 import { ConfigManager } from "./config-manager";
 
@@ -101,7 +101,7 @@ export class MessageHandler {
           spinner.stop();
           process.stdout.write("\r\x1b[K"); // Clear the entire line
           spinner.clear();
-          output.info("\nAssistant: ");
+          ioManager.displayInfo("\nAssistant: ");
         }
         if (chunk.text) {
           process.stdout.write(chunk.text);
diff --git a/packages/qllm-cli/src/chat/utils.ts b/packages/qllm-cli/src/chat/utils.ts
index 263cf1f..6a5af20 100644
--- a/packages/qllm-cli/src/chat/utils.ts
+++ b/packages/qllm-cli/src/chat/utils.ts
@@ -2,7 +2,7 @@
 import fs from "fs/promises";
 import path from "path";
 import { createSpinner } from "nanospinner";
-import { output } from "../utils/output";
+import { ioManager } from "../utils/io-manager";
 
 export const utils = {
   async readLocalFile(filePath: string): Promise<string> {
@@ -72,7 +72,7 @@ export const utils = {
         return await operation();
       } catch (error) {
         if (i === maxRetries - 1) throw error;
-        output.warn(`Operation failed, retrying in ${delayMs}ms...`);
+        ioManager.displayWarning(`Operation failed, retrying in ${delayMs}ms...`);
         await this.delay(delayMs);
       }
     }
diff --git a/packages/qllm-cli/src/commands/ask-command.ts b/packages/qllm-cli/src/commands/ask-command.ts
index 71e24c0..1cd086a 100644
--- a/packages/qllm-cli/src/commands/ask-command.ts
+++ b/packages/qllm-cli/src/commands/ask-command.ts
@@ -3,7 +3,6 @@ import path from "path";
 import { Command } from "commander";
 import { getLLMProvider, ChatMessage, LLMProvider } from "qllm-lib";
 import { createSpinner, Spinner } from "nanospinner";
-import kleur from "kleur";
 import {
   AskCommandOptions,
   AskCommandOptionsPartialSchema,
@@ -15,10 +14,10 @@ import {
   readImageFileAndConvertToBase64,
   isImageFile,
 } from "../utils/image-utils";
-import { output } from "../utils/output";
+import { ioManager } from "../utils/io-manager";
 import { processAndExit } from "../utils/common";
 import { validateOptions } from "../utils/validate-options";
-import { IOManager } from "../chat/io-manager";
+import { IOManager } from "../utils/io-manager";
 import { CliConfigManager } from "../utils/cli-config-manager";
 import { DEFAULT_PROVIDER, DEFAULT_MODEL } from "../constants";
 
@@ -37,7 +36,7 @@ const askCommandAction = async (
     );
   } catch (error) {
     if (error instanceof Error) {
-      output.error(
+      ioManager.displayError(
         `An error occurred while validating the options: ${error.message}`
       );
       process.exit(1);
@@ -84,20 +83,27 @@ const askCommandAction = async (
       usedOptions
     );
 
-    spinner.success({ text: kleur.green("response received successfully!") });
+    spinner.success({
+      text: ioManager.colorize("response received successfully!", "green"),
+    });
 
     if (options.output) {
       await saveResponseToFile(response, options.output);
-      output.success(`Response saved to ${options.output}`);
+      ioManager.displaySuccess(`Response saved to ${options.output}`);
     } else {
-      output.info("Response:");
+      ioManager.displayInfo("Response:");
       console.log(response);
     }
   } catch (error) {
     spinner.error({
-      text: kleur.red("An error occurred while processing your request."),
+      text: ioManager.colorize(
+        "An error occurred while processing your request.",
+        "red"
+      ),
     });
-    output.error(error instanceof Error ? error.message : String(error));
+    ioManager.displayError(
+      error instanceof Error ? error.message : String(error)
+    );
     process.exit(1);
   } finally {
   }
@@ -157,15 +163,14 @@ async function prepareImageInputs({
     try {
       const screenshotCapture = new ScreenshotCapture();
       await screenshotCapture.initialize();
-      const screenshotBase64 = await screenshotCapture.captureAndGetBase64(
-        screenshot
-      );
+      const screenshotBase64 =
+        await screenshotCapture.captureAndGetBase64(screenshot);
       images.push(screenshotBase64);
-      output.success(
+      ioManager.displaySuccess(
         `Screenshot captured successfully from display ${screenshot}`
       );
     } catch (error) {
-      output.error(
+      ioManager.displayError(
         `Failed to capture screenshot from display ${screenshot}: ${error}`
       );
     }
@@ -176,33 +181,33 @@ async function prepareImageInputs({
       if (await isImageFile(item)) {
         const base64Image = await readImageFileAndConvertToBase64(item);
         images.push(base64Image);
-        output.success(`Image loaded successfully: ${item}`);
+        ioManager.displaySuccess(`Image loaded successfully: ${item}`);
       } else {
         // Assume it's a URL
         images.push(item);
-        output.success(`Image URL added: ${item}`);
+        ioManager.displaySuccess(`Image URL added: ${item}`);
       }
     } catch (error) {
-      output.error(`Failed to process image input ${item}: ${error}`);
+      ioManager.displayError(`Failed to process image input ${item}: ${error}`);
     }
   }
 
   if (useClipboard) {
     try {
-      output.info("Checking clipboard for images...");
+      ioManager.displayInfo("Checking clipboard for images...");
       const clipboardImage = await Clipboard.getImageFromClipboard();
       if (clipboardImage) {
         images.push(clipboardImage);
-        output.success(
+        ioManager.displaySuccess(
           `Image found in clipboard, size ${formatBytes(
             clipboardImage.length
           )} bytes`
         );
       } else {
-        output.warn("No image found in clipboard.");
+        ioManager.displayWarning("No image found in clipboard.");
       }
     } catch (error) {
-      output.error(`Failed to get image from clipboard: ${error}`);
+      ioManager.displayError(`Failed to get image from clipboard: ${error}`);
     }
   }
 
diff --git a/packages/qllm-cli/src/commands/chat-command.ts b/packages/qllm-cli/src/commands/chat-command.ts
index d155256..6847837 100644
--- a/packages/qllm-cli/src/commands/chat-command.ts
+++ b/packages/qllm-cli/src/commands/chat-command.ts
@@ -4,13 +4,13 @@ import { Command } from "commander";
 import { getListProviderNames, getLLMProvider } from "qllm-lib";
 import { Chat } from "../chat/chat";
 import { chatConfig } from "../chat/chat-config";
-import { output } from "../utils/output";
+import { ioManager} from "../utils/io-manager";
 import { CliConfigManager } from "../utils/cli-config-manager";
 import {
   ChatCommandOptions,
   ChatCommandOptionsSchema,
 } from "../types/chat-command-options";
-import { IOManager } from "../chat/io-manager";
+import { IOManager } from "../utils/io-manager";
 import { validateOptions } from "../utils/validate-options";
 
 const chatAction = async (options: ChatCommandOptions) => {
@@ -28,7 +28,7 @@ const chatAction = async (options: ChatCommandOptions) => {
       );
     } catch (error) {
       if (error instanceof Error) {
-        output.error(
+       ioManager.displayError(
           `An error occurred while validating the options: ${error.message}`
         );
         process.exit(1);
@@ -46,13 +46,13 @@ const chatAction = async (options: ChatCommandOptions) => {
 
     const availableProviders = getListProviderNames();
     if (!availableProviders.includes(providerName)) {
-      output.warn(
+      ioManager.displayWarning(
         `Invalid provider "${providerName}". Available providers: ${availableProviders.join(
           ", "
         )}`
       );
-      output.info("Use the 'configure' command to set a valid provider.");
-      output.info("Use the '/providers' command to see available providers.");
+      ioManager.displayInfo("Use the 'configure' command to set a valid provider.");
+      ioManager.displayInfo("Use the '/providers' command to see available providers.");
     }
 
     chatConfig.set("maxTokens", validOptions.maxTokens);
@@ -66,20 +66,20 @@ const chatAction = async (options: ChatCommandOptions) => {
     const models = await provider.listModels();
 
     if (!models.some((m) => m.id === modelName)) {
-      output.warn(
+      ioManager.displayWarning(
         `Invalid model "${modelName}" for provider "${providerName}".`
       );
-      output.info("Available models:");
-      models.forEach((m) => output.info(`- ${m.id}`));
-      output.info("Use the 'configure' command to set a valid model.");
-      output.info("Use the '/models' command to see available models.");
+      ioManager.displayInfo("Available models:");
+      models.forEach((m) => ioManager.displayInfo(`- ${m.id}`));
+      ioManager.displayInfo("Use the 'configure' command to set a valid model.");
+      ioManager.displayInfo("Use the '/models' command to see available models.");
     }
 
     const chat = new Chat(providerName, modelName);
 
     await chat.start();
   } catch (error) {
-    output.error("An error occurred while starting the chat:");
+   ioManager.displayError("An error occurred while starting the chat:");
     console.error(error);
   }
 };
diff --git a/packages/qllm-cli/src/commands/configure-command.ts b/packages/qllm-cli/src/commands/configure-command.ts
index 5567588..172e6aa 100644
--- a/packages/qllm-cli/src/commands/configure-command.ts
+++ b/packages/qllm-cli/src/commands/configure-command.ts
@@ -2,7 +2,7 @@
 
 import { Command } from "commander";
 import { CliConfigManager } from "../utils/cli-config-manager";
-import { IOManager } from "../chat/io-manager";
+import { IOManager } from "../utils/io-manager";
 import { validateSingleOption } from "../utils/input-validator";
 import { Config } from "../types/config-types";
 
diff --git a/packages/qllm-cli/src/commands/list-command.ts b/packages/qllm-cli/src/commands/list-command.ts
index 7edde26..0d5f85d 100644
--- a/packages/qllm-cli/src/commands/list-command.ts
+++ b/packages/qllm-cli/src/commands/list-command.ts
@@ -5,9 +5,6 @@ import { createSpinner } from 'nanospinner';
 import { getListProviderNames, getLLMProvider, LLMProvider, Model } from 'qllm-lib';
 import { processAndExit } from '../utils/common';
 
-
-
-
 export const listCommand = new Command('list')
   .description('List providers and models')
   .addCommand(
diff --git a/packages/qllm-cli/src/commands/run-command.ts b/packages/qllm-cli/src/commands/run-command.ts
new file mode 100644
index 0000000..5c872d9
--- /dev/null
+++ b/packages/qllm-cli/src/commands/run-command.ts
@@ -0,0 +1,103 @@
+// packages/qllm-cli/src/commands/run-command.ts
+
+import { Command } from "commander";
+import { RunCommandOptions, RunCommandOptionsSchema } from "../types/run-command-options";
+import { TemplateExecutor, getLLMProvider } from "qllm-lib";
+import { loadTemplate, parseVariables } from "../utils/template-utils";
+import { promptForVariables } from "../utils/variable-utils";
+import { validateOptions } from "../utils/validate-options";
+import { IOManager } from "../utils/io-manager";
+import { CliConfigManager } from "../utils/cli-config-manager";
+import { DEFAULT_PROVIDER, DEFAULT_MODEL } from "../constants";
+
+const runAction = async (templateSource: string, options: RunCommandOptions) => {
+  const ioManager = new IOManager();
+  const cliConfig = CliConfigManager.getInstance();
+
+  try {
+    // Validate options using zod schema
+    const validOptions = await validateOptions(RunCommandOptionsSchema, options, ioManager);
+
+    const providerName = validOptions.provider || cliConfig.get("defaultProvider") || DEFAULT_PROVIDER;
+    const modelName = validOptions.model || cliConfig.get("defaultModel") || DEFAULT_MODEL;
+
+    const spinner = ioManager.createSpinner("Processing template...");
+    spinner.start();
+
+    try {
+      spinner.update({ text: "Loading template..." });
+      const template = await loadTemplate(templateSource, validOptions.type || "file");
+      spinner.update({ text: "" });
+      spinner.stop();
+
+      const variables = parseVariables(validOptions.variables);
+
+      const executor = new TemplateExecutor();
+      const provider = await getLLMProvider(providerName);
+      const result = await executor.execute({
+        template,
+        variables: { ...variables},
+        providerOptions: {
+          model: modelName,
+          maxTokens: validOptions.maxTokens,
+          temperature: validOptions.temperature,
+        },
+        provider,
+        stream: validOptions.stream,
+        onPromptForMissingVariables: async (template, initialVariables) => {
+          return promptForVariables(template, initialVariables);
+        },
+        onOutput: (event) => {
+          if (event.type === 'chunk' && validOptions.stream) {
+            process.stdout.write((event as any).chunk);
+          }
+        },
+      });
+
+      spinner.start();
+      spinner.success({ text: "Template executed successfully" });
+
+      if (validOptions.output) {
+        await saveResponseToFile(result.response, validOptions.output);
+        ioManager.displaySuccess(`Response saved to ${validOptions.output}`);
+      } else {
+        ioManager.displayInfo("Template Execution Result:");
+        console.log(result.response);
+      }
+
+      if (Object.keys(result.outputVariables).length > 0) {
+        ioManager.displayInfo("Extracted Output Variables:");
+        for (const [key, value] of Object.entries(result.outputVariables)) {
+          console.log(`${ioManager.colorize(key, "green")}:`);
+          console.log(`${ioManager.colorize(value, "yellow")}`);
+          console.log('-------------------------');
+        }
+      }
+    } catch (error) {
+      spinner.error({ text: `Error executing template: ${(error as Error).message}` });
+    }
+  } catch (error) {
+   ioManager.displayError(`An error occurred: ${(error as Error).message}`);
+  }
+};
+
+export const runCommand = new Command("run")
+  .description("Execute a template")
+  .argument("<template>", "Template name, file path, or URL")
+  .option("-t, --type <type>", "Template source type (file, url, inline)", "file")
+  .option("-v, --variables <variables>", "Template variables in JSON format")
+  .option("-p, --provider <provider>", "LLM provider to use")
+  .option("-m, --model <model>", "Specific model to use")
+  .option("--max-tokens <maxTokens>", "Maximum number of tokens to generate", parseInt)
+  .option("--temperature <temperature>", "Temperature for response generation", parseFloat)
+  .option("-s, --stream", "Stream the response")
+  .option("-o, --output <output>", "Output file for the response")
+  .action(runAction);
+
+async function saveResponseToFile(response: string, outputPath: string): Promise<void> {
+  try {
+    //await output.writeToFile(outputPath, response);
+  } catch (error) {
+    throw new Error(`Failed to save response to file: ${error}`);
+  }
+}
\ No newline at end of file
diff --git a/packages/qllm-cli/src/qllm.ts b/packages/qllm-cli/src/qllm.ts
index 636e624..a774b6e 100755
--- a/packages/qllm-cli/src/qllm.ts
+++ b/packages/qllm-cli/src/qllm.ts
@@ -6,6 +6,7 @@ import { listCommand } from "./commands/list-command";
 import { chatCommand } from "./commands/chat-command";
 import { CliConfigManager } from "./utils/cli-config-manager";
 import { configureCommand } from "./commands/configure-command";
+import { runCommand } from "./commands/run-command";
 
 const VERSION = "1.8.6";
 
@@ -40,6 +41,9 @@ export async function main() {
     // Add the configure command
     program.addCommand(configureCommand);
 
+    // Add to program commands
+    program.addCommand(runCommand);
+
     // Add other commands here as needed
     // For example:
     // program.addCommand(generateEmbeddingCommand);
diff --git a/packages/qllm-cli/src/types/run-command-options.ts b/packages/qllm-cli/src/types/run-command-options.ts
new file mode 100644
index 0000000..d35f09e
--- /dev/null
+++ b/packages/qllm-cli/src/types/run-command-options.ts
@@ -0,0 +1,14 @@
+import { z } from "zod";
+
+export const RunCommandOptionsSchema = z.object({
+  type: z.enum(["file", "url", "inline"]).default("file"),
+  variables: z.string().optional(),
+  provider: z.string().optional(),
+  model: z.string().optional(),
+  maxTokens: z.number().int().positive().optional(),
+  temperature: z.number().min(0).max(1).optional(),
+  stream: z.boolean().optional(),
+  output: z.string().optional(),
+});
+
+export type RunCommandOptions = z.infer<typeof RunCommandOptionsSchema>;
\ No newline at end of file
diff --git a/packages/qllm-cli/src/utils/io-manager/index.ts b/packages/qllm-cli/src/utils/io-manager/index.ts
new file mode 100644
index 0000000..9e7231b
--- /dev/null
+++ b/packages/qllm-cli/src/utils/io-manager/index.ts
@@ -0,0 +1,377 @@
+// packages/qllm-cli/src/utils/io-manager/io-manager.ts
+
+import readline from "readline";
+import kleur from "kleur";
+import { getBorderCharacters, table } from "table";
+import { createSpinner } from "nanospinner";
+import { Table } from "console-table-printer";
+import prompts from "prompts";
+import gradient from 'gradient-string';
+
+type ColorName = keyof typeof kleur;
+
+interface IOManagerConfig {
+  defaultColor: ColorName;
+  errorColor: ColorName;
+  successColor: ColorName;
+  warningColor: ColorName;
+  infoColor: ColorName;
+  highlightColor: ColorName;
+  dimColor: ColorName;
+  useColors: boolean;
+}
+
+interface Spinner {
+  start: () => void;
+  stop: () => void;
+  update: (options: { text: string }) => void;
+  success: (options: { text: string }) => void;
+  error: (options: { text: string }) => void;
+}
+
+class DisplayManager {
+  constructor(private config: IOManagerConfig) {}
+
+  colorize(text: string, color: ColorName): string {
+    if (!this.config.useColors) return text;
+    if (typeof kleur[color] === "function") {
+      return (kleur[color] as (...text: string[]) => string)(text);
+    }
+    return text; // Fallback to uncolored text if the color function doesn't exist
+  }
+
+  error(message: string): void {
+    console.error(this.colorize(`✖ ${message}`, this.config.errorColor));
+  }
+
+  success(message: string): void {
+    console.log(this.colorize(`✔ ${message}`, this.config.successColor));
+  }
+
+  warning(message: string): void {
+    console.warn(this.colorize(`⚠ ${message}`, this.config.warningColor));
+  }
+
+  info(message: string): void {
+    console.log(this.colorize(message, this.config.infoColor));
+  }
+
+  table(headers: string[], data: string[][]): void {
+    const tableData = [headers.map((h) => this.colorize(h, "cyan")), ...data];
+    console.log(
+      table(tableData, {
+        border: getBorderCharacters("norc"),
+        columnDefault: {
+          paddingLeft: 0,
+          paddingRight: 1,
+        },
+        drawHorizontalLine: () => false,
+      })
+    );
+  }
+
+  list(items: string[]): void {
+    items.forEach((item) => this.info(`• ${item}`));
+  }
+
+  title(text: string): void {
+    console.log(gradient.pastel.multiline(text));
+  }
+
+  codeBlock(code: string, language?: string): void {
+    const formattedCode = language ? this.colorize(code, "cyan") : code;
+    console.log(this.colorize("```" + (language || ""), this.config.dimColor));
+    console.log(formattedCode);
+    console.log(this.colorize("```", this.config.dimColor));
+  }
+
+  sectionHeader(header: string): void {
+    console.log(kleur.bold().underline().yellow(`\n${header}`));
+  }
+
+  json(data: unknown): void {
+    console.log(JSON.stringify(data, null, 2));
+  }
+}
+
+class InputManager {
+  private rl: readline.Interface;
+
+  constructor() {
+    this.rl = readline.createInterface({
+      input: process.stdin,
+      output: process.stdout,
+    });
+  }
+
+  async getUserInput(prompt: string): Promise<string> {
+    const response = await prompts({
+      type: "text",
+      name: "userInput",
+      message: prompt,
+    });
+    return response.userInput;
+  }
+
+  async confirm(message: string): Promise<boolean> {
+    const response = await prompts({
+      type: "confirm",
+      name: "confirmed",
+      message: message,
+      initial: false,
+    });
+    return response.confirmed;
+  }
+
+  close(): void {
+    this.rl.close();
+  }
+}
+
+class SpinnerManager {
+  private currentSpinner: Spinner | null = null;
+
+  create(message: string): Spinner {
+    this.currentSpinner = createSpinner(message);
+    return this.currentSpinner;
+  }
+
+  stop(): void {
+    if (this.currentSpinner) {
+      this.currentSpinner.stop();
+      this.currentSpinner = null;
+    }
+  }
+}
+
+export class IOManager {
+  private display: DisplayManager;
+  private input: InputManager;
+  private spinner: SpinnerManager;
+
+  constructor(config: Partial<IOManagerConfig> = {}) {
+    const fullConfig: IOManagerConfig = {
+      defaultColor: "white",
+      errorColor: "red",
+      successColor: "green",
+      warningColor: "yellow",
+      infoColor: "blue",
+      highlightColor: "magenta",
+      dimColor: "gray",
+      useColors: true,
+      ...config,
+    };
+    this.display = new DisplayManager(fullConfig);
+    this.input = new InputManager();
+    this.spinner = new SpinnerManager();
+  }
+
+  displayGroupHeader(header: string): void {
+    this.display.sectionHeader(header);
+  }
+
+  colorize(text: string, color: ColorName): string {
+    return this.display.colorize(text, color);
+  }
+
+  // Display methods
+  displayError(message: string): void {
+    this.display.error(message);
+  }
+
+  displaySuccess(message: string): void {
+    this.display.success(message);
+  }
+
+  displayWarning(message: string): void {
+    this.display.warning(message);
+  }
+
+  displayInfo(message: string): void {
+    this.display.info(message);
+  }
+
+  displayTable(headers: string[], data: string[][]): void {
+    this.display.table(headers, data);
+  }
+
+  displayList(items: string[]): void {
+    this.display.list(items);
+  }
+
+  displayTitle(title: string): void {
+    this.display.title(title);
+  }
+
+  displayCodeBlock(code: string, language?: string): void {
+    this.display.codeBlock(code, language);
+  }
+
+  displaySectionHeader(header: string): void {
+    this.display.sectionHeader(header);
+  }
+
+  json(data: unknown): void {
+    this.display.json(data);
+  }
+
+  // Input methods
+  async getUserInput(prompt: string): Promise<string> {
+    return this.input.getUserInput(this.display.colorize(prompt, "green"));
+  }
+
+  async confirmAction(message: string): Promise<boolean> {
+    return this.input.confirm(message);
+  }
+
+  // Spinner methods
+  createSpinner(message: string): Spinner {
+    return this.spinner.create(message);
+  }
+
+  stopSpinner(): void {
+    this.spinner.stop();
+  }
+
+  // Utility methods
+  clearLine(): void {
+    process.stdout.write("\r\x1b[K");
+  }
+
+  newLine(): void {
+    console.log();
+  }
+
+  clear(): void {
+    console.clear();
+  }
+
+  write(text: string): void {
+    process.stdout.write(text);
+  }
+
+  close(): void {
+    this.input.close();
+  }
+
+  // Specialized display methods
+  displayUserMessage(message: string): void {
+    this.displayInfo(this.display.colorize(`You: ${message}`, "green"));
+  }
+
+  displayAssistantMessage(message: string): void {
+    this.displayInfo(this.display.colorize(`Assistant: ${message}`, "blue"));
+  }
+
+  displaySystemMessage(message: string): void {
+    this.displayInfo(this.display.colorize(`System: ${message}`, "yellow"));
+  }
+
+  displayConversationList(
+    conversations: { id: string; createdAt: Date }[]
+  ): void {
+    this.displayInfo("Conversations:");
+    conversations.forEach((conv, index) => {
+      this.displayInfo(
+        `${index + 1}. ID: ${conv.id}, Created: ${conv.createdAt.toLocaleString()}`
+      );
+    });
+  }
+
+  displayConversationDetails(
+    id: string,
+    messages: { role: string; content: string }[]
+  ): void {
+    this.displayInfo(`Conversation ${id}:`);
+    messages.forEach((msg, index) => {
+      const roleColor = msg.role === "user" ? "green" : "blue";
+      this.displayInfo(
+        `${index + 1}. ${this.display.colorize(msg.role, roleColor)}: ${msg.content}`
+      );
+    });
+  }
+
+  displayGroupedInfo(title: string, items: string[]): void {
+    this.displayInfo(this.display.colorize(`\n${title}:`, "yellow"));
+    items.forEach((item) => this.displayInfo(` ${item}`));
+  }
+
+  displayModelTable(models: { id: string; description: string }[]): void {
+    const p = new Table({
+      columns: [
+        { name: "id", alignment: "left", color: "cyan" },
+        { name: "description", alignment: "left", color: "white" },
+      ],
+      sort: (row1, row2) => row1.id.localeCompare(row2.id),
+    });
+    models.forEach((model) => {
+      p.addRow({ id: model.id, description: model.description || "N/A" });
+    });
+    p.printTable();
+  }
+
+  displayConfigOptions(options: Array<{ name: string; value: unknown }>): void {
+    this.displaySectionHeader("Current Configuration");
+    const longestNameLength = Math.max(
+      ...options.map((opt) => opt.name.length)
+    );
+    options.forEach(({ name, value }) => {
+      const paddedName = name.padEnd(longestNameLength);
+      const formattedValue = value !== undefined ? String(value) : "Not set";
+      const coloredValue =
+        value !== undefined
+          ? this.display.colorize(formattedValue, "green")
+          : this.display.colorize(formattedValue, "yellow");
+      this.displayInfo(
+        `${this.display.colorize(paddedName, "cyan")}: ${coloredValue}`
+      );
+    });
+    this.newLine();
+    this.displayInfo(
+      this.display.colorize("Use '/set <option> <value>' to change a setting", "dim")
+    );
+    this.displayInfo(
+      this.display.colorize("Example: set provider openai", "dim")
+    );
+  }
+
+  displayProviderList(providers: string[]): void {
+    this.displaySectionHeader("Available Providers");
+    if (providers.length === 0) {
+      this.displayInfo(
+        this.display.colorize("No providers available.", "yellow")
+      );
+    } else {
+      providers.forEach((provider, index) => {
+        this.displayInfo(
+          `${this.display.colorize(`${index + 1}.`, "cyan")} ${this.display.colorize(provider, "green")}`
+        );
+      });
+    }
+    this.newLine();
+    this.displayInfo(
+      this.display.colorize(
+        "To set a provider, use: set provider <provider_name>",
+        "dim"
+      )
+    );
+  }
+
+  // Async utility method
+  async safeExecute<T>(
+    fn: () => Promise<T>,
+    errorMessage: string
+  ): Promise<T | undefined> {
+    try {
+      return await fn();
+    } catch (error) {
+      this.displayError(
+        `${errorMessage}: ${error instanceof Error ? error.message : String(error)}`
+      );
+      return undefined;
+    }
+  }
+}
+
+// Create a singleton instance for easy access
+export const ioManager = new IOManager();
\ No newline at end of file
diff --git a/packages/qllm-cli/src/utils/output.ts b/packages/qllm-cli/src/utils/output.ts
deleted file mode 100644
index 8ba3dc4..0000000
--- a/packages/qllm-cli/src/utils/output.ts
+++ /dev/null
@@ -1,132 +0,0 @@
-import kleur from 'kleur';
-import { table, TableUserConfig } from 'table';
-
-/**
- * OutputFormatter class provides methods for formatting and displaying output in the CLI.
- */
-export class OutputFormatter {
-  /**
-   * Formats and prints a success message.
-   * @param message The success message to display.
-   */
-  static success(message: string): void {
-    console.error(kleur.green('✔ ') + message);
-  }
-
-  /**
-   * Formats and prints an error message.
-   * @param message The error message to display.
-   */
-  static error(message: string): void {
-    console.error(kleur.red('✖ ') + message);
-  }
-
-  /**
-   * Formats and prints a warning message.
-   * @param message The warning message to display.
-   */
-  static warn(message: string): void {
-    console.warn(kleur.yellow('⚠ ') + message);
-  }
-
-  /**
-   * Formats and prints an info message.
-   * @param message The info message to display.
-   */
-  static info(message: string): void {
-    console.error(kleur.blue('ℹ ') + message);
-  }
-
-  /**
-   * Formats and prints a table of data.
-   * @param headers An array of table headers.
-   * @param data An array of arrays representing table rows.
-   */
-  static table(headers: string[], data: string[][]): void {
-    const tableData = [headers, ...data];
-    const config: TableUserConfig = {
-      columns: headers.map(() => ({ alignment: 'left' as const })),
-      header: {
-        alignment: 'center' as const,
-        content: headers.map(header => kleur.bold(header)).join(' | '), // Join the headers
-      },
-    };
-    console.log(table(tableData, config));
-  }
-
-  /**
-   * Formats and prints JSON data.
-   * @param data The data to be formatted as JSON.
-   */
-  static json(data: any): void {
-    console.log(JSON.stringify(data, null, 2));
-  }
-
-  /**
-   * Formats and prints a title.
-   * @param title The title to display.
-   */
-  static title(title: string): void {
-    console.log(kleur.bold().underline(title));
-  }
-
-  /**
-   * Formats and prints a section header.
-   * @param header The section header to display.
-   */
-  static sectionHeader(header: string): void {
-    console.log(kleur.bold('\n' + header));
-  }
-
-  /**
-   * Formats and prints a list of items.
-   * @param items An array of items to display as a list.
-   */
-  static list(items: string[]): void {
-    items.forEach((item) => console.log(`  • ${item}`));
-  }
-
-  /**
-   * Clears the console screen.
-   */
-  static clear(): void {
-    console.clear();
-  }
-
-  /**
-   * Prints a blank line.
-   */
-  static newLine(): void {
-    console.log();
-  }
-
-  /**
-   * Formats and prints a code block.
-   * @param code The code to display.
-   * @param language Optional language for syntax highlighting.
-   */
-  static codeBlock(code: string, language?: string): void {
-    const formattedCode = language ? kleur.cyan(code) : code;
-    console.log(kleur.gray('```' + (language || '')));
-    console.log(formattedCode);
-    console.log(kleur.gray('```'));
-  }
-}
-
-/**
- * A simplified output utility object for quick access to common output methods.
- */
-export const output = {
-  success: OutputFormatter.success,
-  error: OutputFormatter.error,
-  warn: OutputFormatter.warn,
-  info: OutputFormatter.info,
-  table: OutputFormatter.table,
-  json: OutputFormatter.json,
-  title: OutputFormatter.title,
-  sectionHeader: OutputFormatter.sectionHeader,
-  list: OutputFormatter.list,
-  clear: OutputFormatter.clear,
-  newLine: OutputFormatter.newLine,
-  codeBlock: OutputFormatter.codeBlock,
-};
\ No newline at end of file
diff --git a/packages/qllm-cli/src/utils/template-utils.ts b/packages/qllm-cli/src/utils/template-utils.ts
new file mode 100644
index 0000000..9a39659
--- /dev/null
+++ b/packages/qllm-cli/src/utils/template-utils.ts
@@ -0,0 +1,24 @@
+import { Template, TemplateManager } from "qllm-lib";
+import axios from "axios";
+
+export async function loadTemplate(source: string, type: "file" | "url" | "inline"): Promise<Template> {
+  switch (type) {
+    case "file":
+      return Template.fromPath(source);
+    case "url":
+      return Template.fromUrl(source);
+    case "inline":
+      return Template.fromYaml(source);
+    default:
+      throw new Error(`Invalid template source type: ${type}`);
+  }
+}
+
+export function parseVariables(variablesString?: string): Record<string, any> {
+  if (!variablesString) return {};
+  try {
+    return JSON.parse(variablesString);
+  } catch (error) {
+    throw new Error("Invalid JSON format for variables");
+  }
+}
\ No newline at end of file
diff --git a/packages/qllm-cli/src/utils/validate-options.ts b/packages/qllm-cli/src/utils/validate-options.ts
index 47991e1..3855446 100644
--- a/packages/qllm-cli/src/utils/validate-options.ts
+++ b/packages/qllm-cli/src/utils/validate-options.ts
@@ -1,5 +1,5 @@
 import { z } from "zod";
-import { IOManager } from "../chat/io-manager";
+import { IOManager } from "./io-manager";
 
 export async function validateOptions<T extends z.ZodType>(
   schema: T,
diff --git a/packages/qllm-cli/src/utils/variable-utils.ts b/packages/qllm-cli/src/utils/variable-utils.ts
new file mode 100644
index 0000000..34f3a1d
--- /dev/null
+++ b/packages/qllm-cli/src/utils/variable-utils.ts
@@ -0,0 +1,98 @@
+import { TemplateDefinition } from "qllm-lib";
+import prompts from "prompts";
+import kleur from "kleur";
+
+export async function promptForVariables(
+  template: TemplateDefinition,
+  providedVariables: Record<string, any>
+): Promise<Record<string, any>> {
+  const missingVariables: Record<string, any> = {};
+
+  console.log(kleur.cyan().bold('\n🔍 Collecting input variables\n'));
+
+  try {
+    for (const [key, variable] of Object.entries(template.input_variables)) {
+      if (!(key in providedVariables)) {
+        console.log(kleur.cyan(`📝 Input required for: ${kleur.bold(key)}`));
+        
+        const response = await prompts({
+          type: getPromptType(variable) as any,
+          name: "value",
+          message: formatPromptMessage(variable),
+          initial: variable.default,
+          validate: (value) => validateInput(value, variable),
+          choices: getChoices(variable),
+        });
+
+        if (response.value === undefined) {
+          throw new Error("User cancelled the input");
+        }
+
+        missingVariables[key] = castValue(response.value, variable.type);
+        console.log(kleur.green(`✅ Value for ${key} collected successfully\n`));
+      }
+    }
+
+    console.log(kleur.green().bold('✅ All variables collected successfully'));
+    return missingVariables;
+  } catch (error: unknown) {
+    if (error instanceof Error) {
+      console.error(kleur.red(`❌ Error: ${error.message}`));
+    } else {
+      console.error(kleur.red('❌ An unknown error occurred'));
+    }
+    throw error;
+  }
+}
+
+function getPromptType(variable: any): string {
+  if (variable.choices) return 'select';
+  switch (variable.type) {
+    case 'number':
+      return 'number';
+    case 'boolean':
+      return 'confirm';
+    case 'array':
+      return 'list';
+    default:
+      return 'text';
+  }
+}
+
+function formatPromptMessage(variable: any): string {
+  let message = variable.description;
+  if (variable.default !== undefined) {
+    message += kleur.gray(` (Default: ${variable.default})`);
+  }
+  return message;
+}
+
+function validateInput(value: any, variable: any): boolean | string {
+  if (value === '' && !('default' in variable)) {
+    return kleur.red('This field is required');
+  }
+  return true;
+}
+
+function castValue(value: any, type: string): any {
+  switch (type) {
+    case 'number':
+      return Number(value);
+    case 'boolean':
+      return Boolean(value);
+    case 'array':
+      return Array.isArray(value) ? value : value.split(',').map((item: string) => item.trim());
+    default:
+      return value;
+  }
+}
+
+function getChoices(variable: any): { title: string; value: any }[] | undefined {
+  if (!variable.choices) return undefined;
+  return variable.choices.map((choice: string | { title: string; value: any }) => {
+    if (typeof choice === 'string') {
+      return { title: choice, value: choice };
+    }
+    return choice;
+  });
+}
\ No newline at end of file
diff --git a/packages/qllm-lib/package.json b/packages/qllm-lib/package.json
index d34d27b..c1be6a3 100644
--- a/packages/qllm-lib/package.json
+++ b/packages/qllm-lib/package.json
@@ -1,6 +1,6 @@
 {
   "name": "qllm-lib",
-  "version": "1.8.5",
+  "version": "1.8.7",
   "description": "Core library providing robust AI engineering functionalities tailored for Large Language Model (LLM) applications, enabling developers to build, deploy, and optimize AI solutions with ease.",
   "keywords": [
     "ai",
diff --git a/packages/qllm-lib/prompts/create_story.yaml b/packages/qllm-lib/prompts/create_story.yaml
index 1430d03..ec6e9c9 100644
--- a/packages/qllm-lib/prompts/create_story.yaml
+++ b/packages/qllm-lib/prompts/create_story.yaml
@@ -28,5 +28,18 @@ output_variables:
     type: string
     description: The story
 
-content: >
-  {{file:./story.md}}
+content: |
+  Write a story about {{subject}} as {{role}} in {{lang}} with a max length of {{max_length}}.
+
+  1. First, find funny and subversive ideas formatted as a table.
+  2. Select the best 3 ideas in ideas
+  3. Create the outline of the story in outline tags.
+  4. Write the full story in <story> tags.
+
+  Format markdown:
+
+  <ideas>ideas (markdown format)</ideas> 
+  <outline>outline (markdown format)</outline>
+  <story> The story (markdown format)</story>
+
+  END.
\ No newline at end of file
diff --git a/packages/qllm-lib/prompts/story.md b/packages/qllm-lib/prompts/story.md
new file mode 100644
index 0000000..2ae7afc
--- /dev/null
+++ b/packages/qllm-lib/prompts/story.md
@@ -0,0 +1,14 @@
+
+Write a story aboout {{subject}} as {{role}} in {{lang}} max length.
+
+1 - First find funny and subversive ideas format as a table
+2 - Select the best 3 ideas
+3 - Create the outline of the story 
+4 - Write the full story
+
+Format markdown:
+
+<ideas/>
+<outline/>
+<story/>
+
diff --git a/packages/qllm-lib/src/samples/demo-prompt-variable-no-stream.ts b/packages/qllm-lib/src/samples/demo-prompt-variable-no-stream.ts
new file mode 100644
index 0000000..67837c3
--- /dev/null
+++ b/packages/qllm-lib/src/samples/demo-prompt-variable-no-stream.ts
@@ -0,0 +1,131 @@
+import path from "path";
+import { getLLMProvider } from "../providers";
+import { LLMProvider, ChatMessage } from "../types";
+import { Template } from "../templates/template";
+import { TemplateExecutor } from "../templates";
+import { OutputEvent, OutputEventType } from "../templates/types";
+import readline from 'readline';
+
+const runLLMTests = async () => {
+  console.log('🚀 Starting LLM Prompts Tests');
+
+  const awsAnthropicModels = {
+    embeddingModelName: '',
+    visionModelName: 'anthropic.claude-3-haiku-20240307-v1:0',
+    toolModelName: 'anthropic.claude-3-haiku-20240307-v1:0',
+    textModelName: 'anthropic.claude-3-haiku-20240307-v1:0',
+  };
+
+  const groqModels = {
+    embeddingModelName: 'nomic-embed-text',
+    visionModelName: 'llama-3.1-70b-versatile',
+    toolModelName: 'llama3-groq-70b-8192-tool-use-preview',
+    textModelName: 'llama-3.1-70b-versatile',
+  };
+
+  const ollamaModels = {
+    embeddingModelName: 'nomic-embed-text:latest',
+    visionModelName: 'llava:latest',
+    toolModelName: 'mistral:latest',
+    textModelName: 'gemma2:2b',
+  };
+
+  const openaiModels = {
+    embeddingModelName: 'text-embedding-3-small',
+    visionModelName: 'gpt-4o-mini',
+    toolModelName: 'gpt-4o-mini',
+    textModelName: 'gpt-4o-mini',
+  };
+
+  await testLLMModel('aws-anthropic', { maxTokens: 1024 }, awsAnthropicModels);
+  await testLLMModel('groq', { maxTokens: 1024 }, groqModels);
+  await testLLMModel('ollama', { maxTokens: 1024 }, ollamaModels);
+  await testLLMModel('openai', { maxTokens: 1024 }, openaiModels);
+
+  console.log('✅ LLM Tests completed');
+};
+
+const testLLMModel = async (
+  providerName: string,
+  options: { maxTokens: number },
+  models: {
+    embeddingModelName: string;
+    visionModelName: string;
+    toolModelName: string;
+    textModelName: string;
+  },
+) => {
+  console.log(`🧪 Testing LLM model with provider: ${providerName}`);
+
+  const provider = await getLLMProvider(providerName);
+  console.log(`🔧 ${providerName}Provider instance created`);
+
+  await testCompletion(provider, { model: models.textModelName, maxTokens: options.maxTokens });
+
+  console.log(`✅ LLM model test completed for ${providerName}`);
+};
+
+async function testCompletion(
+  provider: LLMProvider,
+  options: { model: string; maxTokens: number },
+) {
+  const filePath = path.join(__dirname, './prompts/chain_of_tought_leader.yaml');
+  const template = await Template.fromPath(filePath);
+
+  console.log('📝 Template definition:');
+  console.dir(template.input_variables, { depth: null });
+
+  const templateExecutor = new TemplateExecutor();
+  const { response, outputVariables } = await templateExecutor.execute({
+    template: template,
+    variables: {},
+    provider: provider,
+    providerOptions: { model: options.model, maxTokens: options.maxTokens },
+    spinner: undefined,
+    stream: false,
+    onPromptForMissingVariables: async (template, initialVariables) => {
+      console.log('🔤 onPromptFromMissingVariables');
+      console.dir(template.input_variables, { depth: null });
+
+      const rl = readline.createInterface({
+        input: process.stdin,
+        output: process.stdout
+      });
+
+      const promptForVariable = (variableName: string, variableDetails: any) => {
+        return new Promise<string>((resolve) => {
+          rl.question(`Please enter a value for ${variableName} (${variableDetails.description}): `, (answer) => {
+            resolve(answer);
+          });
+        });
+      };
+
+      const resolvedVariables = { ...initialVariables };
+      for (const [key, value] of Object.entries(template.input_variables)) {
+        if (!(key in resolvedVariables)) {
+          resolvedVariables[key] = await promptForVariable(key, value);
+        }
+      }
+
+      rl.close();
+      return resolvedVariables;
+    },
+    onOutput: (output: OutputEvent) => {
+      if (output.type === OutputEventType.COMPLETE) {
+        console.log('📝 Completion result:', (output as any).response);
+      } else if (output.type === OutputEventType.CHUNK) {
+        process.stdout.write((output as any).chunk);
+      } else if (output.type === OutputEventType.ERROR) {
+        console.error('❌ Error during completion:', (output as any).error);
+      }
+    }
+  });
+
+  console.log('📝 Template execution result:', response);
+  console.log('📝 Output variables:', outputVariables);
+
+}
+
+runLLMTests()
+  .then(() => console.log('🎉 All LLM Tests executed successfully'))
+  .catch((error) => console.error('❌ Error during LLM tests execution:', error));
\ No newline at end of file
diff --git a/packages/qllm-lib/src/templates/template-executor.ts b/packages/qllm-lib/src/templates/template-executor.ts
index 25c11ac..4f808d4 100644
--- a/packages/qllm-lib/src/templates/template-executor.ts
+++ b/packages/qllm-lib/src/templates/template-executor.ts
@@ -11,12 +11,9 @@ import {
   StartOutputEvent,
   StopOutputEvent,
 } from './types';
-import { ChatMessage } from '../types';
+import { ChatMessage, LLMProvider } from '../types';
+
 
-type VariableResolver = (
-  template: TemplateDefinition,
-  initialVariables: Record<string, any>,
-) => Promise<Record<string, any>>;
 
 export class TemplateExecutor extends EventEmitter {
   constructor() {
@@ -114,7 +111,7 @@ export class TemplateExecutor extends EventEmitter {
   }
 
   private async generateResponse(
-    provider: any,
+    provider: LLMProvider,
     messages: ChatMessage[],
     providerOptions: any,
     stream: boolean,
@@ -127,14 +124,15 @@ export class TemplateExecutor extends EventEmitter {
   }
 
   private async handleNonStreamingResponse(
-    provider: any,
+    provider: LLMProvider,
     messages: ChatMessage[],
     providerOptions: any,
     onOutput?: (event: OutputEvent) => void,
   ): Promise<string> {
-    const response = await provider.generateMessage(messages, providerOptions);
-    onOutput?.(new CompleteOutputEvent(response));
-    return response;
+    const response = await provider.generateChatCompletion({ messages, options: providerOptions });
+    const textResponse = response.text || '';
+    onOutput?.(new CompleteOutputEvent(textResponse));
+    return textResponse;
   }
 
   private async handleStreamingResponse(
diff --git a/packages/qllm-lib/src/templates/template.ts b/packages/qllm-lib/src/templates/template.ts
index 855ffb2..63fc4da 100644
--- a/packages/qllm-lib/src/templates/template.ts
+++ b/packages/qllm-lib/src/templates/template.ts
@@ -97,7 +97,10 @@ export class Template implements TemplateDefinition {
         case 'string':
           return value;
         default:
-          ErrorManager.throw(InputValidationError, `Unknown variable type '${variableType}' for variable '${key}'`);
+          ErrorManager.throw(
+            InputValidationError,
+            `Unknown variable type '${variableType}' for variable '${key}'`,
+          );
       }
     }
     return value;
@@ -106,7 +109,10 @@ export class Template implements TemplateDefinition {
   private castToNumber(value: string, key: string): number {
     const numberValue = Number(value);
     if (isNaN(numberValue)) {
-      ErrorManager.throw(InputValidationError, `Failed to cast '${value}' to number for variable '${key}'`);
+      ErrorManager.throw(
+        InputValidationError,
+        `Failed to cast '${value}' to number for variable '${key}'`,
+      );
     }
     return numberValue;
   }
@@ -114,7 +120,10 @@ export class Template implements TemplateDefinition {
   private castToBoolean(value: string, key: string): boolean {
     const lowerValue = value.toLowerCase();
     if (lowerValue !== 'true' && lowerValue !== 'false') {
-      ErrorManager.throw(InputValidationError, `Failed to cast '${value}' to boolean for variable '${key}'. Use 'true' or 'false'`);
+      ErrorManager.throw(
+        InputValidationError,
+        `Failed to cast '${value}' to boolean for variable '${key}'. Use 'true' or 'false'`,
+      );
     }
     return lowerValue === 'true';
   }
@@ -132,25 +141,54 @@ export class Template implements TemplateDefinition {
       ...this,
     };
   }
-  private extractVariablesFromContent(): void {
-    const variablePattern = /{{([^{}]+)}}/g;
+  private extractVariablesFromContent(
+    options: {
+      allowDotNotation?: boolean;
+      allowBracketNotation?: boolean;
+      allowFunctionCalls?: boolean;
+    } = {},
+  ): void {
+    const {
+      allowDotNotation = true,
+      allowBracketNotation = false,
+      allowFunctionCalls = false,
+    } = options;
+
+    // Build the regex pattern based on options
+    const variableNamePattern = '[a-zA-Z_$][\\w$]*';
+    const dotNotationPattern = allowDotNotation ? `(?:\\.${variableNamePattern})*` : '';
+    const bracketNotationPattern = allowBracketNotation
+      ? '(?:\\[(?:[^\\[\\]]*|\\[[^\\[\\]]*\\])*\\])*'
+      : '';
+    const functionCallPattern = allowFunctionCalls ? '(?:\\([^()]*\\))?' : '';
+
+    const variablePattern = new RegExp(
+      `{{\\s*(${variableNamePattern}${dotNotationPattern}${bracketNotationPattern}${functionCallPattern})\\s*}}`,
+      'g',
+    );
+
     const uniqueVariables = new Set<string>();
     let match;
-  
-    while ((match = variablePattern.exec(this.content)) !== null) {
-      const variableExpression = match[1].trim();
-      const rootVariable = variableExpression.split('.')[0].split('[')[0].split('(')[0].trim();
-      uniqueVariables.add(rootVariable);
-    }
-  
-    uniqueVariables.forEach(variable => {
-      if (!this.input_variables[variable]) {
-        this.input_variables[variable] = {
-          type: 'string',
-          description: `Variable ${variable} found in content`,
-          inferred: true
-        };
+
+    try {
+      while ((match = variablePattern.exec(this.content)) !== null) {
+        const variableExpression = match[1].trim();
+        const rootVariable = variableExpression.split(/[.[(]/)[0];
+        uniqueVariables.add(rootVariable);
       }
-    });
+
+      uniqueVariables.forEach((variable) => {
+        if (!this.input_variables[variable]) {
+          this.input_variables[variable] = {
+            type: 'string',
+            description: `Variable ${variable} found in content`,
+            inferred: true,
+          };
+        }
+      });
+    } catch (error) {
+      console.error('Error extracting variables:', error);
+      // Optionally, you could throw the error or handle it in a way that fits your application's needs
+    }
   }
-}
\ No newline at end of file
+}