@oneuptime/common 11.3.25 → 11.3.28

package/build/dist/Server/Utils/AI/Toolbox/ToolTypes.js

@@ -0,0 +1,142 @@
+import BadDataException from "../../../../Types/Exception/BadDataException";
+import ObjectID from "../../../../Types/ObjectID";
+import OneUptimeDate from "../../../../Types/Date";
+/*
+ * Defensive argument readers. Tool arguments come from the LLM and are
+ * untrusted: everything is type-checked, defaulted and clamped.
+ */
+export class ToolArgs {
+    static getString(args, key) {
+        const value = args[key];
+        if (typeof value === "string" && value.trim().length > 0) {
+            return value.trim();
+        }
+        return undefined;
+    }
+    static getStringArray(args, key) {
+        const value = args[key];
+        if (Array.isArray(value)) {
+            const strings = value.filter((item) => {
+                return typeof item === "string" && item.trim().length > 0;
+            });
+            if (strings.length > 0) {
+                return strings;
+            }
+        }
+        return undefined;
+    }
+    static getNumber(args, key, options) {
+        const value = args[key];
+        let parsed = options.defaultValue;
+        if (typeof value === "number" && Number.isFinite(value)) {
+            parsed = value;
+        }
+        else if (typeof value === "string" && value.trim().length > 0) {
+            const fromString = Number(value);
+            if (Number.isFinite(fromString)) {
+                parsed = fromString;
+            }
+        }
+        return Math.min(options.max, Math.max(options.min, Math.floor(parsed)));
+    }
+    static getBoolean(args, key) {
+        const value = args[key];
+        if (typeof value === "boolean") {
+            return value;
+        }
+        if (value === "true") {
+            return true;
+        }
+        if (value === "false") {
+            return false;
+        }
+        return undefined;
+    }
+    static getObjectID(args, key) {
+        const value = this.getString(args, key);
+        if (!value) {
+            return undefined;
+        }
+        return new ObjectID(value);
+    }
+    /*
+     * Combines the caller's optional serviceId filter with the user's owned-scope
+     * access (from ModelPermission.getAccessibleServiceIdsForAnalyticsModel) into
+     * the `serviceIds` filter to hand an aggregation service.
+     *
+     * `allowed === null` means the user has project-wide access: pass only the
+     * caller's own filter (or undefined for no filter). Otherwise the user is
+     * label/owned-restricted: intersect with any requested service and NEVER
+     * return undefined or an empty array — an empty result is forced to a
+     * no-match sentinel, because the aggregation services treat a missing/empty
+     * serviceIds as "no filter" (which would leak the whole project).
+     */
+    static scopeServiceIds(allowed, requested) {
+        if (allowed === null) {
+            return requested ? [requested] : undefined;
+        }
+        let effective = allowed;
+        if (requested) {
+            effective = allowed.filter((id) => {
+                return id.toString() === requested.toString();
+            });
+        }
+        if (effective.length === 0) {
+            return [ObjectID.getZeroObjectID()];
+        }
+        return effective;
+    }
+    /*
+     * Time range for a tool call: explicit ISO startTime/endTime arguments,
+     * clamped to a maximum window, defaulting to the last hour.
+     */
+    static getTimeRange(args, options) {
+        var _a, _b;
+        const defaultHours = (_a = options === null || options === void 0 ? void 0 : options.defaultHours) !== null && _a !== void 0 ? _a : 1;
+        const maxDays = (_b = options === null || options === void 0 ? void 0 : options.maxDays) !== null && _b !== void 0 ? _b : 30;
+        const endTimeString = this.getString(args, "endTime");
+        const startTimeString = this.getString(args, "startTime");
+        /*
+         * Reject unparseable timestamps instead of silently falling back to the
+         * default window — a silent fallback answers a different question than the
+         * one asked. Throwing lets the model see the error and retry with a valid
+         * ISO 8601 value.
+         */
+        let endTime = OneUptimeDate.getCurrentDate();
+        if (endTimeString) {
+            const parsed = new Date(endTimeString);
+            if (isNaN(parsed.getTime())) {
+                throw new BadDataException(`endTime "${endTimeString}" is not a valid ISO 8601 timestamp (e.g. 2024-01-31T14:00:00Z).`);
+            }
+            endTime = parsed;
+        }
+        let startTime = OneUptimeDate.addRemoveHours(endTime, -1 * defaultHours);
+        if (startTimeString) {
+            const parsed = new Date(startTimeString);
+            if (isNaN(parsed.getTime())) {
+                throw new BadDataException(`startTime "${startTimeString}" is not a valid ISO 8601 timestamp (e.g. 2024-01-31T13:00:00Z).`);
+            }
+            startTime = parsed;
+        }
+        if (startTime.getTime() >= endTime.getTime()) {
+            throw new BadDataException("startTime must be before endTime.");
+        }
+        const maxWindowMs = maxDays * 24 * 60 * 60 * 1000;
+        if (endTime.getTime() - startTime.getTime() > maxWindowMs) {
+            startTime = new Date(endTime.getTime() - maxWindowMs);
+        }
+        return { startTime, endTime };
+    }
+}
+// Shared JSON-Schema fragments for tool input schemas.
+export const TimeRangeSchemaProperties = {
+    startTime: {
+        type: "string",
+        description: "Start of the time range as an ISO 8601 timestamp. Defaults to one hour before endTime.",
+    },
+    endTime: {
+        type: "string",
+        description: "End of the time range as an ISO 8601 timestamp. Defaults to now.",
+    },
+};
+//# sourceMappingURL=ToolTypes.js.map

