@opensearch-project/agent-health 0.3.0 → 0.5.2

package/lib/dist/services/traces/toolSimilarity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolSimilarity.js","sourceRoot":"","sources":["../../../../services/traces/toolSimilarity.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAUH,OAAO,EAAE,qBAAqB,EAAE,MAAM,gDAAgD,CAAC;AAEvF,oEAAoE;AACpE,MAAM,cAAc,GAAG,kBAAkB,CAAC;AAC1C,MAAM,eAAe,GAAG,mBAAmB,CAAC;AAE5C;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CAAC,KAAwB;IAC3D,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;IAEjC,MAAM,WAAW,GAAG,CAAC,QAA2B,EAAE,EAAE;QAClD,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC5B,mCAAmC;YACnC,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;gBAC7B,yCAAyC;gBACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,cAAc,CAAC,CAAC;gBACnD,IAAI,QAAQ,EAAE,CAAC;oBACb,IAAI,CAAC;wBACH,MAAM,OAAO,GAAG,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;wBAC/E,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;4BAC3C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;wBACvD,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;wBACP,uEAAuE;oBACzE,CAAC;gBACH,CAAC;gBAED,0DAA0D;gBAC1D,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,eAAe,CAAC,CAAC;gBACjD,IAAI,KAAK,EAAE,CAAC;oBACV,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;wBACvE,IAAI,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;4BAC7C,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;wBACxD,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;wBACP,wBAAwB;oBAC1B,CAAC;gBACH,CAAC;YACH,CAAC;YAED,+BAA+B;YAC/B,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9C,WAAW,CAAC,IAAI,CAAC,QAA6B,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;IACH,CAAC,CAAC;IAEF,WAAW,CAAC,KAAK,CAAC,CAAC;IACnB,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;AACnC,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,IAAqB;IACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,cAAc,CAAC,CAAC;IACnD,IAAI,QAAQ,EAAE,CAAC;QACb,IAAI,CAAC;YACH,OAAO,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QACxE,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,eAAe,CAAC,CAAC;IACjD,IAAI,KAAK,EAAE,CAAC;QACV,IAAI,CAAC;YACH,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QAC/D,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,OAAO,EAAE,CAAC;AACZ,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,IAAqB;IACxC,OAAO,CACL,IAAI,CAAC,UAAU,EAAE,CAAC,qBAAqB,CAAW;QAClD,IAAI,CAAC,IAAI;QACT,cAAc,CACf,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,QAAgB,EAAE,aAAsC;IAC9E,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,IAAI,EAAE,CAAC;IACrD,MAAM,UAAU,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACnF,OAAO,GAAG,QAAQ,KAAK,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;AACjD,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,cAAc,CAC5B,KAAwB,EACxB,MAA4B;IAE5B,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxD,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;IACjD,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,GAAG,EAAqB,CAAC;IAC5C,MAAM,WAAW,GAAsB,EAAE,CAAC;IAE1C,MAAM,YAAY,GAAG,CAAC,QAA2B,EAAE,KAAa,EAAqB,EAAE;QACrF,MAAM,SAAS,GAAsB,EAAE,CAAC;QAExC,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC5B,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;gBAC7B,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;gBACnC,MAAM,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;gBAElC,iCAAiC;gBACjC,MAAM,aAAa,GAA4B,EAAE,CAAC;gBAClD,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;oBACtC,IAAI,GAAG,IAAI,OAAO,EAAE,CAAC;wBACnB,aAAa,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;oBACpC,CAAC;gBACH,CAAC;gBAED,MAAM,QAAQ,GAAG,cAAc,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;gBAEzD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC1B,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE;wBACnB,QAAQ;wBACR,aAAa;wBACb,KAAK,EAAE,EAAE;wBACT,KAAK,EAAE,CAAC;wBACR,aAAa,EAAE,CAAC;wBAChB,WAAW,EAAE,CAAC;qBACf,CAAC,CAAC;gBACL,CAAC;gBAED,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC;gBACpC,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;gBACvF,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACvB,KAAK,CAAC,KAAK,EAAE,CAAC;gBACd,KAAK,CAAC,aAAa,IAAI,QAAQ,CAAC;gBAChC,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC,KAAK,CAAC;gBAEtD,0FAA0F;gBAC1F,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvB,CAAC;iBAAM,CAAC;gBACN,+CAA+C;gBAC/C,MAAM,aAAa,GAAG,EAAE,GAAG,IAAI,EAAE,CAAC;gBAClC,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC9C,aAAa,CAAC,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAC,QAA6B,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;gBACvF,CAAC;gBACD,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC;IAEF,MAAM,cAAc,GAAG,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAE9C,OAAO;QACL,YAAY,EAAE,cAAc;QAC5B,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;KAC1E,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,uBAAuB,CACrC,IAAqB,EACrB,KAAsB,EACtB,MAA4B;IAE5B,qBAAqB;IACrB,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,IAAI,KAAK,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;QAC1D,OAAO,CAAC,CAAC;IACX,CAAC;IAED,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IACnC,MAAM,SAAS,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;IAErC,sCAAsC;IACtC,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,OAAO,CAAC,CAAC;IACX,CAAC;IAED,0DAA0D;IAC1D,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxD,OAAO,GAAG,CAAC;IACb,CAAC;IAED,8BAA8B;IAC9B,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IACnC,MAAM,SAAS,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;IAErC,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,SAAS,GAAG,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC;IAE3C,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;QAChD,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;YACzB,YAAY,EAAE,CAAC;QACjB,CAAC;IACH,CAAC;IAED,OAAO,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC;AACxD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,MAAmB;IAMnD,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAC/D,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC;IAClC,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,+BAA+B;IAC1F,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,CACnC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,aAAa,GAAG,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EACnE,IAAI,CACL,CAAC;IAEF,OAAO;QACL,UAAU;QACV,WAAW;QACX,YAAY;QACZ,eAAe;KAChB,CAAC;AACJ,CAAC"}

