@superblocksteam/vite-plugin-file-sync 1.99.99-next.2 → 2.0.0-SNAPSHOT.84af65cd215b6ca1

package/dist/ai-service/llm/stream/wait-state-tracker.js

@@ -0,0 +1,109 @@
+function asToolCall(chunk) {
+    // Checking the type narrows it to a tool-call, so toolCallId/toolName are
+    // already typed strings here — no cast needed.
+    if (chunk.type !== "tool-call") {
+        return undefined;
+    }
+    return { toolCallId: chunk.toolCallId, toolName: chunk.toolName };
+}
+function asCompletedToolCallId(chunk) {
+    // tool-error is the AI SDK's completion for a thrown execute(); treating
+    // only tool-result as done leaves the call pending and pauses stall
+    // detection as if the tool were still running.
+    if (chunk.type !== "tool-result" && chunk.type !== "tool-error") {
+        return undefined;
+    }
+    return chunk.toolCallId;
+}
+/**
+ * Watches a stream's events so that, when the stall timer fires, we can say
+ * what the agent was actually stuck on.
+ *
+ * The stall timer only knows that nothing arrived for too long; it can't tell
+ * the model going silent apart from a slow local tool or a gap between steps.
+ * This tracker listens to the same events the timer resets on (tool calls,
+ * results, and thrown-tool errors in the stream, plus step start/finish) and
+ * reports the current wait state plus how many keep-alive pings arrived during
+ * the silence.
+ *
+ * Logs no customer data: it only reads the event type, the tool name, and the
+ * tool call id.
+ */
+export class WaitStateTracker {
+    pendingTools = new Map();
+    betweenSteps = false;
+    rawResetCount = 0;
+    constructor(session) {
+        session.events.on("chunk", this.handleChunk);
+        session.events.on("step:start", this.handleStepStart);
+        session.events.on("step:finish", this.handleStepFinish);
+        session.events.on("retry", this.handleRetry);
+    }
+    getWaitState() {
+        if (this.pendingTools.size > 0) {
+            // Usually only one tool runs at a time, so the last one started is the
+            // one running now. A Map keeps insertion order, so the last entry is it.
+            let latest;
+            for (const tool of this.pendingTools.values()) {
+                latest = tool;
+            }
+            return {
+                state: "tool",
+                toolName: latest.toolName,
+                toolCallId: latest.toolCallId,
+                pendingToolCount: this.pendingTools.size,
+            };
+        }
+        if (this.betweenSteps) {
+            return { state: "step-boundary" };
+        }
+        return { state: "provider" };
+    }
+    /**
+     * How many keep-alive pings arrived during the current stretch of silence
+     * (since the last real progress: content, a tool event, a step start, or a
+     * retry). Reset on each of those, so a stall reports only the pings from the
+     * silence that caused it, not every ping in the whole run.
+     */
+    getRawResetCount() {
+        return this.rawResetCount;
+    }
+    handleChunk = (chunk) => {
+        if (chunk.type === "raw") {
+            this.rawResetCount += 1;
+            return;
+        }
+        // Anything that isn't a keep-alive is real progress, so a new stretch of
+        // silence starts and the ping count restarts. Without this, pings from an
+        // earlier silence would inflate the count at a later, unrelated stall.
+        this.rawResetCount = 0;
+        const toolCall = asToolCall(chunk);
+        if (toolCall) {
+            this.pendingTools.set(toolCall.toolCallId, toolCall);
+            return;
+        }
+        const completedToolCallId = asCompletedToolCallId(chunk);
+        if (completedToolCallId) {
+            this.pendingTools.delete(completedToolCallId);
+            return;
+        }
+        // Real model output means it's producing again, so we're no longer sitting
+        // in the gap between steps.
+        this.betweenSteps = false;
+    };
+    handleStepStart = () => {
+        this.rawResetCount = 0;
+        this.betweenSteps = false;
+    };
+    handleStepFinish = () => {
+        this.betweenSteps = true;
+    };
+    handleRetry = () => {
+        // A retry starts over from scratch: drop any tool the failed attempt had
+        // running, reset the ping count, and go back to waiting on the model.
+        this.rawResetCount = 0;
+        this.pendingTools.clear();
+        this.betweenSteps = false;
+    };
+}
+//# sourceMappingURL=wait-state-tracker.js.map

