@superatomai/sdk-node 0.0.2 → 0.0.4

package/dist/index.mjs

@@ -1866,55 +1866,126 @@ import fs3 from "fs";
 import path3 from "path";
 var PromptLoader = class {
   constructor(config) {
+    this.promptCache = /* @__PURE__ */ new Map();
+    this.isInitialized = false;
     logger.debug("Initializing PromptLoader...", process.cwd());
     this.promptsDir = config?.promptsDir || path3.join(process.cwd(), ".prompts");
+    this.defaultPromptsDir = path3.join(__dirname, "..", "..", ".prompts");
   }
   /**
-   * Load a single prompt file and replace variables using {{VARIABLE_NAME}} pattern
+   * Initialize and cache all prompts into memory
+   * This should be called once at SDK startup
+   */
+  async initialize() {
+    if (this.isInitialized) {
+      logger.debug("PromptLoader already initialized, skipping...");
+      return;
+    }
+    logger.info("Loading prompts into memory...");
+    const promptTypes = [
+      "classify",
+      "match-component",
+      "modify-props",
+      "single-component",
+      "mutli-component",
+      "actions",
+      "container-metadata"
+    ];
+    for (const promptType of promptTypes) {
+      try {
+        const template = await this.loadPromptTemplate(promptType);
+        this.promptCache.set(promptType, template);
+        logger.debug(`Cached prompt: ${promptType}`);
+      } catch (error) {
+        logger.error(`Failed to load prompt '${promptType}':`, error);
+        throw error;
+      }
+    }
+    this.isInitialized = true;
+    logger.info(`Successfully loaded ${this.promptCache.size} prompt templates into memory`);
+  }
+  /**
+   * Load a prompt template from file system (tries custom dir first, then defaults to SDK dir)
    * @param promptName - Name of the prompt folder
-   * @param promptType - Type of prompt ('system' or 'user')
-   * @param variables - Variables to replace in the template
-   * @returns Processed prompt string
+   * @returns Template with system and user prompts
    */
-  async loadPrompt(promptName, promptType, variables) {
-    try {
-      const promptPath = path3.join(
-        this.promptsDir,
-        promptName,
-        `${promptType}.md`
-      );
-      logger.debug(`Loading prompt '${promptName}/${promptType}.md' from ${promptPath}  process path: ${process.cwd()}`);
-      let content = fs3.readFileSync(promptPath, "utf-8");
-      for (const [key, value] of Object.entries(variables)) {
-        const pattern = new RegExp(`{{${key}}}`, "g");
-        const replacementValue = typeof value === "string" ? value : JSON.stringify(value);
-        content = content.replace(pattern, replacementValue);
+  async loadPromptTemplate(promptName) {
+    const tryLoadFromDir = (dir) => {
+      try {
+        const systemPath = path3.join(dir, promptName, "system.md");
+        const userPath = path3.join(dir, promptName, "user.md");
+        if (fs3.existsSync(systemPath) && fs3.existsSync(userPath)) {
+          const system = fs3.readFileSync(systemPath, "utf-8");
+          const user = fs3.readFileSync(userPath, "utf-8");
+          logger.debug(`Loaded prompt '${promptName}' from ${dir}`);
+          return { system, user };
+        }
+        return null;
+      } catch (error) {
+        return null;
       }
-      return content;
-    } catch (error) {
-      console.error(`Error loading prompt '${promptName}/${promptType}.md':`, error);
-      throw error;
+    };
+    let template = tryLoadFromDir(this.promptsDir);
+    if (!template) {
+      logger.warn(`Prompt '${promptName}' not found in ${this.promptsDir}, trying default location...`);
+      template = tryLoadFromDir(this.defaultPromptsDir);
     }
+    if (!template) {
+      throw new Error(`Prompt template '${promptName}' not found in either ${this.promptsDir} or ${this.defaultPromptsDir}`);
+    }
+    return template;
+  }
+  /**
+   * Replace variables in a template string using {{VARIABLE_NAME}} pattern
+   * @param template - Template string with placeholders
+   * @param variables - Variables to replace in the template
+   * @returns Processed string
+   */
+  replaceVariables(template, variables) {
+    let content = template;
+    for (const [key, value] of Object.entries(variables)) {
+      const pattern = new RegExp(`{{${key}}}`, "g");
+      const replacementValue = typeof value === "string" ? value : JSON.stringify(value);
+      content = content.replace(pattern, replacementValue);
+    }
+    return content;
   }
   /**
-   * Load both system and user prompts and replace variables
+   * Load both system and user prompts from cache and replace variables
    * @param promptName - Name of the prompt folder
    * @param variables - Variables to replace in the templates
    * @returns Object containing both system and user prompts
    */
   async loadPrompts(promptName, variables) {
-    const [system, user] = await Promise.all([
-      this.loadPrompt(promptName, "system", variables),
-      this.loadPrompt(promptName, "user", variables)
-    ]);
-    return { system, user };
+    if (!this.isInitialized) {
+      logger.warn("PromptLoader not initialized, loading prompts on-demand (not recommended)");
+      await this.initialize();
+    }
+    const template = this.promptCache.get(promptName);
+    if (!template) {
+      throw new Error(`Prompt template '${promptName}' not found in cache. Available prompts: ${Array.from(this.promptCache.keys()).join(", ")}`);
+    }
+    return {
+      system: this.replaceVariables(template.system, variables),
+      user: this.replaceVariables(template.user, variables)
+    };
   }
   /**
-   * Set custom prompts directory
+   * DEPRECATED: Use loadPrompts instead
+   * Load a single prompt file and replace variables using {{VARIABLE_NAME}} pattern
+   */
+  async loadPrompt(promptName, promptType, variables) {
+    const prompts = await this.loadPrompts(promptName, variables);
+    return promptType === "system" ? prompts.system : prompts.user;
+  }
+  /**
+   * Set custom prompts directory (requires re-initialization)
    * @param dir - Path to the prompts directory
    */
   setPromptsDir(dir) {
     this.promptsDir = dir;
+    this.isInitialized = false;
+    this.promptCache.clear();
   }
   /**
    * Get current prompts directory
@@ -1923,8 +1994,23 @@ var PromptLoader = class {
   getPromptsDir() {
     return this.promptsDir;
   }
+  /**
+   * Check if prompts are loaded in memory
+   */
+  isReady() {
+    return this.isInitialized;
+  }
+  /**
+   * Get the number of cached prompts
+   */
+  getCacheSize() {
+    return this.promptCache.size;
+  }
 };
-var promptLoader = new PromptLoader();
+var defaultPromptsPath = process.env.PROMPTS_DIR || path3.join(process.cwd(), ".prompts");
+var promptLoader = new PromptLoader({
+  promptsDir: defaultPromptsPath
+});
 
 // src/llm.ts
 import Anthropic from "@anthropic-ai/sdk";
@@ -1982,8 +2068,6 @@ var LLM = class {
   // ============================================================
   static async _anthropicText(messages, modelName, options) {
     const apiKey = options.apiKey || process.env.ANTHROPIC_API_KEY || "";
-    console.log("[LLM DEBUG] Anthropic Text - apiKey from options:", options.apiKey ? `${options.apiKey.substring(0, 10)}...` : "NOT SET");
-    console.log("[LLM DEBUG] Anthropic Text - final apiKey:", apiKey ? `${apiKey.substring(0, 10)}...` : "EMPTY STRING");
     const client = new Anthropic({
       apiKey
     });
@@ -2002,9 +2086,6 @@ var LLM = class {
   }
   static async _anthropicStream(messages, modelName, options, json) {
     const apiKey = options.apiKey || process.env.ANTHROPIC_API_KEY || "";
-    console.log("[LLM DEBUG] Anthropic - apiKey from options:", options.apiKey ? `${options.apiKey.substring(0, 10)}...` : "NOT SET");
-    console.log("[LLM DEBUG] Anthropic - apiKey from env:", process.env.ANTHROPIC_API_KEY ? `${process.env.ANTHROPIC_API_KEY.substring(0, 10)}...` : "NOT SET");
-    console.log("[LLM DEBUG] Anthropic - final apiKey:", apiKey ? `${apiKey.substring(0, 10)}...` : "EMPTY STRING");
     const client = new Anthropic({
       apiKey
     });
@@ -2054,9 +2135,6 @@ var LLM = class {
   }
   static async _groqStream(messages, modelName, options, json) {
     const apiKey = options.apiKey || process.env.GROQ_API_KEY || "";
-    console.log("[LLM DEBUG] Groq - apiKey from options:", options.apiKey ? `${options.apiKey.substring(0, 10)}...` : "NOT SET");
-    console.log("[LLM DEBUG] Groq - model:", modelName);
-    console.log("[LLM DEBUG] Groq - final apiKey:", apiKey ? `${apiKey.substring(0, 10)}...` : "EMPTY STRING");
     const client = new Groq({
       apiKey
     });
@@ -2128,11 +2206,8 @@ var BaseLLM = class {
    * Classify user question to determine the type and required visualizations
    */
   async classifyUserQuestion(userPrompt, apiKey, logCollector, conversationHistory) {
-    const schemaDoc = schema.generateSchemaDocumentation();
-    logger.info("Generating prompts...", userPrompt, conversationHistory);
     try {
       const prompts = await promptLoader.loadPrompts("classify", {
-        SCHEMA_DOC: schemaDoc || "No schema available",
         USER_PROMPT: userPrompt,
         CONVERSATION_HISTORY: conversationHistory || "No previous conversation"
       });
@@ -2187,6 +2262,7 @@ var BaseLLM = class {
         CURRENT_PROPS: JSON.stringify(originalProps, null, 2),
         CONVERSATION_HISTORY: conversationHistory || "No previous conversation"
       });
+      logger.debug("props-modification: System prompt\n", prompts.system.substring(0, 100), "\n\n\n", "User prompt:", prompts.user.substring(0, 50));
       const result = await LLM.stream(
         {
           sys: prompts.system,
@@ -2234,21 +2310,47 @@ var BaseLLM = class {
     }
   }
   /**
-   * Generate a dynamic component for analytical questions when no matching component exists
-   * This creates a custom component with appropriate visualization and query
+   * Match and select a component from available components filtered by type
+   * This picks the best matching component based on user prompt and modifies its props
    */
-  async generateAnalyticalComponent(userPrompt, preferredVisualizationType, apiKey, logCollector, conversationHistory) {
-    const schemaDoc = schema.generateSchemaDocumentation();
+  async generateAnalyticalComponent(userPrompt, components, preferredVisualizationType, apiKey, logCollector, conversationHistory) {
     try {
+      const filteredComponents = preferredVisualizationType ? components.filter((c) => c.type === preferredVisualizationType) : components;
+      if (filteredComponents.length === 0) {
+        logCollector?.warn(
+          `No components found of type ${preferredVisualizationType}`,
+          "explanation",
+          { reason: "No matching components available for this visualization type" }
+        );
+        return {
+          component: null,
+          reasoning: `No components available of type ${preferredVisualizationType}`,
+          isGenerated: false
+        };
+      }
+      const componentsText = filteredComponents.map((comp, idx) => {
+        const keywords = comp.keywords ? comp.keywords.join(", ") : "";
+        const category = comp.category || "general";
+        const propsPreview = comp.props ? JSON.stringify(comp.props, null, 2) : "No props";
+        return `${idx + 1}. ID: ${comp.id}
+   Name: ${comp.name}
+   Type: ${comp.type}
+   Category: ${category}
+   Description: ${comp.description || "No description"}
+   Keywords: ${keywords}
+   Props Preview: ${propsPreview}`;
+      }).join("\n\n");
       const visualizationConstraint = preferredVisualizationType ? `
-**IMPORTANT: The user has specifically requested a ${preferredVisualizationType} visualization. You MUST use this type.**
+**IMPORTANT: Components are filtered to type ${preferredVisualizationType}. Select the best match.**
 ` : "";
       const prompts = await promptLoader.loadPrompts("single-component", {
-        SCHEMA_DOC: schemaDoc || "No schema available",
+        COMPONENT_TYPE: preferredVisualizationType || "any",
+        COMPONENTS_LIST: componentsText,
         VISUALIZATION_CONSTRAINT: visualizationConstraint,
         USER_PROMPT: userPrompt,
         CONVERSATION_HISTORY: conversationHistory || "No previous conversation"
       });
+      logger.debug("single-component: System prompt\n", prompts.system.substring(0, 100), "\n\n\n", "User prompt:", prompts.user.substring(0, 50));
       const result = await LLM.stream(
         {
           sys: prompts.system,
@@ -2263,53 +2365,63 @@ var BaseLLM = class {
         true
         // Parse as JSON
       );
-      if (!result.canGenerate) {
+      if (!result.canGenerate || result.confidence < 50) {
         logCollector?.warn(
-          "Cannot generate component",
+          "Cannot match component",
           "explanation",
-          { reason: result.reasoning || "Unable to generate component for this question" }
+          { reason: result.reasoning || "Unable to find matching component for this question" }
         );
         return {
           component: null,
-          reasoning: result.reasoning || "Unable to generate component for this question",
+          reasoning: result.reasoning || "Unable to find matching component for this question",
           isGenerated: false
         };
       }
-      const query = ensureQueryLimit(result.query, this.defaultLimit);
-      logCollector?.logQuery(
-        "Analytical component query generated",
-        query,
-        {
-          componentType: result.componentType,
-          visualization: preferredVisualizationType || result.componentType,
-          title: result.title
-        }
+      const componentIndex = result.componentIndex;
+      const componentId = result.componentId;
+      let matchedComponent = null;
+      if (componentId) {
+        matchedComponent = filteredComponents.find((c) => c.id === componentId);
+      }
+      if (!matchedComponent && componentIndex) {
+        matchedComponent = filteredComponents[componentIndex - 1];
+      }
+      if (!matchedComponent) {
+        logCollector?.warn("Component not found in filtered list");
+        return {
+          component: null,
+          reasoning: "Component not found in filtered list",
+          isGenerated: false
+        };
+      }
+      logCollector?.info(`Matched component: ${matchedComponent.name} (confidence: ${result.confidence}%)`);
+      const propsValidation = await this.validateAndModifyProps(
+        userPrompt,
+        matchedComponent.props,
+        matchedComponent.name,
+        matchedComponent.type,
+        matchedComponent.description,
+        apiKey,
+        logCollector,
+        conversationHistory
       );
+      const modifiedComponent = {
+        ...matchedComponent,
+        props: propsValidation.props
+      };
       logCollector?.logExplanation(
-        "Analytical component generated",
-        result.reasoning || "Generated dynamic component based on analytical question",
+        "Analytical component selected and modified",
+        result.reasoning || "Selected component based on analytical question",
         {
-          componentType: result.componentType,
-          description: result.description
+          componentName: matchedComponent.name,
+          componentType: matchedComponent.type,
+          confidence: result.confidence,
+          propsModified: propsValidation.isModified
         }
       );
-      const dynamicComponent = {
-        id: `dynamic_${Date.now()}`,
-        name: `Dynamic${result.componentType}`,
-        type: result.componentType,
-        description: result.description,
-        category: "dynamic",
-        keywords: [],
-        props: {
-          query,
-          title: result.title,
-          description: result.description,
-          config: result.config || {}
-        }
-      };
       return {
-        component: dynamicComponent,
-        reasoning: result.reasoning || "Generated dynamic component based on analytical question",
+        component: modifiedComponent,
+        reasoning: result.reasoning || "Selected and modified component based on analytical question",
         isGenerated: true
       };
     } catch (error) {
@@ -2317,6 +2429,51 @@ var BaseLLM = class {
       throw error;
     }
   }
+  /**
+   * Generate container metadata (title and description) for multi-component dashboard
+   */
+  async generateContainerMetadata(userPrompt, visualizationTypes, apiKey, logCollector, conversationHistory) {
+    try {
+      const prompts = await promptLoader.loadPrompts("container-metadata", {
+        USER_PROMPT: userPrompt,
+        VISUALIZATION_TYPES: visualizationTypes.join(", "),
+        CONVERSATION_HISTORY: conversationHistory || "No previous conversation"
+      });
+      const result = await LLM.stream(
+        {
+          sys: prompts.system,
+          user: prompts.user
+        },
+        {
+          model: this.model,
+          maxTokens: 500,
+          temperature: 0.3,
+          apiKey: this.getApiKey(apiKey)
+        },
+        true
+        // Parse as JSON
+      );
+      logCollector?.logExplanation(
+        "Container metadata generated",
+        `Generated title and description for multi-component dashboard`,
+        {
+          title: result.title,
+          description: result.description,
+          visualizationTypes
+        }
+      );
+      return {
+        title: result.title || `${userPrompt} - Dashboard`,
+        description: result.description || `Multi-component dashboard showing ${visualizationTypes.join(", ")}`
+      };
+    } catch (error) {
+      console.error("Error generating container metadata:", error);
+      return {
+        title: `${userPrompt} - Dashboard`,
+        description: `Multi-component dashboard showing ${visualizationTypes.join(", ")}`
+      };
+    }
+  }
   /**
    * Match component from a list with enhanced props modification
    */
@@ -2378,12 +2535,12 @@ var BaseLLM = class {
         const noMatchMsg = `No matching component found (confidence: ${confidence}%)`;
         console.log("\u2717", noMatchMsg);
         logCollector?.warn(noMatchMsg);
-        const genMsg = "Attempting to generate dynamic component from analytical question...";
+        const genMsg = "Attempting to match component from analytical question...";
         console.log("\u2713", genMsg);
         logCollector?.info(genMsg);
-        const generatedResult = await this.generateAnalyticalComponent(userPrompt, void 0, apiKey, logCollector, conversationHistory);
+        const generatedResult = await this.generateAnalyticalComponent(userPrompt, components, void 0, apiKey, logCollector, conversationHistory);
         if (generatedResult.component) {
-          const genSuccessMsg = `Successfully generated component: ${generatedResult.component.name}`;
+          const genSuccessMsg = `Successfully matched component: ${generatedResult.component.name}`;
           logCollector?.info(genSuccessMsg);
           return {
             component: generatedResult.component,
@@ -2395,10 +2552,10 @@ var BaseLLM = class {
             queryModified: false
           };
         }
-        logCollector?.error("Failed to generate dynamic component");
+        logCollector?.error("Failed to match component");
         return {
           component: null,
-          reasoning: result.reasoning || "No matching component found and unable to generate dynamic component",
+          reasoning: result.reasoning || "No matching component found and unable to match component",
           method: `${this.getProviderName()}-llm`,
           confidence
         };
@@ -2446,15 +2603,15 @@ var BaseLLM = class {
     }
   }
   /**
-   * Generate multiple dynamic components for analytical questions
+   * Match multiple components for analytical questions by visualization types
    * This is used when the user needs multiple visualizations
    */
-  async generateMultipleAnalyticalComponents(userPrompt, visualizationTypes, apiKey, logCollector, conversationHistory) {
+  async generateMultipleAnalyticalComponents(userPrompt, availableComponents, visualizationTypes, apiKey, logCollector, conversationHistory) {
     try {
-      console.log("\u2713 Generating multiple components:", visualizationTypes);
+      console.log("\u2713 Matching multiple components:", visualizationTypes);
       const components = [];
       for (const vizType of visualizationTypes) {
-        const result = await this.generateAnalyticalComponent(userPrompt, vizType, apiKey, logCollector, conversationHistory);
+        const result = await this.generateAnalyticalComponent(userPrompt, availableComponents, vizType, apiKey, logCollector, conversationHistory);
         if (result.component) {
           components.push(result.component);
         }
@@ -2462,75 +2619,45 @@ var BaseLLM = class {
       if (components.length === 0) {
         return {
           components: [],
-          reasoning: "Failed to generate any components",
+          reasoning: "Failed to match any components",
           isGenerated: false
         };
       }
       return {
         components,
-        reasoning: `Generated ${components.length} components: ${visualizationTypes.join(", ")}`,
+        reasoning: `Matched ${components.length} components: ${visualizationTypes.join(", ")}`,
         isGenerated: true
       };
     } catch (error) {
-      console.error("Error generating multiple analytical components:", error);
+      console.error("Error matching multiple analytical components:", error);
       return {
         components: [],
-        reasoning: "Error occurred while generating components",
+        reasoning: "Error occurred while matching components",
         isGenerated: false
       };
     }
   }
   /**
-   * Generate a complete multi-component response with intelligent container and component props
+   * Match multiple components and wrap them in a container
    */
-  async generateMultiComponentResponse(userPrompt, visualizationTypes, apiKey, logCollector, conversationHistory) {
-    const schemaDoc = schema.generateSchemaDocumentation();
+  async generateMultiComponentResponse(userPrompt, availableComponents, visualizationTypes, apiKey, logCollector, conversationHistory) {
     try {
-      const prompts = await promptLoader.loadPrompts("mutli-component", {
-        SCHEMA_DOC: schemaDoc || "No schema available",
-        DEFAULT_LIMIT: this.defaultLimit,
-        USER_PROMPT: userPrompt,
-        VISUALIZATION_TYPES: visualizationTypes.join(", "),
-        CONVERSATION_HISTORY: conversationHistory || "No previous conversation"
-      });
-      const result = await LLM.stream(
-        {
-          sys: prompts.system,
-          user: prompts.user
-        },
-        {
-          model: this.model,
-          maxTokens: 3e3,
-          temperature: 0.2,
-          apiKey: this.getApiKey(apiKey)
-        },
-        true
-        // Parse as JSON
+      const matchResult = await this.generateMultipleAnalyticalComponents(
+        userPrompt,
+        availableComponents,
+        visualizationTypes,
+        apiKey,
+        logCollector,
+        conversationHistory
       );
-      if (!result.canGenerate || !result.components || result.components.length === 0) {
+      if (!matchResult.isGenerated || matchResult.components.length === 0) {
         return {
           containerComponent: null,
-          reasoning: result.reasoning || "Unable to generate multi-component dashboard",
+          reasoning: matchResult.reasoning || "Unable to match multi-component dashboard",
           isGenerated: false
         };
       }
-      const generatedComponents = result.components.map((compData, index) => {
-        const query = ensureQueryLimit(compData.query, this.defaultLimit);
-        return {
-          id: `dynamic_${compData.componentType.toLowerCase()}_${Date.now()}_${index}`,
-          name: `Dynamic${compData.componentType}`,
-          type: compData.componentType,
-          description: compData.description,
-          category: "dynamic",
-          keywords: [],
-          props: {
-            query,
-            title: compData.title,
-            description: compData.description,
-            config: compData.config || {}
-          }
-        };
-      });
+      const generatedComponents = matchResult.components;
       generatedComponents.forEach((component, index) => {
         if (component.props.query) {
           logCollector?.logQuery(
@@ -2545,21 +2672,24 @@ var BaseLLM = class {
           );
         }
       });
+      const containerTitle = `${userPrompt} - Dashboard`;
+      const containerDescription = `Multi-component dashboard showing ${visualizationTypes.join(", ")}`;
       logCollector?.logExplanation(
-        "Multi-component dashboard generated",
-        result.reasoning || `Generated ${generatedComponents.length} components for comprehensive analysis`,
+        "Multi-component dashboard matched",
+        matchResult.reasoning || `Matched ${generatedComponents.length} components for comprehensive analysis`,
         {
           totalComponents: generatedComponents.length,
           componentTypes: generatedComponents.map((c) => c.type),
-          containerTitle: result.containerTitle,
-          containerDescription: result.containerDescription
+          componentNames: generatedComponents.map((c) => c.name),
+          containerTitle,
+          containerDescription
         }
       );
       const containerComponent = {
         id: `multi_container_${Date.now()}`,
         name: "MultiComponentContainer",
         type: "Container",
-        description: result.containerDescription,
+        description: containerDescription,
         category: "dynamic",
         keywords: ["multi", "container", "dashboard"],
         props: {
@@ -2567,14 +2697,14 @@ var BaseLLM = class {
             components: generatedComponents,
             layout: "grid",
             spacing: 24,
-            title: result.containerTitle,
-            description: result.containerDescription
+            title: containerTitle,
+            description: containerDescription
           }
         }
       };
       return {
         containerComponent,
-        reasoning: result.reasoning || `Generated multi-component dashboard with ${generatedComponents.length} components`,
+        reasoning: matchResult.reasoning || `Matched multi-component dashboard with ${generatedComponents.length} components`,
         isGenerated: true
       };
     } catch (error) {
@@ -2595,41 +2725,99 @@ var BaseLLM = class {
       const classInfo = `Question type: ${classification.questionType}, Visualizations: ${classification.visualizations.join(", ") || "None"}, Multiple components: ${classification.needsMultipleComponents}`;
       logCollector?.info(classInfo);
       if (classification.questionType === "analytical") {
-        if (classification.visualizations.length > 0) {
-          if (classification.needsMultipleComponents && classification.visualizations.length > 1) {
-            const multiMsg = "Generating multi-component dashboard...";
-            logCollector?.info(multiMsg);
-            const result = await this.generateMultiComponentResponse(
+        if (classification.visualizations.length > 1) {
+          const multiMsg = `Matching ${classification.visualizations.length} components for types: ${classification.visualizations.join(", ")}`;
+          logCollector?.info(multiMsg);
+          const componentPromises = classification.visualizations.map((vizType) => {
+            logCollector?.info(`Matching component for type: ${vizType}`);
+            return this.generateAnalyticalComponent(
               userPrompt,
-              classification.visualizations,
+              components,
+              vizType,
               apiKey,
               logCollector,
               conversationHistory
-            );
+            ).then((result) => ({ vizType, result }));
+          });
+          const settledResults = await Promise.allSettled(componentPromises);
+          const matchedComponents = [];
+          for (const settledResult of settledResults) {
+            if (settledResult.status === "fulfilled") {
+              const { vizType, result } = settledResult.value;
+              if (result.component) {
+                matchedComponents.push(result.component);
+                logCollector?.info(`Matched: ${result.component.name}`);
+                logger.info("Component : ", result.component.name, " props: ", result.component.props);
+              } else {
+                logCollector?.warn(`Failed to match component for type: ${vizType}`);
+              }
+            } else {
+              logCollector?.warn(`Error matching component: ${settledResult.reason?.message || "Unknown error"}`);
+            }
+          }
+          console.log("matched components: after multi comp", matchedComponents.length);
+          if (matchedComponents.length === 0) {
             return {
-              component: result.containerComponent,
-              reasoning: result.reasoning,
-              method: "classification-multi-generated",
+              component: null,
+              reasoning: "Failed to match any components for the requested visualization types",
+              method: "classification-multi-failed",
               questionType: classification.questionType,
               needsMultipleComponents: true,
               propsModified: false,
               queryModified: false
             };
-          } else {
-            const vizType = classification.visualizations[0];
-            const result = await this.generateAnalyticalComponent(userPrompt, vizType, apiKey, logCollector, conversationHistory);
-            return {
-              component: result.component,
-              reasoning: result.reasoning,
-              method: "classification-generated",
-              questionType: classification.questionType,
-              needsMultipleComponents: false,
-              propsModified: false,
-              queryModified: false
-            };
           }
+          logCollector?.info("Generating container metadata...");
+          const containerMetadata = await this.generateContainerMetadata(
+            userPrompt,
+            classification.visualizations,
+            apiKey,
+            logCollector,
+            conversationHistory
+          );
+          const containerComponent = {
+            id: `multi_container_${Date.now()}`,
+            name: "MultiComponentContainer",
+            type: "Container",
+            description: containerMetadata.description,
+            category: "dynamic",
+            keywords: ["multi", "container", "dashboard"],
+            props: {
+              config: {
+                components: matchedComponents,
+                layout: "grid",
+                spacing: 24,
+                title: containerMetadata.title,
+                description: containerMetadata.description
+              }
+            }
+          };
+          logCollector?.info(`Created multi-component container with ${matchedComponents.length} components: "${containerMetadata.title}"`);
+          return {
+            component: containerComponent,
+            reasoning: `Matched ${matchedComponents.length} components for visualization types: ${classification.visualizations.join(", ")}`,
+            method: "classification-multi-generated",
+            questionType: classification.questionType,
+            needsMultipleComponents: true,
+            propsModified: false,
+            queryModified: false
+          };
+        } else if (classification.visualizations.length === 1) {
+          const vizType = classification.visualizations[0];
+          logCollector?.info(`Matching single component for type: ${vizType}`);
+          const result = await this.generateAnalyticalComponent(userPrompt, components, vizType, apiKey, logCollector, conversationHistory);
+          return {
+            component: result.component,
+            reasoning: result.reasoning,
+            method: "classification-generated",
+            questionType: classification.questionType,
+            needsMultipleComponents: false,
+            propsModified: false,
+            queryModified: false
+          };
         } else {
-          const result = await this.generateAnalyticalComponent(userPrompt, void 0, apiKey, logCollector, conversationHistory);
+          logCollector?.info("No specific visualization type - matching from all components");
+          const result = await this.generateAnalyticalComponent(userPrompt, components, void 0, apiKey, logCollector, conversationHistory);
           return {
             component: result.component,
             reasoning: result.reasoning,
@@ -2640,7 +2828,7 @@ var BaseLLM = class {
             queryModified: false
           };
         }
-      } else if (classification.questionType === "data_modification") {
+      } else if (classification.questionType === "data_modification" || classification.questionType === "general") {
         const matchMsg = "Using component matching for data modification...";
         logCollector?.info(matchMsg);
         const matchResult = await this.matchComponent(userPrompt, components, apiKey, logCollector, conversationHistory);
@@ -2788,7 +2976,6 @@ var useAnthropicMethod = async (prompt, components, apiKey, logCollector, conver
   }
   try {
     const matchResult = await anthropicLLM.handleUserRequest(prompt, components, apiKey, logCollector, conversationHistory);
-    logger.debug(`Anthropic method success: ${matchResult}`);
     return { success: true, data: matchResult };
   } catch (error) {
     const errorMsg = error instanceof Error ? error.message : String(error);
@@ -2842,7 +3029,6 @@ var get_user_response = async (prompt, components, anthropicApiKey, groqApiKey,
       let result;
       if (provider === "anthropic") {
         result = await useAnthropicMethod(prompt, components, anthropicApiKey, logCollector, conversationHistory);
-        logger.debug("Anthropic result:", result);
       } else if (provider === "groq") {
         result = await useGroqMethod(prompt, components, groqApiKey, logCollector, conversationHistory);
       } else {
@@ -2905,6 +3091,7 @@ var UILogCollector = class {
     };
     this.logs.push(log);
     this.sendLogImmediately(log);
+    console.log("UILogCollector:", log);
   }
   /**
    * Send a single log to runtime immediately
@@ -3043,7 +3230,6 @@ async function handleUserPromptRequest(data, components, sendMessage, anthropicA
     const SA_RUNTIME = payload.SA_RUNTIME;
     const wsId = userPromptRequest.from.id || "unknown";
     logger.info(`[REQUEST ${id}] Processing user prompt: "${prompt.substring(0, 50)}..."`);
-    logger.info(`[REQUEST ${id}] Providers: ${llmProviders?.join(", ")}, Anthropic key: ${anthropicApiKey ? "SET" : "NOT SET"}, Groq key: ${groqApiKey ? "SET" : "NOT SET"}`);
     if (processedMessageIds.has(id)) {
       logger.warn(`[REQUEST ${id}] Duplicate request detected - ignoring`);
       return;
@@ -3093,18 +3279,18 @@ async function handleUserPromptRequest(data, components, sendMessage, anthropicA
       }, sendMessage, wsId);
       return;
     }
-    logCollector.info(`Starting user prompt request with ${components.length} components`);
-    logger.info(`components length: ${components.length}`);
     const threadManager = ThreadManager.getInstance();
     let thread = threadManager.getThread(threadId);
     if (!thread) {
       thread = threadManager.createThread(threadId);
       logger.info(`Created new thread: ${threadId}`);
     }
+    logCollector.info(`Starting user prompt request with ${components.length} components`);
     const conversationHistory = thread.getConversationContext(CONTEXT_CONFIG.MAX_CONVERSATION_CONTEXT_BLOCKS, existingUiBlockId);
+    logger.info("conversationHistory", conversationHistory);
     const userResponse = await get_user_response(prompt, components, anthropicApiKey, groqApiKey, llmProviders, logCollector, conversationHistory);
+    logger.info("llm userResponse", userResponse);
     logCollector.info("User prompt request completed");
-    logger.info(`[REQUEST ${id}] Response success: ${userResponse.success}, reason: ${userResponse.success ? "N/A" : userResponse}`);
     if (userResponse.success && userResponse.data && typeof userResponse.data === "object" && "component" in userResponse.data) {
       const component = userResponse.data.component;
       const uiBlockId = existingUiBlockId;
@@ -3146,10 +3332,7 @@ function sendDataResponse4(id, res, sendMessage, clientId) {
       ...res
     }
   };
-  logger.info(`[REQUEST ${id}] Sending USER_PROMPT_RES with success=${res.success}`);
-  if (!res.success && res.reason) {
-    logger.info(`[REQUEST ${id}] Error reason: ${res.reason}`);
-  }
+  logger.info("sending user prompt response", response);
   sendMessage(response);
 }
 
@@ -4848,6 +5031,9 @@ var SuperatomSDK = class {
     this.userManager = new UserManager(this.projectId, 5e3);
     this.dashboardManager = new DashboardManager(this.projectId);
     this.reportManager = new ReportManager(this.projectId);
+    this.initializePromptLoader(config.promptsDir).catch((error) => {
+      logger.error("Failed to initialize PromptLoader:", error);
+    });
     this.initializeUserManager().catch((error) => {
       logger.error("Failed to initialize UserManager:", error);
     });
@@ -4857,6 +5043,21 @@ var SuperatomSDK = class {
       logger.error("Failed to connect to Superatom:", error);
     });
   }
+  /**
+   * Initialize PromptLoader and load prompts into memory
+   */
+  async initializePromptLoader(promptsDir) {
+    try {
+      if (promptsDir) {
+        promptLoader.setPromptsDir(promptsDir);
+      }
+      await promptLoader.initialize();
+      logger.info(`PromptLoader initialized with ${promptLoader.getCacheSize()} prompts from ${promptLoader.getPromptsDir()}`);
+    } catch (error) {
+      logger.error("Failed to initialize PromptLoader:", error);
+      throw error;
+    }
+  }
   /**
    * Initialize UserManager for the project
    */