@opensearch-project/agent-health 0.4.0 → 0.6.0

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

@@ -0,0 +1,114 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ATTR_GEN_AI_TOOL_NAME } from '@opentelemetry/semantic-conventions/incubating';
+/**
+ * Extract tool name from a span
+ */
+export function extractToolName(span) {
+    // Try gen_ai.tool.name attribute first (OTel semantic convention)
+    const toolName = span.attributes?.[ATTR_GEN_AI_TOOL_NAME];
+    if (toolName)
+        return toolName;
+    // Parse from displayName or name
+    const name = span.displayName || span.name || '';
+    // Look for tool name patterns
+    const patterns = [
+        /execute_tool\s+(\S+)/i,
+        /executeTools,\s*(\S+)/i,
+        /tool\.execute\s+(\S+)/i,
+    ];
+    for (const pattern of patterns) {
+        const match = name.match(pattern);
+        if (match)
+            return match[1];
+    }
+    // Try to get the last meaningful part after comma
+    if (name.includes(',')) {
+        const parts = name.split(',');
+        const lastPart = parts[parts.length - 1].trim();
+        if (lastPart && !lastPart.includes('agent.node')) {
+            return lastPart;
+        }
+    }
+    return null;
+}
+/**
+ * Flatten span tree and collect all spans
+ */
+export function flattenSpans(spans) {
+    const result = [];
+    const collect = (spanList) => {
+        for (const span of spanList) {
+            result.push(span);
+            if (span.children && span.children.length > 0) {
+                collect(span.children);
+            }
+        }
+    };
+    collect(spans);
+    return result;
+}
+/**
+ * Calculate category statistics from spans
+ *
+ * Percentages are calculated relative to the sum of all category durations,
+ * not trace duration, because spans can overlap (LLM calls inside AGENT spans).
+ * This ensures percentages naturally sum to 100%.
+ */
+export function calculateCategoryStats(spans, _totalDuration) {
+    const categoryMap = new Map();
+    for (const span of spans) {
+        const existing = categoryMap.get(span.category) || { count: 0, duration: 0 };
+        categoryMap.set(span.category, {
+            count: existing.count + 1,
+            duration: existing.duration + (span.duration || 0),
+        });
+    }
+    // Calculate sum of all category durations for percentage calculation
+    let sumOfAllDurations = 0;
+    categoryMap.forEach((data) => {
+        sumOfAllDurations += data.duration;
+    });
+    const stats = [];
+    categoryMap.forEach((data, category) => {
+        stats.push({
+            category,
+            count: data.count,
+            totalDuration: data.duration,
+            percentage: sumOfAllDurations > 0 ? (data.duration / sumOfAllDurations) * 100 : 0,
+        });
+    });
+    // Sort by duration descending
+    return stats.sort((a, b) => b.totalDuration - a.totalDuration);
+}
+/**
+ * Extract unique tools with usage stats
+ */
+export function extractToolStats(spans) {
+    const toolMap = new Map();
+    for (const span of spans) {
+        if (span.category === 'TOOL') {
+            const toolName = extractToolName(span);
+            if (toolName) {
+                const existing = toolMap.get(toolName) || { count: 0, duration: 0 };
+                toolMap.set(toolName, {
+                    count: existing.count + 1,
+                    duration: existing.duration + (span.duration || 0),
+                });
+            }
+        }
+    }
+    const tools = [];
+    toolMap.forEach((data, name) => {
+        tools.push({
+            name,
+            count: data.count,
+            totalDuration: data.duration,
+        });
+    });
+    // Sort by usage count descending
+    return tools.sort((a, b) => b.count - a.count);
+}
+//# sourceMappingURL=traceStats.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"traceStats.js","sourceRoot":"","sources":["../../../../services/traces/traceStats.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAUH,OAAO,EAAE,qBAAqB,EAAE,MAAM,gDAAgD,CAAC;AAqBvF;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,IAAqB;IACnD,kEAAkE;IAClE,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,qBAAqB,CAAC,CAAC;IAC1D,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAE9B,iCAAiC;IACjC,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;IAEjD,8BAA8B;IAC9B,MAAM,QAAQ,GAAG;QACf,uBAAuB;QACvB,wBAAwB;QACxB,wBAAwB;KACzB,CAAC;IAEF,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAClC,IAAI,KAAK;YAAE,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC;IAC7B,CAAC;IAED,kDAAkD;IAClD,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC9B,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAChD,IAAI,QAAQ,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YACjD,OAAO,QAAQ,CAAC;QAClB,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,KAAwB;IACnD,MAAM,MAAM,GAAsB,EAAE,CAAC;IAErC,MAAM,OAAO,GAAG,CAAC,QAA2B,EAAE,EAAE;QAC9C,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC5B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClB,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9C,OAAO,CAAC,IAAI,CAAC,QAA6B,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;IACH,CAAC,CAAC;IAEF,OAAO,CAAC,KAAK,CAAC,CAAC;IACf,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,sBAAsB,CAAC,KAAwB,EAAE,cAAsB;IACrF,MAAM,WAAW,GAAG,IAAI,GAAG,EAAqD,CAAC;IAEjF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QAC7E,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE;YAC7B,KAAK,EAAE,QAAQ,CAAC,KAAK,GAAG,CAAC;YACzB,QAAQ,EAAE,QAAQ,CAAC,QAAQ,GAAG,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,CAAC;SACnD,CAAC,CAAC;IACL,CAAC;IAED,qEAAqE;IACrE,IAAI,iBAAiB,GAAG,CAAC,CAAC;IAC1B,WAAW,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAC3B,iBAAiB,IAAI,IAAI,CAAC,QAAQ,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,MAAM,KAAK,GAAoB,EAAE,CAAC;IAClC,WAAW,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,QAAQ,EAAE,EAAE;QACrC,KAAK,CAAC,IAAI,CAAC;YACT,QAAQ;YACR,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,aAAa,EAAE,IAAI,CAAC,QAAQ;YAC5B,UAAU,EAAE,iBAAiB,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,GAAG,iBAAiB,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;SAClF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,8BAA8B;IAC9B,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,GAAG,CAAC,CAAC,aAAa,CAAC,CAAC;AACjE,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAwB;IACvD,MAAM,OAAO,GAAG,IAAI,GAAG,EAA+C,CAAC;IAEvE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;YACvC,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;gBACpE,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE;oBACpB,KAAK,EAAE,QAAQ,CAAC,KAAK,GAAG,CAAC;oBACzB,QAAQ,EAAE,QAAQ,CAAC,QAAQ,GAAG,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,CAAC;iBACnD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,KAAK,GAAe,EAAE,CAAC;IAC7B,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE;QAC7B,KAAK,CAAC,IAAI,CAAC;YACT,IAAI;YACJ,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,aAAa,EAAE,IAAI,CAAC,QAAQ;SAC7B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,iCAAiC;IACjC,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;AACjD,CAAC"}

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

