@superatomai/sdk-node 0.0.15 → 0.0.16

package/dist/index.mjs

@@ -525,10 +525,14 @@ var ReportsRequestMessageSchema = z3.object({
 var UIBlockSchema = z3.object({
   id: z3.string().optional(),
   userQuestion: z3.string().optional(),
+  user_prompt: z3.string().optional(),
+  // DB field name
   text: z3.string().optional(),
   textResponse: z3.string().optional(),
+  analysis: z3.string().optional(),
+  // DB field name for textResponse
   component: ComponentSchema.optional(),
-  // Legacy field
+  // DB field name / Legacy field
   generatedComponentMetadata: ComponentSchema.optional(),
   // Actual field used by UIBlock class
   componentData: z3.record(z3.any()).optional(),
@@ -3774,8 +3778,9 @@ var BaseLLM = class {
    * @param collections - Collection registry for executing database queries via database.execute
    * @param components - Optional list of available components for matching suggestions
    * @param externalTools - Optional array of external tools (email, calendar, etc.) that can be called
+   * @param category - Question category ('data_analysis' | 'data_modification' | 'general'). For data_modification, answer component streaming is skipped. For general, component generation is skipped entirely.
    */
-  async generateTextResponse(userPrompt, apiKey, logCollector, conversationHistory, streamCallback, collections, components, externalTools) {
+  async generateTextResponse(userPrompt, apiKey, logCollector, conversationHistory, streamCallback, collections, components, externalTools, category) {
     const errors = [];
     logger.debug(`[${this.getProviderName()}] Starting text response generation`);
     logger.debug(`[${this.getProviderName()}] User prompt: "${userPrompt.substring(0, 50)}..."`);
@@ -4169,16 +4174,19 @@ ${errorMsg}
           textLength: textResponse.length
         }
       );
-      if (wrappedStreamCallback && components && components.length > 0) {
+      if (wrappedStreamCallback && components && components.length > 0 && category !== "general") {
         wrappedStreamCallback("__TEXT_COMPLETE__COMPONENT_GENERATION_START__");
       }
       let matchedComponents = [];
       let layoutTitle = "Dashboard";
       let layoutDescription = "Multi-component dashboard";
       let actions = [];
-      if (components && components.length > 0) {
+      if (category === "general") {
+        logger.info(`[${this.getProviderName()}] Skipping component generation for general/conversational question`);
+        logCollector?.info("Skipping component generation for general question");
+      } else if (components && components.length > 0) {
         logger.info(`[${this.getProviderName()}] Matching components from text response...`);
-        const componentStreamCallback = wrappedStreamCallback ? (component) => {
+        const componentStreamCallback = wrappedStreamCallback && category !== "data_modification" ? (component) => {
           const answerMarker = `__ANSWER_COMPONENT_START__${JSON.stringify(component)}__ANSWER_COMPONENT_END__`;
           wrappedStreamCallback(answerMarker);
           logger.info(`[${this.getProviderName()}] Streamed answer component to frontend: ${component.name} (${component.type})`);
@@ -4277,52 +4285,78 @@ ${errorMsg}
         logCollector?.info(
           `\u2713 Found similar conversation (${(conversationMatch.similarity * 100).toFixed(2)}% match)`
         );
-        if (conversationMatch.similarity >= 0.99) {
-          const elapsedTime2 = Date.now() - startTime;
-          logger.info(`[${this.getProviderName()}] \u2713 100% match - returning UI block directly without adaptation`);
-          logCollector?.info(`\u2713 Exact match (${(conversationMatch.similarity * 100).toFixed(2)}%) - returning cached result`);
-          logCollector?.info(`Total time taken: ${elapsedTime2}ms (${(elapsedTime2 / 1e3).toFixed(2)}s)`);
-          const component = conversationMatch.uiBlock?.generatedComponentMetadata || conversationMatch.uiBlock?.component;
-          return {
-            success: true,
-            data: {
-              component,
-              reasoning: `Exact match from previous conversation (${(conversationMatch.similarity * 100).toFixed(2)}% similarity)`,
-              method: `${this.getProviderName()}-semantic-match-exact`,
-              semanticSimilarity: conversationMatch.similarity
-            },
-            errors: []
-          };
-        }
-        logCollector?.info(`Adapting parameters for similar question...`);
-        const originalPrompt = conversationMatch.metadata?.userPrompt || "Previous question";
-        const adaptResult = await this.adaptUIBlockParameters(
-          userPrompt,
-          originalPrompt,
-          conversationMatch.uiBlock,
-          apiKey,
-          logCollector
-        );
-        if (adaptResult.success && adaptResult.adaptedComponent) {
-          const elapsedTime2 = Date.now() - startTime;
-          logger.info(`[${this.getProviderName()}] \u2713 Successfully adapted UI block parameters`);
-          logger.info(`[${this.getProviderName()}] Total time taken: ${elapsedTime2}ms (${(elapsedTime2 / 1e3).toFixed(2)}s)`);
-          logCollector?.info(`\u2713 UI block adapted successfully`);
-          logCollector?.info(`Total time taken: ${elapsedTime2}ms (${(elapsedTime2 / 1e3).toFixed(2)}s)`);
-          return {
-            success: true,
-            data: {
-              component: adaptResult.adaptedComponent,
-              reasoning: `Adapted from previous conversation: ${originalPrompt}`,
-              method: `${this.getProviderName()}-semantic-match`,
-              semanticSimilarity: conversationMatch.similarity,
-              parametersChanged: adaptResult.parametersChanged
-            },
-            errors: []
-          };
+        const component = conversationMatch.uiBlock?.component || conversationMatch.uiBlock?.generatedComponentMetadata;
+        const cachedTextResponse = conversationMatch.uiBlock?.analysis || conversationMatch.uiBlock?.textResponse || conversationMatch.uiBlock?.text || "";
+        if (!component) {
+          if (conversationMatch.similarity >= 0.99) {
+            const elapsedTime2 = Date.now() - startTime;
+            logger.info(`[${this.getProviderName()}] \u2713 Exact match for general question - returning cached text response`);
+            logCollector?.info(`\u2713 Exact match for general question - returning cached response`);
+            logCollector?.info(`Total time taken: ${elapsedTime2}ms (${(elapsedTime2 / 1e3).toFixed(2)}s)`);
+            return {
+              success: true,
+              data: {
+                text: cachedTextResponse,
+                component: null,
+                matchedComponents: [],
+                actions: conversationMatch.uiBlock?.actions || [],
+                reasoning: `Exact match from previous general conversation`,
+                method: `${this.getProviderName()}-semantic-match-general`,
+                semanticSimilarity: conversationMatch.similarity
+              },
+              errors: []
+            };
+          } else {
+            logger.info(`[${this.getProviderName()}] Similar match but no component (general question) - processing fresh`);
+            logCollector?.info("Similar match found but was a general conversation - processing as new question");
+          }
         } else {
-          logger.info(`[${this.getProviderName()}] Could not adapt matched conversation, continuing to category classification`);
-          logCollector?.warn(`Could not adapt matched conversation: ${adaptResult.explanation}`);
+          if (conversationMatch.similarity >= 0.99) {
+            const elapsedTime2 = Date.now() - startTime;
+            logger.info(`[${this.getProviderName()}] \u2713 100% match - returning UI block directly without adaptation`);
+            logCollector?.info(`\u2713 Exact match (${(conversationMatch.similarity * 100).toFixed(2)}%) - returning cached result`);
+            logCollector?.info(`Total time taken: ${elapsedTime2}ms (${(elapsedTime2 / 1e3).toFixed(2)}s)`);
+            return {
+              success: true,
+              data: {
+                component,
+                reasoning: `Exact match from previous conversation (${(conversationMatch.similarity * 100).toFixed(2)}% similarity)`,
+                method: `${this.getProviderName()}-semantic-match-exact`,
+                semanticSimilarity: conversationMatch.similarity
+              },
+              errors: []
+            };
+          }
+          logCollector?.info(`Adapting parameters for similar question...`);
+          const originalPrompt = conversationMatch.metadata?.userPrompt || "Previous question";
+          const adaptResult = await this.adaptUIBlockParameters(
+            userPrompt,
+            originalPrompt,
+            conversationMatch.uiBlock,
+            apiKey,
+            logCollector
+          );
+          if (adaptResult.success && adaptResult.adaptedComponent) {
+            const elapsedTime2 = Date.now() - startTime;
+            logger.info(`[${this.getProviderName()}] \u2713 Successfully adapted UI block parameters`);
+            logger.info(`[${this.getProviderName()}] Total time taken: ${elapsedTime2}ms (${(elapsedTime2 / 1e3).toFixed(2)}s)`);
+            logCollector?.info(`\u2713 UI block adapted successfully`);
+            logCollector?.info(`Total time taken: ${elapsedTime2}ms (${(elapsedTime2 / 1e3).toFixed(2)}s)`);
+            return {
+              success: true,
+              data: {
+                component: adaptResult.adaptedComponent,
+                reasoning: `Adapted from previous conversation: ${originalPrompt}`,
+                method: `${this.getProviderName()}-semantic-match`,
+                semanticSimilarity: conversationMatch.similarity,
+                parametersChanged: adaptResult.parametersChanged
+              },
+              errors: []
+            };
+          } else {
+            logger.info(`[${this.getProviderName()}] Could not adapt matched conversation, continuing to category classification`);
+            logCollector?.warn(`Could not adapt matched conversation: ${adaptResult.explanation}`);
+          }
         }
       } else {
         logger.info(`[${this.getProviderName()}] No matching previous conversations found, proceeding to category classification`);
@@ -4364,7 +4398,10 @@ ${errorMsg}
           })()
         })) || [];
       }
-      if (categoryClassification.category === "data_analysis") {
+      if (categoryClassification.category === "general") {
+        logger.info(`[${this.getProviderName()}] Routing to general conversation (no database operations)`);
+        logCollector?.info("Routing to general conversation...");
+      } else if (categoryClassification.category === "data_analysis") {
         logger.info(`[${this.getProviderName()}] Routing to data analysis (SELECT operations)`);
         logCollector?.info("Routing to data analysis...");
       } else if (categoryClassification.category === "data_modification") {
@@ -4379,7 +4416,8 @@ ${errorMsg}
         streamCallback,
         collections,
         components,
-        toolsToUse
+        toolsToUse,
+        categoryClassification.category
       );
       const elapsedTime = Date.now() - startTime;
       logger.info(`[${this.getProviderName()}] Total time taken: ${elapsedTime}ms (${(elapsedTime / 1e3).toFixed(2)}s)`);
@@ -5036,8 +5074,10 @@ var get_user_request = async (data, components, sendMessage, anthropicApiKey, gr
       if ("component" in userResponse.data) {
         component = userResponse.data.component;
       }
-      if ("textResponse" in userResponse.data) {
+      if ("text" in userResponse.data) {
         textResponse = userResponse.data.text;
+      } else if ("textResponse" in userResponse.data) {
+        textResponse = userResponse.data.textResponse;
       }
       if ("actions" in userResponse.data) {
         actions = userResponse.data.actions || [];