@superatomai/sdk-node 0.0.19-mds → 0.0.20-mds

package/dist/index.mjs

@@ -7700,10 +7700,10 @@ var DEFAULT_AGENT_CONFIG = {
   // will use the provider's default model
   sourceAgentModel: "",
   // will use the provider's default model
-  maxRetries: 3,
-  // 3 retries = 4 total query attempts (1 initial + 3 retries for SQL errors)
-  maxIterations: 15
-  // must accommodate: schema searches (2-3) + query attempts (4) + LLM responses + final
+  maxRetries: 1,
+  // 1 retry = 2 total query attempts (1 initial + 1 retry for SQL errors)
+  maxIterations: 8
+  // schema search (2-3) + query attempts (2) + LLM responses + final
 };
 
 // src/userResponse/utils/component-props-processor.ts
@@ -7867,9 +7867,20 @@ ${executedToolsText}`);
       true
       // Parse as JSON
     );
-    const matchedComponents = result.matchedComponents || [];
+    let matchedComponents = result.matchedComponents || [];
     const layoutTitle = result.layoutTitle || "Dashboard";
     const layoutDescription = result.layoutDescription || "Multi-component dashboard";
+    const complexityMatch = analysisContent.match(/\[COMPLEXITY:\s*(simple|medium|complex)\]/i);
+    const complexity = complexityMatch ? complexityMatch[1].toLowerCase() : "medium";
+    const maxVisualizations = complexity === "simple" ? 1 : complexity === "medium" ? 3 : 5;
+    const markdownComps = matchedComponents.filter((mc) => mc.componentName === "DynamicMarkdownBlock");
+    const vizComps = matchedComponents.filter((mc) => mc.componentName !== "DynamicMarkdownBlock");
+    if (vizComps.length > maxVisualizations) {
+      logger.info(`[AgentComponentGen] Trimming ${vizComps.length} visualizations to ${maxVisualizations} (complexity: ${complexity})`);
+      matchedComponents = [...vizComps.slice(0, maxVisualizations), ...markdownComps];
+    } else {
+      logger.info(`[AgentComponentGen] Complexity: ${complexity} | ${vizComps.length} visualizations (limit: ${maxVisualizations})`);
+    }
     if (result.hasAnswerComponent && result.answerComponent?.componentId) {
       const answer = result.answerComponent;
       const answerSql = answer.props?.externalTool?.parameters?.sql || "";
@@ -10264,7 +10275,7 @@ function getLLMInstance(provider) {
       return anthropicLLM;
   }
 }
-var get_agent_user_response = async (prompt, components, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, conversationHistory, streamCallback, collections, externalTools, userId) => {
+var get_agent_user_response = async (prompt, components, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, conversationHistory, streamCallback, collections, externalTools, userId, mainAgentModel, sourceAgentModel) => {
   const startTime = Date.now();
   const providers = llmProviders || ["anthropic"];
   const provider = providers[0];
@@ -10360,6 +10371,8 @@ var get_agent_user_response = async (prompt, components, anthropicApiKey, groqAp
     const agentConfig = {
       ...DEFAULT_AGENT_CONFIG,
       apiKey: apiKey || void 0,
+      mainAgentModel: mainAgentModel || DEFAULT_AGENT_CONFIG.mainAgentModel,
+      sourceAgentModel: sourceAgentModel || DEFAULT_AGENT_CONFIG.sourceAgentModel,
       globalKnowledgeBase,
       knowledgeBaseContext
     };
@@ -10628,7 +10641,7 @@ var CONTEXT_CONFIG = {
 };
 
 // src/handlers/user-prompt-request.ts
-var get_user_request = async (data, components, sendMessage, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, collections, externalTools) => {
+var get_user_request = async (data, components, sendMessage, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, collections, externalTools, mainAgentModel, sourceAgentModel) => {
   const errors = [];
   const parseResult = UserPromptRequestMessageSchema.safeParse(data);
   if (!parseResult.success) {
@@ -10711,7 +10724,9 @@ var get_user_request = async (data, components, sendMessage, anthropicApiKey, gr
     streamCallback,
     collections,
     externalTools,
-    userId
+    userId,
+    mainAgentModel,
+    sourceAgentModel
   );
   logger.info("User prompt request completed");
   const uiBlockId = existingUiBlockId;
@@ -10811,8 +10826,8 @@ var get_user_request = async (data, components, sendMessage, anthropicApiKey, gr
     wsId
   };
 };
-async function handleUserPromptRequest(data, components, sendMessage, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, collections, externalTools) {
-  const response = await get_user_request(data, components, sendMessage, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, collections, externalTools);
+async function handleUserPromptRequest(data, components, sendMessage, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, collections, externalTools, mainAgentModel, sourceAgentModel) {
+  const response = await get_user_request(data, components, sendMessage, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, collections, externalTools, mainAgentModel, sourceAgentModel);
   if (response.data?.component?.props?.config?.components) {
     response.data.component.props.config.components = response.data.component.props.config.components.map((comp) => ({
       ...comp,
@@ -16550,6 +16565,8 @@ var SuperatomSDK = class {
     this.llmProviders = config.LLM_PROVIDERS || getLLMProviders();
     this.databaseType = config.databaseType || "postgresql";
     this.modelStrategy = config.modelStrategy || "fast";
+    this.mainAgentModel = config.mainAgentModel || "anthropic/claude-sonnet-4-5-20250929";
+    this.sourceAgentModel = config.sourceAgentModel || "anthropic/claude-sonnet-4-5-20250929";
     this.conversationSimilarityThreshold = config.conversationSimilarityThreshold ?? 0.8;
     this.applyModelStrategy(this.modelStrategy);
     this.applyConversationSimilarityThreshold(this.conversationSimilarityThreshold);
@@ -16707,7 +16724,7 @@ var SuperatomSDK = class {
           });
           break;
         case "USER_PROMPT_REQ":
-          handleUserPromptRequest(parsed, this.components, (msg) => this.send(msg), this.anthropicApiKey, this.groqApiKey, this.geminiApiKey, this.openaiApiKey, this.llmProviders, this.collections, this.tools).catch((error) => {
+          handleUserPromptRequest(parsed, this.components, (msg) => this.send(msg), this.anthropicApiKey, this.groqApiKey, this.geminiApiKey, this.openaiApiKey, this.llmProviders, this.collections, this.tools, this.mainAgentModel, this.sourceAgentModel).catch((error) => {
             logger.error("Failed to handle user prompt request:", error);
           });
           break;