@superatomai/sdk-node 0.0.41 → 0.0.42

package/dist/index.d.mts

@@ -628,19 +628,19 @@ declare const ComponentSchema: z.ZodObject<{
         description: z.ZodOptional<z.ZodString>;
         config: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
         actions: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>;
-    }, "strip", z.ZodTypeAny, {
-        description?: string | undefined;
-        query?: string | {} | undefined;
-        title?: string | undefined;
-        config?: Record<string, unknown> | undefined;
-        actions?: any[] | undefined;
-    }, {
-        description?: string | undefined;
-        query?: string | {} | undefined;
-        title?: string | undefined;
-        config?: Record<string, unknown> | undefined;
-        actions?: any[] | undefined;
-    }>;
+    }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+        query: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>]>>;
+        title: z.ZodOptional<z.ZodString>;
+        description: z.ZodOptional<z.ZodString>;
+        config: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+        actions: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>;
+    }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+        query: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>]>>;
+        title: z.ZodOptional<z.ZodString>;
+        description: z.ZodOptional<z.ZodString>;
+        config: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+        actions: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>;
+    }, z.ZodTypeAny, "passthrough">>;
     category: z.ZodOptional<z.ZodString>;
     keywords: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
 }, "strip", z.ZodTypeAny, {
@@ -654,6 +654,8 @@ declare const ComponentSchema: z.ZodObject<{
         title?: string | undefined;
         config?: Record<string, unknown> | undefined;
         actions?: any[] | undefined;
+    } & {
+        [k: string]: unknown;
     };
     displayName?: string | undefined;
     isDisplayComp?: boolean | undefined;
@@ -670,6 +672,8 @@ declare const ComponentSchema: z.ZodObject<{
         title?: string | undefined;
         config?: Record<string, unknown> | undefined;
         actions?: any[] | undefined;
+    } & {
+        [k: string]: unknown;
     };
     displayName?: string | undefined;
     isDisplayComp?: boolean | undefined;
@@ -709,6 +713,22 @@ type DatabaseType = 'postgresql' | 'mssql' | 'snowflake' | 'mysql';
  * - 'balanced': Use best model for complex tasks, fast model for simple tasks (default)
  */
 type ModelStrategy = 'best' | 'fast' | 'balanced';
+/**
+ * Model configuration for DASH_COMP flow (dashboard component picking)
+ * Allows separate control of models used for component selection
+ */
+interface DashCompModelConfig {
+    /**
+     * Primary model for DASH_COMP requests
+     * Format: "provider/model-name" (e.g., "anthropic/claude-sonnet-4-5-20250929")
+     */
+    model?: string;
+    /**
+     * Fast model for simpler DASH_COMP tasks (optional)
+     * Format: "provider/model-name" (e.g., "anthropic/claude-haiku-4-5-20251001")
+     */
+    fastModel?: string;
+}
 interface SuperatomSDKConfig {
     url?: string;
     apiKey?: string;
@@ -731,6 +751,11 @@ interface SuperatomSDKConfig {
      * - 'balanced': Use best model for complex tasks, fast model for simple tasks (default)
      */
     modelStrategy?: ModelStrategy;
+    /**
+     * Separate model configuration for DASH_COMP flow (dashboard component picking)
+     * If not provided, falls back to provider-based model selection
+     */
+    dashCompModels?: DashCompModelConfig;
 }
 
 declare const KbNodesQueryFiltersSchema: z.ZodObject<{

package/dist/index.d.ts

@@ -628,19 +628,19 @@ declare const ComponentSchema: z.ZodObject<{
         description: z.ZodOptional<z.ZodString>;
         config: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
         actions: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>;
-    }, "strip", z.ZodTypeAny, {
-        description?: string | undefined;
-        query?: string | {} | undefined;
-        title?: string | undefined;
-        config?: Record<string, unknown> | undefined;
-        actions?: any[] | undefined;
-    }, {
-        description?: string | undefined;
-        query?: string | {} | undefined;
-        title?: string | undefined;
-        config?: Record<string, unknown> | undefined;
-        actions?: any[] | undefined;
-    }>;
+    }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+        query: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>]>>;
+        title: z.ZodOptional<z.ZodString>;
+        description: z.ZodOptional<z.ZodString>;
+        config: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+        actions: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>;
+    }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+        query: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>]>>;
+        title: z.ZodOptional<z.ZodString>;
+        description: z.ZodOptional<z.ZodString>;
+        config: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+        actions: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>;
+    }, z.ZodTypeAny, "passthrough">>;
     category: z.ZodOptional<z.ZodString>;
     keywords: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
 }, "strip", z.ZodTypeAny, {
@@ -654,6 +654,8 @@ declare const ComponentSchema: z.ZodObject<{
         title?: string | undefined;
         config?: Record<string, unknown> | undefined;
         actions?: any[] | undefined;
+    } & {
+        [k: string]: unknown;
     };
     displayName?: string | undefined;
     isDisplayComp?: boolean | undefined;
@@ -670,6 +672,8 @@ declare const ComponentSchema: z.ZodObject<{
         title?: string | undefined;
         config?: Record<string, unknown> | undefined;
         actions?: any[] | undefined;
+    } & {
+        [k: string]: unknown;
     };
     displayName?: string | undefined;
     isDisplayComp?: boolean | undefined;
@@ -709,6 +713,22 @@ type DatabaseType = 'postgresql' | 'mssql' | 'snowflake' | 'mysql';
  * - 'balanced': Use best model for complex tasks, fast model for simple tasks (default)
  */
 type ModelStrategy = 'best' | 'fast' | 'balanced';
+/**
+ * Model configuration for DASH_COMP flow (dashboard component picking)
+ * Allows separate control of models used for component selection
+ */
+interface DashCompModelConfig {
+    /**
+     * Primary model for DASH_COMP requests
+     * Format: "provider/model-name" (e.g., "anthropic/claude-sonnet-4-5-20250929")
+     */
+    model?: string;
+    /**
+     * Fast model for simpler DASH_COMP tasks (optional)
+     * Format: "provider/model-name" (e.g., "anthropic/claude-haiku-4-5-20251001")
+     */
+    fastModel?: string;
+}
 interface SuperatomSDKConfig {
     url?: string;
     apiKey?: string;
@@ -731,6 +751,11 @@ interface SuperatomSDKConfig {
      * - 'balanced': Use best model for complex tasks, fast model for simple tasks (default)
      */
     modelStrategy?: ModelStrategy;
+    /**
+     * Separate model configuration for DASH_COMP flow (dashboard component picking)
+     * If not provided, falls back to provider-based model selection
+     */
+    dashCompModels?: DashCompModelConfig;
 }
 
 declare const KbNodesQueryFiltersSchema: z.ZodObject<{

package/dist/index.js

@@ -1933,7 +1933,7 @@ var ComponentPropsSchema = import_zod3.z.object({
   description: import_zod3.z.string().optional(),
   config: import_zod3.z.record(import_zod3.z.unknown()).optional(),
   actions: import_zod3.z.array(import_zod3.z.any()).optional()
-});
+}).passthrough();
 var ComponentSchema = import_zod3.z.object({
   id: import_zod3.z.string(),
   name: import_zod3.z.string(),
@@ -3458,9 +3458,9 @@ var PRICING = {
   "gpt-4": { input: 30, output: 60 },
   "gpt-3.5-turbo": { input: 0.5, output: 1.5 },
   // Google Gemini (December 2025)
-  "gemini-3-pro": { input: 2, output: 8 },
+  "gemini-3-pro-preview": { input: 2, output: 12 },
   // New Gemini 3
-  "gemini-2.5-pro": { input: 1.25, output: 10 },
+  "gemini-3-flash-preview": { input: 0.5, output: 3 },
   // For prompts ≤200K tokens, 2x for >200K
   "gemini-2.5-flash": { input: 0.15, output: 0.6 },
   // Standard mode (thinking disabled: $0.60, thinking enabled: $3.50)
@@ -4645,6 +4645,7 @@ var LLM = class {
     }
   }
   static async _geminiStreamWithTools(messages, tools, toolHandler, modelName, options, maxIterations) {
+    const methodStartTime = Date.now();
     const apiKey = options.apiKey || process.env.GEMINI_API_KEY || "";
     const genAI = new import_generative_ai.GoogleGenerativeAI(apiKey);
     const systemPrompt = typeof messages.sys === "string" ? messages.sys : messages.sys.map((block) => block.text).join("\n");
@@ -4673,11 +4674,18 @@ var LLM = class {
     let iterations = 0;
     let finalText = "";
     let currentUserMessage = messages.user;
+    let totalToolCalls = 0;
+    let totalInputTokens = 0;
+    let totalOutputTokens = 0;
     while (iterations < maxIterations) {
       iterations++;
+      const iterationStartTime = Date.now();
+      const requestId = llmUsageLogger.generateRequestId();
       const result = await chat.sendMessageStream(currentUserMessage);
       let responseText = "";
       const functionCalls = [];
+      let inputTokens = 0;
+      let outputTokens = 0;
       for await (const chunk of result.stream) {
         const candidate = chunk.candidates?.[0];
         if (!candidate) continue;
@@ -4694,7 +4702,37 @@ var LLM = class {
             });
           }
         }
+        if (chunk.usageMetadata) {
+          inputTokens = chunk.usageMetadata.promptTokenCount || 0;
+          outputTokens = chunk.usageMetadata.candidatesTokenCount || 0;
+        }
       }
+      const iterationDuration = Date.now() - iterationStartTime;
+      const toolCallsInIteration = functionCalls.length;
+      totalToolCalls += toolCallsInIteration;
+      if (inputTokens === 0) {
+        const userMsg = typeof currentUserMessage === "string" ? currentUserMessage : JSON.stringify(currentUserMessage);
+        inputTokens = Math.ceil((systemPrompt.length + userMsg.length) / 4);
+      }
+      if (outputTokens === 0) {
+        outputTokens = Math.ceil(responseText.length / 4);
+      }
+      totalInputTokens += inputTokens;
+      totalOutputTokens += outputTokens;
+      llmUsageLogger.log({
+        timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+        requestId,
+        provider: "gemini",
+        model: modelName,
+        method: `streamWithTools[iter=${iterations}]`,
+        inputTokens,
+        outputTokens,
+        totalTokens: inputTokens + outputTokens,
+        costUSD: llmUsageLogger.calculateCost(modelName, inputTokens, outputTokens),
+        durationMs: iterationDuration,
+        toolCalls: toolCallsInIteration,
+        success: true
+      });
       if (functionCalls.length === 0) {
         finalText = responseText;
         break;
@@ -4727,6 +4765,23 @@ var LLM = class {
       }));
       currentUserMessage = functionResponseParts;
     }
+    const totalDuration = Date.now() - methodStartTime;
+    if (iterations > 1) {
+      llmUsageLogger.log({
+        timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+        requestId: llmUsageLogger.generateRequestId(),
+        provider: "gemini",
+        model: modelName,
+        method: `streamWithTools[TOTAL:${iterations}iters]`,
+        inputTokens: totalInputTokens,
+        outputTokens: totalOutputTokens,
+        totalTokens: totalInputTokens + totalOutputTokens,
+        costUSD: llmUsageLogger.calculateCost(modelName, totalInputTokens, totalOutputTokens),
+        durationMs: totalDuration,
+        toolCalls: totalToolCalls,
+        success: true
+      });
+    }
     if (iterations >= maxIterations) {
       throw new Error(`Max iterations (${maxIterations}) reached in tool calling loop`);
     }
@@ -6781,10 +6836,10 @@ var GeminiLLM = class extends BaseLLM {
     super(config);
   }
   getDefaultModel() {
-    return "gemini/gemini-2.5-flash";
+    return "gemini/gemini-3-pro-preview";
   }
   getDefaultFastModel() {
-    return "gemini/gemini-2.0-flash-exp";
+    return "gemini/gemini-3-flash-preview";
   }
   getDefaultApiKey() {
     return process.env.GEMINI_API_KEY;
@@ -10164,7 +10219,13 @@ function sendResponse8(id, res, sendMessage, clientId) {
 // src/handlers/dash-comp-request.ts
 init_logger();
 init_prompt_loader();
-async function pickComponentWithLLM(prompt, components, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, _collections, tools) {
+var DEFAULT_DASH_COMP_MODELS = {
+  anthropic: "anthropic/claude-haiku-4-5-20251001",
+  gemini: "gemini/gemini-3-flash-preview",
+  openai: "openai/gpt-4o-mini",
+  groq: "groq/llama-3.3-70b-versatile"
+};
+async function pickComponentWithLLM(prompt, components, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, _collections, tools, dashCompModels) {
   const errors = [];
   let availableComponentsText = "No components available";
   if (components && components.length > 0) {
@@ -10202,27 +10263,37 @@ async function pickComponentWithLLM(prompt, components, anthropicApiKey, groqApi
     logger.debug("[DASH_COMP_REQ] Loaded dash-comp-picker prompts with schema and tools");
     const providers = llmProviders || ["anthropic", "gemini", "openai", "groq"];
     let apiKey;
-    let model = "anthropic/claude-sonnet-4-5-20250929";
-    for (const provider of providers) {
-      if (provider === "anthropic" && anthropicApiKey) {
-        apiKey = anthropicApiKey;
-        model = "anthropic/claude-sonnet-4-5-20250929";
-        break;
-      } else if (provider === "openai" && openaiApiKey) {
-        apiKey = openaiApiKey;
-        model = "openai/gpt-4o-mini";
-        break;
-      } else if (provider === "gemini" && geminiApiKey) {
-        apiKey = geminiApiKey;
-        model = "google/gemini-2.0-flash-001";
-        break;
-      } else if (provider === "groq" && groqApiKey) {
-        apiKey = groqApiKey;
-        model = "groq/llama-3.3-70b-versatile";
-        break;
+    let model;
+    if (dashCompModels?.model) {
+      model = dashCompModels.model;
+      const modelProvider = model.split("/")[0];
+      if (modelProvider === "anthropic") apiKey = anthropicApiKey;
+      else if (modelProvider === "gemini") apiKey = geminiApiKey;
+      else if (modelProvider === "openai") apiKey = openaiApiKey;
+      else if (modelProvider === "groq") apiKey = groqApiKey;
+      logger.info(`[DASH_COMP_REQ] Using configured model: ${model}`);
+    } else {
+      for (const provider of providers) {
+        if (provider === "anthropic" && anthropicApiKey) {
+          apiKey = anthropicApiKey;
+          model = DEFAULT_DASH_COMP_MODELS.anthropic;
+          break;
+        } else if (provider === "gemini" && geminiApiKey) {
+          apiKey = geminiApiKey;
+          model = DEFAULT_DASH_COMP_MODELS.gemini;
+          break;
+        } else if (provider === "openai" && openaiApiKey) {
+          apiKey = openaiApiKey;
+          model = DEFAULT_DASH_COMP_MODELS.openai;
+          break;
+        } else if (provider === "groq" && groqApiKey) {
+          apiKey = groqApiKey;
+          model = DEFAULT_DASH_COMP_MODELS.groq;
+          break;
+        }
       }
     }
-    if (!apiKey) {
+    if (!apiKey || !model) {
       errors.push("No API key available for any LLM provider");
       return { success: false, errors };
     }
@@ -10245,11 +10316,21 @@ async function pickComponentWithLLM(prompt, components, anthropicApiKey, groqApi
     logger.file("[DASH_COMP_REQ] LLM response:", JSON.stringify(result, null, 2));
     if (!result.componentId || !result.props) {
       errors.push("Invalid LLM response: missing componentId or props");
+      userPromptErrorLogger.logError("DASH_COMP_REQ", "Invalid LLM response structure", {
+        prompt,
+        result,
+        missingFields: { componentId: !result.componentId, props: !result.props }
+      });
       return { success: false, errors };
     }
     const originalComponent = components.find((c) => c.id === result.componentId);
     if (!originalComponent) {
       errors.push(`Component ${result.componentId} not found in available components`);
+      userPromptErrorLogger.logError("DASH_COMP_REQ", "Component not found", {
+        prompt,
+        componentId: result.componentId,
+        availableComponentIds: components.map((c) => c.id)
+      });
       return { success: false, errors };
     }
     const finalComponent = {
@@ -10278,11 +10359,16 @@ async function pickComponentWithLLM(prompt, components, anthropicApiKey, groqApi
   } catch (error) {
     const errorMsg = error instanceof Error ? error.message : String(error);
     logger.error(`[DASH_COMP_REQ] Error picking component: ${errorMsg}`);
+    userPromptErrorLogger.logError("DASH_COMP_REQ", error instanceof Error ? error : new Error(errorMsg), {
+      prompt,
+      componentsCount: components.length,
+      toolsCount: tools?.length || 0
+    });
     errors.push(errorMsg);
     return { success: false, errors };
   }
 }
-var processDashCompRequest = async (data, components, _sendMessage, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, collections, tools) => {
+var processDashCompRequest = async (data, components, _sendMessage, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, collections, tools, dashCompModels) => {
   const errors = [];
   logger.debug("[DASH_COMP_REQ] Parsing incoming message data");
   const parseResult = DashCompRequestMessageSchema.safeParse(data);
@@ -10297,6 +10383,8 @@ var processDashCompRequest = async (data, components, _sendMessage, anthropicApi
   const { id, payload } = dashCompRequest;
   const prompt = payload.prompt;
   const wsId = dashCompRequest.from.id || "unknown";
+  const promptContext = `DASH_COMP: ${prompt?.substring(0, 50)}${(prompt?.length || 0) > 50 ? "..." : ""}`;
+  llmUsageLogger.resetLogFile(promptContext);
   if (!prompt) {
     errors.push("Prompt is required");
   }
@@ -10323,8 +10411,10 @@ var processDashCompRequest = async (data, components, _sendMessage, anthropicApi
     openaiApiKey,
     llmProviders,
     collections,
-    tools
+    tools,
+    dashCompModels
   );
+  llmUsageLogger.logSessionSummary(`DASH_COMP: ${prompt?.substring(0, 30)}`);
   return {
     success: llmResponse.success,
     data: llmResponse.data,
@@ -10333,7 +10423,7 @@ var processDashCompRequest = async (data, components, _sendMessage, anthropicApi
     wsId
   };
 };
-async function handleDashCompRequest(data, components, sendMessage, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, collections, tools) {
+async function handleDashCompRequest(data, components, sendMessage, anthropicApiKey, groqApiKey, geminiApiKey, openaiApiKey, llmProviders, collections, tools, dashCompModels) {
   const response = await processDashCompRequest(
     data,
     components,
@@ -10344,7 +10434,8 @@ async function handleDashCompRequest(data, components, sendMessage, anthropicApi
     openaiApiKey,
     llmProviders,
     collections,
-    tools
+    tools,
+    dashCompModels
   );
   sendDashCompResponse(
     response.id || data.id,