@superatomai/sdk-node 0.0.6 → 0.0.7

package/dist/index.mjs

@@ -438,6 +438,7 @@ var require_main = __commonJS({
 // src/userResponse/utils.ts
 var utils_exports = {};
 __export(utils_exports, {
+  convertQuestionsToActions: () => convertQuestionsToActions,
   convertTopToLimit: () => convertTopToLimit,
   ensureQueryLimit: () => ensureQueryLimit,
   fixScalarSubqueries: () => fixScalarSubqueries,
@@ -539,6 +540,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");
@@ -841,7 +850,8 @@ var ComponentPropsSchema = z3.object({
   query: z3.string().optional(),
   title: z3.string().optional(),
   description: z3.string().optional(),
-  config: z3.record(z3.unknown()).optional()
+  config: z3.record(z3.unknown()).optional(),
+  actions: z3.array(z3.any()).optional()
 });
 var ComponentSchema = z3.object({
   id: z3.string(),
@@ -1224,6 +1234,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)
    */
@@ -2586,6 +2602,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) {
@@ -3125,13 +3177,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 {
@@ -3173,7 +3226,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 || "";
@@ -3216,7 +3268,8 @@ var BaseLLM = class {
             selectedLayout: "MultiComponentContainer",
             selectedLayoutId: "",
             selectedLayoutComponent: null,
-            layoutReasoning: "No layout selected"
+            layoutReasoning: "No layout selected",
+            actions: []
           };
         }
       }
@@ -3224,6 +3277,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;
@@ -3234,6 +3289,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}`);
@@ -3248,6 +3304,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) {
@@ -3267,7 +3329,8 @@ var BaseLLM = class {
         selectedLayout,
         selectedLayoutId,
         selectedLayoutComponent,
-        layoutReasoning
+        layoutReasoning,
+        actions
       };
     } catch (error) {
       const errorMsg = error instanceof Error ? error.message : String(error);
@@ -3279,7 +3342,8 @@ var BaseLLM = class {
         selectedLayout: "MultiComponentContainer",
         selectedLayoutId: "",
         selectedLayoutComponent: null,
-        layoutReasoning: "Error occurred during component matching"
+        layoutReasoning: "Error occurred during component matching",
+        actions: []
       };
     }
   }
@@ -3298,10 +3362,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);
@@ -3502,6 +3573,7 @@ ${errorMsg}
             text: textResponse,
             // Include the streamed text showing all attempts
             matchedComponents: [],
+            actions: [],
             method: `${this.getProviderName()}-text-response-max-attempts`
           }
         };
@@ -3517,6 +3589,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(
@@ -3528,6 +3601,7 @@ ${errorMsg}
         matchedComponents = matchResult.components;
         selectedLayoutComponent = matchResult.selectedLayoutComponent;
         layoutReasoning = matchResult.layoutReasoning;
+        actions = matchResult.actions;
       }
       let container_componet = null;
       if (matchedComponents.length > 0) {
@@ -3541,11 +3615,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()}`,
@@ -3557,11 +3632,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 {
@@ -3571,6 +3647,7 @@ ${errorMsg}
           matchedComponents,
           component: container_componet,
           layoutReasoning,
+          actions,
           method: `${this.getProviderName()}-text-response`
         },
         errors: []
@@ -3586,6 +3663,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`
         }
       };
@@ -4336,6 +4414,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) {
@@ -4344,6 +4423,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;
@@ -4354,11 +4436,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 {
@@ -4596,7 +4682,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) {