@superatomai/sdk-node 0.0.41 → 0.0.42

package/dist/index.mjs

@@ -1883,7 +1883,7 @@ var ComponentPropsSchema = z3.object({
   description: z3.string().optional(),
   config: z3.record(z3.unknown()).optional(),
   actions: z3.array(z3.any()).optional()
-});
+}).passthrough();
 var ComponentSchema = z3.object({
   id: z3.string(),
   name: z3.string(),
@@ -3408,9 +3408,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)
@@ -4595,6 +4595,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 GoogleGenerativeAI(apiKey);
     const systemPrompt = typeof messages.sys === "string" ? messages.sys : messages.sys.map((block) => block.text).join("\n");
@@ -4623,11 +4624,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;
@@ -4644,7 +4652,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;
@@ -4677,6 +4715,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`);
     }
@@ -6731,10 +6786,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;
@@ -10114,7 +10169,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) {
@@ -10152,27 +10213,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 };
     }
@@ -10195,11 +10266,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 = {
@@ -10228,11 +10309,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);
@@ -10247,6 +10333,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");
   }
@@ -10273,8 +10361,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,
@@ -10283,7 +10373,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,
@@ -10294,7 +10384,8 @@ async function handleDashCompRequest(data, components, sendMessage, anthropicApi
     openaiApiKey,
     llmProviders,
     collections,
-    tools
+    tools,
+    dashCompModels
   );
   sendDashCompResponse(
     response.id || data.id,