package/build/dist/Server/Utils/AI/Toolbox/ToolTypes.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolTypes.js","sourceRoot":"","sources":["../../../../../../Server/Utils/AI/Toolbox/ToolTypes.ts"],"names":[],"mappings":"AACA,OAAO,gBAAgB,MAAM,8CAA8C,CAAC;AAE5E,OAAO,QAAQ,MAAM,4BAA4B,CAAC;AAClD,OAAO,aAAa,MAAM,wBAAwB,CAAC;AA+DnD;;;GAGG;AACH,MAAM,OAAO,QAAQ;IACZ,MAAM,CAAC,SAAS,CAAC,IAAgB,EAAE,GAAW;QACnD,MAAM,KAAK,GAAY,IAAI,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzD,OAAO,KAAK,CAAC,IAAI,EAAE,CAAC;QACtB,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,MAAM,CAAC,cAAc,CAC1B,IAAgB,EAChB,GAAW;QAEX,MAAM,KAAK,GAAY,IAAI,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,MAAM,OAAO,GAAkB,KAAK,CAAC,MAAM,CAAC,CAAC,IAAa,EAAE,EAAE;gBAC5D,OAAO,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;YAC5D,CAAC,CAAkB,CAAC;YACpB,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvB,OAAO,OAAO,CAAC;YACjB,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,MAAM,CAAC,SAAS,CACrB,IAAgB,EAChB,GAAW,EACX,OAA2D;QAE3D,MAAM,KAAK,GAAY,IAAI,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,MAAM,GAAW,OAAO,CAAC,YAAY,CAAC;QAE1C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACxD,MAAM,GAAG,KAAK,CAAC;QACjB,CAAC;aAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChE,MAAM,UAAU,GAAW,MAAM,CAAC,KAAK,CAAC,CAAC;YACzC,IAAI,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;gBAChC,MAAM,GAAG,UAAU,CAAC;YACtB,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IAC1E,CAAC;IAEM,MAAM,CAAC,UAAU,CAAC,IAAgB,EAAE,GAAW;QACpD,MAAM,KAAK,GAAY,IAAI,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE,CAAC;YAC/B,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,KAAK,KAAK,MAAM,EAAE,CAAC;YACrB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;YACtB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,MAAM,CAAC,WAAW,CACvB,IAAgB,EAChB,GAAW;QAEX,MAAM,KAAK,GAAuB,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QAC5D,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IAED;;;;;;;;;;;OAWG;IACI,MAAM,CAAC,eAAe,CAC3B,OAA+B,EAC/B,SAA+B;QAE/B,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;YACrB,OAAO,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC7C,CAAC;QAED,IAAI,SAAS,GAAoB,OAAO,CAAC;QACzC,IAAI,SAAS,EAAE,CAAC;YACd,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,EAAY,EAAE,EAAE;gBAC1C,OAAO,EAAE,CAAC,QAAQ,EAAE,KAAK,SAAS,CAAC,QAAQ,EAAE,CAAC;YAChD,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,OAAO,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC,CAAC;QACtC,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,YAAY,CACxB,IAAgB,EAChB,OAAqD;;QAErD,MAAM,YAAY,GAAW,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,YAAY,mCAAI,CAAC,CAAC;QACxD,MAAM,OAAO,GAAW,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,OAAO,mCAAI,EAAE,CAAC;QAE/C,MAAM,aAAa,GAAuB,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAC1E,MAAM,eAAe,GAAuB,IAAI,CAAC,SAAS,CACxD,IAAI,EACJ,WAAW,CACZ,CAAC;QAEF;;;;;WAKG;QACH,IAAI,OAAO,GAAS,aAAa,CAAC,cAAc,EAAE,CAAC;QACnD,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM,MAAM,GAAS,IAAI,IAAI,CAAC,aAAa,CAAC,CAAC;YAC7C,IAAI,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;gBAC5B,MAAM,IAAI,gBAAgB,CACxB,YAAY,aAAa,kEAAkE,CAC5F,CAAC;YACJ,CAAC;YACD,OAAO,GAAG,MAAM,CAAC;QACnB,CAAC;QAED,IAAI,SAAS,GAAS,aAAa,CAAC,cAAc,CAChD,OAAO,EACP,CAAC,CAAC,GAAG,YAAY,CAClB,CAAC;QACF,IAAI,eAAe,EAAE,CAAC;YACpB,MAAM,MAAM,GAAS,IAAI,IAAI,CAAC,eAAe,CAAC,CAAC;YAC/C,IAAI,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;gBAC5B,MAAM,IAAI,gBAAgB,CACxB,cAAc,eAAe,kEAAkE,CAChG,CAAC;YACJ,CAAC;YACD,SAAS,GAAG,MAAM,CAAC;QACrB,CAAC;QAED,IAAI,SAAS,CAAC,OAAO,EAAE,IAAI,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YAC7C,MAAM,IAAI,gBAAgB,CAAC,mCAAmC,CAAC,CAAC;QAClE,CAAC;QAED,MAAM,WAAW,GAAW,OAAO,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QAC1D,IAAI,OAAO,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,WAAW,EAAE,CAAC;YAC1D,SAAS,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,WAAW,CAAC,CAAC;QACxD,CAAC;QAED,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;IAChC,CAAC;CACF;AAED,uDAAuD;AACvD,MAAM,CAAC,MAAM,yBAAyB,GAAe;IACnD,SAAS,EAAE;QACT,IAAI,EAAE,QAAQ;QACd,WAAW,EACT,wFAAwF;KAC3F;IACD,OAAO,EAAE;QACP,IAAI,EAAE,QAAQ;QACd,WAAW,EACT,kEAAkE;KACrE;CACF,CAAC"}

package/build/dist/Server/Utils/AI/Toolbox/TraceTools.js