@@ -0,0 +1,47 @@
+/**
+ * traceSummary
+ *
+ * Computes a compact summary for a trace's span tree — the same set of
+ * non-redundant signals shown in the Agent Traces inline-expansion header
+ * strip and the fullscreen header. Keeping the computation in one place
+ * means inline and fullscreen render exactly the same numbers from the
+ * same source-of-truth (`categorizeSpanTree` for category buckets,
+ * `gen_ai.usage.*` attribute aggregation for tokens, and dedup over
+ * `gen_ai.request.model` / `gen_ai.response.model` / `model` for the
+ * model name list).
+ */
+import { Span } from '../../types/index.js';
+export interface TraceSummary {
+    llm: number;
+    tool: number;
+    agent: number;
+    evalCount: number;
+    errors: number;
+    inputTokens: number;
+    outputTokens: number;
+    totalTokens: number;
+    /** Largest single-span input-token count in the trace. Used for
+        context-window utilization (Ctx%) — the peak request, not the sum. */
+    peakInputTokens: number;
+    models: string[];
+}
+/**
+ * Aggregate the headline signals from a span tree.
+ *
+ * The category counts come from {@link countByCategory} on a categorized
+ * tree. The token counters look at every span (including children) and
+ * sum the OTel GenAI usage attributes, with `prompt_tokens` /
+ * `completion_tokens` accepted as fallbacks for older instrumentation
+ * that pre-dates the input/output rename. Models are collected and
+ * deduplicated across the trace because some agents fan out to multiple
+ * models in one trace (e.g. a planner + a tool-using model).
+ */
+export declare function computeTraceSummary(spanTree: Span[]): TraceSummary;
+/**
+ * True when no headline signal is present (no LLM/tool/agent/eval span,
+ * no errors, no token counters, no model). Callers can use this to fall
+ * back to a "no summary attributes available" placeholder instead of
+ * collapsing to an empty bar.
+ */
+export declare function isEmptyTraceSummary(s: TraceSummary): boolean;
+//# sourceMappingURL=traceSummary.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"traceSummary.d.ts","sourceRoot":"","sources":["../../../../services/traces/traceSummary.ts"],"names":[],"mappings":"AAKA;;;;;;;;;;;GAWG;AACH,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAI/B,MAAM,WAAW,YAAY;IAC3B,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB;6EACyE;IACzE,eAAe,EAAE,MAAM,CAAC;IACxB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,mBAAmB,CAAC,QAAQ,EAAE,IAAI,EAAE,GAAG,YAAY,CAqClE;AAED;;;;;GAKG;AACH,wBAAgB,mBAAmB,CAAC,CAAC,EAAE,YAAY,GAAG,OAAO,CAU5D"}

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

