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

package/dist/index.d.mts

@@ -860,6 +860,18 @@ interface SuperatomSDKConfig {
      * - 'balanced': Use best model for complex tasks, fast model for simple tasks (default)
      */
     modelStrategy?: ModelStrategy;
+    /**
+     * Model for the main agent (routing + analysis).
+     * Format: "provider/model-name" (e.g., "anthropic/claude-haiku-4-5-20251001")
+     * If not set, uses the provider's default model.
+     */
+    mainAgentModel?: string;
+    /**
+     * Model for source agents (per-source query generation).
+     * Format: "provider/model-name" (e.g., "anthropic/claude-haiku-4-5-20251001")
+     * If not set, uses the provider's default model.
+     */
+    sourceAgentModel?: string;
     /**
      * Separate model configuration for DASH_COMP flow (dashboard component picking)
      * If not provided, falls back to provider-based model selection
@@ -2754,6 +2766,8 @@ declare class SuperatomSDK {
     private llmProviders;
     private databaseType;
     private modelStrategy;
+    private mainAgentModel;
+    private sourceAgentModel;
     private conversationSimilarityThreshold;
     private userManager;
     private dashboardManager;

package/dist/index.d.ts

@@ -860,6 +860,18 @@ interface SuperatomSDKConfig {
      * - 'balanced': Use best model for complex tasks, fast model for simple tasks (default)
      */
     modelStrategy?: ModelStrategy;
+    /**
+     * Model for the main agent (routing + analysis).
+     * Format: "provider/model-name" (e.g., "anthropic/claude-haiku-4-5-20251001")
+     * If not set, uses the provider's default model.
+     */
+    mainAgentModel?: string;
+    /**
+     * Model for source agents (per-source query generation).
+     * Format: "provider/model-name" (e.g., "anthropic/claude-haiku-4-5-20251001")
+     * If not set, uses the provider's default model.
+     */
+    sourceAgentModel?: string;
     /**
      * Separate model configuration for DASH_COMP flow (dashboard component picking)
      * If not provided, falls back to provider-based model selection
@@ -2754,6 +2766,8 @@ declare class SuperatomSDK {
     private llmProviders;
     private databaseType;
     private modelStrategy;
+    private mainAgentModel;
+    private sourceAgentModel;
     private conversationSimilarityThreshold;
     private userManager;
     private dashboardManager;

package/dist/index.js

@@ -7760,10 +7760,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
@@ -7927,9 +7927,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 || "";
@@ -10324,7 +10335,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];
@@ -10420,6 +10431,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
     };
@@ -10688,7 +10701,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) {
@@ -10771,7 +10784,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;
@@ -10871,8 +10886,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,
@@ -16610,6 +16625,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);
@@ -16767,7 +16784,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;