@superblocksteam/vite-plugin-file-sync 2.0.41-next.5 → 2.0.41-next.51

package/dist/ai-service/llmobs.js

@@ -1,216 +0,0 @@
-import { AsyncLocalStorage } from "async_hooks";
-import ddTrace from "dd-trace";
-class IndexedStack {
-    items = [];
-    push(item) {
-        this.items.push(item);
-    }
-    top() {
-        return this.items[this.items.length - 1];
-    }
-    remove(item) {
-        const index = this.items.indexOf(item);
-        if (index !== -1) {
-            this.items.splice(index, 1);
-            return true;
-        }
-        return false;
-    }
-    length() {
-        return this.items.length;
-    }
-}
-class State extends IndexedStack {
-}
-/**
- * Flexible wrapper around DataDog's LLMObs tracing API.
- *
- * Provides four ways to create spans:
- * - `trace()`: Execute function within auto-closing span
- * - `wrap()`: Wrap function with automatic tracing
- * - `startActiveSpan()`: Create manually-managed span
- * - `decorate()`: Create decorator function for automatic tracing
- */
-export class LLMObsTracer {
-    globalState = new State();
-    localStorage = new AsyncLocalStorage();
-    globalTags = {};
-    static _instance = new LLMObsTracer();
-    constructor() { }
-    static get instance() {
-        return LLMObsTracer._instance;
-    }
-    /**
-     * Execute a function within a traced span context
-     * Automatically closes the span when the function completes
-     * Supports both synchronous and asynchronous functions
-     */
-    trace(options, fn) {
-        return this._trace({ finish: "auto", ...options }, fn);
-    }
-    /**
-     * Wrap a function with automatic tracing
-     * The function name is used as the span name if not explicitly provided,
-     * unless the function is anonymous, in which case options.kind is used
-     */
-    wrap(options, fn) {
-        const name = options.name || fn.name || options.kind;
-        return ((...args) => {
-            return this.trace({ ...options, name }, (_span) => fn(...args));
-        });
-    }
-    /**
-     * Start a new active span that must be manually closed
-     *
-     * Activation modes (for manual spans only):
-     * - activation: "local" (default): Span becomes inactive at end of current scope
-     * - activation: "global": Span remains active across scopes until finished
-     *
-     * @param options Span configuration including name, kind, and activation mode
-     * @returns A span that must be manually finished
-     */
-    startActiveSpan(options) {
-        const { activation, ...rest } = options;
-        return this._trace({ finish: "manual", activation, ...rest }, (span) => span);
-    }
-    /**
-     * Export a span for submission to DataDog
-     * Defaults to the currently active span if none provided
-     */
-    exportSpan(span) {
-        return this.llmobs.exportSpan(span || this.active());
-    }
-    annotate(spanOrAnnotations, annotations) {
-        if (annotations === undefined) {
-            // First overload: annotate(annotations)
-            return this.llmobs.annotate(this.active(), spanOrAnnotations);
-        }
-        else {
-            // Second overload: annotate(span, annotations)
-            return this.llmobs.annotate(spanOrAnnotations, annotations);
-        }
-    }
-    /**
-     * Submit feedback or evaluation metrics for a completed span
-     * Use for user feedback (accept/reject) or quality scores
-     */
-    submitEvaluation(spanContext, option) {
-        return this.llmobs.submitEvaluation(spanContext, option);
-    }
-    /**
-     * Create a decorator function for automatic tracing
-     * Use as a TypeScript decorator or function wrapper
-     */
-    decorate(options) {
-        return (target) => {
-            return this.wrap(options, target);
-        };
-    }
-    /**
-     * Get the currently active span
-     */
-    active() {
-        return this.localState?.top() || this.globalState.top();
-    }
-    /**
-     * Add global tags to be included on all spans
-     * Existing tags with the same key will be overwritten
-     */
-    setGlobalTags(tags) {
-        this.globalTags = { ...this.globalTags, ...tags };
-    }
-    /**
-     * Register a span and set up cleanup
-     *
-     * @param span The span to register
-     * @param activation Whether span becomes inactive at scope end or remains active
-     */
-    register(span, activation) {
-        this.annotate(span, { tags: this.globalTags });
-        const originalFinish = span.finish.bind(span);
-        // Override original finish to maintain stack integrity
-        span.finish = (finishTime) => {
-            // Always try to remove from both
-            this.localState?.remove(span);
-            this.globalState.remove(span);
-            return originalFinish(finishTime);
-        };
-        // Add span to appropriate state containers
-        const localState = this.localState;
-        if (localState) {
-            localState.push(span);
-        }
-        if (!localState || activation === "global") {
-            this.globalState.push(span);
-        }
-    }
-    /**
-     * Core tracing implementation
-     *
-     * Finish modes:
-     * - finish: "auto": Span finishes automatically at end of scope
-     * - finish: "manual": Span must be manually finished
-     *
-     * Activation modes (for manual spans only):
-     * - activation: "local": Span becomes inactive at end of scope
-     * - activation: "global": Span remains active across scopes
-     *
-     * @param options Span configuration including finish and activation modes
-     * @param fn Function to execute within span context
-     * @returns Result of executing fn within the span context
-     */
-    _trace(options, fn) {
-        const { tags, finish, activation = "local", ...rest } = options;
-        const setup = (span) => {
-            this.register(span, activation);
-            this.annotate(span, { tags });
-        };
-        return this.activate(this.active(), () => {
-            if (finish == "auto") {
-                return this.llmobs.trace(rest, (span) => {
-                    return this.localStorage.run(new State(), () => {
-                        setup(span);
-                        return fn(span);
-                    });
-                });
-            }
-            else {
-                return this.llmobs.trace(rest, (span, _done) => {
-                    setup(span);
-                    return fn(span);
-                });
-            }
-        });
-    }
-    /**
-     * Activate a span as the current parent context for DataDog tracing
-     *
-     * This is a thin wrapper around DataDog's internal _activate method,
-     * which establishes the parent-child relationship for spans created
-     * within the executed function.
-     *
-     * @param span The span to set as parent context (null/undefined = no parent)
-     * @param fn Function to execute with the span as active parent
-     * @returns Result of executing fn with the span context active
-     */
-    activate(span, fn) {
-        return this.llmobs._activate(span, null, fn);
-    }
-    get localState() {
-        return this.localStorage.getStore();
-    }
-    // Lazy initialization to avoid import order sensitivity
-    static _llmobs = null;
-    get llmobs() {
-        if (!LLMObsTracer._llmobs) {
-            LLMObsTracer._llmobs = ddTrace.llmobs;
-        }
-        return LLMObsTracer._llmobs;
-    }
-}
-/**
- * Default singleton tracer instance
- */
-const tracer = LLMObsTracer.instance;
-export default tracer;
-//# sourceMappingURL=llmobs.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"llmobs.js","sourceRoot":"","sources":["../../src/ai-service/llmobs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAChD,OAAO,OAAO,MAAM,UAAU,CAAC;AA4C/B,MAAM,YAAY;IACR,KAAK,GAAQ,EAAE,CAAC;IAExB,IAAI,CAAC,IAAO;QACV,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAED,GAAG;QACD,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,CAAC,IAAO;QACZ,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;YACjB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAC5B,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;IAC3B,CAAC;CACF;AAED,MAAM,KAAM,SAAQ,YAAwB;CAAG;AAE/C;;;;;;;;GAQG;AACH,MAAM,OAAO,YAAY;IACf,WAAW,GAAU,IAAI,KAAK,EAAE,CAAC;IACjC,YAAY,GAA6B,IAAI,iBAAiB,EAAE,CAAC;IACjE,UAAU,GAA2B,EAAE,CAAC;IACxC,MAAM,CAAC,SAAS,GAAiB,IAAI,YAAY,EAAE,CAAC;IAE5D,gBAAyB,CAAC;IAEnB,MAAM,KAAK,QAAQ;QACxB,OAAO,YAAY,CAAC,SAAS,CAAC;IAChC,CAAC;IAED;;;;OAIG;IACI,KAAK,CACV,OAA+B,EAC/B,EAA2B;QAE3B,OAAO,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;IACzD,CAAC;IAED;;;;OAIG;IACI,IAAI,CACT,OAAkC,EAClC,EAAK;QAEL,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;QACrD,OAAO,CAAC,CAAC,GAAG,IAAmB,EAAiB,EAAE;YAChD,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE,GAAG,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;QAClE,CAAC,CAAM,CAAC;IACV,CAAC;IAED;;;;;;;;;OASG;IACI,eAAe,CAAC,OAAgC;QACrD,MAAM,EAAE,UAAU,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;QACxC,OAAO,IAAI,CAAC,MAAM,CAChB,EAAE,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,GAAG,IAAI,EAAE,EACzC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CACf,CAAC;IACJ,CAAC;IAED;;;OAGG;IACI,UAAU,CAAC,IAAiB;QACjC,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IACvD,CAAC;IAQM,QAAQ,CACb,iBAAuD,EACvD,WAAqC;QAErC,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YAC9B,wCAAwC;YACxC,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CACzB,IAAI,CAAC,MAAM,EAAE,EACb,iBAA4C,CAC7C,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,+CAA+C;YAC/C,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAA+B,EAAE,WAAW,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;IAED;;;OAGG;IACI,gBAAgB,CACrB,WAA+B,EAC/B,MAA+B;QAE/B,OAAO,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;IAC3D,CAAC;IAED;;;OAGG;IACI,QAAQ,CACb,OAAkC;QAElC,OAAO,CAAC,MAAS,EAAE,EAAE;YACnB,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACpC,CAAC,CAAC;IACJ,CAAC;IAED;;OAEG;IACI,MAAM;QACX,OAAO,IAAI,CAAC,UAAU,EAAE,GAAG,EAAE,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC;IAC1D,CAAC;IAED;;;OAGG;IACI,aAAa,CAAC,IAA4B;QAC/C,IAAI,CAAC,UAAU,GAAG,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,GAAG,IAAI,EAAE,CAAC;IACpD,CAAC;IAED;;;;;OAKG;IACK,QAAQ,CAAC,IAAgB,EAAE,UAAiC;QAClE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QAC/C,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE9C,uDAAuD;QACvD,IAAI,CAAC,MAAM,GAAG,CAAC,UAAmB,EAAE,EAAE;YACpC,iCAAiC;YACjC,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YAC9B,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAE9B,OAAO,cAAc,CAAC,UAAU,CAAC,CAAC;QACpC,CAAC,CAAC;QAEF,2CAA2C;QAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QACnC,IAAI,UAAU,EAAE,CAAC;YACf,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxB,CAAC;QAED,IAAI,CAAC,UAAU,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;YAC3C,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACK,MAAM,CACZ,OAA4B,EAC5B,EAA2B;QAE3B,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,GAAG,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;QAEhE,MAAM,KAAK,GAAG,CAAC,IAAgB,EAAE,EAAE;YACjC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;YAChC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QAChC,CAAC,CAAC;QAEF,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE;YACvC,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;gBACrB,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE;oBACtC,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,KAAK,EAAE,EAAE,GAAG,EAAE;wBAC7C,KAAK,CAAC,IAAI,CAAC,CAAC;wBACZ,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC;oBAClB,CAAC,CAAC,CAAC;gBACL,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;oBAC7C,KAAK,CAAC,IAAI,CAAC,CAAC;oBACZ,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC;gBAClB,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAM,CAAC;IACV,CAAC;IAED;;;;;;;;;;OAUG;IACK,QAAQ,CAAI,IAA4B,EAAE,EAAW;QAC3D,OAAQ,IAAI,CAAC,MAAc,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;IACxD,CAAC;IAED,IAAY,UAAU;QACpB,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC;IACtC,CAAC;IACD,wDAAwD;IAChD,MAAM,CAAC,OAAO,GAAiC,IAAI,CAAC;IAC5D,IAAY,MAAM;QAChB,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;YAC1B,YAAY,CAAC,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;QACxC,CAAC;QACD,OAAO,YAAY,CAAC,OAAO,CAAC;IAC9B,CAAC;;AAOH;;GAEG;AACH,MAAM,MAAM,GAAG,YAAY,CAAC,QAAQ,CAAC;AACrC,eAAe,MAAM,CAAC"}