package/dist/ai-service/llm/stream/wait-state-tracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wait-state-tracker.js","sourceRoot":"","sources":["../../../../src/ai-service/llm/stream/wait-state-tracker.ts"],"names":[],"mappings":"AA6BA,SAAS,UAAU,CAAC,KAAkB;IACpC,0EAA0E;IAC1E,+CAA+C;IAC/C,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;QAC/B,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC;AACpE,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAkB;IAC/C,yEAAyE;IACzE,oEAAoE;IACpE,+CAA+C;IAC/C,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;QAChE,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,KAAK,CAAC,UAAU,CAAC;AAC1B,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,MAAM,OAAO,gBAAgB;IACV,YAAY,GAAG,IAAI,GAAG,EAAuB,CAAC;IACvD,YAAY,GAAG,KAAK,CAAC;IACrB,aAAa,GAAG,CAAC,CAAC;IAE1B,YAAY,OAAsB;QAChC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAC7C,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QACtD,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,aAAa,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACxD,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IAC/C,CAAC;IAED,YAAY;QACV,IAAI,IAAI,CAAC,YAAY,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC/B,uEAAuE;YACvE,yEAAyE;YACzE,IAAI,MAA+B,CAAC;YACpC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC9C,MAAM,GAAG,IAAI,CAAC;YAChB,CAAC;YACD,OAAO;gBACL,KAAK,EAAE,MAAM;gBACb,QAAQ,EAAE,MAAO,CAAC,QAAQ;gBAC1B,UAAU,EAAE,MAAO,CAAC,UAAU;gBAC9B,gBAAgB,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;aACzC,CAAC;QACJ,CAAC;QACD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,OAAO,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC;QACpC,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;IAC/B,CAAC;IAED;;;;;OAKG;IACH,gBAAgB;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAEO,WAAW,GAAG,CAAC,KAAkB,EAAQ,EAAE;QACjD,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,aAAa,IAAI,CAAC,CAAC;YACxB,OAAO;QACT,CAAC;QAED,yEAAyE;QACzE,0EAA0E;QAC1E,uEAAuE;QACvE,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC;QAEvB,MAAM,QAAQ,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;QACnC,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YACrD,OAAO;QACT,CAAC;QAED,MAAM,mBAAmB,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;QACzD,IAAI,mBAAmB,EAAE,CAAC;YACxB,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;YAC9C,OAAO;QACT,CAAC;QAED,2EAA2E;QAC3E,4BAA4B;QAC5B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;IAC5B,CAAC,CAAC;IAEM,eAAe,GAAG,GAAS,EAAE;QACnC,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC;QACvB,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;IAC5B,CAAC,CAAC;IAEM,gBAAgB,GAAG,GAAS,EAAE;QACpC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;IAC3B,CAAC,CAAC;IAEM,WAAW,GAAG,GAAS,EAAE;QAC/B,yEAAyE;QACzE,sEAAsE;QACtE,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC;QACvB,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC1B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;IAC5B,CAAC,CAAC;CACH"}

package/dist/ai-service/llm/tool-context-integrity.d.ts

@@ -0,0 +1,25 @@
+import type { ModelMessage } from "ai";
+type ToolContextIntegritySource = "ctx.initial" | "ctx.prepareStep" | "provider.streamText";
+type ToolReference = {
+    readonly messageIndex: number;
+    readonly toolCallId: string;
+    readonly toolName?: string;
+};
+type ToolContextIntegritySummary = {
+    readonly source: ToolContextIntegritySource;
+    readonly messageCount: number;
+    readonly orphanToolCallCount: number;
+    readonly orphanToolResultCount: number;
+    readonly orphanToolCalls: ReadonlyArray<ToolReference>;
+    readonly orphanToolResults: ReadonlyArray<ToolReference>;
+};
+type AnalyzeOptions = {
+    maxDetails?: number;
+};
+type WarnLogger = {
+    warn: (...messages: unknown[]) => void;
+};
+export declare function analyzeToolContextIntegrity(source: ToolContextIntegritySource, messages: ModelMessage[], options?: AnalyzeOptions): ToolContextIntegritySummary;
+export declare function logToolContextIntegrityIssues(logger: WarnLogger, source: ToolContextIntegritySource, messages: ModelMessage[]): void;
+export {};
+//# sourceMappingURL=tool-context-integrity.d.ts.map

package/dist/ai-service/llm/tool-context-integrity.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-context-integrity.d.ts","sourceRoot":"","sources":["../../../src/ai-service/llm/tool-context-integrity.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAEvC,KAAK,0BAA0B,GAC3B,aAAa,GACb,iBAAiB,GACjB,qBAAqB,CAAC;AAE1B,KAAK,aAAa,GAAG;IACnB,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,KAAK,2BAA2B,GAAG;IACjC,QAAQ,CAAC,MAAM,EAAE,0BAA0B,CAAC;IAC5C,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,mBAAmB,EAAE,MAAM,CAAC;IACrC,QAAQ,CAAC,qBAAqB,EAAE,MAAM,CAAC;IACvC,QAAQ,CAAC,eAAe,EAAE,aAAa,CAAC,aAAa,CAAC,CAAC;IACvD,QAAQ,CAAC,iBAAiB,EAAE,aAAa,CAAC,aAAa,CAAC,CAAC;CAC1D,CAAC;AAEF,KAAK,cAAc,GAAG;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF,KAAK,UAAU,GAAG;IAChB,IAAI,EAAE,CAAC,GAAG,QAAQ,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;CACxC,CAAC;AA2FF,wBAAgB,2BAA2B,CACzC,MAAM,EAAE,0BAA0B,EAClC,QAAQ,EAAE,YAAY,EAAE,EACxB,OAAO,GAAE,cAAmB,GAC3B,2BAA2B,CA8D7B;AAED,wBAAgB,6BAA6B,CAC3C,MAAM,EAAE,UAAU,EAClB,MAAM,EAAE,0BAA0B,EAClC,QAAQ,EAAE,YAAY,EAAE,GACvB,IAAI,CA2BN"}

package/dist/ai-service/llm/tool-context-integrity.js