@@ -0,0 +1,68 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { categorizeSpanTree, countByCategory } from './spanCategorization.js';
+import { flattenSpans } from './traceStats.js';
+/**
+ * Aggregate the headline signals from a span tree.
+ *
+ * The category counts come from {@link countByCategory} on a categorized
+ * tree. The token counters look at every span (including children) and
+ * sum the OTel GenAI usage attributes, with `prompt_tokens` /
+ * `completion_tokens` accepted as fallbacks for older instrumentation
+ * that pre-dates the input/output rename. Models are collected and
+ * deduplicated across the trace because some agents fan out to multiple
+ * models in one trace (e.g. a planner + a tool-using model).
+ */
+export function computeTraceSummary(spanTree) {
+    const categorized = categorizeSpanTree(spanTree);
+    const counts = countByCategory(categorized);
+    const flat = flattenSpans(categorized);
+    let inputTokens = 0;
+    let outputTokens = 0;
+    let totalTokens = 0;
+    let peakInputTokens = 0;
+    const modelSet = new Set();
+    for (const s of flat) {
+        const a = s.attributes || {};
+        const it = Number(a['gen_ai.usage.input_tokens'] ?? a['gen_ai.usage.prompt_tokens'] ?? a['input_tokens'] ?? 0) || 0;
+        const ot = Number(a['gen_ai.usage.output_tokens'] ?? a['gen_ai.usage.completion_tokens'] ?? a['output_tokens'] ?? 0) || 0;
+        inputTokens += it;
+        outputTokens += ot;
+        totalTokens += it + ot;
+        if (it > peakInputTokens)
+            peakInputTokens = it;
+        const m = a['gen_ai.request.model'] || a['gen_ai.response.model'] || a['model'];
+        if (typeof m === 'string' && m.trim())
+            modelSet.add(m.trim());
+    }
+    return {
+        llm: counts.LLM,
+        tool: counts.TOOL,
+        agent: counts.AGENT,
+        evalCount: counts.EVAL,
+        errors: counts.ERROR,
+        inputTokens,
+        outputTokens,
+        totalTokens,
+        peakInputTokens,
+        models: Array.from(modelSet),
+    };
+}
+/**
+ * True when no headline signal is present (no LLM/tool/agent/eval span,
+ * no errors, no token counters, no model). Callers can use this to fall
+ * back to a "no summary attributes available" placeholder instead of
+ * collapsing to an empty bar.
+ */
+export function isEmptyTraceSummary(s) {
+    return (s.llm === 0 &&
+        s.tool === 0 &&
+        s.agent === 0 &&
+        s.evalCount === 0 &&
+        s.errors === 0 &&
+        s.totalTokens === 0 &&
+        s.models.length === 0);
+}
+//# sourceMappingURL=traceSummary.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"traceSummary.js","sourceRoot":"","sources":["../../../../services/traces/traceSummary.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAeH,OAAO,EAAE,kBAAkB,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAC3E,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAiB5C;;;;;;;;;;GAUG;AACH,MAAM,UAAU,mBAAmB,CAAC,QAAgB;IAClD,MAAM,WAAW,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IACjD,MAAM,MAAM,GAAG,eAAe,CAAC,WAAW,CAAC,CAAC;IAC5C,MAAM,IAAI,GAAG,YAAY,CAAC,WAAW,CAAC,CAAC;IAEvC,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,eAAe,GAAG,CAAC,CAAC;IACxB,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;IAEnC,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;QACrB,MAAM,CAAC,GAAG,CAAC,CAAC,UAAU,IAAI,EAAE,CAAC;QAC7B,MAAM,EAAE,GACN,MAAM,CAAC,CAAC,CAAC,2BAA2B,CAAC,IAAI,CAAC,CAAC,4BAA4B,CAAC,IAAI,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3G,MAAM,EAAE,GACN,MAAM,CAAC,CAAC,CAAC,4BAA4B,CAAC,IAAI,CAAC,CAAC,gCAAgC,CAAC,IAAI,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC;QACjH,WAAW,IAAI,EAAE,CAAC;QAClB,YAAY,IAAI,EAAE,CAAC;QACnB,WAAW,IAAI,EAAE,GAAG,EAAE,CAAC;QACvB,IAAI,EAAE,GAAG,eAAe;YAAE,eAAe,GAAG,EAAE,CAAC;QAC/C,MAAM,CAAC,GAAG,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC;QAChF,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,EAAE;YAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAChE,CAAC;IAED,OAAO;QACL,GAAG,EAAE,MAAM,CAAC,GAAG;QACf,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,SAAS,EAAE,MAAM,CAAC,IAAI;QACtB,MAAM,EAAE,MAAM,CAAC,KAAK;QACpB,WAAW;QACX,YAAY;QACZ,WAAW;QACX,eAAe;QACf,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC;KAC7B,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,mBAAmB,CAAC,CAAe;IACjD,OAAO,CACL,CAAC,CAAC,GAAG,KAAK,CAAC;QACX,CAAC,CAAC,IAAI,KAAK,CAAC;QACZ,CAAC,CAAC,KAAK,KAAK,CAAC;QACb,CAAC,CAAC,SAAS,KAAK,CAAC;QACjB,CAAC,CAAC,MAAM,KAAK,CAAC;QACd,CAAC,CAAC,WAAW,KAAK,CAAC;QACnB,CAAC,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,CACtB,CAAC;AACJ,CAAC"}

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