@@ -0,0 +1,309 @@
+import Span from "../../../../Models/AnalyticsModels/Span";
+import DatabaseRequestType from "../../../Types/BaseDatabase/DatabaseRequestType";
+import ModelPermission from "../../../Types/AnalyticsDatabase/ModelPermission";
+import SortOrder from "../../../../Types/BaseDatabase/SortOrder";
+import BadDataException from "../../../../Types/Exception/BadDataException";
+import Permission from "../../../../Types/Permission";
+import { AIChatCitationTargetType, } from "../../../../Types/AI/AIChatTypes";
+import SpanService from "../../../Services/SpanService";
+import TraceAggregationService from "../../../Services/TraceAggregationService";
+import ToolResultSerializer from "./Serializer";
+import WidgetBuilder from "./WidgetBuilder";
+import { TimeRangeSchemaProperties, ToolArgs, } from "./ToolTypes";
+const TRACE_READ_PERMISSIONS = [
+    Permission.ProjectOwner,
+    Permission.ProjectAdmin,
+    Permission.ProjectMember,
+    Permission.Viewer,
+    Permission.TelemetryAdmin,
+    Permission.TelemetryMember,
+    Permission.TelemetryViewer,
+    Permission.ReadTelemetryServiceTraces,
+];
+const VALID_METRICS = [
+    "count",
+    "errorCount",
+    "avgDuration",
+    "minDuration",
+    "maxDuration",
+    "p50Duration",
+    "p90Duration",
+    "p95Duration",
+    "p99Duration",
+];
+const VALID_GROUP_BY = [
+    "name",
+    "primaryEntityId",
+    "kind",
+    "statusCode",
+    "isRootSpan",
+];
+/*
+ * Upper bound on spans fetched for a single trace waterfall. High enough for
+ * almost every real trace; when a trace exceeds it we say so explicitly rather
+ * than silently dropping spans (which also orphans their children into fake
+ * roots).
+ */
+const MAX_TRACE_SPANS = 500;
+export const QueryTracesTool = {
+    name: "query_traces",
+    description: "Analyze spans/traces: latency profile (count, errorCount, avg and p50/p90/p95/p99 durations in ms) grouped by one dimension such as span name or service. Use this to answer 'why is X slow', 'error rate by operation' or 'slowest endpoints'. Then drill into a single trace with get_trace.",
+    inputSchema: {
+        type: "object",
+        properties: Object.assign(Object.assign({}, TimeRangeSchemaProperties), { groupBy: {
+                type: "string",
+                enum: ["name", "primaryEntityId", "kind", "statusCode", "isRootSpan"],
+                description: "Dimension to group by (default name = span/operation name; primaryEntityId = service).",
+            }, metric: {
+                type: "string",
+                enum: [
+                    "count",
+                    "errorCount",
+                    "avgDuration",
+                    "minDuration",
+                    "maxDuration",
+                    "p50Duration",
+                    "p90Duration",
+                    "p95Duration",
+                    "p99Duration",
+                ],
+                description: "Metric used to rank the groups (default p95Duration). The full latency profile is always returned per group.",
+            }, nameSearchText: {
+                type: "string",
+                description: "Only spans whose name contains this text.",
+            }, serviceId: {
+                type: "string",
+                description: "Only spans from this telemetry service.",
+            }, hasException: {
+                type: "boolean",
+                description: "Only spans that recorded an exception.",
+            }, rootOnly: {
+                type: "boolean",
+                description: "Only root spans (whole-request latency).",
+            }, limit: {
+                type: "number",
+                description: "Maximum groups to return (default 10, max 25).",
+            } }),
+    },
+    requiredPermissions: TRACE_READ_PERMISSIONS,
+    execute: async (args, ctx) => {
+        const { startTime, endTime } = ToolArgs.getTimeRange(args, {
+            defaultHours: 24,
+            maxDays: 30,
+        });
+        const metric = ToolArgs.getString(args, "metric") || "p95Duration";
+        if (!VALID_METRICS.includes(metric)) {
+            throw new BadDataException(`Invalid metric: ${metric}`);
+        }
+        const groupBy = ToolArgs.getString(args, "groupBy") || "name";
+        if (!VALID_GROUP_BY.includes(groupBy)) {
+            throw new BadDataException(`Invalid groupBy: ${groupBy}`);
+        }
+        const limit = ToolArgs.getNumber(args, "limit", {
+            defaultValue: 10,
+            min: 1,
+            max: 25,
+        });
+        const serviceId = ToolArgs.getObjectID(args, "serviceId");
+        const windowMinutes = (endTime.getTime() - startTime.getTime()) / (60 * 1000);
+        /*
+         * getAnalyticsTable builds raw aggregation SQL and skips the model layer's
+         * owned-scope filter, so a label-restricted user would otherwise see
+         * project-wide trace analytics. Constrain to the services this user may
+         * read (spans are owned through Service, same as logs).
+         */
+        const accessibleServiceIds = await ModelPermission.getAccessibleServiceIdsForAnalyticsModel(Span, ctx.props, DatabaseRequestType.Read);
+        const tableRows = await TraceAggregationService.getAnalyticsTable({
+            projectId: ctx.projectId,
+            startTime: startTime,
+            endTime: endTime,
+            bucketSizeInMinutes: Math.max(1, Math.round(windowMinutes / 60)),
+            chartType: "table",
+            metric: metric,
+            groupBy: [groupBy],
+            limit: limit,
+            nameSearchText: ToolArgs.getString(args, "nameSearchText"),
+            serviceIds: ToolArgs.scopeServiceIds(accessibleServiceIds, serviceId),
+            hasException: ToolArgs.getBoolean(args, "hasException"),
+            rootOnly: ToolArgs.getBoolean(args, "rootOnly"),
+        });
+        const rows = tableRows.map((row) => {
+            return {
+                [groupBy]: Object.values(row.groupValues).join(" / "),
+                count: row.count,
+                errorCount: row.errorCount,
+                avgMs: Math.round(row.avgDurationMs * 100) / 100,
+                p50Ms: Math.round(row.p50DurationMs * 100) / 100,
+                p90Ms: Math.round(row.p90DurationMs * 100) / 100,
+                p95Ms: Math.round(row.p95DurationMs * 100) / 100,
+                p99Ms: Math.round(row.p99DurationMs * 100) / 100,
+            };
+        });
+        const serialized = ToolResultSerializer.serializeRows(rows);
+        return {
+            dataForLlm: serialized.text,
+            rowCount: serialized.rowCount,
+            citationLabel: `Trace analytics by ${groupBy} (${metric}), ${startTime.toISOString()} – ${endTime.toISOString()}`,
+            citationTarget: {
+                type: AIChatCitationTargetType.Traces,
+            },
+            redactionCount: serialized.redactionCount,
+            isTruncated: serialized.isTruncated,
+            widget: rows.length > 0
+                ? WidgetBuilder.table({
+                    title: `Trace analytics by ${groupBy}`,
+                    description: `Ranked by ${metric} · ${startTime.toISOString()} – ${endTime.toISOString()}`,
+                    columns: [
+                        { key: groupBy, title: groupBy, type: "text" },
+                        { key: "count", title: "Count", type: "number" },
+                        { key: "errorCount", title: "Errors", type: "number" },
+                        { key: "avgMs", title: "Avg (ms)", type: "number" },
+                        { key: "p90Ms", title: "p90 (ms)", type: "number" },
+                        { key: "p95Ms", title: "p95 (ms)", type: "number" },
+                        { key: "p99Ms", title: "p99 (ms)", type: "number" },
+                    ],
+                    rows: rows,
+                    link: { type: AIChatCitationTargetType.Traces },
+                })
+                : undefined,
+        };
+    },
+};
+export const GetTraceTool = {
+    name: "get_trace",
+    description: "Get the span tree (waterfall) of one trace by traceId: every span with its duration, status and whether it recorded an exception. Use after query_traces or search_logs surfaces an interesting traceId.",
+    inputSchema: {
+        type: "object",
+        properties: {
+            traceId: {
+                type: "string",
+                description: "The trace ID (required).",
+            },
+        },
+        required: ["traceId"],
+    },
+    requiredPermissions: TRACE_READ_PERMISSIONS,
+    execute: async (args, ctx) => {
+        var _a, _b;
+        const traceId = ToolArgs.getString(args, "traceId");
+        if (!traceId) {
+            throw new BadDataException("traceId is required.");
+        }
+        const spans = await SpanService.findBy({
+            query: {
+                traceId: traceId,
+            },
+            select: {
+                spanId: true,
+                parentSpanId: true,
+                name: true,
+                startTimeUnixNano: true,
+                durationUnixNano: true,
+                statusCode: true,
+                kind: true,
+            },
+            sort: {
+                startTimeUnixNano: SortOrder.Ascending,
+            },
+            limit: MAX_TRACE_SPANS,
+            skip: 0,
+            props: ctx.props,
+        });
+        const isSpanLimitHit = spans.length >= MAX_TRACE_SPANS;
+        // Build the tree.
+        const nodesBySpanId = new Map();
+        const roots = [];
+        for (const span of spans) {
+            nodesBySpanId.set(((_a = span.spanId) === null || _a === void 0 ? void 0 : _a.toString()) || "", {
+                span,
+                children: [],
+            });
+        }
+        for (const node of nodesBySpanId.values()) {
+            const parentId = ((_b = node.span.parentSpanId) === null || _b === void 0 ? void 0 : _b.toString()) || "";
+            const parent = nodesBySpanId.get(parentId);
+            if (parent && parent !== node) {
+                parent.children.push(node);
+            }
+            else {
+                roots.push(node);
+            }
+        }
+        const lines = [];
+        const renderNode = (node, depth) => {
+            const durationMs = Math.round((Number(node.span.durationUnixNano) / 1000000) * 100) /
+                100;
+            const status = Number(node.span.statusCode) === 2 ? " [ERROR]" : "";
+            lines.push(`${"  ".repeat(depth)}- ${node.span.name} (${durationMs}ms)${status}`);
+            for (const child of node.children) {
+                renderNode(child, depth + 1);
+            }
+        };
+        for (const root of roots) {
+            renderNode(root, 0);
+        }
+        if (isSpanLimitHit) {
+            lines.push(`… trace truncated at ${MAX_TRACE_SPANS} spans; deeper spans are omitted and some shown here may appear as roots because their parent was cut off.`);
+        }
+        const serialized = ToolResultSerializer.serializeText(lines.join("\n"), spans.length);
+        /*
+         * Build the waterfall widget: each span's start is expressed as an offset in
+         * ms from the earliest span start (the trace start), so the renderer can lay
+         * out proportional bars without any wall-clock math.
+         */
+        let traceStartNano = undefined;
+        let traceEndNano = 0;
+        for (const span of spans) {
+            const start = Number(span.startTimeUnixNano);
+            const end = start + Number(span.durationUnixNano);
+            if (traceStartNano === undefined || start < traceStartNano) {
+                traceStartNano = start;
+            }
+            if (end > traceEndNano) {
+                traceEndNano = end;
+            }
+        }
+        const nanoToMs = (nano) => {
+            return Math.round((nano / 1000000) * 100) / 100;
+        };
+        const widgetSpans = spans.map((span) => {
+            var _a, _b;
+            const start = Number(span.startTimeUnixNano);
+            return {
+                spanId: ((_a = span.spanId) === null || _a === void 0 ? void 0 : _a.toString()) || "",
+                parentSpanId: ((_b = span.parentSpanId) === null || _b === void 0 ? void 0 : _b.toString()) || undefined,
+                name: span.name || "(unnamed span)",
+                startOffsetMs: nanoToMs(start - (traceStartNano !== null && traceStartNano !== void 0 ? traceStartNano : start)),
+                durationMs: nanoToMs(Number(span.durationUnixNano)),
+                isError: Number(span.statusCode) === 2,
+            };
+        });
+        const totalDurationMs = traceStartNano !== undefined
+            ? nanoToMs(traceEndNano - traceStartNano)
+            : 0;
+        return {
+            dataForLlm: serialized.text,
+            rowCount: serialized.rowCount,
+            citationLabel: `Trace ${traceId} (${spans.length}${isSpanLimitHit ? "+" : ""} spans)`,
+            citationTarget: {
+                type: AIChatCitationTargetType.TraceView,
+                params: { traceId: traceId },
+            },
+            redactionCount: serialized.redactionCount,
+            isTruncated: serialized.isTruncated || isSpanLimitHit,
+            widget: widgetSpans.length > 0
+                ? WidgetBuilder.traceWaterfall({
+                    title: `Trace waterfall (${spans.length}${isSpanLimitHit ? "+" : ""} spans)`,
+                    description: `${totalDurationMs} ms total`,
+                    spans: widgetSpans,
+                    totalDurationMs: totalDurationMs,
+                    link: {
+                        type: AIChatCitationTargetType.TraceView,
+                        params: { traceId: traceId },
+                    },
+                })
+                : undefined,
+        };
+    },
+};
+//# sourceMappingURL=TraceTools.js.map

