@apifuse/provider-sdk 2.2.0-beta.53 → 2.2.0-beta.55

package/dist/runtime/auth-flow.d.ts

@@ -15,4 +15,5 @@ export declare function createFlowContext(options: {
     initialContext?: Record<string, unknown>;
     ocr?: OcrContext;
     stt?: SttContext;
+    trace?: FlowContext["trace"];
 }): FlowContext;

package/dist/runtime/auth-flow.js

@@ -3,6 +3,7 @@ import { createAuthFlowHelpers } from "../auth.js";
 import { createUnsupportedOcrClient } from "./ocr.js";
 import { createUnsupportedResolverClient } from "./resolver-shared.js";
 import { createUnsupportedSttClient } from "./stt.js";
+import { createTraceContext } from "./trace.js";
 function normalizeAllowedKeys(allowedKeys) {
     return new Set(allowedKeys.filter((key) => key.trim().length > 0));
 }
@@ -39,6 +40,7 @@ export function createFlowContext(options) {
         externalRef: options.externalRef,
         tenantId: options.tenantId,
         providerId: options.providerId,
+        trace: options.trace ?? createTraceContext(),
         http: options.http,
         state: options.state,
         stealth: options.stealth,

package/dist/runtime/trace.d.ts

@@ -10,6 +10,8 @@ export interface CreateTraceContextOptions {
     onSpan?: (span: Span) => void;
     exportOptions?: OTLPExportOptions;
     resourceAttributes?: Record<string, string>;
+    /** W3C-compatible 32-character lowercase hexadecimal trace id used for export. */
+    traceId?: string;
     /**
      * Applied to a detached copy of each span immediately before OTLP export; never touches
      * getSpans() or onSpan. Returning nothing (or throwing) drops that export batch.
@@ -25,6 +27,11 @@ export interface TraceRecorder {
     runSpan<T>(name: string, fn: () => Promise<T> | T, options?: SpanHookOptions<T>): Promise<T>;
 }
 export declare const TRACE_RECORDER: unique symbol;
+/** Updates request metadata before its pending root span completes and exports. */
+export declare function updateTraceContextExportMetadata(trace: BaseTraceContext, input: {
+    traceId?: string;
+    resourceAttributes?: Record<string, string>;
+}): void;
 export declare function resolveTraceContextOptions(config?: TraceConfig): CreateTraceContextOptions;
 export declare function getTraceRecorder(trace: BaseTraceContext): TraceRecorder | null;
 export declare function createTraceContext(options?: CreateTraceContextOptions): TraceContext;

package/dist/runtime/trace.js

@@ -1,6 +1,16 @@
 import { AsyncLocalStorage } from "node:async_hooks";
 import { exportSpansOTLP } from "./otlp.js";
 export const TRACE_RECORDER = Symbol.for("@apifuse/provider-sdk/runtime/trace-recorder");
+const TRACE_EXPORT_METADATA = Symbol.for("@apifuse/provider-sdk/runtime/trace-export-metadata");
+function assertValidTraceId(traceId) {
+    if (!/^[0-9a-f]{32}$/.test(traceId) || /^0{32}$/.test(traceId)) {
+        throw new TypeError("traceId must be 32 lowercase hexadecimal characters and non-zero");
+    }
+}
+/** Updates request metadata before its pending root span completes and exports. */
+export function updateTraceContextExportMetadata(trace, input) {
+    trace[TRACE_EXPORT_METADATA]?.update(input);
+}
 function buildOTLPExportOptions(config) {
     if (config?.exporter !== "otlp") {
         return undefined;
@@ -67,6 +77,8 @@ export function getTraceRecorder(trace) {
     return trace[TRACE_RECORDER] ?? null;
 }
 export function createTraceContext(options = {}) {
+    if (options.traceId !== undefined)
+        assertValidTraceId(options.traceId);
     const maxSpans = options.maxSpans ?? 1000;
     const completed = [];
     const activeSpanStorage = new AsyncLocalStorage();
@@ -79,7 +91,7 @@ export function createTraceContext(options = {}) {
         : undefined;
     // One trace id per context so every export batch of this request shares it and
     // two processes can never mint the same id.
-    const exportTraceId = crypto.randomUUID().replace(/-/g, "");
+    let exportTraceId = options.traceId ?? crypto.randomUUID().replace(/-/g, "");
     let exportScheduled = false;
     // One pending batch per context: roots completing before the flush share it, and a span is
     // handed to the exporter exactly once, so later roots never re-send earlier spans.
@@ -168,6 +180,17 @@ export function createTraceContext(options = {}) {
             return completed.map((entry) => ({ ...entry.span }));
         },
         [TRACE_RECORDER]: recorder,
+        [TRACE_EXPORT_METADATA]: {
+            update(input) {
+                if (input.traceId !== undefined) {
+                    assertValidTraceId(input.traceId);
+                    exportTraceId = input.traceId;
+                }
+                if (input.resourceAttributes !== undefined && exportResourceAttributes) {
+                    Object.assign(exportResourceAttributes, input.resourceAttributes);
+                }
+            },
+        },
     };
     return traceContext;
 }

package/dist/server/serve-implementation.d.ts

@@ -79,6 +79,10 @@ type ProviderServerLogEventBase = ProviderRequestCost & {
     kind: "operation" | "auth";
     route: string;
     requestId?: string;
+    connectionId?: string;
+    flowId?: string;
+    tenantId?: string;
+    requestedProviderId?: string;
     status: number;
     proxy?: ProxyTelemetryLogPayload;
 };