@@ -0,0 +1,141 @@
+import { createHash } from "node:crypto";
+const DEFAULT_MAX_DETAILS = 10;
+const EMPTY_TOOL_IDS = new Set();
+const MAX_DIAGNOSTIC_STRING_LENGTH = 128;
+const MAX_LOGGED_TOOL_REFERENCES = 10;
+function getContentParts(msg) {
+    if (!Array.isArray(msg.content)) {
+        return [];
+    }
+    return msg.content.filter((part) => part && typeof part === "object");
+}
+function getToolName(part) {
+    return typeof part.toolName === "string" ? part.toolName : undefined;
+}
+function getToolCallId(part) {
+    return typeof part.toolCallId === "string" ? part.toolCallId : undefined;
+}
+function getMessageToolParts(msg) {
+    const parts = getContentParts(msg);
+    const toolCallIds = new Set();
+    const toolResultIds = new Set();
+    for (const part of parts) {
+        const toolCallId = getToolCallId(part);
+        if (!toolCallId) {
+            continue;
+        }
+        if (msg.role === "assistant" && part.type === "tool-call") {
+            toolCallIds.add(toolCallId);
+        }
+        if ((msg.role === "assistant" || msg.role === "tool") &&
+            part.type === "tool-result") {
+            toolResultIds.add(toolCallId);
+        }
+    }
+    return { parts, toolCallIds, toolResultIds };
+}
+function getMaxDetails(options) {
+    if (options.maxDetails === undefined) {
+        return DEFAULT_MAX_DETAILS;
+    }
+    if (!Number.isFinite(options.maxDetails)) {
+        return MAX_LOGGED_TOOL_REFERENCES;
+    }
+    return Math.min(Math.max(Math.floor(options.maxDetails), 0), MAX_LOGGED_TOOL_REFERENCES);
+}
+function toDiagnosticString(value) {
+    if (value.length <= MAX_DIAGNOSTIC_STRING_LENGTH) {
+        return value;
+    }
+    return `${value.slice(0, MAX_DIAGNOSTIC_STRING_LENGTH - 3)}...`;
+}
+function toDiagnosticToolCallId(value) {
+    return `sha256:${createHash("sha256").update(value).digest("hex").slice(0, 16)}`;
+}
+function toToolReference(messageIndex, toolCallId, part) {
+    const toolName = getToolName(part);
+    return {
+        messageIndex,
+        toolCallId: toDiagnosticToolCallId(toolCallId),
+        ...(toolName ? { toolName: toDiagnosticString(toolName) } : {}),
+    };
+}
+export function analyzeToolContextIntegrity(source, messages, options = {}) {
+    const maxDetails = getMaxDetails(options);
+    const orphanToolCalls = [];
+    const orphanToolResults = [];
+    let orphanToolCallCount = 0;
+    let orphanToolResultCount = 0;
+    const messageToolParts = messages.map(getMessageToolParts);
+    messages.forEach((msg, index) => {
+        const currentToolParts = messageToolParts[index];
+        if (msg.role === "assistant") {
+            const nextToolResultIds = messageToolParts[index + 1]?.toolResultIds ?? EMPTY_TOOL_IDS;
+            const validToolResultIds = new Set([
+                ...nextToolResultIds,
+                ...currentToolParts.toolResultIds,
+            ]);
+            for (const part of currentToolParts.parts) {
+                if (part.type !== "tool-call") {
+                    continue;
+                }
+                const toolCallId = getToolCallId(part);
+                if (!toolCallId || validToolResultIds.has(toolCallId)) {
+                    continue;
+                }
+                orphanToolCallCount++;
+                if (orphanToolCalls.length < maxDetails) {
+                    orphanToolCalls.push(toToolReference(index, toolCallId, part));
+                }
+            }
+        }
+        if (msg.role === "tool") {
+            const previousToolCallIds = messageToolParts[index - 1]?.toolCallIds ?? EMPTY_TOOL_IDS;
+            for (const part of currentToolParts.parts) {
+                if (part.type !== "tool-result") {
+                    continue;
+                }
+                const toolCallId = getToolCallId(part);
+                if (!toolCallId || previousToolCallIds.has(toolCallId)) {
+                    continue;
+                }
+                orphanToolResultCount++;
+                if (orphanToolResults.length < maxDetails) {
+                    orphanToolResults.push(toToolReference(index, toolCallId, part));
+                }
+            }
+        }
+    });
+    return {
+        source,
+        messageCount: messages.length,
+        orphanToolCallCount,
+        orphanToolResultCount,
+        orphanToolCalls,
+        orphanToolResults,
+    };
+}
+export function logToolContextIntegrityIssues(logger, source, messages) {
+    const messagesSnapshot = Array.isArray(messages) ? [...messages] : messages;
+    queueMicrotask(() => {
+        try {
+            const summary = analyzeToolContextIntegrity(source, messagesSnapshot);
+            if (summary.orphanToolCallCount === 0 &&
+                summary.orphanToolResultCount === 0) {
+                return;
+            }
+            logger.warn("[ai-service] tool context integrity issue", summary);
+        }
+        catch {
+            try {
+                logger.warn("[ai-service] tool context integrity analysis failed", {
+                    source,
+                });
+            }
+            catch {
+                // Diagnostic logging must never disrupt the LLM request path.
+            }
+        }
+    });
+}
+//# sourceMappingURL=tool-context-integrity.js.map