package/build/dist/Server/Utils/AI/Toolbox/TraceTools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TraceTools.js","sourceRoot":"","sources":["../../../../../../Server/Utils/AI/Toolbox/TraceTools.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,yCAAyC,CAAC;AAC3D,OAAO,mBAAmB,MAAM,iDAAiD,CAAC;AAClF,OAAO,eAAe,MAAM,kDAAkD,CAAC;AAC/E,OAAO,SAAS,MAAM,0CAA0C,CAAC;AACjE,OAAO,gBAAgB,MAAM,8CAA8C,CAAC;AAG5E,OAAO,UAAU,MAAM,8BAA8B,CAAC;AACtD,OAAO,EACL,wBAAwB,GAEzB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,WAAW,MAAM,+BAA+B,CAAC;AACxD,OAAO,uBAEN,MAAM,2CAA2C,CAAC;AACnD,OAAO,oBAA0C,MAAM,cAAc,CAAC;AACtE,OAAO,aAAa,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAEL,yBAAyB,EACzB,QAAQ,GAGT,MAAM,aAAa,CAAC;AAErB,MAAM,sBAAsB,GAAsB;IAChD,UAAU,CAAC,YAAY;IACvB,UAAU,CAAC,YAAY;IACvB,UAAU,CAAC,aAAa;IACxB,UAAU,CAAC,MAAM;IACjB,UAAU,CAAC,cAAc;IACzB,UAAU,CAAC,eAAe;IAC1B,UAAU,CAAC,eAAe;IAC1B,UAAU,CAAC,0BAA0B;CACtC,CAAC;AAEF,MAAM,aAAa,GAAkB;IACnC,OAAO;IACP,YAAY;IACZ,aAAa;IACb,aAAa;IACb,aAAa;IACb,aAAa;IACb,aAAa;IACb,aAAa;IACb,aAAa;CACd,CAAC;AAEF,MAAM,cAAc,GAAkB;IACpC,MAAM;IACN,iBAAiB;IACjB,MAAM;IACN,YAAY;IACZ,YAAY;CACb,CAAC;AAEF;;;;;GAKG;AACH,MAAM,eAAe,GAAW,GAAG,CAAC;AAEpC,MAAM,CAAC,MAAM,eAAe,GAAsB;IAChD,IAAI,EAAE,cAAc;IACpB,WAAW,EACT,gSAAgS;IAClS,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,kCACL,yBAAyB,KAC5B,OAAO,EAAE;gBACP,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,MAAM,EAAE,iBAAiB,EAAE,MAAM,EAAE,YAAY,EAAE,YAAY,CAAC;gBACrE,WAAW,EACT,wFAAwF;aAC3F,EACD,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE;oBACJ,OAAO;oBACP,YAAY;oBACZ,aAAa;oBACb,aAAa;oBACb,aAAa;oBACb,aAAa;oBACb,aAAa;oBACb,aAAa;oBACb,aAAa;iBACd;gBACD,WAAW,EACT,8GAA8G;aACjH,EACD,cAAc,EAAE;gBACd,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,2CAA2C;aACzD,EACD,SAAS,EAAE;gBACT,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,yCAAyC;aACvD,EACD,YAAY,EAAE;gBACZ,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,wCAAwC;aACtD,EACD,QAAQ,EAAE;gBACR,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,0CAA0C;aACxD,EACD,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,gDAAgD;aAC9D,GACF;KACF;IACD,mBAAmB,EAAE,sBAAsB;IAC3C,OAAO,EAAE,KAAK,EACZ,IAAgB,EAChB,GAAgB,EACc,EAAE;QAChC,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC,YAAY,CAAC,IAAI,EAAE;YACzD,YAAY,EAAE,EAAE;YAChB,OAAO,EAAE,EAAE;SACZ,CAAC,CAAC;QAEH,MAAM,MAAM,GAAW,QAAQ,CAAC,SAAS,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,aAAa,CAAC;QAC3E,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YACpC,MAAM,IAAI,gBAAgB,CAAC,mBAAmB,MAAM,EAAE,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,OAAO,GAAW,QAAQ,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC;QACtE,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YACtC,MAAM,IAAI,gBAAgB,CAAC,oBAAoB,OAAO,EAAE,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,KAAK,GAAW,QAAQ,CAAC,SAAS,CAAC,IAAI,EAAE,OAAO,EAAE;YACtD,YAAY,EAAE,EAAE;YAChB,GAAG,EAAE,CAAC;YACN,GAAG,EAAE,EAAE;SACR,CAAC,CAAC;QAEH,MAAM,SAAS,GAAyB,QAAQ,CAAC,WAAW,CAC1D,IAAI,EACJ,WAAW,CACZ,CAAC;QAEF,MAAM,aAAa,GACjB,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAE1D;;;;;WAKG;QACH,MAAM,oBAAoB,GACxB,MAAM,eAAe,CAAC,wCAAwC,CAC5D,IAAI,EACJ,GAAG,CAAC,KAAK,EACT,mBAAmB,CAAC,IAAI,CACzB,CAAC;QAEJ,MAAM,SAAS,GACb,MAAM,uBAAuB,CAAC,iBAAiB,CAAC;YAC9C,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,SAAS,EAAE,SAAS;YACpB,OAAO,EAAE,OAAO;YAChB,mBAAmB,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa,GAAG,EAAE,CAAC,CAAC;YAChE,SAAS,EAAE,OAAO;YAClB,MAAM,EAAE,MAAe;YACvB,OAAO,EAAE,CAAC,OAAO,CAAC;YAClB,KAAK,EAAE,KAAK;YACZ,cAAc,EAAE,QAAQ,CAAC,SAAS,CAAC,IAAI,EAAE,gBAAgB,CAAC;YAC1D,UAAU,EAAE,QAAQ,CAAC,eAAe,CAAC,oBAAoB,EAAE,SAAS,CAAC;YACrE,YAAY,EAAE,QAAQ,CAAC,UAAU,CAAC,IAAI,EAAE,cAAc,CAAC;YACvD,QAAQ,EAAE,QAAQ,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC;SAChD,CAAC,CAAC;QAEL,MAAM,IAAI,GAAsB,SAAS,CAAC,GAAG,CAC3C,CAAC,GAA2B,EAAE,EAAE;YAC9B,OAAO;gBACL,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;gBACrD,KAAK,EAAE,GAAG,CAAC,KAAK;gBAChB,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,GAAG,GAAG,CAAC,GAAG,GAAG;gBAChD,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,GAAG,GAAG,CAAC,GAAG,GAAG;gBAChD,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,GAAG,GAAG,CAAC,GAAG,GAAG;gBAChD,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,GAAG,GAAG,CAAC,GAAG,GAAG;gBAChD,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,GAAG,GAAG,CAAC,GAAG,GAAG;aACjD,CAAC;QACJ,CAAC,CACF,CAAC;QAEF,MAAM,UAAU,GACd,oBAAoB,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QAE3C,OAAO;YACL,UAAU,EAAE,UAAU,CAAC,IAAI;YAC3B,QAAQ,EAAE,UAAU,CAAC,QAAQ;YAC7B,aAAa,EAAE,sBAAsB,OAAO,KAAK,MAAM,MAAM,SAAS,CAAC,WAAW,EAAE,MAAM,OAAO,CAAC,WAAW,EAAE,EAAE;YACjH,cAAc,EAAE;gBACd,IAAI,EAAE,wBAAwB,CAAC,MAAM;aACtC;YACD,cAAc,EAAE,UAAU,CAAC,cAAc;YACzC,WAAW,EAAE,UAAU,CAAC,WAAW;YACnC,MAAM,EACJ,IAAI,CAAC,MAAM,GAAG,CAAC;gBACb,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC;oBAClB,KAAK,EAAE,sBAAsB,OAAO,EAAE;oBACtC,WAAW,EAAE,aAAa,MAAM,MAAM,SAAS,CAAC,WAAW,EAAE,MAAM,OAAO,CAAC,WAAW,EAAE,EAAE;oBAC1F,OAAO,EAAE;wBACP,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE;wBAC9C,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE;wBAChD,EAAE,GAAG,EAAE,YAAY,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE;wBACtD,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE;wBACnD,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE;wBACnD,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE;wBACnD,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE;qBACpD;oBACD,IAAI,EAAE,IAAI;oBACV,IAAI,EAAE,EAAE,IAAI,EAAE,wBAAwB,CAAC,MAAM,EAAE;iBAChD,CAAC;gBACJ,CAAC,CAAC,SAAS;SAChB,CAAC;IACJ,CAAC;CACF,CAAC;AAOF,MAAM,CAAC,MAAM,YAAY,GAAsB;IAC7C,IAAI,EAAE,WAAW;IACjB,WAAW,EACT,0MAA0M;IAC5M,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,OAAO,EAAE;gBACP,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,0BAA0B;aACxC;SACF;QACD,QAAQ,EAAE,CAAC,SAAS,CAAC;KACtB;IACD,mBAAmB,EAAE,sBAAsB;IAC3C,OAAO,EAAE,KAAK,EACZ,IAAgB,EAChB,GAAgB,EACc,EAAE;;QAChC,MAAM,OAAO,GAAuB,QAAQ,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAExE,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,gBAAgB,CAAC,sBAAsB,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,KAAK,GAAgB,MAAM,WAAW,CAAC,MAAM,CAAC;YAClD,KAAK,EAAE;gBACL,OAAO,EAAE,OAAO;aACR;YACV,MAAM,EAAE;gBACN,MAAM,EAAE,IAAI;gBACZ,YAAY,EAAE,IAAI;gBAClB,IAAI,EAAE,IAAI;gBACV,iBAAiB,EAAE,IAAI;gBACvB,gBAAgB,EAAE,IAAI;gBACtB,UAAU,EAAE,IAAI;gBAChB,IAAI,EAAE,IAAI;aACF;YACV,IAAI,EAAE;gBACJ,iBAAiB,EAAE,SAAS,CAAC,SAAS;aAC9B;YACV,KAAK,EAAE,eAAe;YACtB,IAAI,EAAE,CAAC;YACP,KAAK,EAAE,GAAG,CAAC,KAAK;SACjB,CAAC,CAAC;QAEH,MAAM,cAAc,GAAY,KAAK,CAAC,MAAM,IAAI,eAAe,CAAC;QAEhE,kBAAkB;QAClB,MAAM,aAAa,GAA8B,IAAI,GAAG,EAAE,CAAC;QAC3D,MAAM,KAAK,GAAwB,EAAE,CAAC;QAEtC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,aAAa,CAAC,GAAG,CAAC,CAAA,MAAA,IAAI,CAAC,MAAM,0CAAE,QAAQ,EAAE,KAAI,EAAE,EAAE;gBAC/C,IAAI;gBACJ,QAAQ,EAAE,EAAE;aACb,CAAC,CAAC;QACL,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1C,MAAM,QAAQ,GAAW,CAAA,MAAA,IAAI,CAAC,IAAI,CAAC,YAAY,0CAAE,QAAQ,EAAE,KAAI,EAAE,CAAC;YAClE,MAAM,MAAM,GAA6B,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACrE,IAAI,MAAM,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;gBAC9B,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7B,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnB,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GAAkB,EAAE,CAAC;QAEhC,MAAM,UAAU,GAAgD,CAC9D,IAAkB,EAClB,KAAa,EACP,EAAE;YACR,MAAM,UAAU,GACd,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,OAAS,CAAC,GAAG,GAAG,CAAC;gBAClE,GAAG,CAAC;YACN,MAAM,MAAM,GACV,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;YACvD,KAAK,CAAC,IAAI,CACR,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,UAAU,MAAM,MAAM,EAAE,CACtE,CAAC;YACF,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClC,UAAU,CAAC,KAAK,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC,CAAC;QAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QACtB,CAAC;QAED,IAAI,cAAc,EAAE,CAAC;YACnB,KAAK,CAAC,IAAI,CACR,wBAAwB,eAAe,4GAA4G,CACpJ,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GAAqB,oBAAoB,CAAC,aAAa,CACrE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAChB,KAAK,CAAC,MAAM,CACb,CAAC;QAEF;;;;WAIG;QACH,IAAI,cAAc,GAAuB,SAAS,CAAC;QACnD,IAAI,YAAY,GAAW,CAAC,CAAC;QAC7B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,KAAK,GAAW,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACrD,MAAM,GAAG,GAAW,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAC1D,IAAI,cAAc,KAAK,SAAS,IAAI,KAAK,GAAG,cAAc,EAAE,CAAC;gBAC3D,cAAc,GAAG,KAAK,CAAC;YACzB,CAAC;YACD,IAAI,GAAG,GAAG,YAAY,EAAE,CAAC;gBACvB,YAAY,GAAG,GAAG,CAAC;YACrB,CAAC;QACH,CAAC;QAED,MAAM,QAAQ,GAA6B,CAAC,IAAY,EAAU,EAAE;YAClE,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,GAAG,OAAS,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;QACpD,CAAC,CAAC;QAEF,MAAM,WAAW,GAA4B,KAAK,CAAC,GAAG,CAAC,CAAC,IAAU,EAAE,EAAE;;YACpE,MAAM,KAAK,GAAW,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACrD,OAAO;gBACL,MAAM,EAAE,CAAA,MAAA,IAAI,CAAC,MAAM,0CAAE,QAAQ,EAAE,KAAI,EAAE;gBACrC,YAAY,EAAE,CAAA,MAAA,IAAI,CAAC,YAAY,0CAAE,QAAQ,EAAE,KAAI,SAAS;gBACxD,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,gBAAgB;gBACnC,aAAa,EAAE,QAAQ,CAAC,KAAK,GAAG,CAAC,cAAc,aAAd,cAAc,cAAd,cAAc,GAAI,KAAK,CAAC,CAAC;gBAC1D,UAAU,EAAE,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;gBACnD,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;aACvC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,MAAM,eAAe,GACnB,cAAc,KAAK,SAAS;YAC1B,CAAC,CAAC,QAAQ,CAAC,YAAY,GAAG,cAAc,CAAC;YACzC,CAAC,CAAC,CAAC,CAAC;QAER,OAAO;YACL,UAAU,EAAE,UAAU,CAAC,IAAI;YAC3B,QAAQ,EAAE,UAAU,CAAC,QAAQ;YAC7B,aAAa,EAAE,SAAS,OAAO,KAAK,KAAK,CAAC,MAAM,GAAG,cAAc,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,SAAS;YACrF,cAAc,EAAE;gBACd,IAAI,EAAE,wBAAwB,CAAC,SAAS;gBACxC,MAAM,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;aAC7B;YACD,cAAc,EAAE,UAAU,CAAC,cAAc;YACzC,WAAW,EAAE,UAAU,CAAC,WAAW,IAAI,cAAc;YACrD,MAAM,EACJ,WAAW,CAAC,MAAM,GAAG,CAAC;gBACpB,CAAC,CAAC,aAAa,CAAC,cAAc,CAAC;oBAC3B,KAAK,EAAE,oBAAoB,KAAK,CAAC,MAAM,GAAG,cAAc,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,SAAS;oBAC5E,WAAW,EAAE,GAAG,eAAe,WAAW;oBAC1C,KAAK,EAAE,WAAW;oBAClB,eAAe,EAAE,eAAe;oBAChC,IAAI,EAAE;wBACJ,IAAI,EAAE,wBAAwB,CAAC,SAAS;wBACxC,MAAM,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;qBAC7B;iBACF,CAAC;gBACJ,CAAC,CAAC,SAAS;SAChB,CAAC;IACJ,CAAC;CACF,CAAC"}