package/lib/dist/services/traces/traceComparison.d.ts

@@ -0,0 +1,31 @@
+/**
+ * Trace Comparison Service
+ *
+ * Aligns two span trees using LCS-based algorithm for side-by-side comparison.
+ * Follows the same pattern as trajectoryDiffService but works with hierarchical spans.
+ */
+import { Span, CategorizedSpan, AlignedSpanPair, TraceComparisonResult, ToolSimilarityConfig } from '../../types/index.js';
+/**
+ * Calculate similarity between two spans (0-1).
+ * Takes into account span name, category, attributes, and duration.
+ */
+export declare function calculateSpanSimilarity(left: CategorizedSpan, right: CategorizedSpan, toolConfig?: ToolSimilarityConfig): number;
+/**
+ * Compare two span trees and return aligned result with statistics.
+ * Main entry point for trace comparison.
+ */
+export declare function compareTraces(leftSpans: Span[], rightSpans: Span[], toolConfig?: ToolSimilarityConfig): TraceComparisonResult;
+/**
+ * Get a flat list of all aligned pairs for easier iteration
+ */
+export declare function flattenAlignedTree(aligned: AlignedSpanPair[]): AlignedSpanPair[];
+/**
+ * Get display info for a comparison type
+ */
+export declare function getComparisonTypeInfo(type: AlignedSpanPair['type']): {
+    label: string;
+    color: string;
+    bgColor: string;
+    borderColor: string;
+};
+//# sourceMappingURL=traceComparison.d.ts.map

package/lib/dist/services/traces/traceComparison.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"traceComparison.d.ts","sourceRoot":"","sources":["../../../../services/traces/traceComparison.ts"],"names":[],"mappings":"AAKA;;;;;GAKG;AAEH,OAAO,EACL,IAAI,EACJ,eAAe,EACf,eAAe,EACf,qBAAqB,EACrB,oBAAoB,EACrB,MAAM,SAAS,CAAC;AAcjB;;;GAGG;AACH,wBAAgB,uBAAuB,CACrC,IAAI,EAAE,eAAe,EACrB,KAAK,EAAE,eAAe,EACtB,UAAU,CAAC,EAAE,oBAAoB,GAChC,MAAM,CAiDR;AA6MD;;;GAGG;AACH,wBAAgB,aAAa,CAC3B,SAAS,EAAE,IAAI,EAAE,EACjB,UAAU,EAAE,IAAI,EAAE,EAClB,UAAU,CAAC,EAAE,oBAAoB,GAChC,qBAAqB,CAoBvB;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,eAAe,EAAE,GAAG,eAAe,EAAE,CAWhF;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,eAAe,CAAC,MAAM,CAAC,GAAG;IACpE,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;CACrB,CA+BA"}

package/lib/dist/services/traces/traceComparison.js

