@superatomai/sdk-node 0.0.44-mds → 0.0.45-mds

package/dist/index.mjs

@@ -3767,6 +3767,12 @@ You MUST respond with ONLY a valid JSON object (no markdown, no code blocks):
 ---
 
 ## CONTEXT
+
+### Global Knowledge Base
+{{GLOBAL_KNOWLEDGE_BASE}}
+
+### Knowledge Base Context
+{{KNOWLEDGE_BASE_CONTEXT}}
 `,
     user: `{{USER_PROMPT}}`
   },
@@ -14594,7 +14600,7 @@ function sendDashCompResponse(id, res, sendMessage, clientId) {
 }
 
 // src/dashComp/pick-component.ts
-async function pickComponentWithLLM(prompt, components, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, collections, tools, dashCompModels, conversationHistory) {
+async function pickComponentWithLLM(prompt, components, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, collections, tools, dashCompModels, conversationHistory, userId) {
   const errors = [];
   const availableComponentsText = formatComponentsForPrompt(components);
   const availableToolsText = formatToolsForPrompt(tools);
@@ -14610,6 +14616,25 @@ async function pickComponentWithLLM(prompt, components, anthropicApiKey, groqApi
       schemaDoc = schema.generateSchemaDocumentation();
     }
     const databaseRules = await promptLoader.loadDatabaseRules();
+    let globalKnowledgeBase = "No global knowledge base available.";
+    let knowledgeBaseContext = "No additional knowledge base context available.";
+    if (collections) {
+      const kbResult = await knowledge_base_default.getAllKnowledgeBase({
+        prompt,
+        collections,
+        userId,
+        topK: KNOWLEDGE_BASE_TOP_K
+      });
+      globalKnowledgeBase = kbResult.globalContext || globalKnowledgeBase;
+      const dynamicParts = [];
+      if (kbResult.userContext) {
+        dynamicParts.push("## User-Specific Knowledge Base\n" + kbResult.userContext);
+      }
+      if (kbResult.queryContext) {
+        dynamicParts.push("## Relevant Knowledge Base (Query-Matched)\n" + kbResult.queryContext);
+      }
+      knowledgeBaseContext = dynamicParts.join("\n\n") || knowledgeBaseContext;
+    }
     const prompts = await promptLoader.loadPrompts("dash-comp-picker", {
       USER_PROMPT: prompt,
       AVAILABLE_COMPONENTS: availableComponentsText,
@@ -14617,8 +14642,12 @@ async function pickComponentWithLLM(prompt, components, anthropicApiKey, groqApi
       DATABASE_RULES: databaseRules,
       AVAILABLE_TOOLS: availableToolsText,
       CURRENT_DATETIME: getCurrentDateTimeForPrompt(),
-      CONVERSATION_HISTORY: conversationHistory || "No previous conversation"
+      CONVERSATION_HISTORY: conversationHistory || "No previous conversation",
+      GLOBAL_KNOWLEDGE_BASE: globalKnowledgeBase,
+      KNOWLEDGE_BASE_CONTEXT: knowledgeBaseContext
     });
+    logger.logLLMPrompt("dashCompPicker", "system", extractPromptText(prompts.system));
+    logger.logLLMPrompt("dashCompPicker", "user", prompts.user);
     logger.debug("[DASH_COMP_REQ] Loaded dash-comp-picker prompts with schema and tools");
     const { apiKey, model } = getApiKeyAndModel(
       anthropicApiKey,
@@ -15446,7 +15475,8 @@ var processDashCompRequest = async (data, components, _sendMessage, anthropicApi
       collections,
       tools,
       dashCompModels,
-      conversationHistory
+      conversationHistory,
+      userId
     );
   }
   if (llmResponse.success && dashboardId && prompt) {