@superatomai/sdk-node 0.0.36-mds → 0.0.37-mds

package/dist/index.js

@@ -10724,6 +10724,77 @@ var get_agent_user_response = async (prompt, components, anthropicApiKey, groqAp
   }
 };
 
+// src/utils/analytics-client.ts
+var AnalyticsClient = class {
+  constructor() {
+    this.apiUrl = null;
+    this.orgId = null;
+    this.enabled = true;
+    this.initialized = false;
+  }
+  init() {
+    if (this.initialized) return;
+    this.initialized = true;
+    this.apiUrl = process.env.SA_API_URL || null;
+    this.orgId = process.env.SA_ORG_ID || null;
+    this.enabled = process.env.SA_ANALYTICS_ENABLED !== "false";
+    if (!this.apiUrl) {
+      if (this.enabled) {
+        logger.warn("[Analytics] Disabled \u2014 SA_API_URL not configured");
+      }
+      this.enabled = false;
+    }
+  }
+  /**
+   * Track a chat analytics event. Fire-and-forget — never blocks or throws.
+   */
+  async track(event) {
+    this.init();
+    if (!this.enabled) return;
+    try {
+      const response = await fetch(`${this.apiUrl}/analytics/chat`, {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/json"
+        },
+        body: JSON.stringify(event)
+      });
+      if (!response.ok) {
+        const errorText = await response.text().catch(() => "unknown");
+        logger.warn(`[Analytics] Failed to track event: ${response.status} ${errorText}`);
+      }
+    } catch (err) {
+      logger.warn(`[Analytics] Failed to send event: ${err instanceof Error ? err.message : String(err)}`);
+    }
+  }
+  /**
+   * Build a ChatAnalyticsEvent from the available request context.
+   * Reads session stats from llmUsageLogger automatically.
+   */
+  buildEvent(params) {
+    this.init();
+    const stats = llmUsageLogger.getSessionStats();
+    return {
+      userId: params.userId,
+      orgId: this.orgId || void 0,
+      projectId: params.projectId,
+      threadId: params.threadId,
+      messageIndex: params.messageIndex,
+      question: params.question,
+      sourcesUsed: params.sourcesUsed,
+      sqlGenerated: params.sqlGenerated,
+      model: params.model,
+      inputTokens: stats.totalInputTokens,
+      outputTokens: stats.totalOutputTokens,
+      cost: stats.totalCostUSD.toFixed(6),
+      latencyMs: params.latencyMs,
+      status: params.success ? "success" : "error",
+      errorMessage: params.errorMessage
+    };
+  }
+};
+var analyticsClient = new AnalyticsClient();
+
 // src/utils/conversation-saver.ts
 function transformUIBlockForDB(uiblock, userPrompt, uiBlockId) {
   const component = uiblock?.generatedComponentMetadata && Object.keys(uiblock.generatedComponentMetadata).length > 0 ? uiblock.generatedComponentMetadata : null;
@@ -10886,6 +10957,7 @@ var get_user_request = async (data, components, sendMessage, anthropicApiKey, gr
     thread = threadManager.createThread(threadId);
     logger.info(`Created new thread: ${threadId}`);
   }
+  const requestStartTime = Date.now();
   logger.info(`Starting user prompt request with ${components.length} components`);
   const conversationHistory = thread.getConversationContext(CONTEXT_CONFIG.MAX_CONVERSATION_CONTEXT_BLOCKS, existingUiBlockId);
   const responseMode = payload.responseMode || "component";
@@ -10943,6 +11015,23 @@ var get_user_request = async (data, components, sendMessage, anthropicApiKey, gr
     });
     userPromptErrorLogger.writeSummary();
     llmUsageLogger.logSessionSummary(`FAILED: ${prompt?.substring(0, 30)}`);
+    if (userId) {
+      const latencyMs = Date.now() - requestStartTime;
+      const model = mainAgentModel || "unknown";
+      const event = analyticsClient.buildEvent({
+        userId,
+        projectId: process.env.SUPERATOM_PROJECT_ID || "",
+        threadId,
+        messageIndex: 0,
+        question: prompt,
+        success: false,
+        latencyMs,
+        model,
+        errorMessage: userResponse.errors.join("; ")
+      });
+      analyticsClient.track(event).catch(() => {
+      });
+    }
     return {
       success: false,
       data: userResponse.data,
@@ -11019,6 +11108,42 @@ var get_user_request = async (data, components, sendMessage, anthropicApiKey, gr
     }
   }
   llmUsageLogger.logSessionSummary(prompt?.substring(0, 50));
+  if (userId) {
+    const latencyMs = Date.now() - requestStartTime;
+    const model = mainAgentModel || "unknown";
+    const componentConfig = userResponse.data?.component?.props?.config;
+    const queryMap = componentConfig?._queryMap || {};
+    const sqlStatements = Object.values(queryMap).filter((v) => typeof v === "string");
+    const childComponents = componentConfig?.components || [];
+    const sourceSet = /* @__PURE__ */ new Map();
+    for (const child of childComponents) {
+      const toolId = child?.props?.externalTool?.toolId;
+      const toolName = child?.props?.externalTool?.name;
+      if (toolId && !sourceSet.has(toolId)) {
+        sourceSet.set(toolId, {
+          sourceId: toolId,
+          sourceName: toolName || toolId,
+          sourceType: toolId.split("-")[0] || ""
+          // e.g. "postgres" from "postgres-xxx_query"
+        });
+      }
+    }
+    const sourcesUsed = Array.from(sourceSet.values());
+    const event = analyticsClient.buildEvent({
+      userId,
+      projectId: process.env.SUPERATOM_PROJECT_ID || "",
+      threadId,
+      messageIndex: thread.getUIBlockCount(),
+      question: prompt,
+      success: true,
+      latencyMs,
+      model,
+      sourcesUsed: sourcesUsed.length > 0 ? sourcesUsed : void 0,
+      sqlGenerated: sqlStatements.length > 0 ? sqlStatements.join(";\n") : void 0
+    });
+    analyticsClient.track(event).catch(() => {
+    });
+  }
   return {
     success: userResponse.success,
     data: userResponse.data,