package/dist/ai-service/util/ddog-llmobs.d.ts

@@ -1,20 +0,0 @@
-import { streamText, type ToolSet, type StepResult } from "ai";
-import tracer, { type Span } from "dd-trace";
-type LLMObsNamedSpanOptions = tracer.llmobs.LLMObsNamedSpanOptions;
-type LLMObsSpanKind = tracer.llmobs.spanKind;
-export declare function traceWorkflow<T>(name: string, operation: (span: any) => Promise<T>, tags?: Record<string, string | boolean>): Promise<T>;
-export declare function traceTask<T>(name: string, operation: (span: any) => Promise<T>, tags?: Record<string, string | boolean>): Promise<T>;
-export declare function traceLLM<T>(name: string, modelName: string, modelProvider: string, operation: (span: Span) => Promise<T>, tags?: Record<string, string | boolean>): Promise<T>;
-export declare const traceToolSet: (toolset: ToolSet) => ToolSet;
-export declare function getJwtTraceTags(jwt: string | undefined): Record<string, string>;
-export declare function getCommonTraceTags(jwt: string | undefined, applicationId: string, organizationId: string, additionalTags?: Record<string, string | boolean>): Record<string, string | boolean>;
-/**
- * Builds LLMObsNamedSpanOptions from StepResult for tracing
- */
-export declare function buildSpanOptions<T extends ToolSet>(kind: LLMObsSpanKind, step: StepResult<T>, modelName?: string, modelProvider?: string): LLMObsNamedSpanOptions;
-/**
- * Wrapper around streamText that automatically handles tracing for onStepFinish, onFinish, and prepareStep
- */
-export declare function tracedStreamText<T extends ToolSet>(options: Parameters<typeof streamText<T>>[0]): import("ai").StreamTextResult<T, never>;
-export {};
-//# sourceMappingURL=ddog-llmobs.d.ts.map