package/dist/ai-service/llm/tool-context-integrity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-context-integrity.js","sourceRoot":"","sources":["../../../src/ai-service/llm/tool-context-integrity.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAsCzC,MAAM,mBAAmB,GAAG,EAAE,CAAC;AAC/B,MAAM,cAAc,GAAwB,IAAI,GAAG,EAAE,CAAC;AACtD,MAAM,4BAA4B,GAAG,GAAG,CAAC;AACzC,MAAM,0BAA0B,GAAG,EAAE,CAAC;AAEtC,SAAS,eAAe,CAAC,GAAiB;IACxC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QAChC,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,GAAG,CAAC,OAAO,CAAC,MAAM,CACvB,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,CACT,CAAC;AACtC,CAAC;AAED,SAAS,WAAW,CAAC,IAA6B;IAChD,OAAO,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;AACvE,CAAC;AAED,SAAS,aAAa,CAAC,IAA6B;IAClD,OAAO,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;AAC3E,CAAC;AAED,SAAS,mBAAmB,CAAC,GAAiB;IAC5C,MAAM,KAAK,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC;IACnC,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IACtC,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;IAExC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,SAAS;QACX,CAAC;QACD,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC1D,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC9B,CAAC;QACD,IACE,CAAC,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,CAAC;YACjD,IAAI,CAAC,IAAI,KAAK,aAAa,EAC3B,CAAC;YACD,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC;AAC/C,CAAC;AAED,SAAS,aAAa,CAAC,OAAuB;IAC5C,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QACrC,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACzC,OAAO,0BAA0B,CAAC;IACpC,CAAC;IACD,OAAO,IAAI,CAAC,GAAG,CACb,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,EAC3C,0BAA0B,CAC3B,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAa;IACvC,IAAI,KAAK,CAAC,MAAM,IAAI,4BAA4B,EAAE,CAAC;QACjD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,4BAA4B,GAAG,CAAC,CAAC,KAAK,CAAC;AAClE,CAAC;AAED,SAAS,sBAAsB,CAAC,KAAa;IAC3C,OAAO,UAAU,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;AACnF,CAAC;AAED,SAAS,eAAe,CACtB,YAAoB,EACpB,UAAkB,EAClB,IAA6B;IAE7B,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IACnC,OAAO;QACL,YAAY;QACZ,UAAU,EAAE,sBAAsB,CAAC,UAAU,CAAC;QAC9C,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,kBAAkB,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAChE,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,MAAkC,EAClC,QAAwB,EACxB,UAA0B,EAAE;IAE5B,MAAM,UAAU,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IAC1C,MAAM,eAAe,GAAoB,EAAE,CAAC;IAC5C,MAAM,iBAAiB,GAAoB,EAAE,CAAC;IAC9C,IAAI,mBAAmB,GAAG,CAAC,CAAC;IAC5B,IAAI,qBAAqB,GAAG,CAAC,CAAC;IAC9B,MAAM,gBAAgB,GAAG,QAAQ,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;IAE3D,QAAQ,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;QAC9B,MAAM,gBAAgB,GAAG,gBAAgB,CAAC,KAAK,CAAE,CAAC;QAClD,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC7B,MAAM,iBAAiB,GACrB,gBAAgB,CAAC,KAAK,GAAG,CAAC,CAAC,EAAE,aAAa,IAAI,cAAc,CAAC;YAC/D,MAAM,kBAAkB,GAAG,IAAI,GAAG,CAAC;gBACjC,GAAG,iBAAiB;gBACpB,GAAG,gBAAgB,CAAC,aAAa;aAClC,CAAC,CAAC;YAEH,KAAK,MAAM,IAAI,IAAI,gBAAgB,CAAC,KAAK,EAAE,CAAC;gBAC1C,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC9B,SAAS;gBACX,CAAC;gBACD,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;gBACvC,IAAI,CAAC,UAAU,IAAI,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;oBACtD,SAAS;gBACX,CAAC;gBAED,mBAAmB,EAAE,CAAC;gBACtB,IAAI,eAAe,CAAC,MAAM,GAAG,UAAU,EAAE,CAAC;oBACxC,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACxB,MAAM,mBAAmB,GACvB,gBAAgB,CAAC,KAAK,GAAG,CAAC,CAAC,EAAE,WAAW,IAAI,cAAc,CAAC;YAC7D,KAAK,MAAM,IAAI,IAAI,gBAAgB,CAAC,KAAK,EAAE,CAAC;gBAC1C,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;oBAChC,SAAS;gBACX,CAAC;gBACD,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;gBACvC,IAAI,CAAC,UAAU,IAAI,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;oBACvD,SAAS;gBACX,CAAC;gBAED,qBAAqB,EAAE,CAAC;gBACxB,IAAI,iBAAiB,CAAC,MAAM,GAAG,UAAU,EAAE,CAAC;oBAC1C,iBAAiB,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC;gBACnE,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO;QACL,MAAM;QACN,YAAY,EAAE,QAAQ,CAAC,MAAM;QAC7B,mBAAmB;QACnB,qBAAqB;QACrB,eAAe;QACf,iBAAiB;KAClB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,6BAA6B,CAC3C,MAAkB,EAClB,MAAkC,EAClC,QAAwB;IAExB,MAAM,gBAAgB,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;IAE5E,cAAc,CAAC,GAAG,EAAE;QAClB,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,2BAA2B,CACzC,MAAM,EACN,gBAAkC,CACnC,CAAC;YACF,IACE,OAAO,CAAC,mBAAmB,KAAK,CAAC;gBACjC,OAAO,CAAC,qBAAqB,KAAK,CAAC,EACnC,CAAC;gBACD,OAAO;YACT,CAAC;YAED,MAAM,CAAC,IAAI,CAAC,2CAA2C,EAAE,OAAO,CAAC,CAAC;QACpE,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC;gBACH,MAAM,CAAC,IAAI,CAAC,qDAAqD,EAAE;oBACjE,MAAM;iBACP,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;gBACP,8DAA8D;YAChE,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/ai-service/llm/types.d.ts