@@ -0,0 +1,318 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { debug } from '../../lib/debug.js';
+import { categorizeSpanTree } from './spanCategorization.js';
+import { calculateToolSimilarity } from './toolSimilarity.js';
+import { ATTR_GEN_AI_OPERATION_NAME, ATTR_GEN_AI_AGENT_NAME, ATTR_GEN_AI_REQUEST_MODEL, ATTR_GEN_AI_TOOL_NAME, } from '@opentelemetry/semantic-conventions/incubating';
+const MATCH_THRESHOLD = 0.6;
+const MODIFIED_THRESHOLD = 0.4;
+/**
+ * Calculate similarity between two spans (0-1).
+ * Takes into account span name, category, attributes, and duration.
+ */
+export function calculateSpanSimilarity(left, right, toolConfig) {
+    let score = 0;
+    // Category match: 0.3 weight
+    if (left.category === right.category) {
+        score += 0.3;
+    }
+    // Name/operation match: 0.3 weight
+    const leftOp = left.attributes?.[ATTR_GEN_AI_OPERATION_NAME] || left.name;
+    const rightOp = right.attributes?.[ATTR_GEN_AI_OPERATION_NAME] || right.name;
+    if (leftOp === rightOp) {
+        score += 0.3;
+    }
+    // For TOOL spans, use tool similarity config if available
+    if (left.category === 'TOOL' && right.category === 'TOOL' && toolConfig) {
+        const toolSim = calculateToolSimilarity(left, right, toolConfig);
+        score += toolSim * 0.25;
+    }
+    else {
+        // For non-tool spans, compare relevant attributes
+        const leftAgent = left.attributes?.[ATTR_GEN_AI_AGENT_NAME];
+        const rightAgent = right.attributes?.[ATTR_GEN_AI_AGENT_NAME];
+        const leftModel = left.attributes?.[ATTR_GEN_AI_REQUEST_MODEL];
+        const rightModel = right.attributes?.[ATTR_GEN_AI_REQUEST_MODEL];
+        const leftTool = left.attributes?.[ATTR_GEN_AI_TOOL_NAME];
+        const rightTool = right.attributes?.[ATTR_GEN_AI_TOOL_NAME];
+        if ((leftAgent && rightAgent && leftAgent === rightAgent) ||
+            (leftModel && rightModel && leftModel === rightModel) ||
+            (leftTool && rightTool && leftTool === rightTool)) {
+            score += 0.25;
+        }
+    }
+    // Duration similarity: 0.15 weight (within 50% range is similar)
+    const leftDuration = new Date(left.endTime).getTime() - new Date(left.startTime).getTime();
+    const rightDuration = new Date(right.endTime).getTime() - new Date(right.startTime).getTime();
+    const maxDuration = Math.max(leftDuration, rightDuration);
+    if (maxDuration > 0) {
+        const durationRatio = Math.min(leftDuration, rightDuration) / maxDuration;
+        score += durationRatio * 0.15;
+    }
+    else {
+        score += 0.15;
+    }
+    return Math.min(score, 1);
+}
+/**
+ * Align two span arrays using LCS-based dynamic programming.
+ * Returns aligned pairs with type annotations.
+ */
+function alignSpanArrays(left, right, toolConfig) {
+    // Handle edge cases
+    if (left.length === 0 && right.length === 0) {
+        return [];
+    }
+    if (left.length === 0) {
+        return right.map(span => ({
+            type: 'added',
+            rightSpan: span,
+            children: span.children && span.children.length > 0
+                ? alignSpanArrays([], span.children, toolConfig)
+                : undefined,
+        }));
+    }
+    if (right.length === 0) {
+        return left.map(span => ({
+            type: 'removed',
+            leftSpan: span,
+            children: span.children && span.children.length > 0
+                ? alignSpanArrays(span.children, [], toolConfig)
+                : undefined,
+        }));
+    }
+    const n = left.length;
+    const m = right.length;
+    // Build similarity matrix
+    const similarity = [];
+    for (let i = 0; i < n; i++) {
+        similarity[i] = [];
+        for (let j = 0; j < m; j++) {
+            similarity[i][j] = calculateSpanSimilarity(left[i], right[j], toolConfig);
+        }
+    }
+    // DP for optimal alignment
+    const dp = Array(n + 1).fill(null).map(() => Array(m + 1).fill(0));
+    const path = Array(n + 1).fill(null).map(() => Array(m + 1).fill('skip_left'));
+    for (let i = 1; i <= n; i++) {
+        for (let j = 1; j <= m; j++) {
+            const matchScore = similarity[i - 1][j - 1];
+            // Match option
+            const matchOption = dp[i - 1][j - 1] + (matchScore >= MATCH_THRESHOLD ? matchScore : -0.5);
+            // Skip left (removed)
+            const skipLeftOption = dp[i - 1][j] - 0.1;
+            // Skip right (added)
+            const skipRightOption = dp[i][j - 1] - 0.1;
+            if (matchOption >= skipLeftOption && matchOption >= skipRightOption) {
+                dp[i][j] = matchOption;
+                path[i][j] = 'match';
+            }
+            else if (skipLeftOption >= skipRightOption) {
+                dp[i][j] = skipLeftOption;
+                path[i][j] = 'skip_left';
+            }
+            else {
+                dp[i][j] = skipRightOption;
+                path[i][j] = 'skip_right';
+            }
+        }
+    }
+    // Backtrack
+    const result = [];
+    let i = n, j = m;
+    while (i > 0 || j > 0) {
+        if (i > 0 && j > 0 && path[i][j] === 'match') {
+            const sim = similarity[i - 1][j - 1];
+            const leftSpan = left[i - 1];
+            const rightSpan = right[j - 1];
+            // Recursively align children
+            const leftChildren = (leftSpan.children || []);
+            const rightChildren = (rightSpan.children || []);
+            const alignedChildren = leftChildren.length > 0 || rightChildren.length > 0
+                ? alignSpanArrays(leftChildren, rightChildren, toolConfig)
+                : undefined;
+            if (sim >= MATCH_THRESHOLD) {
+                const isExactMatch = leftSpan.name === rightSpan.name &&
+                    leftSpan.category === rightSpan.category &&
+                    JSON.stringify(leftSpan.attributes) === JSON.stringify(rightSpan.attributes);
+                result.push({
+                    type: isExactMatch ? 'matched' : 'modified',
+                    leftSpan,
+                    rightSpan,
+                    similarity: sim,
+                    children: alignedChildren,
+                });
+            }
+            else if (sim >= MODIFIED_THRESHOLD) {
+                result.push({
+                    type: 'modified',
+                    leftSpan,
+                    rightSpan,
+                    similarity: sim,
+                    children: alignedChildren,
+                });
+            }
+            else {
+                // Too different, separate entries
+                result.push({
+                    type: 'removed',
+                    leftSpan,
+                    children: leftChildren.length > 0
+                        ? alignSpanArrays(leftChildren, [], toolConfig)
+                        : undefined,
+                });
+                result.push({
+                    type: 'added',
+                    rightSpan,
+                    children: rightChildren.length > 0
+                        ? alignSpanArrays([], rightChildren, toolConfig)
+                        : undefined,
+                });
+            }
+            i--;
+            j--;
+        }
+        else if (i > 0 && (j === 0 || path[i][j] === 'skip_left')) {
+            const leftSpan = left[i - 1];
+            const leftChildren = (leftSpan.children || []);
+            result.push({
+                type: 'removed',
+                leftSpan,
+                children: leftChildren.length > 0
+                    ? alignSpanArrays(leftChildren, [], toolConfig)
+                    : undefined,
+            });
+            i--;
+        }
+        else {
+            const rightSpan = right[j - 1];
+            const rightChildren = (rightSpan.children || []);
+            result.push({
+                type: 'added',
+                rightSpan,
+                children: rightChildren.length > 0
+                    ? alignSpanArrays([], rightChildren, toolConfig)
+                    : undefined,
+            });
+            j--;
+        }
+    }
+    // Reverse since we backtracked
+    result.reverse();
+    return result;
+}
+/**
+ * Count stats recursively from aligned tree
+ */
+function countAlignedStats(aligned) {
+    let matched = 0, added = 0, removed = 0, modified = 0;
+    for (const pair of aligned) {
+        switch (pair.type) {
+            case 'matched':
+                matched++;
+                break;
+            case 'added':
+                added++;
+                break;
+            case 'removed':
+                removed++;
+                break;
+            case 'modified':
+                modified++;
+                break;
+        }
+        if (pair.children) {
+            const childStats = countAlignedStats(pair.children);
+            matched += childStats.matched;
+            added += childStats.added;
+            removed += childStats.removed;
+            modified += childStats.modified;
+        }
+    }
+    return { matched, added, removed, modified };
+}
+/**
+ * Count total spans in a tree
+ */
+function countSpans(spans) {
+    let count = spans.length;
+    for (const span of spans) {
+        if (span.children && span.children.length > 0) {
+            count += countSpans(span.children);
+        }
+    }
+    return count;
+}
+/**
+ * Compare two span trees and return aligned result with statistics.
+ * Main entry point for trace comparison.
+ */
+export function compareTraces(leftSpans, rightSpans, toolConfig) {
+    debug('TraceComparison', 'Comparing traces:', leftSpans.length, 'vs', rightSpans.length, 'spans');
+    // Categorize both trees
+    const leftCategorized = categorizeSpanTree(leftSpans);
+    const rightCategorized = categorizeSpanTree(rightSpans);
+    // Align the trees
+    const alignedTree = alignSpanArrays(leftCategorized, rightCategorized, toolConfig);
+    // Calculate statistics
+    const stats = countAlignedStats(alignedTree);
+    return {
+        alignedTree,
+        stats: {
+            totalLeft: countSpans(leftCategorized),
+            totalRight: countSpans(rightCategorized),
+            ...stats,
+        },
+    };
+}
+/**
+ * Get a flat list of all aligned pairs for easier iteration
+ */
+export function flattenAlignedTree(aligned) {
+    const result = [];
+    for (const pair of aligned) {
+        result.push(pair);
+        if (pair.children) {
+            result.push(...flattenAlignedTree(pair.children));
+        }
+    }
+    return result;
+}
+/**
+ * Get display info for a comparison type
+ */
+export function getComparisonTypeInfo(type) {
+    switch (type) {
+        case 'matched':
+            return {
+                label: 'Matched',
+                color: 'text-slate-400',
+                bgColor: 'bg-slate-500/10',
+                borderColor: 'border-slate-500/30',
+            };
+        case 'added':
+            return {
+                label: 'Added',
+                color: 'text-green-400',
+                bgColor: 'bg-green-500/10',
+                borderColor: 'border-green-500/30',
+            };
+        case 'removed':
+            return {
+                label: 'Removed',
+                color: 'text-red-400',
+                bgColor: 'bg-red-500/10',
+                borderColor: 'border-red-500/30',
+            };
+        case 'modified':
+            return {
+                label: 'Modified',
+                color: 'text-amber-400',
+                bgColor: 'bg-amber-500/10',
+                borderColor: 'border-amber-500/30',
+            };
+    }
+}
+//# sourceMappingURL=traceComparison.js.map