package/dist/ai-service/util/ddog-llmobs.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"ddog-llmobs.d.ts","sourceRoot":"","sources":["../../../src/ai-service/util/ddog-llmobs.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,UAAU,EACV,KAAK,OAAO,EACZ,KAAK,UAAU,EAGhB,MAAM,IAAI,CAAC;AACZ,OAAO,MAAM,EAAE,EAAE,KAAK,IAAI,EAAE,MAAM,UAAU,CAAC;AAG7C,KAAK,sBAAsB,GAAG,MAAM,CAAC,MAAM,CAAC,sBAAsB,CAAC;AACnE,KAAK,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC;AAG7C,wBAAgB,aAAa,CAAC,CAAC,EAC7B,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,EACpC,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,GACtC,OAAO,CAAC,CAAC,CAAC,CAeZ;AAGD,wBAAgB,SAAS,CAAC,CAAC,EACzB,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,EACpC,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,GACtC,OAAO,CAAC,CAAC,CAAC,CAeZ;AAGD,wBAAgB,QAAQ,CAAC,CAAC,EACxB,IAAI,EAAE,MAAM,EACZ,SAAS,EAAE,MAAM,EACjB,aAAa,EAAE,MAAM,EACrB,SAAS,EAAE,CAAC,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,EACrC,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,GACtC,OAAO,CAAC,CAAC,CAAC,CAiBZ;AAGD,eAAO,MAAM,YAAY,GAAI,SAAS,OAAO,KAAG,OAsC/C,CAAC;AAGF,wBAAgB,eAAe,CAC7B,GAAG,EAAE,MAAM,GAAG,SAAS,GACtB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAoBxB;AAGD,wBAAgB,kBAAkB,CAChC,GAAG,EAAE,MAAM,GAAG,SAAS,EACvB,aAAa,EAAE,MAAM,EACrB,cAAc,EAAE,MAAM,EACtB,cAAc,GAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAM,GACpD,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,CAclC;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,CAAC,SAAS,OAAO,EAChD,IAAI,EAAE,cAAc,EACpB,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC,EACnB,SAAS,CAAC,EAAE,MAAM,EAClB,aAAa,CAAC,EAAE,MAAM,GACrB,sBAAsB,CA6BxB;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,CAAC,SAAS,OAAO,EAChD,OAAO,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,2CAiE7C"}

