weflayr 0.26.1 → 0.28.0

package/dist/src/instrumentation/vercel/get_structured_output_schema_telemetry.js

@@ -1,84 +0,0 @@
-"use strict";
-/**
- * Structured-output schema capture for Vercel AI SDK (ai >= 7) model-call spans.
- *
- * A `generateObject` call's schema is only ever emitted on the operation's ROOT
- * span (`onObjectOperationStart` stamps `ai.schema` there under the
- * supplemental-attributes opt-in) — and weflayr's span filter drops that
- * aggregate span, keeping one span per model call. So the schema never reaches
- * the exported span, even though the gateway bills it as input.
- *
- * The operation and its nested model call share one `callId` in the
- * dispatcher's call state. Wrap `onObjectOperationStart` to remember the
- * event's schema by callId, and `onObjectStepEnd` — the object operation's own
- * model-call hook, where the call state's `inferenceSpan` exists (object
- * operations never fire `onLanguageModelCallEnd`; verified live) — to stamp it
- * as `ai.schema` on that span, the attribute the worker already reads.
- * `cleanupCallState` is wrapped so an operation that errors before its model
- * call does not leak its entry.
- */
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.VercelStructuredOutputSchemaInstrumentor = void 0;
-const SCHEMA_ATTR = 'ai.schema';
-class VercelStructuredOutputSchemaInstrumentor {
-    // [owner, methodName, originalFunction] for uninstrument().
-    _patched = [];
-    _schemasByCallId = new Map();
-    // No-op (and never throws) if the interface is missing — the worst case is
-    // the schema capture staying empty, exactly the pre-complement behaviour.
-    instrument(dispatcher) {
-        if (!dispatcher ||
-            typeof dispatcher.onObjectOperationStart !== 'function' ||
-            typeof dispatcher.onObjectStepEnd !== 'function') {
-            return;
-        }
-        const schemasByCallId = this._schemasByCallId;
-        const originalOperationStart = dispatcher.onObjectOperationStart;
-        dispatcher.onObjectOperationStart = function onObjectOperationStartWithSchema(event) {
-            try {
-                if (event && event.callId && event.schema && typeof event.schema === 'object') {
-                    schemasByCallId.set(event.callId, event.schema);
-                }
-            }
-            catch {
-                // Never break span creation over an enrichment failure.
-            }
-            return originalOperationStart.call(this, event);
-        };
-        this._patched.push([dispatcher, 'onObjectOperationStart', originalOperationStart]);
-        const originalStepEnd = dispatcher.onObjectStepEnd;
-        dispatcher.onObjectStepEnd = function onObjectStepEndWithSchema(event) {
-            try {
-                const schema = event && schemasByCallId.get(event.callId);
-                if (schema) {
-                    const callState = this.getCallState && this.getCallState(event.callId);
-                    const span = callState && callState.inferenceSpan;
-                    if (span && typeof span.setAttribute === 'function') {
-                        span.setAttribute(SCHEMA_ATTR, JSON.stringify(schema));
-                    }
-                }
-            }
-            catch {
-                // Never break span finalization over an enrichment failure.
-            }
-            return originalStepEnd.call(this, event);
-        };
-        this._patched.push([dispatcher, 'onObjectStepEnd', originalStepEnd]);
-        const originalCleanup = dispatcher.cleanupCallState;
-        if (typeof originalCleanup === 'function') {
-            dispatcher.cleanupCallState = function cleanupCallStateWithSchema(callId) {
-                schemasByCallId.delete(callId);
-                return originalCleanup.call(this, callId);
-            };
-            this._patched.push([dispatcher, 'cleanupCallState', originalCleanup]);
-        }
-    }
-    uninstrument() {
-        for (const [owner, methodName, original] of this._patched) {
-            owner[methodName] = original;
-        }
-        this._patched = [];
-        this._schemasByCallId.clear();
-    }
-}
-exports.VercelStructuredOutputSchemaInstrumentor = VercelStructuredOutputSchemaInstrumentor;

package/dist/src/otel/vercel-ai-sdk-span-filter.d.ts

@@ -1,12 +0,0 @@
-import type { ReadableSpan } from '@opentelemetry/sdk-trace-base';
-export declare class VercelAiSdkSpanFilter {
-    private _openSpanOperations;
-    private _openLegacySchemas;
-    onStart(span: ReadableSpan): void;
-    /**
-     * Export verdict for a Vercel AI SDK span, or null when the span is not
-     * one (the caller falls back to the default filter).
-     */
-    shouldExport(span: ReadableSpan): boolean | null;
-}
-export declare function stampVercelGatewayProviderName(span: ReadableSpan): void;