package/build/dist/Server/Utils/AI/Toolbox/WidgetBuilder.js

@@ -0,0 +1,120 @@
+import { AIChatWidgetType, } from "../../../../Types/AI/AIChatTypes";
+/*
+ * Builds the inline widgets a tool attaches to its result. Widgets are built
+ * from the RAW rows a tool fetched (not the redacted LLM payload) and rendered
+ * back to the same user who already has RBAC access to the data.
+ *
+ * `id` and `citationId` are left blank here — ChatAgentRunner mints them
+ * (W1, W2, … and the matching C1, C2, …) after the tool executes so they line
+ * up with the server-minted citations.
+ */
+export default class WidgetBuilder {
+    static table(params) {
+        return {
+            id: "",
+            type: AIChatWidgetType.Table,
+            title: params.title,
+            description: params.description,
+            data: {
+                columns: params.columns,
+                rows: params.rows,
+                link: params.link,
+            },
+        };
+    }
+    static timeSeries(params) {
+        return {
+            id: "",
+            type: AIChatWidgetType.TimeSeriesChart,
+            title: params.title,
+            description: params.description,
+            data: {
+                series: params.series,
+                xIsTime: true,
+                unit: params.unit,
+                valueLabel: params.valueLabel,
+                link: params.link,
+            },
+        };
+    }
+    static bars(params) {
+        var _a, _b;
+        return {
+            id: "",
+            type: AIChatWidgetType.BarChart,
+            title: params.title,
+            description: params.description,
+            data: {
+                series: params.series,
+                stacked: (_a = params.stacked) !== null && _a !== void 0 ? _a : false,
+                xIsTime: (_b = params.xIsTime) !== null && _b !== void 0 ? _b : false,
+                unit: params.unit,
+                link: params.link,
+            },
+        };
+    }
+    static traceWaterfall(params) {
+        return {
+            id: "",
+            type: AIChatWidgetType.TraceWaterfall,
+            title: params.title,
+            description: params.description,
+            data: {
+                spans: params.spans,
+                totalDurationMs: params.totalDurationMs,
+                link: params.link,
+            },
+        };
+    }
+    static incidentList(params) {
+        return {
+            id: "",
+            type: AIChatWidgetType.IncidentList,
+            title: params.title,
+            description: params.description,
+            data: { items: params.items, link: params.link },
+        };
+    }
+    static alertList(params) {
+        return {
+            id: "",
+            type: AIChatWidgetType.AlertList,
+            title: params.title,
+            description: params.description,
+            data: { items: params.items, link: params.link },
+        };
+    }
+    static exceptionList(params) {
+        return {
+            id: "",
+            type: AIChatWidgetType.ExceptionList,
+            title: params.title,
+            description: params.description,
+            data: { items: params.items, link: params.link },
+        };
+    }
+    static resourceCard(params) {
+        return {
+            id: "",
+            type: AIChatWidgetType.ResourceCard,
+            title: params.title,
+            data: {
+                resourceType: params.resourceType,
+                heading: params.heading,
+                subheading: params.subheading,
+                fields: params.fields,
+                link: params.link,
+            },
+        };
+    }
+    static stats(params) {
+        return {
+            id: "",
+            type: AIChatWidgetType.StatCards,
+            title: params.title,
+            description: params.description,
+            data: { stats: params.stats },
+        };
+    }
+}
+//# sourceMappingURL=WidgetBuilder.js.map