@@ -0,0 +1,33 @@
+/**
+ * Trace Utility Functions
+ */
+import { Span } from '../../types/index.js';
+/**
+ * Format duration in human-readable format
+ */
+export declare function formatDuration(ms: number | null | undefined): string;
+/**
+ * Extract key attributes based on span type
+ */
+export declare function getKeyAttributes(span: Span): Record<string, string | number | null>;
+/**
+ * Estimate token count from text (rough approximation)
+ */
+export declare function estimateTokens(text: string | object | null | undefined): number;
+/**
+ * Truncate text with ellipsis
+ */
+export declare function truncateText(text: string, maxLength: number): string;
+/**
+ * Format timestamp for display
+ */
+export declare function formatTimestamp(timestamp: string): string;
+/**
+ * Format large numbers in compact notation: 999 → "999", 1200 → "1.2k", 1500000 → "1.5M"
+ */
+export declare function formatCompact(n: number): string;
+/**
+ * Parse JSON safely
+ */
+export declare function safeParseJSON(str: string): object | string;
+//# sourceMappingURL=utils.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../../services/traces/utils.ts"],"names":[],"mappings":"AAKA;;GAEG;AAEH,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAU/B;;GAEG;AACH,wBAAgB,cAAc,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,GAAG,MAAM,CASpE;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,IAAI,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC,CAuCnF;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,GAAG,SAAS,GAAG,MAAM,CAI/E;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAGpE;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CASzD;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,CAAC,EAAE,MAAM,GAAG,MAAM,CAQ/C;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAM1D"}

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