package/lib/dist/services/traces/traceComparison.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"traceComparison.js","sourceRoot":"","sources":["../../../../services/traces/traceComparison.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAgBH,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAC3D,OAAO,EACL,0BAA0B,EAC1B,sBAAsB,EACtB,yBAAyB,EACzB,qBAAqB,GACtB,MAAM,gDAAgD,CAAC;AAExD,MAAM,eAAe,GAAG,GAAG,CAAC;AAC5B,MAAM,kBAAkB,GAAG,GAAG,CAAC;AAE/B;;;GAGG;AACH,MAAM,UAAU,uBAAuB,CACrC,IAAqB,EACrB,KAAsB,EACtB,UAAiC;IAEjC,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,6BAA6B;IAC7B,IAAI,IAAI,CAAC,QAAQ,KAAK,KAAK,CAAC,QAAQ,EAAE,CAAC;QACrC,KAAK,IAAI,GAAG,CAAC;IACf,CAAC;IAED,mCAAmC;IACnC,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,0BAA0B,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC;IAC1E,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC,0BAA0B,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC;IAC7E,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;QACvB,KAAK,IAAI,GAAG,CAAC;IACf,CAAC;IAED,0DAA0D;IAC1D,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,IAAI,KAAK,CAAC,QAAQ,KAAK,MAAM,IAAI,UAAU,EAAE,CAAC;QACxE,MAAM,OAAO,GAAG,uBAAuB,CAAC,IAAI,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;QACjE,KAAK,IAAI,OAAO,GAAG,IAAI,CAAC;IAC1B,CAAC;SAAM,CAAC;QACN,kDAAkD;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,sBAAsB,CAAC,CAAC;QAC5D,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC,sBAAsB,CAAC,CAAC;QAC9D,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,yBAAyB,CAAC,CAAC;QAC/D,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC,yBAAyB,CAAC,CAAC;QACjE,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,qBAAqB,CAAC,CAAC;QAC1D,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC,qBAAqB,CAAC,CAAC;QAE5D,IACE,CAAC,SAAS,IAAI,UAAU,IAAI,SAAS,KAAK,UAAU,CAAC;YACrD,CAAC,SAAS,IAAI,UAAU,IAAI,SAAS,KAAK,UAAU,CAAC;YACrD,CAAC,QAAQ,IAAI,SAAS,IAAI,QAAQ,KAAK,SAAS,CAAC,EACjD,CAAC;YACD,KAAK,IAAI,IAAI,CAAC;QAChB,CAAC;IACH,CAAC;IAED,iEAAiE;IACjE,MAAM,YAAY,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;IAC3F,MAAM,aAAa,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;IAC9F,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;IAC1D,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;QACpB,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,aAAa,CAAC,GAAG,WAAW,CAAC;QAC1E,KAAK,IAAI,aAAa,GAAG,IAAI,CAAC;IAChC,CAAC;SAAM,CAAC;QACN,KAAK,IAAI,IAAI,CAAC;IAChB,CAAC;IAED,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;AAC5B,CAAC;AAED;;;GAGG;AACH,SAAS,eAAe,CACtB,IAAuB,EACvB,KAAwB,EACxB,UAAiC;IAEjC,oBAAoB;IACpB,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5C,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACtB,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACxB,IAAI,EAAE,OAAgB;YACtB,SAAS,EAAE,IAAI;YACf,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC;gBACjD,CAAC,CAAC,eAAe,CAAC,EAAE,EAAE,IAAI,CAAC,QAA6B,EAAE,UAAU,CAAC;gBACrE,CAAC,CAAC,SAAS;SACd,CAAC,CAAC,CAAC;IACN,CAAC;IACD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACvB,IAAI,EAAE,SAAkB;YACxB,QAAQ,EAAE,IAAI;YACd,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC;gBACjD,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,QAA6B,EAAE,EAAE,EAAE,UAAU,CAAC;gBACrE,CAAC,CAAC,SAAS;SACd,CAAC,CAAC,CAAC;IACN,CAAC;IAED,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;IACtB,MAAM,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC;IAEvB,0BAA0B;IAC1B,MAAM,UAAU,GAAe,EAAE,CAAC;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,UAAU,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;QACnB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,uBAAuB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;IAED,2BAA2B;IAC3B,MAAM,EAAE,GAAe,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IAC/E,MAAM,IAAI,GACR,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;IAEpE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,UAAU,GAAG,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAE5C,eAAe;YACf,MAAM,WAAW,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,UAAU,IAAI,eAAe,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YAE3F,sBAAsB;YACtB,MAAM,cAAc,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;YAE1C,qBAAqB;YACrB,MAAM,eAAe,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC;YAE3C,IAAI,WAAW,IAAI,cAAc,IAAI,WAAW,IAAI,eAAe,EAAE,CAAC;gBACpE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,WAAW,CAAC;gBACvB,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC;YACvB,CAAC;iBAAM,IAAI,cAAc,IAAI,eAAe,EAAE,CAAC;gBAC7C,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC;gBAC1B,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,WAAW,CAAC;YAC3B,CAAC;iBAAM,CAAC;gBACN,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,eAAe,CAAC;gBAC3B,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,YAAY,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;IAED,YAAY;IACZ,MAAM,MAAM,GAAsB,EAAE,CAAC;IACrC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC;IAEjB,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QACtB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,OAAO,EAAE,CAAC;YAC7C,MAAM,GAAG,GAAG,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAE/B,6BAA6B;YAC7B,MAAM,YAAY,GAAG,CAAC,QAAQ,CAAC,QAAQ,IAAI,EAAE,CAAsB,CAAC;YACpE,MAAM,aAAa,GAAG,CAAC,SAAS,CAAC,QAAQ,IAAI,EAAE,CAAsB,CAAC;YACtE,MAAM,eAAe,GAAG,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC;gBACzE,CAAC,CAAC,eAAe,CAAC,YAAY,EAAE,aAAa,EAAE,UAAU,CAAC;gBAC1D,CAAC,CAAC,SAAS,CAAC;YAEd,IAAI,GAAG,IAAI,eAAe,EAAE,CAAC;gBAC3B,MAAM,YAAY,GAChB,QAAQ,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI;oBAChC,QAAQ,CAAC,QAAQ,KAAK,SAAS,CAAC,QAAQ;oBACxC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;gBAE/E,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU;oBAC3C,QAAQ;oBACR,SAAS;oBACT,UAAU,EAAE,GAAG;oBACf,QAAQ,EAAE,eAAe;iBAC1B,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,GAAG,IAAI,kBAAkB,EAAE,CAAC;gBACrC,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,UAAU;oBAChB,QAAQ;oBACR,SAAS;oBACT,UAAU,EAAE,GAAG;oBACf,QAAQ,EAAE,eAAe;iBAC1B,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,kCAAkC;gBAClC,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,SAAS;oBACf,QAAQ;oBACR,QAAQ,EAAE,YAAY,CAAC,MAAM,GAAG,CAAC;wBAC/B,CAAC,CAAC,eAAe,CAAC,YAAY,EAAE,EAAE,EAAE,UAAU,CAAC;wBAC/C,CAAC,CAAC,SAAS;iBACd,CAAC,CAAC;gBACH,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,OAAO;oBACb,SAAS;oBACT,QAAQ,EAAE,aAAa,CAAC,MAAM,GAAG,CAAC;wBAChC,CAAC,CAAC,eAAe,CAAC,EAAE,EAAE,aAAa,EAAE,UAAU,CAAC;wBAChD,CAAC,CAAC,SAAS;iBACd,CAAC,CAAC;YACL,CAAC;YACD,CAAC,EAAE,CAAC;YACJ,CAAC,EAAE,CAAC;QACN,CAAC;aAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,WAAW,CAAC,EAAE,CAAC;YAC5D,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7B,MAAM,YAAY,GAAG,CAAC,QAAQ,CAAC,QAAQ,IAAI,EAAE,CAAsB,CAAC;YACpE,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,SAAS;gBACf,QAAQ;gBACR,QAAQ,EAAE,YAAY,CAAC,MAAM,GAAG,CAAC;oBAC/B,CAAC,CAAC,eAAe,CAAC,YAAY,EAAE,EAAE,EAAE,UAAU,CAAC;oBAC/C,CAAC,CAAC,SAAS;aACd,CAAC,CAAC;YACH,CAAC,EAAE,CAAC;QACN,CAAC;aAAM,CAAC;YACN,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAC/B,MAAM,aAAa,GAAG,CAAC,SAAS,CAAC,QAAQ,IAAI,EAAE,CAAsB,CAAC;YACtE,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,OAAO;gBACb,SAAS;gBACT,QAAQ,EAAE,aAAa,CAAC,MAAM,GAAG,CAAC;oBAChC,CAAC,CAAC,eAAe,CAAC,EAAE,EAAE,aAAa,EAAE,UAAU,CAAC;oBAChD,CAAC,CAAC,SAAS;aACd,CAAC,CAAC;YACH,CAAC,EAAE,CAAC;QACN,CAAC;IACH,CAAC;IAED,+BAA+B;IAC/B,MAAM,CAAC,OAAO,EAAE,CAAC;IACjB,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CACxB,OAA0B;IAE1B,IAAI,OAAO,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,QAAQ,GAAG,CAAC,CAAC;IAEtD,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;QAC3B,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,SAAS;gBAAE,OAAO,EAAE,CAAC;gBAAC,MAAM;YACjC,KAAK,OAAO;gBAAE,KAAK,EAAE,CAAC;gBAAC,MAAM;YAC7B,KAAK,SAAS;gBAAE,OAAO,EAAE,CAAC;gBAAC,MAAM;YACjC,KAAK,UAAU;gBAAE,QAAQ,EAAE,CAAC;gBAAC,MAAM;QACrC,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,UAAU,GAAG,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACpD,OAAO,IAAI,UAAU,CAAC,OAAO,CAAC;YAC9B,KAAK,IAAI,UAAU,CAAC,KAAK,CAAC;YAC1B,OAAO,IAAI,UAAU,CAAC,OAAO,CAAC;YAC9B,QAAQ,IAAI,UAAU,CAAC,QAAQ,CAAC;QAClC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;AAC/C,CAAC;AAED;;GAEG;AACH,SAAS,UAAU,CAAC,KAAwB;IAC1C,IAAI,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC;IACzB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,KAAK,IAAI,UAAU,CAAC,IAAI,CAAC,QAA6B,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,aAAa,CAC3B,SAAiB,EACjB,UAAkB,EAClB,UAAiC;IAEjC,KAAK,CAAC,iBAAiB,EAAE,mBAAmB,EAAE,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClG,wBAAwB;IACxB,MAAM,eAAe,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;IACtD,MAAM,gBAAgB,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAC;IAExD,kBAAkB;IAClB,MAAM,WAAW,GAAG,eAAe,CAAC,eAAe,EAAE,gBAAgB,EAAE,UAAU,CAAC,CAAC;IAEnF,uBAAuB;IACvB,MAAM,KAAK,GAAG,iBAAiB,CAAC,WAAW,CAAC,CAAC;IAE7C,OAAO;QACL,WAAW;QACX,KAAK,EAAE;YACL,SAAS,EAAE,UAAU,CAAC,eAAe,CAAC;YACtC,UAAU,EAAE,UAAU,CAAC,gBAAgB,CAAC;YACxC,GAAG,KAAK;SACT;KACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,OAA0B;IAC3D,MAAM,MAAM,GAAsB,EAAE,CAAC;IAErC,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;QAC3B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,IAA6B;IAMjE,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,SAAS;YACZ,OAAO;gBACL,KAAK,EAAE,SAAS;gBAChB,KAAK,EAAE,gBAAgB;gBACvB,OAAO,EAAE,iBAAiB;gBAC1B,WAAW,EAAE,qBAAqB;aACnC,CAAC;QACJ,KAAK,OAAO;YACV,OAAO;gBACL,KAAK,EAAE,OAAO;gBACd,KAAK,EAAE,gBAAgB;gBACvB,OAAO,EAAE,iBAAiB;gBAC1B,WAAW,EAAE,qBAAqB;aACnC,CAAC;QACJ,KAAK,SAAS;YACZ,OAAO;gBACL,KAAK,EAAE,SAAS;gBAChB,KAAK,EAAE,cAAc;gBACrB,OAAO,EAAE,eAAe;gBACxB,WAAW,EAAE,mBAAmB;aACjC,CAAC;QACJ,KAAK,UAAU;YACb,OAAO;gBACL,KAAK,EAAE,UAAU;gBACjB,KAAK,EAAE,gBAAgB;gBACvB,OAAO,EAAE,iBAAiB;gBAC1B,WAAW,EAAE,qBAAqB;aACnC,CAAC;IACN,CAAC;AACH,CAAC"}

