Feat/run template (#26)

* feat/template-utils-and-variable-utils(packages/qllm-cli/src/utils): add new utility modules

* feat/run-template(packages/qllm-cli/src): add run command to execute templates

* feat(packages/qllm-cli): update dependencies and package version

* feat/run-template(packages/qllm-cli/src/utils): improve variable prompt experience

* feat/run-template(packages/qllm-cli/src/commands): add spinner update and start for template execution

* update

* update

* feat(packages/qllm-cli): add mime-types types and improve io-manager

The commit message should be:

feat(packages/qllm-cli): add mime-types types and improve io-manager

* feat(packages/qllm-cli/src/utils/io-manager): Refactor IOManager class with DisplayManager, InputManager, and SpinnerManager

* update

* feat(packages/qllm-cli): update dependencies and add configuration management

* feat(packages/qllm-lib): update package.json version and template variable extraction

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",

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,79 +62,81 @@ 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 {
     return this.config.provider;
   }
 
   public setProvider(provider: string): void {
-    this.config.provider = provider;
+    this.set("provider", provider);
   }
 
   public getModel(): string | undefined {
     return this.config.model;
   }
 
   public setModel(model: string): void {
-    this.config.model = model;
+    this.set("model", model);
   }
 
   public getTemperature(): number | undefined {
     return this.config.temperature;
   }
 
   public setTemperature(temperature: number): void {
-    this.config.temperature = temperature;
+    this.set("temperature", temperature);
   }
 
   public getMaxTokens(): number | undefined {
     return this.config.maxTokens;
   }
 
   public setMaxTokens(maxTokens: number): void {
-    this.config.maxTokens = maxTokens;
+    this.set("maxTokens", maxTokens);
   }
 
   public getTopP(): number | undefined {
     return this.config.topP;
   }
 
   public setTopP(topP: number): void {
-    this.config.topP = topP;
+    this.set("topP", topP);
   }
 
   public getFrequencyPenalty(): number | undefined {
     return this.config.frequencyPenalty;
   }
 
   public setFrequencyPenalty(frequencyPenalty: number): void {
-    this.config.frequencyPenalty = frequencyPenalty;
+    this.set("frequencyPenalty", frequencyPenalty);
   }
 
   public getPresencePenalty(): number | undefined {
     return this.config.presencePenalty;
   }
 
   public setPresencePenalty(presencePenalty: number): void {
-    this.config.presencePenalty = presencePenalty;
+    this.set("presencePenalty", presencePenalty);
   }
 
   public getStopSequence(): string[] | undefined {
     return this.config.stopSequence;
   }
 
   public setStopSequence(stopSequence: string[]): void {
-    this.config.stopSequence = stopSequence;
+    this.set("stopSequence", stopSequence);
   }
 
   public getCurrentConversationId(): string | undefined {
     return this.config.currentConversationId;
   }
 
   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();

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();

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";

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;

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,15 +80,15 @@ export class ConfigManager {
   }
 
   private showValidOptions(): void {
-    output.info("Valid options are:");
+    ioManager.displayInfo("Valid options are:");
     [
       "temperature",
       "max_tokens",
       "top_p",
       "frequency_penalty",
       "presence_penalty",
       "stop_sequence",
-    ].forEach((opt) => output.info(`- ${opt}`));
+    ].forEach((opt) => ioManager.displayInfo(`- ${opt}`));
   }
 
   async initialize(): Promise<void> {

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)}`);
     });
   }
 }