package/dist/src/otel/vercel-ai-sdk-span-filter.js

@@ -1,88 +0,0 @@
-"use strict";
-// The Vercel AI SDK's OTel integrations emit aggregate spans around each model
-// call, all repeating the same usage counts:
-//   - `@ai-sdk/otel` (ai >= 7, scope "gen_ai"): per-call "chat" spans are
-//     wrapped in "agent_step" + "invoke_agent" spans, and embed/embedMany
-//     emits an outer "embeddings" operation span around the model-call
-//     "embeddings" span(s).
-//   - the legacy integration (ai <= 6, scope "ai"): `ai.<fn>.do<Call>` spans
-//     are wrapped in an outer `ai.<fn>` span.
-// Exactly one span per model call may reach the exporter — exporting an
-// aggregate alongside its children would bill every call twice.
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.VercelAiSdkSpanFilter = void 0;
-exports.stampVercelGatewayProviderName = stampVercelGatewayProviderName;
-const VERCEL_SEMCONV_SCOPE = 'gen_ai';
-const VERCEL_LEGACY_SCOPE = 'ai';
-const LEGACY_MODEL_CALL_SPAN_NAME = /\.do(Generate|Stream|Embed)$/;
-function _parentSpanId(span) {
-    return span.parentSpanContext?.spanId ?? span.parentSpanId;
-}
-class VercelAiSdkSpanFilter {
-    // spanId -> gen_ai.operation.name for open "gen_ai"-scope spans, so a
-    // span can check what its parent is (children end before their parent).
-    _openSpanOperations = new Map();
-    // spanId -> ai.schema of open legacy generateObject/streamObject ROOT spans:
-    // ai <= 6 records the schema only there, and this filter drops the root, so
-    // the schema is copied onto the model-call child when it starts (the root's
-    // creation attributes are already set by then).
-    _openLegacySchemas = new Map();
-    onStart(span) {
-        const scope = span.instrumentationScope?.name;
-        if (scope === VERCEL_LEGACY_SCOPE) {
-            const schema = span.attributes['ai.schema'];
-            if (schema !== undefined && !LEGACY_MODEL_CALL_SPAN_NAME.test(span.name)) {
-                this._openLegacySchemas.set(span.spanContext().spanId, schema);
-            }
-            const parentSpanId = _parentSpanId(span);
-            if (parentSpanId &&
-                span.attributes['ai.schema'] === undefined &&
-                this._openLegacySchemas.has(parentSpanId)) {
-                span.attributes['ai.schema'] = this._openLegacySchemas.get(parentSpanId);
-            }
-            return;
-        }
-        if (scope !== VERCEL_SEMCONV_SCOPE)
-            return;
-        this._openSpanOperations.set(span.spanContext().spanId, span.attributes['gen_ai.operation.name']);
-    }
-    /**
-     * Export verdict for a Vercel AI SDK span, or null when the span is not
-     * one (the caller falls back to the default filter).
-     */
-    shouldExport(span) {
-        const scope = span.instrumentationScope?.name;
-        if (scope === VERCEL_LEGACY_SCOPE) {
-            this._openLegacySchemas.delete(span.spanContext().spanId);
-            return LEGACY_MODEL_CALL_SPAN_NAME.test(span.name);
-        }
-        if (scope !== VERCEL_SEMCONV_SCOPE) {
-            return null;
-        }
-        const operation = span.attributes['gen_ai.operation.name'];
-        this._openSpanOperations.delete(span.spanContext().spanId);
-        if (operation === 'chat') {
-            return true;
-        }
-        if (operation === 'embeddings') {
-            // Keep the outer aggregate (one event per embed/embedMany call, with the
-            // total usage) and drop the nested model-call span(s) repeating it.
-            const parentSpanId = _parentSpanId(span);
-            return (parentSpanId && this._openSpanOperations.get(parentSpanId)) !== 'embeddings';
-        }
-        // invoke_agent / agent_step (usage aggregates) and execute_tool (no usage).
-        return false;
-    }
-}
-exports.VercelAiSdkSpanFilter = VercelAiSdkSpanFilter;
-// `@ai-sdk/otel` reports the Vercel AI Gateway as the bare provider id
-// "gateway" — too generic a name to special-case downstream. Stamp the weflayr
-// provider override (which the normalizer prefers) with the canonical name.
-// A providerNameOverride set through propagateMetadata was stamped at span start and
-// wins.
-function stampVercelGatewayProviderName(span) {
-    if (span.attributes['gen_ai.provider.name'] === 'gateway' &&
-        span.attributes['weflayr.provider_name'] === undefined) {
-        span.attributes['weflayr.provider_name'] = 'vercel_ai_gateway';
-    }
-}