package/lib/dist/services/traces/traceGrouping.d.ts

@@ -0,0 +1,19 @@
+/**
+ * Trace Grouping Utility
+ *
+ * Groups flat spans by traceId and calculates summary statistics
+ * for trace list display.
+ */
+import { Span, TraceSummary } from '../../types/index.js';
+/**
+ * Group flat spans by traceId and calculate summary statistics
+ *
+ * @param spans - Flat array of spans from API
+ * @returns Array of TraceSummary objects, sorted by startTime descending (newest first)
+ */
+export declare function groupSpansByTrace(spans: Span[]): TraceSummary[];
+/**
+ * Get spans for a specific trace from grouped data
+ */
+export declare function getSpansForTrace(summaries: TraceSummary[], traceId: string): Span[];
+//# sourceMappingURL=traceGrouping.d.ts.map

package/lib/dist/services/traces/traceGrouping.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"traceGrouping.d.ts","sourceRoot":"","sources":["../../../../services/traces/traceGrouping.ts"],"names":[],"mappings":"AAKA;;;;;GAKG;AAEH,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AA8D7C;;;;;GAKG;AACH,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG,YAAY,EAAE,CA6C/D;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,YAAY,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,EAAE,CAGnF"}

package/lib/dist/services/traces/traceGrouping.js