@@ -0,0 +1,114 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ATTR_GEN_AI_REQUEST_MODEL, ATTR_GEN_AI_USAGE_INPUT_TOKENS, ATTR_GEN_AI_USAGE_OUTPUT_TOKENS, ATTR_GEN_AI_REQUEST_TEMPERATURE, ATTR_GEN_AI_TOOL_NAME, ATTR_GEN_AI_TOOL_CALL_ID, } from '@opentelemetry/semantic-conventions/incubating';
+/**
+ * Format duration in human-readable format
+ */
+export function formatDuration(ms) {
+    if (ms == null || isNaN(ms))
+        return '0ms';
+    if (ms >= 60000) {
+        const mins = Math.floor(ms / 60000);
+        const secs = ((ms % 60000) / 1000).toFixed(1);
+        return `${mins}m ${secs}s`;
+    }
+    if (ms >= 1000)
+        return `${(ms / 1000).toFixed(2)}s`;
+    return `${Math.round(ms)}ms`;
+}
+/**
+ * Extract key attributes based on span type
+ */
+export function getKeyAttributes(span) {
+    const attrs = span.attributes || {};
+    const name = span.name || '';
+    // Eval spans
+    const opName = attrs['gen_ai.operation.name'];
+    if (opName === 'evaluation' || name.includes('test_case') || name.includes('test_suite_run')) {
+        return {
+            'Test Case': attrs['test.case.name'],
+            'Result': attrs['test.case.result.status'],
+            'Suite': attrs['test.suite.name'],
+            'Service': attrs['service.name'] || attrs['serviceName'],
+        };
+    }
+    // Bedrock/LLM spans
+    if (name.startsWith('bedrock.') || name.includes('llm') || name.includes('converse')) {
+        return {
+            'Model': attrs[ATTR_GEN_AI_REQUEST_MODEL]?.split('.').pop() || attrs[ATTR_GEN_AI_REQUEST_MODEL],
+            'Input Tokens': attrs[ATTR_GEN_AI_USAGE_INPUT_TOKENS]?.toLocaleString(),
+            'Output Tokens': attrs[ATTR_GEN_AI_USAGE_OUTPUT_TOKENS]?.toLocaleString(),
+            'Temperature': attrs[ATTR_GEN_AI_REQUEST_TEMPERATURE],
+        };
+    }
+    // Tool spans
+    if (name.includes('tool')) {
+        return {
+            'Tool': attrs[ATTR_GEN_AI_TOOL_NAME] || name,
+            'Tool Call ID': attrs[ATTR_GEN_AI_TOOL_CALL_ID],
+            'MCP Server': attrs['tool.mcp_server'],
+        };
+    }
+    // Default attributes
+    return {
+        'Service': attrs['service.name'] || attrs['serviceName'],
+        'Kind': attrs['spanKind'],
+    };
+}
+/**
+ * Estimate token count from text (rough approximation)
+ */
+export function estimateTokens(text) {
+    if (!text)
+        return 0;
+    const charCount = typeof text === 'string' ? text.length : JSON.stringify(text).length;
+    return Math.ceil(charCount / 4);
+}
+/**
+ * Truncate text with ellipsis
+ */
+export function truncateText(text, maxLength) {
+    if (text.length <= maxLength)
+        return text;
+    return text.substring(0, maxLength - 3) + '...';
+}
+/**
+ * Format timestamp for display
+ */
+export function formatTimestamp(timestamp) {
+    const date = new Date(timestamp);
+    return date.toLocaleTimeString('en-US', {
+        hour12: false,
+        hour: '2-digit',
+        minute: '2-digit',
+        second: '2-digit',
+        fractionalSecondDigits: 3
+    });
+}
+/**
+ * Format large numbers in compact notation: 999 → "999", 1200 → "1.2k", 1500000 → "1.5M"
+ */
+export function formatCompact(n) {
+    if (n < 1000)
+        return n.toString();
+    if (n < 1_000_000) {
+        const v = n / 1000;
+        return v >= 100 ? `${Math.round(v)}k` : `${parseFloat(v.toFixed(1))}k`;
+    }
+    const v = n / 1_000_000;
+    return v >= 100 ? `${Math.round(v)}M` : `${parseFloat(v.toFixed(1))}M`;
+}
+/**
+ * Parse JSON safely
+ */
+export function safeParseJSON(str) {
+    try {
+        return JSON.parse(str);
+    }
+    catch {
+        return str;
+    }
+}
+//# sourceMappingURL=utils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../../services/traces/utils.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAOH,OAAO,EACL,yBAAyB,EACzB,8BAA8B,EAC9B,+BAA+B,EAC/B,+BAA+B,EAC/B,qBAAqB,EACrB,wBAAwB,GACzB,MAAM,gDAAgD,CAAC;AAExD;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,EAA6B;IAC1D,IAAI,EAAE,IAAI,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;QAAE,OAAO,KAAK,CAAC;IAC1C,IAAI,EAAE,IAAI,KAAK,EAAE,CAAC;QAChB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,GAAG,KAAK,CAAC,CAAC;QACpC,MAAM,IAAI,GAAG,CAAC,CAAC,EAAE,GAAG,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAC9C,OAAO,GAAG,IAAI,KAAK,IAAI,GAAG,CAAC;IAC7B,CAAC;IACD,IAAI,EAAE,IAAI,IAAI;QAAE,OAAO,GAAG,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IACpD,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC;AAC/B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,IAAU;IACzC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC;IACpC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;IAE7B,aAAa;IACb,MAAM,MAAM,GAAG,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC9C,IAAI,MAAM,KAAK,YAAY,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;QAC7F,OAAO;YACL,WAAW,EAAE,KAAK,CAAC,gBAAgB,CAAC;YACpC,QAAQ,EAAE,KAAK,CAAC,yBAAyB,CAAC;YAC1C,OAAO,EAAE,KAAK,CAAC,iBAAiB,CAAC;YACjC,SAAS,EAAE,KAAK,CAAC,cAAc,CAAC,IAAI,KAAK,CAAC,aAAa,CAAC;SACzD,CAAC;IACJ,CAAC;IAED,oBAAoB;IACpB,IAAI,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QACrF,OAAO;YACL,OAAO,EAAE,KAAK,CAAC,yBAAyB,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,KAAK,CAAC,yBAAyB,CAAC;YAC/F,cAAc,EAAE,KAAK,CAAC,8BAA8B,CAAC,EAAE,cAAc,EAAE;YACvE,eAAe,EAAE,KAAK,CAAC,+BAA+B,CAAC,EAAE,cAAc,EAAE;YACzE,aAAa,EAAE,KAAK,CAAC,+BAA+B,CAAC;SACtD,CAAC;IACJ,CAAC;IAED,aAAa;IACb,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO;YACL,MAAM,EAAE,KAAK,CAAC,qBAAqB,CAAC,IAAI,IAAI;YAC5C,cAAc,EAAE,KAAK,CAAC,wBAAwB,CAAC;YAC/C,YAAY,EAAE,KAAK,CAAC,iBAAiB,CAAC;SACvC,CAAC;IACJ,CAAC;IAED,qBAAqB;IACrB,OAAO;QACL,SAAS,EAAE,KAAK,CAAC,cAAc,CAAC,IAAI,KAAK,CAAC,aAAa,CAAC;QACxD,MAAM,EAAE,KAAK,CAAC,UAAU,CAAC;KAC1B,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,IAAwC;IACrE,IAAI,CAAC,IAAI;QAAE,OAAO,CAAC,CAAC;IACpB,MAAM,SAAS,GAAG,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;IACvF,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC;AAClC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,IAAY,EAAE,SAAiB;IAC1D,IAAI,IAAI,CAAC,MAAM,IAAI,SAAS;QAAE,OAAO,IAAI,CAAC;IAC1C,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;AAClD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,SAAiB;IAC/C,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC;IACjC,OAAO,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE;QACtC,MAAM,EAAE,KAAK;QACb,IAAI,EAAE,SAAS;QACf,MAAM,EAAE,SAAS;QACjB,MAAM,EAAE,SAAS;QACjB,sBAAsB,EAAE,CAAC;KAC1B,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa,CAAC,CAAS;IACrC,IAAI,CAAC,GAAG,IAAI;QAAE,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;IAClC,IAAI,CAAC,GAAG,SAAS,EAAE,CAAC;QAClB,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;QACnB,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;IACzE,CAAC;IACD,MAAM,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC;IACxB,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;AACzE,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa,CAAC,GAAW;IACvC,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,GAAG,CAAC;IACb,CAAC;AACH,CAAC"}