package/build/dist/Server/Utils/AI/Toolbox/WidgetBuilder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WidgetBuilder.js","sourceRoot":"","sources":["../../../../../../Server/Utils/AI/Toolbox/WidgetBuilder.ts"],"names":[],"mappings":"AACA,OAAO,EAOL,gBAAgB,GACjB,MAAM,kCAAkC,CAAC;AAE1C;;;;;;;;GAQG;AACH,MAAM,CAAC,OAAO,OAAO,aAAa;IACzB,MAAM,CAAC,KAAK,CAAC,MAMnB;QACC,OAAO;YACL,EAAE,EAAE,EAAE;YACN,IAAI,EAAE,gBAAgB,CAAC,KAAK;YAC5B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,IAAI,EAAE;gBACJ,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,IAAI,EAAE,MAAM,CAAC,IAAI;aAClB;SACF,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,UAAU,CAAC,MAOxB;QACC,OAAO;YACL,EAAE,EAAE,EAAE;YACN,IAAI,EAAE,gBAAgB,CAAC,eAAe;YACtC,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,IAAI,EAAE;gBACJ,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,OAAO,EAAE,IAAI;gBACb,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,IAAI,EAAE,MAAM,CAAC,IAAI;aAClB;SACF,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,IAAI,CAAC,MAQlB;;QACC,OAAO;YACL,EAAE,EAAE,EAAE;YACN,IAAI,EAAE,gBAAgB,CAAC,QAAQ;YAC/B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,IAAI,EAAE;gBACJ,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,OAAO,EAAE,MAAA,MAAM,CAAC,OAAO,mCAAI,KAAK;gBAChC,OAAO,EAAE,MAAA,MAAM,CAAC,OAAO,mCAAI,KAAK;gBAChC,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,IAAI,EAAE,MAAM,CAAC,IAAI;aAClB;SACF,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,cAAc,CAAC,MAM5B;QACC,OAAO;YACL,EAAE,EAAE,EAAE;YACN,IAAI,EAAE,gBAAgB,CAAC,cAAc;YACrC,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,IAAI,EAAE;gBACJ,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,eAAe,EAAE,MAAM,CAAC,eAAe;gBACvC,IAAI,EAAE,MAAM,CAAC,IAAI;aAClB;SACF,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,YAAY,CAAC,MAK1B;QACC,OAAO;YACL,EAAE,EAAE,EAAE;YACN,IAAI,EAAE,gBAAgB,CAAC,YAAY;YACnC,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,IAAI,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE;SACjD,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,SAAS,CAAC,MAKvB;QACC,OAAO;YACL,EAAE,EAAE,EAAE;YACN,IAAI,EAAE,gBAAgB,CAAC,SAAS;YAChC,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,IAAI,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE;SACjD,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,aAAa,CAAC,MAK3B;QACC,OAAO;YACL,EAAE,EAAE,EAAE;YACN,IAAI,EAAE,gBAAgB,CAAC,aAAa;YACpC,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,IAAI,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE;SACjD,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,YAAY,CAAC,MAO1B;QACC,OAAO;YACL,EAAE,EAAE,EAAE;YACN,IAAI,EAAE,gBAAgB,CAAC,YAAY;YACnC,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,IAAI,EAAE;gBACJ,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,IAAI,EAAE,MAAM,CAAC,IAAI;aAClB;SACF,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,MAInB;QACC,OAAO;YACL,EAAE,EAAE,EAAE;YACN,IAAI,EAAE,gBAAgB,CAAC,SAAS;YAChC,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,IAAI,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE;SAC9B,CAAC;IACJ,CAAC;CACF"}