@@ -0,0 +1,107 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { debug } from '../../lib/debug.js';
+/**
+ * Extract service name from span attributes
+ */
+function extractServiceName(span) {
+    return (span.attributes?.['service.name'] ||
+        span.attributes?.['gen_ai.system'] ||
+        'unknown');
+}
+/**
+ * Find the root span (no parentSpanId) from a list of spans
+ */
+function findRootSpan(spans) {
+    return spans.find(span => !span.parentSpanId);
+}
+/**
+ * Calculate total duration from spans (max endTime - min startTime)
+ */
+function calculateDuration(spans) {
+    if (spans.length === 0)
+        return 0;
+    let minStart = Infinity;
+    let maxEnd = -Infinity;
+    for (const span of spans) {
+        const startTime = new Date(span.startTime).getTime();
+        const endTime = new Date(span.endTime).getTime();
+        if (startTime < minStart)
+            minStart = startTime;
+        if (endTime > maxEnd)
+            maxEnd = endTime;
+    }
+    return maxEnd - minStart;
+}
+/**
+ * Check if any span in the list has an error status
+ */
+function hasErrors(spans) {
+    return spans.some(span => span.status === 'ERROR');
+}
+/**
+ * Get earliest start time from spans
+ */
+function getEarliestStartTime(spans) {
+    if (spans.length === 0)
+        return new Date().toISOString();
+    let earliest = spans[0].startTime;
+    for (const span of spans) {
+        if (new Date(span.startTime).getTime() < new Date(earliest).getTime()) {
+            earliest = span.startTime;
+        }
+    }
+    return earliest;
+}
+/**
+ * Group flat spans by traceId and calculate summary statistics
+ *
+ * @param spans - Flat array of spans from API
+ * @returns Array of TraceSummary objects, sorted by startTime descending (newest first)
+ */
+export function groupSpansByTrace(spans) {
+    if (!spans || spans.length === 0)
+        return [];
+    debug('TraceGrouping', 'Grouping', spans.length, 'spans by traceId');
+    // Group spans by traceId
+    const traceGroups = new Map();
+    for (const span of spans) {
+        const existing = traceGroups.get(span.traceId) || [];
+        existing.push(span);
+        traceGroups.set(span.traceId, existing);
+    }
+    // Convert to TraceSummary objects
+    const summaries = [];
+    for (const [traceId, traceSpans] of traceGroups) {
+        const rootSpan = findRootSpan(traceSpans);
+        const serviceName = rootSpan
+            ? extractServiceName(rootSpan)
+            : extractServiceName(traceSpans[0]);
+        summaries.push({
+            traceId,
+            serviceName,
+            spanCount: traceSpans.length,
+            rootSpanName: rootSpan?.name || traceSpans[0]?.name || 'Unknown',
+            startTime: getEarliestStartTime(traceSpans),
+            duration: calculateDuration(traceSpans),
+            hasErrors: hasErrors(traceSpans),
+            hasEvalSpans: traceSpans.some(s => s.attributes?.['gen_ai.operation.name'] === 'evaluation' ||
+                s.name?.includes('test_case')),
+            spans: traceSpans,
+        });
+    }
+    // Sort by startTime descending (newest first)
+    summaries.sort((a, b) => new Date(b.startTime).getTime() - new Date(a.startTime).getTime());
+    debug('TraceGrouping', 'Grouped into', summaries.length, 'traces');
+    return summaries;
+}
+/**
+ * Get spans for a specific trace from grouped data
+ */
+export function getSpansForTrace(summaries, traceId) {
+    const summary = summaries.find(s => s.traceId === traceId);
+    return summary?.spans || [];
+}
+//# sourceMappingURL=traceGrouping.js.map