package/lib/dist/types/agui.d.ts

@@ -0,0 +1,13 @@
+/**
+ * AG UI Protocol Event Types
+ *
+ * Most types are imported directly from @ag-ui/core.
+ * This file provides:
+ * - AGUIEvent union type for type-safe event handling
+ * - AGUIEventType alias for the EventType enum
+ */
+import { EventType } from '@ag-ui/core';
+import type { RunStartedEvent, RunFinishedEvent, RunErrorEvent, TextMessageStartEvent, TextMessageContentEvent, TextMessageEndEvent, ThinkingStartEvent, ThinkingEndEvent, ThinkingTextMessageStartEvent, ThinkingTextMessageContentEvent, ThinkingTextMessageEndEvent, ToolCallStartEvent, ToolCallArgsEvent, ToolCallEndEvent, ToolCallResultEvent, StateSnapshotEvent, StateDeltaEvent, ActivitySnapshotEvent, ActivityDeltaEvent, MessagesSnapshotEvent } from '@ag-ui/core';
+export declare const AGUIEventType: typeof EventType;
+export type AGUIEvent = RunStartedEvent | RunFinishedEvent | RunErrorEvent | TextMessageStartEvent | TextMessageContentEvent | TextMessageEndEvent | ThinkingStartEvent | ThinkingEndEvent | ThinkingTextMessageStartEvent | ThinkingTextMessageContentEvent | ThinkingTextMessageEndEvent | ToolCallStartEvent | ToolCallArgsEvent | ToolCallEndEvent | ToolCallResultEvent | StateSnapshotEvent | StateDeltaEvent | ActivitySnapshotEvent | ActivityDeltaEvent | MessagesSnapshotEvent;
+//# sourceMappingURL=agui.d.ts.map

