whale-code 6.5.4 → 6.5.6

package/dist/server/handlers/platform.js

@@ -1,73 +1,95 @@
 import { getClickHouseClient } from "../lib/clickhouse-client.js";
+
 /** Escape single quotes for ClickHouse SQL */
 function esc(s) {
-    return s.replace(/'/g, "\\'");
+  return s.replace(/'/g, "\\'");
 }
 export async function handleWebSearch(sb, args, _storeId) {
-    const query = args.query;
-    const numResults = args.num_results || 5;
-    // Validate input first (cheap, no DB call)
-    if (!query) {
-        return { success: false, error: "Query parameter is required" };
-    }
-    // Read from platform_secrets table first, fall back to env var
-    const { data: secret } = await sb.from("platform_secrets").select("value").eq("key", "exa_api_key").single();
-    const exaApiKey = secret?.value || process.env["EXA_API_KEY"];
-    if (!exaApiKey) {
-        return { success: false, error: "Exa API key not configured. Add 'exa_api_key' to platform_secrets table." };
-    }
-    try {
-        const response = await fetch("https://api.exa.ai/search", {
-            method: "POST",
-            headers: {
-                "Content-Type": "application/json",
-                "x-api-key": exaApiKey
-            },
-            body: JSON.stringify({
-                query,
-                numResults,
-                useAutoprompt: true,
-                type: "auto"
-            })
-        });
-        if (!response.ok) {
-            const errorText = await response.text();
-            return { success: false, error: `Exa API error: ${response.status} - ${errorText}` };
-        }
-        const data = await response.json();
-        return {
-            success: true,
-            data: {
-                query,
-                results: data.results || [],
-                autopromptString: data.autopromptString
-            }
-        };
-    }
-    catch (err) {
-        return { success: false, error: `Web search failed: ${err}` };
+  const query = args.query;
+  const numResults = args.num_results || 5;
+
+  // Validate input first (cheap, no DB call)
+  if (!query) {
+    return {
+      success: false,
+      error: "Query parameter is required"
+    };
+  }
+
+  // Read from platform_secrets table first, fall back to env var
+  const {
+    data: secret
+  } = await sb.from("platform_secrets").select("value").eq("key", "exa_api_key").single();
+  const exaApiKey = secret?.value || process.env["EXA_API_KEY"];
+  if (!exaApiKey) {
+    return {
+      success: false,
+      error: "Exa API key not configured. Add 'exa_api_key' to platform_secrets table."
+    };
+  }
+  try {
+    const response = await fetch("https://api.exa.ai/search", {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "x-api-key": exaApiKey
+      },
+      body: JSON.stringify({
+        query,
+        numResults,
+        useAutoprompt: true,
+        type: "auto"
+      })
+    });
+    if (!response.ok) {
+      const errorText = await response.text();
+      return {
+        success: false,
+        error: `Exa API error: ${response.status} - ${errorText}`
+      };
     }
+    const data = await response.json();
+    return {
+      success: true,
+      data: {
+        query,
+        results: data.results || [],
+        autopromptString: data.autopromptString
+      }
+    };
+  } catch (err) {
+    return {
+      success: false,
+      error: `Web search failed: ${err}`
+    };
+  }
 }
 export async function handleTelemetry(sb, args, storeId) {
-    const sid = storeId;
-    const hoursBack = args.hours_back || 24;
-    const limit = Math.min(args.limit || 50, 200);
-    switch (args.action) {
-        // ---- conversation_detail: Full conversation with messages + ClickHouse spans ----
-        case "conversation_detail": {
-            const convId = args.conversation_id;
-            if (!convId)
-                return { success: false, error: "conversation_id is required" };
-            // Verify conversation belongs to this store first
-            const convResult = await sb.from("ai_conversations").select("*").eq("id", convId).eq("store_id", sid).single();
-            if (convResult.error)
-                return { success: false, error: convResult.error.message };
-            // Fetch messages from Postgres + spans from ClickHouse in parallel
-            const ch = getClickHouseClient();
-            const [msgResult, spans] = await Promise.all([
-                sb.from("ai_messages").select("*").eq("conversation_id", convId).order("created_at", { ascending: true }),
-                ch.isEnabled
-                    ? ch.query(`
+  const sid = storeId;
+  const hoursBack = args.hours_back || 24;
+  const limit = Math.min(args.limit || 50, 200);
+  switch (args.action) {
+    // ---- conversation_detail: Full conversation with messages + ClickHouse spans ----
+    case "conversation_detail":
+      {
+        const convId = args.conversation_id;
+        if (!convId) return {
+          success: false,
+          error: "conversation_id is required"
+        };
+
+        // Verify conversation belongs to this store first
+        const convResult = await sb.from("ai_conversations").select("*").eq("id", convId).eq("store_id", sid).single();
+        if (convResult.error) return {
+          success: false,
+          error: convResult.error.message
+        };
+
+        // Fetch messages from Postgres + spans from ClickHouse in parallel
+        const ch = getClickHouseClient();
+        const [msgResult, spans] = await Promise.all([sb.from("ai_messages").select("*").eq("conversation_id", convId).order("created_at", {
+          ascending: true
+        }), ch.isEnabled ? ch.query(`
               SELECT
                 span_id AS id, operation_name AS action, severity,
                 duration_ms, status_code, error_message,
@@ -78,55 +100,87 @@ export async function handleTelemetry(sb, args, storeId) {
               WHERE conversation_id = '${esc(convId)}'
               ORDER BY started_at ASC
               LIMIT 200
-            `)
-                    : Promise.resolve([]),
-            ]);
-            return {
-                success: true,
-                data: {
-                    conversation: convResult.data,
-                    messages: msgResult.data || [],
-                    audit_entries: spans,
-                    message_count: msgResult.data?.length || 0,
-                    audit_count: spans.length,
-                },
-            };
-        }
-        // ---- conversations: List recent conversations ----
-        case "conversations": {
-            const convLimit = Math.min(args.limit || 20, 100);
-            let q = sb.from("ai_conversations")
-                .select("*")
-                .eq("store_id", sid)
-                .gte("created_at", new Date(Date.now() - hoursBack * 3600_000).toISOString())
-                .order("created_at", { ascending: false })
-                .limit(convLimit);
-            if (args.agent_id)
-                q = q.eq("agent_id", args.agent_id);
-            const { data, error } = await q;
-            return error ? { success: false, error: error.message } : { success: true, data: { count: data?.length || 0, conversations: data } };
-        }
-        // ---- agent_performance: Agent-level analytics via RPC ----
-        case "agent_performance": {
-            const agentId = args.agent_id;
-            if (!agentId)
-                return { success: false, error: "agent_id is required" };
-            // Verify agent belongs to this store
-            const { data: conv } = await sb.from("ai_conversations").select("id").eq("agent_id", agentId).eq("store_id", sid).limit(1);
-            if (!conv?.length)
-                return { success: false, error: "Agent not found for this store" };
-            const days = args.days || 7;
-            const { data, error } = await sb.rpc("get_agent_analytics", { p_agent_id: agentId, p_days: days });
-            return error ? { success: false, error: error.message } : { success: true, data };
-        }
-        // ---- tool_analytics: Per-tool performance metrics via direct ClickHouse ----
-        case "tool_analytics": {
-            const ch = getClickHouseClient();
-            if (!ch.isEnabled)
-                return { success: false, error: "ClickHouse not configured" };
-            const toolFilter = args.tool_name ? `AND tool_name = '${esc(args.tool_name)}'` : "";
-            const storeF = sid ? `AND store_id = '${esc(sid)}'` : "";
-            const tools = await ch.query(`
+            `) : Promise.resolve([])]);
+        return {
+          success: true,
+          data: {
+            conversation: convResult.data,
+            messages: msgResult.data || [],
+            audit_entries: spans,
+            message_count: msgResult.data?.length || 0,
+            audit_count: spans.length
+          }
+        };
+      }
+
+    // ---- conversations: List recent conversations ----
+    case "conversations":
+      {
+        const convLimit = Math.min(args.limit || 20, 100);
+        let q = sb.from("ai_conversations").select("*").eq("store_id", sid).gte("created_at", new Date(Date.now() - hoursBack * 3600_000).toISOString()).order("created_at", {
+          ascending: false
+        }).limit(convLimit);
+        if (args.agent_id) q = q.eq("agent_id", args.agent_id);
+        const {
+          data,
+          error
+        } = await q;
+        return error ? {
+          success: false,
+          error: error.message
+        } : {
+          success: true,
+          data: {
+            count: data?.length || 0,
+            conversations: data
+          }
+        };
+      }
+
+    // ---- agent_performance: Agent-level analytics via RPC ----
+    case "agent_performance":
+      {
+        const agentId = args.agent_id;
+        if (!agentId) return {
+          success: false,
+          error: "agent_id is required"
+        };
+        // Verify agent belongs to this store
+        const {
+          data: conv
+        } = await sb.from("ai_conversations").select("id").eq("agent_id", agentId).eq("store_id", sid).limit(1);
+        if (!conv?.length) return {
+          success: false,
+          error: "Agent not found for this store"
+        };
+        const days = args.days || 7;
+        const {
+          data,
+          error
+        } = await sb.rpc("get_agent_analytics", {
+          p_agent_id: agentId,
+          p_days: days
+        });
+        return error ? {
+          success: false,
+          error: error.message
+        } : {
+          success: true,
+          data
+        };
+      }
+
+    // ---- tool_analytics: Per-tool performance metrics via direct ClickHouse ----
+    case "tool_analytics":
+      {
+        const ch = getClickHouseClient();
+        if (!ch.isEnabled) return {
+          success: false,
+          error: "ClickHouse not configured"
+        };
+        const toolFilter = args.tool_name ? `AND tool_name = '${esc(args.tool_name)}'` : "";
+        const storeF = sid ? `AND store_id = '${esc(sid)}'` : "";
+        const tools = await ch.query(`
         SELECT
           tool_name,
           count() AS total_calls,
@@ -153,29 +207,39 @@ export async function handleTelemetry(sb, args, storeId) {
         GROUP BY tool_name
         ORDER BY total_calls DESC
       `);
-            const totalCalls = tools.reduce((s, t) => s + (t.total_calls || 0), 0);
-            const totalErrors = tools.reduce((s, t) => s + (t.error_count || 0), 0);
-            const totalTimeouts = tools.reduce((s, t) => s + (t.timeout_count || 0), 0);
-            const totalCost = tools.reduce((s, t) => s + (t.total_marginal_cost || 0), 0);
-            return { success: true, data: {
-                    tools,
-                    summary: {
-                        total_calls: totalCalls, total_errors: totalErrors, total_timeouts: totalTimeouts,
-                        overall_error_rate: totalCalls > 0 ? Math.round(10000 * totalErrors / totalCalls) / 100 : 0,
-                        unique_tools: tools.length, total_marginal_cost: Math.round(totalCost * 1e6) / 1e6,
-                        hours_analyzed: hoursBack,
-                    },
-                } };
-        }
-        // ---- tool_timeline: Time-bucketed tool metrics via direct ClickHouse ----
-        case "tool_timeline": {
-            const ch = getClickHouseClient();
-            if (!ch.isEnabled)
-                return { success: false, error: "ClickHouse not configured" };
-            const bucketMinutes = args.bucket_minutes || 15;
-            const toolFilter = args.tool_name ? `AND tool_name = '${esc(args.tool_name)}'` : "";
-            const storeF = sid ? `AND store_id = '${esc(sid)}'` : "";
-            const buckets = await ch.query(`
+        const totalCalls = tools.reduce((s, t) => s + (t.total_calls || 0), 0);
+        const totalErrors = tools.reduce((s, t) => s + (t.error_count || 0), 0);
+        const totalTimeouts = tools.reduce((s, t) => s + (t.timeout_count || 0), 0);
+        const totalCost = tools.reduce((s, t) => s + (t.total_marginal_cost || 0), 0);
+        return {
+          success: true,
+          data: {
+            tools,
+            summary: {
+              total_calls: totalCalls,
+              total_errors: totalErrors,
+              total_timeouts: totalTimeouts,
+              overall_error_rate: totalCalls > 0 ? Math.round(10000 * totalErrors / totalCalls) / 100 : 0,
+              unique_tools: tools.length,
+              total_marginal_cost: Math.round(totalCost * 1e6) / 1e6,
+              hours_analyzed: hoursBack
+            }
+          }
+        };
+      }
+
+    // ---- tool_timeline: Time-bucketed tool metrics via direct ClickHouse ----
+    case "tool_timeline":
+      {
+        const ch = getClickHouseClient();
+        if (!ch.isEnabled) return {
+          success: false,
+          error: "ClickHouse not configured"
+        };
+        const bucketMinutes = args.bucket_minutes || 15;
+        const toolFilter = args.tool_name ? `AND tool_name = '${esc(args.tool_name)}'` : "";
+        const storeF = sid ? `AND store_id = '${esc(sid)}'` : "";
+        const buckets = await ch.query(`
         SELECT
           toStartOfInterval(started_at, INTERVAL ${bucketMinutes} MINUTE) AS time,
           tool_name AS tool,
@@ -192,17 +256,30 @@ export async function handleTelemetry(sb, args, storeId) {
         GROUP BY time, tool
         ORDER BY time ASC, tool
       `);
-            return { success: true, data: { bucket_minutes: bucketMinutes, hours_back: hoursBack, buckets } };
-        }
-        // ---- trace: Full trace reconstruction via direct ClickHouse ----
-        case "trace": {
-            const traceId = args.trace_id;
-            if (!traceId)
-                return { success: false, error: "trace_id is required" };
-            const ch = getClickHouseClient();
-            if (!ch.isEnabled)
-                return { success: false, error: "ClickHouse not configured" };
-            const spans = await ch.query(`
+        return {
+          success: true,
+          data: {
+            bucket_minutes: bucketMinutes,
+            hours_back: hoursBack,
+            buckets
+          }
+        };
+      }
+
+    // ---- trace: Full trace reconstruction via direct ClickHouse ----
+    case "trace":
+      {
+        const traceId = args.trace_id;
+        if (!traceId) return {
+          success: false,
+          error: "trace_id is required"
+        };
+        const ch = getClickHouseClient();
+        if (!ch.isEnabled) return {
+          success: false,
+          error: "ClickHouse not configured"
+        };
+        const spans = await ch.query(`
         SELECT
           span_id, parent_span_id, trace_id, operation_name,
           service_name, duration_ms AS duration,
@@ -213,17 +290,29 @@ export async function handleTelemetry(sb, args, storeId) {
         WHERE trace_id = '${esc(traceId)}'
         ORDER BY started_at ASC
       `);
-            return { success: true, data: { trace_id: traceId, spans } };
-        }
-        // ---- span_detail: Individual span deep-dive via direct ClickHouse ----
-        case "span_detail": {
-            const spanId = args.span_id;
-            if (!spanId)
-                return { success: false, error: "span_id is required" };
-            const ch = getClickHouseClient();
-            if (!ch.isEnabled)
-                return { success: false, error: "ClickHouse not configured" };
-            const spans = await ch.query(`
+        return {
+          success: true,
+          data: {
+            trace_id: traceId,
+            spans
+          }
+        };
+      }
+
+    // ---- span_detail: Individual span deep-dive via direct ClickHouse ----
+    case "span_detail":
+      {
+        const spanId = args.span_id;
+        if (!spanId) return {
+          success: false,
+          error: "span_id is required"
+        };
+        const ch = getClickHouseClient();
+        if (!ch.isEnabled) return {
+          success: false,
+          error: "ClickHouse not configured"
+        };
+        const spans = await ch.query(`
         SELECT
           span_id, operation_name, tool_name, severity, duration_ms,
           status_code, error_message, started_at, ended_at,
@@ -233,14 +322,17 @@ export async function handleTelemetry(sb, args, storeId) {
         WHERE span_id = '${esc(spanId)}'
         LIMIT 1
       `);
-            if (spans.length === 0)
-                return { success: false, error: "Span not found" };
-            const span = spans[0];
-            // Comparison metrics for this tool (24h lookback)
-            const toolName = span.tool_name;
-            let comparison = {};
-            if (toolName) {
-                const comp = await ch.query(`
+        if (spans.length === 0) return {
+          success: false,
+          error: "Span not found"
+        };
+        const span = spans[0];
+
+        // Comparison metrics for this tool (24h lookback)
+        const toolName = span.tool_name;
+        let comparison = {};
+        if (toolName) {
+          const comp = await ch.query(`
           SELECT
             round(avg(duration_ms), 1) AS avg_ms,
             round(quantile(0.95)(duration_ms), 1) AS p95_ms,
@@ -250,21 +342,31 @@ export async function handleTelemetry(sb, args, storeId) {
           WHERE tool_name = '${esc(toolName)}'
             AND started_at >= now() - INTERVAL 24 HOUR
         `);
-                if (comp.length > 0) {
-                    comparison = {
-                        ...comp[0],
-                        is_slow: span.duration_ms > (comp[0].p95_ms || 999999),
-                    };
-                }
-            }
-            return { success: true, data: { span, comparison } };
+          if (comp.length > 0) {
+            comparison = {
+              ...comp[0],
+              is_slow: span.duration_ms > (comp[0].p95_ms || 999999)
+            };
+          }
         }
-        // ---- error_patterns: Error correlation + burst detection via ClickHouse ----
-        case "error_patterns": {
-            const ch = getClickHouseClient();
-            if (!ch.isEnabled)
-                return { success: false, error: "ClickHouse not configured" };
-            const patterns = await ch.query(`
+        return {
+          success: true,
+          data: {
+            span,
+            comparison
+          }
+        };
+      }
+
+    // ---- error_patterns: Error correlation + burst detection via ClickHouse ----
+    case "error_patterns":
+      {
+        const ch = getClickHouseClient();
+        if (!ch.isEnabled) return {
+          success: false,
+          error: "ClickHouse not configured"
+        };
+        const patterns = await ch.query(`
         SELECT
           fingerprint,
           any(error_type) AS error_type,
@@ -283,19 +385,26 @@ export async function handleTelemetry(sb, args, storeId) {
         ORDER BY occurrence_count DESC
         LIMIT 50
       `);
-            return { success: true, data: patterns };
+        return {
+          success: true,
+          data: patterns
+        };
+      }
+
+    // ---- token_usage: Token consumption by model/day via ClickHouse ----
+    case "token_usage":
+      {
+        const ch = getClickHouseClient();
+        if (!ch.isEnabled) return {
+          success: false,
+          error: "ClickHouse not configured"
+        };
+        const storeF = sid ? `AND store_id = '${esc(sid)}'` : "";
+        let agentFilter = "";
+        if (args.agent_id) {
+          agentFilter = `AND agent_id = '${esc(args.agent_id)}'`;
         }
-        // ---- token_usage: Token consumption by model/day via ClickHouse ----
-        case "token_usage": {
-            const ch = getClickHouseClient();
-            if (!ch.isEnabled)
-                return { success: false, error: "ClickHouse not configured" };
-            const storeF = sid ? `AND store_id = '${esc(sid)}'` : "";
-            let agentFilter = "";
-            if (args.agent_id) {
-                agentFilter = `AND agent_id = '${esc(args.agent_id)}'`;
-            }
-            const rows = await ch.query(`
+        const rows = await ch.query(`
         SELECT
           model_name AS model,
           toDate(hour) AS day,
@@ -309,21 +418,37 @@ export async function handleTelemetry(sb, args, storeId) {
         GROUP BY model_name, toDate(hour)
         ORDER BY day DESC, total_cost DESC
       `);
-            const summary = rows.reduce((acc, r) => ({
-                total_input: acc.total_input + r.input_tokens,
-                total_output: acc.total_output + r.output_tokens,
-                total_cost: acc.total_cost + r.total_cost,
-                total_requests: acc.total_requests + r.requests,
-            }), { total_input: 0, total_output: 0, total_cost: 0, total_requests: 0 });
-            return { success: true, data: { rows, summary, hours_back: hoursBack } };
-        }
-        // ---- sources: List all telemetry sources with counts via ClickHouse ----
-        case "sources": {
-            const ch = getClickHouseClient();
-            if (!ch.isEnabled)
-                return { success: false, error: "ClickHouse not configured" };
-            const storeF = sid ? `AND store_id = '${esc(sid)}'` : "";
-            const sources = await ch.query(`
+        const summary = rows.reduce((acc, r) => ({
+          total_input: acc.total_input + r.input_tokens,
+          total_output: acc.total_output + r.output_tokens,
+          total_cost: acc.total_cost + r.total_cost,
+          total_requests: acc.total_requests + r.requests
+        }), {
+          total_input: 0,
+          total_output: 0,
+          total_cost: 0,
+          total_requests: 0
+        });
+        return {
+          success: true,
+          data: {
+            rows,
+            summary,
+            hours_back: hoursBack
+          }
+        };
+      }
+
+    // ---- sources: List all telemetry sources with counts via ClickHouse ----
+    case "sources":
+      {
+        const ch = getClickHouseClient();
+        if (!ch.isEnabled) return {
+          success: false,
+          error: "ClickHouse not configured"
+        };
+        const storeF = sid ? `AND store_id = '${esc(sid)}'` : "";
+        const sources = await ch.query(`
         SELECT
           source,
           count() AS count,
@@ -336,9 +461,20 @@ export async function handleTelemetry(sb, args, storeId) {
         GROUP BY source
         ORDER BY count DESC
       `);
-            return { success: true, data: { sources, total_entries: sources.reduce((s, r) => s + r.count, 0), hours_back: hoursBack } };
-        }
-        default:
-            return { success: false, error: `Unknown telemetry action: ${args.action}. Available: conversation_detail, conversations, agent_performance, tool_analytics, tool_timeline, trace, span_detail, error_patterns, token_usage, sources. For activity logs and inventory changes, use the audit_trail tool instead.` };
-    }
+        return {
+          success: true,
+          data: {
+            sources,
+            total_entries: sources.reduce((s, r) => s + r.count, 0),
+            hours_back: hoursBack
+          }
+        };
+      }
+    default:
+      return {
+        success: false,
+        error: `Unknown telemetry action: ${args.action}. Available: conversation_detail, conversations, agent_performance, tool_analytics, tool_timeline, trace, span_detail, error_patterns, token_usage, sources. For activity logs and inventory changes, use the audit_trail tool instead.`
+      };
+  }
 }
+//# sourceMappingURL=platform.js.map