package/dist/ai-service/util/ddog-llmobs.js

@@ -1,197 +0,0 @@
-import { streamText, } from "ai";
-import tracer, {} from "dd-trace";
-// Helper function for workflow tracing with common patterns
-export function traceWorkflow(name, operation, tags) {
-    const { llmobs } = tracer;
-    return llmobs.trace({
-        kind: "workflow",
-        name,
-    }, async (span) => {
-        if (tags) {
-            llmobs.annotate({ tags });
-        }
-        return operation(span);
-    });
-}
-// Helper function for task tracing with common patterns
-export function traceTask(name, operation, tags) {
-    const { llmobs } = tracer;
-    return llmobs.trace({
-        kind: "task",
-        name,
-    }, async (span) => {
-        if (tags) {
-            llmobs.annotate({ tags });
-        }
-        return operation(span);
-    });
-}
-// Helper function for LLM tracing with common patterns
-export function traceLLM(name, modelName, modelProvider, operation, tags) {
-    const { llmobs } = tracer;
-    return llmobs.trace({
-        kind: "llm",
-        name,
-        modelName,
-        modelProvider,
-    }, async (span) => {
-        if (tags) {
-            llmobs.annotate({ tags });
-        }
-        return operation(span);
-    });
-}
-// Helper function for tool tracing with common patterns
-export const traceToolSet = (toolset) => {
-    const tracedToolset = {};
-    const { llmobs } = tracer;
-    for (const [name, tool] of Object.entries(toolset)) {
-        tracedToolset[name] = {
-            ...tool,
-            execute: async (input, options) => {
-                try {
-                    return await llmobs.trace({
-                        kind: "tool",
-                        name,
-                    }, async () => {
-                        if (!tool.execute) {
-                            throw new Error(`Tool ${name} has no execute method`);
-                        }
-                        const result = await tool.execute(input, options);
-                        llmobs.annotate({
-                            inputData: input,
-                            outputData: result,
-                            metadata: {
-                                toolCallId: options?.toolCallId,
-                            },
-                        });
-                        return result;
-                    });
-                }
-                catch (error) {
-                    console.error(`Error executing tool ${name}:`, error);
-                    throw error;
-                }
-            },
-        };
-    }
-    return tracedToolset;
-};
-// Helper to extract JWT trace tags
-export function getJwtTraceTags(jwt) {
-    const result = {};
-    if (jwt) {
-        try {
-            const parts = jwt.split(".");
-            if (parts.length === 3 && parts[1]) {
-                const payload = JSON.parse(Buffer.from(parts[1], "base64").toString());
-                if (payload.org_id)
-                    result["org.id"] = payload.org_id;
-                if (payload.org_name)
-                    result["org.name"] = payload.org_name;
-                if (payload.user_id)
-                    result["user.id"] = payload.user_id;
-                if (payload.user_email)
-                    result["user.email"] = payload.user_email;
-            }
-        }
-        catch {
-            // Silently handle JWT parsing errors
-        }
-    }
-    return result;
-}
-// Helper to combine common tags with enhanced user information
-export function getCommonTraceTags(jwt, applicationId, organizationId, additionalTags = {}) {
-    const result = {};
-    // Extract and add JWT tags using the extracted function
-    Object.assign(result, getJwtTraceTags(jwt));
-    // Add required tags (no if checks needed since they're required)
-    result["application.id"] = applicationId;
-    result["organization.id"] = organizationId;
-    // Add additional tags
-    Object.assign(result, additionalTags);
-    return result;
-}
-/**
- * Builds LLMObsNamedSpanOptions from StepResult for tracing
- */
-export function buildSpanOptions(kind, step, modelName, modelProvider) {
-    const getSpanName = (finishReason, content) => {
-        switch (finishReason) {
-            case "tool-calls": {
-                const toolName = Array.isArray(content)
-                    ? content.find((item) => item?.type === "tool-call" && item.toolName)
-                        ?.toolName
-                    : undefined;
-                return toolName ? `llm_${toolName}` : "llm_tool_calls";
-            }
-            case "stop":
-                return "llm_text_generation";
-            case "length":
-                return "llm_token_limit";
-            case "content-filter":
-                return "llm_content_filtered";
-            case "error":
-                return "llm_error";
-            default:
-                return "llm_step";
-        }
-    };
-    return {
-        kind,
-        name: getSpanName(step.finishReason, step.content),
-        ...(modelName && { modelName }),
-        ...(modelProvider && { modelProvider }),
-    };
-}
-/**
- * Wrapper around streamText that automatically handles tracing for onStepFinish, onFinish, and prepareStep
- */
-export function tracedStreamText(options) {
-    const { llmobs } = tracer;
-    let startTime;
-    const tracedPrepareStep = async (step) => {
-        startTime = Date.now();
-        if (options.prepareStep) {
-            return await options.prepareStep(step);
-        }
-    };
-    const tracedOnStepFinish = async (step) => {
-        return llmobs.trace(buildSpanOptions("llm", step, typeof options.model === "string"
-            ? options.model
-            : options.model.modelId, typeof options.model === "string" ? "unknown" : options.model.provider), async () => {
-            if (options.onStepFinish) {
-                await options.onStepFinish(step);
-            }
-            console.log("Usage:", JSON.stringify(step.usage));
-            llmobs.annotate({
-                inputData: JSON.stringify(step.request?.body?.messages || []),
-                outputData: JSON.stringify(step.content),
-                metrics: {
-                    inputTokens: step.usage?.inputTokens ?? 0,
-                    outputTokens: step.usage?.outputTokens ?? 0,
-                    totalTokens: step.usage?.totalTokens ?? 0,
-                },
-                metadata: {
-                    request: step.request,
-                    response: step.response,
-                },
-                tags: {
-                    duration: Date.now() - startTime,
-                    cachedInputTokens: step.usage?.cachedInputTokens ?? 0,
-                },
-            });
-        });
-    };
-    // Trace all tools
-    if (options.tools) {
-        options.tools = traceToolSet(options.tools);
-    }
-    return streamText({
-        ...options,
-        prepareStep: tracedPrepareStep,
-        onStepFinish: tracedOnStepFinish,
-    });
-}
-//# sourceMappingURL=ddog-llmobs.js.map