package/lib/dist/services/traces/traceGrouping.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"traceGrouping.js","sourceRoot":"","sources":["../../../../services/traces/traceGrouping.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAUH,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAEpC;;GAEG;AACH,SAAS,kBAAkB,CAAC,IAAU;IACpC,OAAO,CACL,IAAI,CAAC,UAAU,EAAE,CAAC,cAAc,CAAC;QACjC,IAAI,CAAC,UAAU,EAAE,CAAC,eAAe,CAAC;QAClC,SAAS,CACV,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,YAAY,CAAC,KAAa;IACjC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;AAChD,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,KAAa;IACtC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IAEjC,IAAI,QAAQ,GAAG,QAAQ,CAAC;IACxB,IAAI,MAAM,GAAG,CAAC,QAAQ,CAAC;IAEvB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;QACrD,MAAM,OAAO,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;QACjD,IAAI,SAAS,GAAG,QAAQ;YAAE,QAAQ,GAAG,SAAS,CAAC;QAC/C,IAAI,OAAO,GAAG,MAAM;YAAE,MAAM,GAAG,OAAO,CAAC;IACzC,CAAC;IAED,OAAO,MAAM,GAAG,QAAQ,CAAC;AAC3B,CAAC;AAED;;GAEG;AACH,SAAS,SAAS,CAAC,KAAa;IAC9B,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,OAAO,CAAC,CAAC;AACrD,CAAC;AAED;;GAEG;AACH,SAAS,oBAAoB,CAAC,KAAa;IACzC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAExD,IAAI,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAClC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;YACtE,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;QAC5B,CAAC;IACH,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,iBAAiB,CAAC,KAAa;IAC7C,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAC5C,KAAK,CAAC,eAAe,EAAE,UAAU,EAAE,KAAK,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC;IAErE,yBAAyB;IACzB,MAAM,WAAW,GAAG,IAAI,GAAG,EAAkB,CAAC;IAE9C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QACrD,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpB,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC1C,CAAC;IAED,kCAAkC;IAClC,MAAM,SAAS,GAAmB,EAAE,CAAC;IAErC,KAAK,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,WAAW,EAAE,CAAC;QAChD,MAAM,QAAQ,GAAG,YAAY,CAAC,UAAU,CAAC,CAAC;QAC1C,MAAM,WAAW,GAAG,QAAQ;YAC1B,CAAC,CAAC,kBAAkB,CAAC,QAAQ,CAAC;YAC9B,CAAC,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;QAEtC,SAAS,CAAC,IAAI,CAAC;YACb,OAAO;YACP,WAAW;YACX,SAAS,EAAE,UAAU,CAAC,MAAM;YAC5B,YAAY,EAAE,QAAQ,EAAE,IAAI,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,SAAS;YAChE,SAAS,EAAE,oBAAoB,CAAC,UAAU,CAAC;YAC3C,QAAQ,EAAE,iBAAiB,CAAC,UAAU,CAAC;YACvC,SAAS,EAAE,SAAS,CAAC,UAAU,CAAC;YAChC,YAAY,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAChC,CAAC,CAAC,UAAU,EAAE,CAAC,uBAAuB,CAAC,KAAK,YAAY;gBACxD,CAAC,CAAC,IAAI,EAAE,QAAQ,CAAC,WAAW,CAAC,CAC9B;YACD,KAAK,EAAE,UAAU;SAClB,CAAC,CAAC;IACL,CAAC;IAED,8CAA8C;IAC9C,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACtB,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAClE,CAAC;IAEF,KAAK,CAAC,eAAe,EAAE,cAAc,EAAE,SAAS,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACnE,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,SAAyB,EAAE,OAAe;IACzE,MAAM,OAAO,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,OAAO,CAAC,CAAC;IAC3D,OAAO,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;AAC9B,CAAC"}

