@superatomai/sdk-node 0.0.6 → 0.0.7

package/dist/index.d.mts

@@ -1042,6 +1042,10 @@ declare class UIBlock {
      * @returns Promise resolving to Action[]
      */
     getOrFetchActions(generateFn: () => Promise<Action[]>): Promise<Action[]>;
+    /**
+     * Set or replace all actions
+     */
+    setActions(actions: Action[]): void;
     /**
      * Add a single action (only if actions are resolved)
      */

package/dist/index.d.ts

@@ -1042,6 +1042,10 @@ declare class UIBlock {
      * @returns Promise resolving to Action[]
      */
     getOrFetchActions(generateFn: () => Promise<Action[]>): Promise<Action[]>;
+    /**
+     * Set or replace all actions
+     */
+    setActions(actions: Action[]): void;
     /**
      * Add a single action (only if actions are resolved)
      */

package/dist/index.js

@@ -434,6 +434,7 @@ var require_main = __commonJS({
 // src/userResponse/utils.ts
 var utils_exports = {};
 __export(utils_exports, {
+  convertQuestionsToActions: () => convertQuestionsToActions,
   convertTopToLimit: () => convertTopToLimit,
   ensureQueryLimit: () => ensureQueryLimit,
   fixScalarSubqueries: () => fixScalarSubqueries,
@@ -535,6 +536,14 @@ function fixScalarSubqueries(query) {
   }
   return modifiedQuery;
 }
+function convertQuestionsToActions(questions) {
+  return questions.map((question, index) => ({
+    id: `action_${index}_${Date.now()}`,
+    name: question,
+    type: "next_question",
+    question
+  }));
+}
 function getJsonSizeInBytes(obj) {
   const jsonString = JSON.stringify(obj);
   return Buffer.byteLength(jsonString, "utf8");
@@ -855,7 +864,8 @@ var ComponentPropsSchema = import_zod3.z.object({
   query: import_zod3.z.string().optional(),
   title: import_zod3.z.string().optional(),
   description: import_zod3.z.string().optional(),
-  config: import_zod3.z.record(import_zod3.z.unknown()).optional()
+  config: import_zod3.z.record(import_zod3.z.unknown()).optional(),
+  actions: import_zod3.z.array(import_zod3.z.any()).optional()
 });
 var ComponentSchema = import_zod3.z.object({
   id: import_zod3.z.string(),
@@ -1238,6 +1248,12 @@ var UIBlock = class {
       throw error;
     }
   }
+  /**
+   * Set or replace all actions
+   */
+  setActions(actions) {
+    this.actions = actions;
+  }
   /**
    * Add a single action (only if actions are resolved)
    */
@@ -2600,6 +2616,42 @@ var LLM = class {
   }
 };
 
+// src/userResponse/knowledge-base.ts
+var getKnowledgeBase = async ({
+  prompt,
+  collections,
+  topK = 1
+}) => {
+  try {
+    if (!collections || !collections["knowledge-base"] || !collections["knowledge-base"]["query"]) {
+      logger.info("[KnowledgeBase] knowledge-base.query collection not registered, skipping");
+      return "";
+    }
+    logger.info(`[KnowledgeBase] Querying knowledge base for: "${prompt.substring(0, 50)}..."`);
+    const result = await collections["knowledge-base"]["query"]({
+      prompt,
+      topK
+    });
+    if (!result || !result.content) {
+      logger.error("[KnowledgeBase] No knowledge base results returned");
+      return "";
+    }
+    logger.info(`[KnowledgeBase] Retrieved knowledge base context (${result.content.length} chars)`);
+    if (result.metadata?.sources && result.metadata.sources.length > 0) {
+      logger.debug(`[KnowledgeBase] Sources: ${result.metadata.sources.map((s) => s.title).join(", ")}`);
+    }
+    return result.content;
+  } catch (error) {
+    const errorMsg = error instanceof Error ? error.message : String(error);
+    logger.warn(`[KnowledgeBase] Error querying knowledge base: ${errorMsg}`);
+    return "";
+  }
+};
+var KB = {
+  getKnowledgeBase
+};
+var knowledge_base_default = KB;
+
 // src/userResponse/base-llm.ts
 var BaseLLM = class {
   constructor(config) {
@@ -3139,13 +3191,14 @@ var BaseLLM = class {
     }
   }
   /**
-   * Match components from text response suggestions
+   * Match components from text response suggestions and generate follow-up questions
    * Takes a text response with component suggestions (c1:type format) and matches with available components
+   * Also generates intelligent follow-up questions (actions) based on the analysis
    * @param textResponse - The text response containing component suggestions
    * @param components - List of available components
    * @param apiKey - Optional API key
    * @param logCollector - Optional log collector
-   * @returns Object containing matched components, selected layout, and reasoning
+   * @returns Object containing matched components, selected layout, reasoning, and follow-up actions
    */
   async matchComponentsFromTextResponse(textResponse, components, apiKey, logCollector) {
     try {
@@ -3187,7 +3240,6 @@ var BaseLLM = class {
         // Don't parse as JSON yet, get raw response
       );
       logger.debug(`[${this.getProviderName()}] Raw component matching response length: ${rawResponse?.length || 0}`);
-      logger.file(`[${this.getProviderName()}] Component matching raw response:`, rawResponse);
       let result;
       try {
         let cleanedResponse = rawResponse || "";
@@ -3230,7 +3282,8 @@ var BaseLLM = class {
             selectedLayout: "MultiComponentContainer",
             selectedLayoutId: "",
             selectedLayoutComponent: null,
-            layoutReasoning: "No layout selected"
+            layoutReasoning: "No layout selected",
+            actions: []
           };
         }
       }
@@ -3238,6 +3291,8 @@ var BaseLLM = class {
       const selectedLayout = result.selectedLayout || "MultiComponentContainer";
       const selectedLayoutId = result.selectedLayoutId || "";
       const layoutReasoning = result.layoutReasoning || "No layout reasoning provided";
+      const rawActions = result.actions || [];
+      const actions = convertQuestionsToActions(rawActions);
       let selectedLayoutComponent = null;
       if (selectedLayoutId) {
         selectedLayoutComponent = components.find((c) => c.id === selectedLayoutId) || null;
@@ -3248,6 +3303,7 @@ var BaseLLM = class {
       logger.info(`[${this.getProviderName()}] Matched ${matchedComponents.length} components from text response`);
       logger.info(`[${this.getProviderName()}] Selected layout: ${selectedLayout} (ID: ${selectedLayoutId})`);
       logger.info(`[${this.getProviderName()}] Layout reasoning: ${layoutReasoning}`);
+      logger.info(`[${this.getProviderName()}] Generated ${actions.length} follow-up actions`);
       if (matchedComponents.length > 0) {
         logCollector?.info(`Matched ${matchedComponents.length} components for visualization using ${selectedLayout}`);
         logCollector?.info(`Layout reasoning: ${layoutReasoning}`);
@@ -3262,6 +3318,12 @@ var BaseLLM = class {
           }
         });
       }
+      if (actions.length > 0) {
+        logCollector?.info(`Generated ${actions.length} follow-up questions`);
+        actions.forEach((action, idx) => {
+          logCollector?.info(`  ${idx + 1}. ${action.name}`);
+        });
+      }
       const finalComponents = matchedComponents.map((mc) => {
         const originalComponent = components.find((c) => c.id === mc.componentId);
         if (!originalComponent) {
@@ -3281,7 +3343,8 @@ var BaseLLM = class {
         selectedLayout,
         selectedLayoutId,
         selectedLayoutComponent,
-        layoutReasoning
+        layoutReasoning,
+        actions
       };
     } catch (error) {
       const errorMsg = error instanceof Error ? error.message : String(error);
@@ -3293,7 +3356,8 @@ var BaseLLM = class {
         selectedLayout: "MultiComponentContainer",
         selectedLayoutId: "",
         selectedLayoutComponent: null,
-        layoutReasoning: "Error occurred during component matching"
+        layoutReasoning: "Error occurred during component matching",
+        actions: []
       };
     }
   }
@@ -3312,10 +3376,17 @@ var BaseLLM = class {
     logger.debug(`[${this.getProviderName()}] User prompt: "${userPrompt.substring(0, 50)}..."`);
     try {
       const schemaDoc = schema.generateSchemaDocumentation();
+      const knowledgeBaseContext = await knowledge_base_default.getKnowledgeBase({
+        prompt: userPrompt,
+        collections,
+        topK: 1
+      });
+      logger.file("\n=============================\nknowledge base context:", knowledgeBaseContext);
       const prompts = await promptLoader.loadPrompts("text-response", {
         USER_PROMPT: userPrompt,
         CONVERSATION_HISTORY: conversationHistory || "No previous conversation",
-        SCHEMA_DOC: schemaDoc
+        SCHEMA_DOC: schemaDoc,
+        KNOWLEDGE_BASE_CONTEXT: knowledgeBaseContext || "No additional knowledge base context available."
       });
       logger.file("\n=============================\nsystem prompt:", prompts.system);
       logger.file("\n=============================\nuser prompt:", prompts.user);
@@ -3516,6 +3587,7 @@ ${errorMsg}
             text: textResponse,
             // Include the streamed text showing all attempts
             matchedComponents: [],
+            actions: [],
             method: `${this.getProviderName()}-text-response-max-attempts`
           }
         };
@@ -3531,6 +3603,7 @@ ${errorMsg}
       let matchedComponents = [];
       let selectedLayoutComponent = null;
       let layoutReasoning = "No layout selected";
+      let actions = [];
       if (components && components.length > 0) {
         logger.info(`[${this.getProviderName()}] Matching components from text response...`);
         const matchResult = await this.matchComponentsFromTextResponse(
@@ -3542,6 +3615,7 @@ ${errorMsg}
         matchedComponents = matchResult.components;
         selectedLayoutComponent = matchResult.selectedLayoutComponent;
         layoutReasoning = matchResult.layoutReasoning;
+        actions = matchResult.actions;
       }
       let container_componet = null;
       if (matchedComponents.length > 0) {
@@ -3555,11 +3629,12 @@ ${errorMsg}
               config: {
                 ...selectedLayoutComponent.props?.config || {},
                 components: matchedComponents
-              }
+              },
+              actions
             }
           };
-          logger.info(`[${this.getProviderName()}] Created ${selectedLayoutComponent.name} (${selectedLayoutComponent.type}) container with ${matchedComponents.length} components`);
-          logCollector?.info(`Created ${selectedLayoutComponent.name} with ${matchedComponents.length} components: ${layoutReasoning}`);
+          logger.info(`[${this.getProviderName()}] Created ${selectedLayoutComponent.name} (${selectedLayoutComponent.type}) container with ${matchedComponents.length} components and ${actions.length} actions`);
+          logCollector?.info(`Created ${selectedLayoutComponent.name} with ${matchedComponents.length} components and ${actions.length} actions: ${layoutReasoning}`);
         } else {
           container_componet = {
             id: `multi_container_${Date.now()}`,
@@ -3571,11 +3646,12 @@ ${errorMsg}
             props: {
               config: {
                 components: matchedComponents
-              }
+              },
+              actions
             }
           };
-          logger.info(`[${this.getProviderName()}] Created fallback MultiComponentContainer with ${matchedComponents.length} components`);
-          logCollector?.info(`Created MultiComponentContainer with ${matchedComponents.length} components: ${layoutReasoning}`);
+          logger.info(`[${this.getProviderName()}] Created fallback MultiComponentContainer with ${matchedComponents.length} components and ${actions.length} actions`);
+          logCollector?.info(`Created MultiComponentContainer with ${matchedComponents.length} components and ${actions.length} actions: ${layoutReasoning}`);
         }
       }
       return {
@@ -3585,6 +3661,7 @@ ${errorMsg}
           matchedComponents,
           component: container_componet,
           layoutReasoning,
+          actions,
           method: `${this.getProviderName()}-text-response`
         },
         errors: []
@@ -3600,6 +3677,7 @@ ${errorMsg}
         data: {
           text: "I apologize, but I encountered an error while processing your question. Please try rephrasing or ask something else.",
           matchedComponents: [],
+          actions: [],
           method: `${this.getProviderName()}-text-response-error`
         }
       };
@@ -4350,6 +4428,7 @@ var get_user_request = async (data, components, sendMessage, anthropicApiKey, gr
   }
   let component = null;
   let textResponse = null;
+  let actions = [];
   if (userResponse.data) {
     if (typeof userResponse.data === "object") {
       if ("component" in userResponse.data) {
@@ -4358,6 +4437,9 @@ var get_user_request = async (data, components, sendMessage, anthropicApiKey, gr
       if ("textResponse" in userResponse.data) {
         textResponse = userResponse.data.text;
       }
+      if ("actions" in userResponse.data) {
+        actions = userResponse.data.actions || [];
+      }
     }
   }
   const finalTextResponse = responseMode === "text" && accumulatedStreamResponse ? accumulatedStreamResponse : textResponse;
@@ -4368,11 +4450,15 @@ var get_user_request = async (data, components, sendMessage, anthropicApiKey, gr
     component,
     // generatedComponentMetadata: full component object (ComponentSchema)
     [],
-    // actions: empty initially
+    // actions: empty initially, will be set below
     uiBlockId,
     finalTextResponse
     // textResponse: FULL streaming response including all intermediate messages
   );
+  if (actions.length > 0) {
+    uiBlock.setActions(actions);
+    logger.info(`Stored ${actions.length} actions in UIBlock: ${uiBlockId}`);
+  }
   thread.addUIBlock(uiBlock);
   logger.info(`Created UIBlock: ${uiBlockId} in Thread: ${threadId}`);
   return {
@@ -4610,7 +4696,6 @@ async function handleActionsRequest(data, sendMessage, anthropicApiKey, groqApiK
     const uiBlockId = SA_RUNTIME.uiBlockId;
     const threadId = SA_RUNTIME.threadId;
     logger.debug(`[ACTIONS_REQ ${id}] SA_RUNTIME validated - threadId: ${threadId}, uiBlockId: ${uiBlockId}`);
-    logger.debug(`[ACTIONS_REQ ${id}] Retrieving thread: ${threadId}`);
     const threadManager = ThreadManager.getInstance();
     const thread = threadManager.getThread(threadId);
     if (!thread) {