@@ -0,0 +1,44 @@
+import type { LanguageModelV3 } from "@ai-sdk/provider";
+import type { ClarkModelFactory, ClarkProviderSettings } from "@superblocksteam/clark";
+import type { LLMProviderConfig } from "@superblocksteam/library-shared/types";
+import type { CoreModelType, ProviderType } from "./stream/config.js";
+export type LLMProviderSettings = {
+    clark: ClarkProviderSettings;
+    /**
+     * Optional override for how the clark provider builds its `LanguageModelV3`s.
+     * When set, `createClark` uses this factory instead of the default HTTP
+     * wire-format factory. Used by evals to back the clark provider with
+     * direct-SDK models (bedrock/vertex/snowflake) without routing through the
+     * Superblocks server.
+     */
+    clarkModelFactory?: ClarkModelFactory;
+    /**
+     * Whether to write the Superblocks JWT to the standard Authorization header.
+     * Disable for direct-provider SDK calls that manage their own Authorization.
+     */
+    useSuperblocksAuthorizationHeader?: boolean;
+};
+export type LLMTask = "classifyPrompt" | "summarizeMessages" | "explain" | "generateBroadEdit" | "fast";
+export type LLMTaskModels = {
+    [K in LLMTask]: LanguageModelV3;
+};
+export type LLMTaskModel = {
+    model: LanguageModelV3;
+    providerConfig: {
+        provider: ProviderType;
+        model: CoreModelType;
+    };
+};
+export type LLMProvider = {
+    name: LLMProviderConfig["provider"];
+    /**
+     * The org's configured core model (from `llmConfig`). Optional so existing
+     * test mocks needn't set it. Used to build the streamText `providerConfig`
+     * (retry-budget profile + the initial telemetry label) for internal one-shot
+     * calls; the real per-call model/provider still come from the response
+     * headers (x-model-id / x-model-provider).
+     */
+    coreModel?: LLMProviderConfig["coreModel"];
+    modelForTask: (task: LLMTask) => LLMTaskModel;
+};
+//# sourceMappingURL=types.d.ts.map

package/dist/ai-service/llm/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/ai-service/llm/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAExD,OAAO,KAAK,EACV,iBAAiB,EACjB,qBAAqB,EACtB,MAAM,wBAAwB,CAAC;AAChC,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAE/E,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAEtE,MAAM,MAAM,mBAAmB,GAAG;IAChC,KAAK,EAAE,qBAAqB,CAAC;IAC7B;;;;;;OAMG;IACH,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;IACtC;;;OAGG;IACH,iCAAiC,CAAC,EAAE,OAAO,CAAC;CAC7C,CAAC;AAEF,MAAM,MAAM,OAAO,GACf,gBAAgB,GAChB,mBAAmB,GACnB,SAAS,GACT,mBAAmB,GACnB,MAAM,CAAC;AAEX,MAAM,MAAM,aAAa,GAAG;KACzB,CAAC,IAAI,OAAO,GAAG,eAAe;CAChC,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG;IACzB,KAAK,EAAE,eAAe,CAAC;IACvB,cAAc,EAAE;QACd,QAAQ,EAAE,YAAY,CAAC;QACvB,KAAK,EAAE,aAAa,CAAC;KACtB,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG;IACxB,IAAI,EAAE,iBAAiB,CAAC,UAAU,CAAC,CAAC;IACpC;;;;;;OAMG;IACH,SAAS,CAAC,EAAE,iBAAiB,CAAC,WAAW,CAAC,CAAC;IAC3C,YAAY,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,YAAY,CAAC;CAC/C,CAAC"}

package/dist/ai-service/llm/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/ai-service/llm/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/ai-service/llm/types.ts"],"names":[],"mappings":""}

package/dist/ai-service/llm/workflow-metrics.d.ts

@@ -0,0 +1,14 @@
+/**
+ * Extracts workflow metrics from LLM stream result steps.
+ */
+export declare function extractWorkflowMetrics(steps: Array<{
+    toolCalls?: unknown[];
+    content?: unknown[];
+}>): {
+    toolCalls: number;
+    stepCount: number;
+    linesAdded: number;
+    linesEdited: number;
+    linesRemoved: number;
+};
+//# sourceMappingURL=workflow-metrics.d.ts.map

package/dist/ai-service/llm/workflow-metrics.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-metrics.d.ts","sourceRoot":"","sources":["../../../src/ai-service/llm/workflow-metrics.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,KAAK,EAAE,KAAK,CAAC;IAAE,SAAS,CAAC,EAAE,OAAO,EAAE,CAAC;IAAC,OAAO,CAAC,EAAE,OAAO,EAAE,CAAA;CAAE,CAAC;;;;;;EAiB7D"}

package/dist/ai-service/llm/workflow-metrics.js

@@ -0,0 +1,19 @@
+/**
+ * Extracts workflow metrics from LLM stream result steps.
+ */
+export function extractWorkflowMetrics(steps) {
+    let toolCalls = 0;
+    for (const step of steps) {
+        toolCalls += step.toolCalls?.length ?? 0;
+    }
+    return {
+        toolCalls,
+        stepCount: steps.length,
+        // TODO: Implement accurate line counting with proper diff algorithm
+        // Currently returning 0 for all line metrics
+        linesAdded: 0,
+        linesEdited: 0,
+        linesRemoved: 0,
+    };
+}
+//# sourceMappingURL=workflow-metrics.js.map