package/dist/ai-service/util/ddog-llmobs.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"ddog-llmobs.js","sourceRoot":"","sources":["../../../src/ai-service/util/ddog-llmobs.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,UAAU,GAKX,MAAM,IAAI,CAAC;AACZ,OAAO,MAAM,EAAE,EAAa,MAAM,UAAU,CAAC;AAM7C,4DAA4D;AAC5D,MAAM,UAAU,aAAa,CAC3B,IAAY,EACZ,SAAoC,EACpC,IAAuC;IAEvC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAE1B,OAAO,MAAM,CAAC,KAAK,CACjB;QACE,IAAI,EAAE,UAAU;QAChB,IAAI;KACL,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC,CACF,CAAC;AACJ,CAAC;AAED,wDAAwD;AACxD,MAAM,UAAU,SAAS,CACvB,IAAY,EACZ,SAAoC,EACpC,IAAuC;IAEvC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAE1B,OAAO,MAAM,CAAC,KAAK,CACjB;QACE,IAAI,EAAE,MAAM;QACZ,IAAI;KACL,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC,CACF,CAAC;AACJ,CAAC;AAED,uDAAuD;AACvD,MAAM,UAAU,QAAQ,CACtB,IAAY,EACZ,SAAiB,EACjB,aAAqB,EACrB,SAAqC,EACrC,IAAuC;IAEvC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAE1B,OAAO,MAAM,CAAC,KAAK,CACjB;QACE,IAAI,EAAE,KAAK;QACX,IAAI;QACJ,SAAS;QACT,aAAa;KACd,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC,CACF,CAAC;AACJ,CAAC;AAED,wDAAwD;AACxD,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,OAAgB,EAAW,EAAE;IACxD,MAAM,aAAa,GAAY,EAAE,CAAC;IAClC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAE1B,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACnD,aAAa,CAAC,IAAI,CAAC,GAAG;YACpB,GAAG,IAAI;YACP,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;gBAChC,IAAI,CAAC;oBACH,OAAO,MAAM,MAAM,CAAC,KAAK,CACvB;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI;qBACL,EACD,KAAK,IAAI,EAAE;wBACT,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;4BAClB,MAAM,IAAI,KAAK,CAAC,QAAQ,IAAI,wBAAwB,CAAC,CAAC;wBACxD,CAAC;wBAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;wBAClD,MAAM,CAAC,QAAQ,CAAC;4BACd,SAAS,EAAE,KAAK;4BAChB,UAAU,EAAE,MAAM;4BAClB,QAAQ,EAAE;gCACR,UAAU,EAAE,OAAO,EAAE,UAAU;6BAChC;yBACF,CAAC,CAAC;wBACH,OAAO,MAAM,CAAC;oBAChB,CAAC,CACF,CAAC;gBACJ,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,wBAAwB,IAAI,GAAG,EAAE,KAAK,CAAC,CAAC;oBACtD,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;SACF,CAAC;IACJ,CAAC;IACD,OAAO,aAAa,CAAC;AACvB,CAAC,CAAC;AAEF,mCAAmC;AACnC,MAAM,UAAU,eAAe,CAC7B,GAAuB;IAEvB,MAAM,MAAM,GAA2B,EAAE,CAAC;IAE1C,IAAI,GAAG,EAAE,CAAC;QACR,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC7B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAEvE,IAAI,OAAO,CAAC,MAAM;oBAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;gBACtD,IAAI,OAAO,CAAC,QAAQ;oBAAE,MAAM,CAAC,UAAU,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC;gBAC5D,IAAI,OAAO,CAAC,OAAO;oBAAE,MAAM,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;gBACzD,IAAI,OAAO,CAAC,UAAU;oBAAE,MAAM,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC,UAAU,CAAC;YACpE,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,qCAAqC;QACvC,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,+DAA+D;AAC/D,MAAM,UAAU,kBAAkB,CAChC,GAAuB,EACvB,aAAqB,EACrB,cAAsB,EACtB,iBAAmD,EAAE;IAErD,MAAM,MAAM,GAAqC,EAAE,CAAC;IAEpD,wDAAwD;IACxD,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC;IAE5C,iEAAiE;IACjE,MAAM,CAAC,gBAAgB,CAAC,GAAG,aAAa,CAAC;IACzC,MAAM,CAAC,iBAAiB,CAAC,GAAG,cAAc,CAAC;IAE3C,sBAAsB;IACtB,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IAEtC,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAC9B,IAAoB,EACpB,IAAmB,EACnB,SAAkB,EAClB,aAAsB;IAEtB,MAAM,WAAW,GAAG,CAAC,YAAgC,EAAE,OAAc,EAAE,EAAE;QACvE,QAAQ,YAAY,EAAE,CAAC;YACrB,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;oBACrC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,KAAK,WAAW,IAAI,IAAI,CAAC,QAAQ,CAAC;wBACjE,EAAE,QAAQ;oBACd,CAAC,CAAC,SAAS,CAAC;gBACd,OAAO,QAAQ,CAAC,CAAC,CAAC,OAAO,QAAQ,EAAE,CAAC,CAAC,CAAC,gBAAgB,CAAC;YACzD,CAAC;YACD,KAAK,MAAM;gBACT,OAAO,qBAAqB,CAAC;YAC/B,KAAK,QAAQ;gBACX,OAAO,iBAAiB,CAAC;YAC3B,KAAK,gBAAgB;gBACnB,OAAO,sBAAsB,CAAC;YAChC,KAAK,OAAO;gBACV,OAAO,WAAW,CAAC;YACrB;gBACE,OAAO,UAAU,CAAC;QACtB,CAAC;IACH,CAAC,CAAC;IAEF,OAAO;QACL,IAAI;QACJ,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC;QAClD,GAAG,CAAC,SAAS,IAAI,EAAE,SAAS,EAAE,CAAC;QAC/B,GAAG,CAAC,aAAa,IAAI,EAAE,aAAa,EAAE,CAAC;KACxC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAC9B,OAA4C;IAE5C,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAE1B,IAAI,SAAiB,CAAC;IAEtB,MAAM,iBAAiB,GAA2B,KAAK,EAAE,IAAI,EAAE,EAAE;QAC/D,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YACxB,OAAO,MAAM,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,kBAAkB,GAAsC,KAAK,EACjE,IAAI,EACJ,EAAE;QACF,OAAO,MAAM,CAAC,KAAK,CACjB,gBAAgB,CACd,KAAK,EACL,IAAI,EACJ,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ;YAC/B,CAAC,CAAC,OAAO,CAAC,KAAK;YACf,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,EACzB,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CACvE,EACD,KAAK,IAAI,EAAE;YACT,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;gBACzB,MAAM,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YACnC,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YAElD,MAAM,CAAC,QAAQ,CAAC;gBACd,SAAS,EAAE,IAAI,CAAC,SAAS,CACtB,IAAI,CAAC,OAAO,EAAE,IAAY,EAAE,QAAQ,IAAI,EAAE,CAC5C;gBACD,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC;gBACxC,OAAO,EAAE;oBACP,WAAW,EAAE,IAAI,CAAC,KAAK,EAAE,WAAW,IAAI,CAAC;oBACzC,YAAY,EAAE,IAAI,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC;oBAC3C,WAAW,EAAE,IAAI,CAAC,KAAK,EAAE,WAAW,IAAI,CAAC;iBAC1C;gBACD,QAAQ,EAAE;oBACR,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ;iBACxB;gBACD,IAAI,EAAE;oBACJ,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;oBAChC,iBAAiB,EAAE,IAAI,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;iBACtD;aACF,CAAC,CAAC;QACL,CAAC,CACF,CAAC;IACJ,CAAC,CAAC;IAEF,kBAAkB;IAClB,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QAClB,OAAO,CAAC,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,CAAM,CAAC;IACnD,CAAC;IAED,OAAO,UAAU,CAAC;QAChB,GAAG,OAAO;QACV,WAAW,EAAE,iBAAiB;QAC9B,YAAY,EAAE,kBAAkB;KACjC,CAAC,CAAC;AACL,CAAC"}