package/lib/dist/services/traces/tracePoller.d.ts

@@ -0,0 +1,84 @@
+/**
+ * Trace Polling Service
+ *
+ * Manages polling for trace availability after a trace-mode run completes.
+ * Traces take ~5 minutes to propagate to OpenSearch after agent execution.
+ */
+import { Span, EvaluationReport, AgentConfig } from '../../types/index.js';
+export interface PollState {
+    reportId: string;
+    runId: string;
+    attempts: number;
+    maxAttempts: number;
+    intervalMs: number;
+    lastAttempt: string | null;
+    running: boolean;
+    timerId?: ReturnType<typeof setTimeout>;
+    agentConfig?: AgentConfig;
+}
+export interface PollCallbacks {
+    onTracesFound: (spans: Span[], report: EvaluationReport) => Promise<void>;
+    onAttempt?: (attempt: number, maxAttempts: number) => void;
+    onError: (error: Error) => void;
+}
+/**
+ * Trace Polling Manager
+ *
+ * Singleton that manages active polling for trace availability.
+ * State is in-memory only - polling is short-lived (~10 min max).
+ *
+ * Polling runs in two places for redundancy:
+ * - Server (experimentRunner.ts): Primary - starts immediately after agent execution
+ * - Browser (RunDetailsContent.tsx): Recovery - starts when viewing a pending report
+ */
+declare class TracePollingManager {
+    private polls;
+    private callbacks;
+    private completionPromises;
+    /**
+     * Start polling for traces for a specific report
+     */
+    startPolling(reportId: string, runId: string, callbacks: PollCallbacks, options?: {
+        intervalMs?: number;
+        maxAttempts?: number;
+        agentConfig?: AgentConfig;
+    }): void;
+    /**
+     * Stop polling for a specific report.
+     * If a completion promise exists (from startPollingAsync), it is rejected
+     * so callers awaiting it are unblocked.
+     */
+    stopPolling(reportId: string): void;
+    /**
+     * Get the state for a specific poll
+     */
+    getState(reportId: string): PollState | undefined;
+    /**
+     * Get all active polls
+     */
+    getAllActivePolls(): Map<string, PollState>;
+    /**
+     * Start polling and return a Promise that resolves when polling completes.
+     * This allows callers (e.g., benchmark runner) to await trace availability
+     * instead of firing and forgetting.
+     *
+     * If polling is already active for this reportId, returns the existing
+     * completion promise (no duplicate poll started).
+     */
+    startPollingAsync(reportId: string, runId: string, callbacks: PollCallbacks, options?: {
+        intervalMs?: number;
+        maxAttempts?: number;
+        agentConfig?: AgentConfig;
+    }): Promise<void>;
+    /**
+     * Execute a single poll attempt
+     */
+    private poll;
+    /**
+     * Build trajectory from spans with proper error handling
+     */
+    private buildTrajectory;
+}
+export declare const tracePollingManager: TracePollingManager;
+export {};
+//# sourceMappingURL=tracePoller.d.ts.map

package/lib/dist/services/traces/tracePoller.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracePoller.d.ts","sourceRoot":"","sources":["../../../../services/traces/tracePoller.ts"],"names":[],"mappings":"AAKA;;;;;GAKG;AAEH,OAAO,EAAE,IAAI,EAAE,gBAAgB,EAAE,WAAW,EAA0B,MAAM,SAAS,CAAC;AA0BtF,MAAM,WAAW,SAAS;IACxB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,CAAC;IACxC,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AAED,MAAM,WAAW,aAAa;IAC5B,aAAa,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,gBAAgB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1E,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,KAAK,IAAI,CAAC;IAC3D,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CACjC;AAED;;;;;;;;;GASG;AACH,cAAM,mBAAmB;IACvB,OAAO,CAAC,KAAK,CAAqC;IAClD,OAAO,CAAC,SAAS,CAAyC;IAC1D,OAAO,CAAC,kBAAkB,CAAiF;IAE3G;;OAEG;IACH,YAAY,CACV,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,aAAa,EACxB,OAAO,CAAC,EAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,WAAW,CAAA;KAAE,GACjF,IAAI;IAyBP;;;;OAIG;IACH,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAmBnC;;OAEG;IACH,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAIjD;;OAEG;IACH,iBAAiB,IAAI,GAAG,CAAC,MAAM,EAAE,SAAS,CAAC;IAU3C;;;;;;;OAOG;IACH,iBAAiB,CACf,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,aAAa,EACxB,OAAO,CAAC,EAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,WAAW,CAAA;KAAE,GACjF,OAAO,CAAC,IAAI,CAAC;IAyChB;;OAEG;YACW,IAAI;IA4IlB;;OAEG;YACW,eAAe;CAgC9B;AAGD,eAAO,MAAM,mBAAmB,qBAA4B,CAAC"}