package/dist/ai-service/llm/workflow-metrics.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-metrics.js","sourceRoot":"","sources":["../../../src/ai-service/llm/workflow-metrics.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,UAAU,sBAAsB,CACpC,KAA4D;IAE5D,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,SAAS,IAAI,IAAI,CAAC,SAAS,EAAE,MAAM,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,OAAO;QACL,SAAS;QACT,SAAS,EAAE,KAAK,CAAC,MAAM;QACvB,oEAAoE;QACpE,6CAA6C;QAC7C,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,CAAC;QACd,YAAY,EAAE,CAAC;KAChB,CAAC;AACJ,CAAC"}

package/dist/ai-service/llmobs/constants.d.ts

@@ -0,0 +1,5 @@
+export declare const DD_INTERNAL_TAGS: {
+    MODEL_NAME: string;
+    MODEL_PROVIDER: string;
+};
+//# sourceMappingURL=constants.d.ts.map

package/dist/ai-service/llmobs/constants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../../src/ai-service/llmobs/constants.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,gBAAgB;;;CAG5B,CAAC"}

package/dist/ai-service/llmobs/constants.js

@@ -0,0 +1,5 @@
+export const DD_INTERNAL_TAGS = {
+    MODEL_NAME: "_ml_obs.meta.model_name",
+    MODEL_PROVIDER: "_ml_obs.meta.model_provider",
+};
+//# sourceMappingURL=constants.js.map

package/dist/ai-service/llmobs/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../src/ai-service/llmobs/constants.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B,UAAU,EAAE,yBAAyB;IACrC,cAAc,EAAE,6BAA6B;CAC9C,CAAC"}

package/dist/ai-service/llmobs/context-registry.d.ts

@@ -0,0 +1,62 @@
+import type { Context } from "@opentelemetry/api";
+/**
+ * Bridges the OTEL context active at LLMObs span-creation time to the
+ * async flush callback where OTEL spans are actually emitted.
+ *
+ * Two maps:
+ * - `captured`: OTEL context snapshotted when each LLMObs ROOT span starts.
+ *   Only root spans are stored here; child spans get their parent context
+ *   from `crossBatch` or `perFlushCtx` in processBatch. Consumed once.
+ * - `crossBatch`: OTEL context of already-emitted gen_ai OTEL spans,
+ *   so children that flush in a later batch can still find their parent.
+ *   TTL-evicted after 10 minutes.
+ *
+ * Memory safety:
+ * - Both maps are capped at MAX_CAPTURED / MAX_CROSS_BATCH entries;
+ *   oldest entry is evicted when the cap is hit.
+ * - A background reaper (startReaper) evicts stale entries every 30 s,
+ *   independently of flush cycles so cleanup runs even if flush stalls.
+ */
+export declare class OtelContextRegistry {
+    private captured;
+    private crossBatch;
+    private _reaper;
+    /**
+     * Start the background reaper that evicts stale entries independently
+     * of flush cycles. Safe to call multiple times — subsequent calls are no-ops.
+     * The timer is unref'd so it does not prevent Node.js from exiting.
+     */
+    startReaper(intervalMs?: number): void;
+    /** Stop the background reaper. Call in tests to prevent timer leaks. */
+    stopReaper(): void;
+    /**
+     * Store the active OTEL context at the moment a LLMObs ROOT span starts.
+     * Must only be called for root spans (no LLMObs parent); child spans are
+     * handled via crossBatch / perFlushCtx and do not need an entry here.
+     */
+    captureAtSpanStart(spanId: string, ctx: Context): void;
+    /**
+     * Retrieve and remove the captured context for a span.
+     * Used for root spans: the context holds the ambient OTEL context at
+     * span-start time, which becomes the OTEL trace root.
+     */
+    consumeCaptured(spanId: string): Context | undefined;
+    /**
+     * Store the OTEL context of an already-emitted gen_ai span so children
+     * flushed in a later batch can find it via getCreated().
+     */
+    setCreated(spanId: string, ctx: Context): void;
+    /** Look up the OTEL context of a previously emitted gen_ai span. */
+    getCreated(spanId: string): Context | undefined;
+    /**
+     * Evict entries older than TTL_MS from both maps.
+     * Called by the background reaper and at the start of each flush batch.
+     */
+    cleanup(): void;
+    /** Exposed for testing only. */
+    get _capturedSize(): number;
+    /** Exposed for testing only. */
+    get _crossBatchSize(): number;
+}
+export declare const otelContextRegistry: OtelContextRegistry;
+//# sourceMappingURL=context-registry.d.ts.map

package/dist/ai-service/llmobs/context-registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-registry.d.ts","sourceRoot":"","sources":["../../../src/ai-service/llmobs/context-registry.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAYlD;;;;;;;;;;;;;;;;;GAiBG;AACH,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,QAAQ,CAA4B;IAC5C,OAAO,CAAC,UAAU,CAA4B;IAC9C,OAAO,CAAC,OAAO,CAA6C;IAE5D;;;;OAIG;IACH,WAAW,CAAC,UAAU,SAAsB,GAAG,IAAI;IAKnD,wEAAwE;IACxE,UAAU,IAAI,IAAI;IAOlB;;;;OAIG;IACH,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,GAAG,IAAI;IAWtD;;;;OAIG;IACH,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,GAAG,SAAS;IAOpD;;;OAGG;IACH,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,GAAG,IAAI;IAY9C,oEAAoE;IACpE,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,GAAG,SAAS;IAI/C;;;OAGG;IACH,OAAO,IAAI,IAAI;IAUf,gCAAgC;IAChC,IAAI,aAAa,IAAI,MAAM,CAE1B;IAED,gCAAgC;IAChC,IAAI,eAAe,IAAI,MAAM,CAE5B;CACF;AAED,eAAO,MAAM,mBAAmB,qBAA4B,CAAC"}