package/lib/dist/types/agui.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agui.d.ts","sourceRoot":"","sources":["../../../types/agui.ts"],"names":[],"mappings":"AAKA;;;;;;;GAOG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,KAAK,EACV,eAAe,EACf,gBAAgB,EAChB,aAAa,EACb,qBAAqB,EACrB,uBAAuB,EACvB,mBAAmB,EACnB,kBAAkB,EAClB,gBAAgB,EAChB,6BAA6B,EAC7B,+BAA+B,EAC/B,2BAA2B,EAC3B,kBAAkB,EAClB,iBAAiB,EACjB,gBAAgB,EAChB,mBAAmB,EACnB,kBAAkB,EAClB,eAAe,EACf,qBAAqB,EACrB,kBAAkB,EAClB,qBAAqB,EACtB,MAAM,aAAa,CAAC;AAGrB,eAAO,MAAM,aAAa,kBAAY,CAAC;AAGvC,MAAM,MAAM,SAAS,GACjB,eAAe,GACf,gBAAgB,GAChB,aAAa,GACb,qBAAqB,GACrB,uBAAuB,GACvB,mBAAmB,GACnB,kBAAkB,GAClB,gBAAgB,GAChB,6BAA6B,GAC7B,+BAA+B,GAC/B,2BAA2B,GAC3B,kBAAkB,GAClB,iBAAiB,GACjB,gBAAgB,GAChB,mBAAmB,GACnB,kBAAkB,GAClB,eAAe,GACf,qBAAqB,GACrB,kBAAkB,GAClB,qBAAqB,CAAC"}

package/lib/dist/types/agui.js

@@ -0,0 +1,16 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * AG UI Protocol Event Types
+ *
+ * Most types are imported directly from @ag-ui/core.
+ * This file provides:
+ * - AGUIEvent union type for type-safe event handling
+ * - AGUIEventType alias for the EventType enum
+ */
+import { EventType } from '@ag-ui/core';
+// Alias for EventType enum - used throughout the codebase
+export const AGUIEventType = EventType;
+//# sourceMappingURL=agui.js.map

package/lib/dist/types/agui.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agui.js","sourceRoot":"","sources":["../../../types/agui.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAwBxC,0DAA0D;AAC1D,MAAM,CAAC,MAAM,aAAa,GAAG,SAAS,CAAC"}