package/dist/ai-service/llmobs/context-registry.js

@@ -0,0 +1,115 @@
+const TTL_MS = 10 * 60 * 1000; // 10 minutes
+const CLEANUP_INTERVAL_MS = 30 * 1000; // 30 seconds — independent of flush cadence
+const MAX_CAPTURED = 500; // root spans only; safety net against runaway sessions
+const MAX_CROSS_BATCH = 2000; // one entry per emitted gen_ai span
+/**
+ * Bridges the OTEL context active at LLMObs span-creation time to the
+ * async flush callback where OTEL spans are actually emitted.
+ *
+ * Two maps:
+ * - `captured`: OTEL context snapshotted when each LLMObs ROOT span starts.
+ *   Only root spans are stored here; child spans get their parent context
+ *   from `crossBatch` or `perFlushCtx` in processBatch. Consumed once.
+ * - `crossBatch`: OTEL context of already-emitted gen_ai OTEL spans,
+ *   so children that flush in a later batch can still find their parent.
+ *   TTL-evicted after 10 minutes.
+ *
+ * Memory safety:
+ * - Both maps are capped at MAX_CAPTURED / MAX_CROSS_BATCH entries;
+ *   oldest entry is evicted when the cap is hit.
+ * - A background reaper (startReaper) evicts stale entries every 30 s,
+ *   independently of flush cycles so cleanup runs even if flush stalls.
+ */
+export class OtelContextRegistry {
+    captured = new Map();
+    crossBatch = new Map();
+    _reaper;
+    /**
+     * Start the background reaper that evicts stale entries independently
+     * of flush cycles. Safe to call multiple times — subsequent calls are no-ops.
+     * The timer is unref'd so it does not prevent Node.js from exiting.
+     */
+    startReaper(intervalMs = CLEANUP_INTERVAL_MS) {
+        if (this._reaper !== undefined)
+            return;
+        this._reaper = setInterval(() => this.cleanup(), intervalMs).unref();
+    }
+    /** Stop the background reaper. Call in tests to prevent timer leaks. */
+    stopReaper() {
+        if (this._reaper !== undefined) {
+            clearInterval(this._reaper);
+            this._reaper = undefined;
+        }
+    }
+    /**
+     * Store the active OTEL context at the moment a LLMObs ROOT span starts.
+     * Must only be called for root spans (no LLMObs parent); child spans are
+     * handled via crossBatch / perFlushCtx and do not need an entry here.
+     */
+    captureAtSpanStart(spanId, ctx) {
+        if (this.captured.size >= MAX_CAPTURED && !this.captured.has(spanId)) {
+            // Evict the oldest entry (Maps preserve insertion order in V8).
+            // Skip eviction when overwriting an existing key — Map.set() on an
+            // existing key does not increase size, so no room needs to be made.
+            const oldest = this.captured.keys().next().value;
+            if (oldest !== undefined)
+                this.captured.delete(oldest);
+        }
+        this.captured.set(spanId, { ctx, createdAt: Date.now() });
+    }
+    /**
+     * Retrieve and remove the captured context for a span.
+     * Used for root spans: the context holds the ambient OTEL context at
+     * span-start time, which becomes the OTEL trace root.
+     */
+    consumeCaptured(spanId) {
+        const entry = this.captured.get(spanId);
+        if (!entry)
+            return undefined;
+        this.captured.delete(spanId);
+        return entry.ctx;
+    }
+    /**
+     * Store the OTEL context of an already-emitted gen_ai span so children
+     * flushed in a later batch can find it via getCreated().
+     */
+    setCreated(spanId, ctx) {
+        if (this.crossBatch.size >= MAX_CROSS_BATCH &&
+            !this.crossBatch.has(spanId)) {
+            // Same rationale as captureAtSpanStart: skip eviction on overwrite.
+            const oldest = this.crossBatch.keys().next().value;
+            if (oldest !== undefined)
+                this.crossBatch.delete(oldest);
+        }
+        this.crossBatch.set(spanId, { ctx, createdAt: Date.now() });
+    }
+    /** Look up the OTEL context of a previously emitted gen_ai span. */
+    getCreated(spanId) {
+        return this.crossBatch.get(spanId)?.ctx;
+    }
+    /**
+     * Evict entries older than TTL_MS from both maps.
+     * Called by the background reaper and at the start of each flush batch.
+     */
+    cleanup() {
+        const cutoff = Date.now() - TTL_MS;
+        for (const [id, entry] of this.captured) {
+            if (entry.createdAt < cutoff)
+                this.captured.delete(id);
+        }
+        for (const [id, entry] of this.crossBatch) {
+            if (entry.createdAt < cutoff)
+                this.crossBatch.delete(id);
+        }
+    }
+    /** Exposed for testing only. */
+    get _capturedSize() {
+        return this.captured.size;
+    }
+    /** Exposed for testing only. */
+    get _crossBatchSize() {
+        return this.crossBatch.size;
+    }
+}
+export const otelContextRegistry = new OtelContextRegistry();
+//# sourceMappingURL=context-registry.js.map

package/dist/ai-service/llmobs/context-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-registry.js","sourceRoot":"","sources":["../../../src/ai-service/llmobs/context-registry.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,aAAa;AAC5C,MAAM,mBAAmB,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,4CAA4C;AACnF,MAAM,YAAY,GAAG,GAAG,CAAC,CAAC,uDAAuD;AACjF,MAAM,eAAe,GAAG,IAAI,CAAC,CAAC,oCAAoC;AAOlE;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,OAAO,mBAAmB;IACtB,QAAQ,GAAG,IAAI,GAAG,EAAiB,CAAC;IACpC,UAAU,GAAG,IAAI,GAAG,EAAiB,CAAC;IACtC,OAAO,CAA6C;IAE5D;;;;OAIG;IACH,WAAW,CAAC,UAAU,GAAG,mBAAmB;QAC1C,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS;YAAE,OAAO;QACvC,IAAI,CAAC,OAAO,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,UAAU,CAAC,CAAC,KAAK,EAAE,CAAC;IACvE,CAAC;IAED,wEAAwE;IACxE,UAAU;QACR,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAC/B,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC5B,IAAI,CAAC,OAAO,GAAG,SAAS,CAAC;QAC3B,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,kBAAkB,CAAC,MAAc,EAAE,GAAY;QAC7C,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,YAAY,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YACrE,gEAAgE;YAChE,mEAAmE;YACnE,oEAAoE;YACpE,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;YACjD,IAAI,MAAM,KAAK,SAAS;gBAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACzD,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED;;;;OAIG;IACH,eAAe,CAAC,MAAc;QAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACxC,IAAI,CAAC,KAAK;YAAE,OAAO,SAAS,CAAC;QAC7B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC7B,OAAO,KAAK,CAAC,GAAG,CAAC;IACnB,CAAC;IAED;;;OAGG;IACH,UAAU,CAAC,MAAc,EAAE,GAAY;QACrC,IACE,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,eAAe;YACvC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAC5B,CAAC;YACD,oEAAoE;YACpE,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;YACnD,IAAI,MAAM,KAAK,SAAS;gBAAE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED,oEAAoE;IACpE,UAAU,CAAC,MAAc;QACvB,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,GAAG,CAAC;IAC1C,CAAC;IAED;;;OAGG;IACH,OAAO;QACL,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC;QACnC,KAAK,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACxC,IAAI,KAAK,CAAC,SAAS,GAAG,MAAM;gBAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACzD,CAAC;QACD,KAAK,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAC1C,IAAI,KAAK,CAAC,SAAS,GAAG,MAAM;gBAAE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;IAED,gCAAgC;IAChC,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;IAC5B,CAAC;IAED,gCAAgC;IAChC,IAAI,eAAe;QACjB,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;IAC9B,CAAC;CACF;AAED,MAAM,CAAC,MAAM,mBAAmB,GAAG,IAAI,mBAAmB,EAAE,CAAC"}

package/dist/ai-service/llmobs/helpers.d.ts

@@ -0,0 +1,5 @@
+import { LLMObsTracer } from "./tracer.js";
+import type { LLMObsSpan } from "./types.js";
+export declare function traceLLM<T>(name: string, modelName: string, modelProvider: string, operation: (span: LLMObsSpan) => Promise<T>, tags?: Record<string, string | boolean>, llmobs?: LLMObsTracer): Promise<T>;
+export declare function getJwtTraceTags(jwt: string | undefined): Record<string, string>;
+//# sourceMappingURL=helpers.d.ts.map

package/dist/ai-service/llmobs/helpers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"helpers.d.ts","sourceRoot":"","sources":["../../../src/ai-service/llmobs/helpers.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAG7C,wBAAgB,QAAQ,CAAC,CAAC,EACxB,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,MAAM,EACjB,aAAa,EAAE,MAAM,EACrB,SAAS,EAAE,CAAC,IAAI,EAAE,UAAU,KAAK,OAAO,CAAC,CAAC,CAAC,EAC3C,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,EACvC,MAAM,GAAE,YAAoC,GAC3C,OAAO,CAAC,CAAC,CAAC,CAeZ;AAGD,wBAAgB,eAAe,CAC7B,GAAG,EAAE,MAAM,GAAG,SAAS,GACtB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAUxB"}

package/dist/ai-service/llmobs/helpers.js

@@ -0,0 +1,31 @@
+import { parseJwt } from "../util/parse-jwt.js";
+import { LLMObsTracer } from "./tracer.js";
+// Helper function for LLM tracing with common patterns
+export function traceLLM(name, modelName, modelProvider, operation, tags, llmobs = LLMObsTracer.instance) {
+    return llmobs.trace({
+        kind: "llm",
+        name,
+        modelName,
+        modelProvider,
+    }, async (span) => {
+        if (tags) {
+            llmobs.annotate({ tags });
+        }
+        return operation(span);
+    });
+}
+// Helper to extract JWT trace tags
+export function getJwtTraceTags(jwt) {
+    const result = {};
+    const parsed = parseJwt(jwt);
+    if (parsed.orgId)
+        result["org.id"] = parsed.orgId;
+    if (parsed.orgName)
+        result["org.name"] = parsed.orgName;
+    if (parsed.userId)
+        result["user.id"] = parsed.userId;
+    if (parsed.userEmail)
+        result["user.email"] = parsed.userEmail;
+    return result;
+}
+//# sourceMappingURL=helpers.js.map