assistant-cloud 0.2.0 → 0.2.2

package/dist/instrumentMcpSampling.js.map

@@ -1 +1 @@
-{"version":3,"file":"instrumentMcpSampling.js","names":[],"sources":["../src/instrumentMcpSampling.ts"],"sourcesContent":["/**\n * MCP sampling instrumentation utility.\n *\n * Wraps an MCP client's sampling handler to capture nested LLM calls\n * (sampling/createMessage requests) made during tool execution.\n * The captured data can be reported as child generation spans.\n */\n\nexport type SamplingCallData = {\n  model_id?: string;\n  input_tokens?: number;\n  output_tokens?: number;\n  reasoning_tokens?: number;\n  cached_input_tokens?: number;\n  duration_ms?: number;\n};\n\nexport type McpSamplingHandler = (\n  request: McpSamplingRequest,\n) => Promise<McpSamplingResponse>;\n\nexport type McpSamplingRequest = {\n  method: \"sampling/createMessage\";\n  params: {\n    messages: unknown[];\n    modelPreferences?: { hints?: { name?: string }[] };\n    maxTokens?: number;\n    [key: string]: unknown;\n  };\n};\n\nexport type McpSamplingResponse = {\n  model?: string;\n  content: unknown;\n  usage?: {\n    inputTokens?: number;\n    outputTokens?: number;\n    promptTokens?: number;\n    completionTokens?: number;\n    reasoningTokens?: number;\n    cachedInputTokens?: number;\n  };\n  [key: string]: unknown;\n};\n\n/**\n * Wraps an MCP sampling handler to intercept and measure sampling calls.\n *\n * @param handler - The original sampling handler from the MCP client\n * @param onSamplingCall - Callback invoked with metrics for each sampling call\n * @returns A wrapped handler that transparently captures sampling metrics\n *\n * @example\n * ```ts\n * const samplingCalls: SamplingCallData[] = [];\n * const wrapped = wrapSamplingHandler(\n *   originalHandler,\n *   (data) => samplingCalls.push(data),\n * );\n * // Use `wrapped` as the MCP client's sampling handler\n * // After tool execution, `samplingCalls` contains metrics for all nested LLM calls\n * ```\n */\nexport function wrapSamplingHandler(\n  handler: McpSamplingHandler,\n  onSamplingCall: (data: SamplingCallData) => void,\n): McpSamplingHandler {\n  return async (request) => {\n    const startTime = Date.now();\n    const response = await handler(request);\n    const durationMs = Date.now() - startTime;\n\n    const modelId =\n      response.model ?? request.params.modelPreferences?.hints?.[0]?.name;\n\n    const inputTokens =\n      response.usage?.inputTokens ?? response.usage?.promptTokens;\n    const outputTokens =\n      response.usage?.outputTokens ?? response.usage?.completionTokens;\n    const reasoningTokens = response.usage?.reasoningTokens;\n    const cachedInputTokens = response.usage?.cachedInputTokens;\n\n    onSamplingCall({\n      ...(modelId ? { model_id: modelId } : undefined),\n      ...(inputTokens != null ? { input_tokens: inputTokens } : undefined),\n      ...(outputTokens != null ? { output_tokens: outputTokens } : undefined),\n      ...(reasoningTokens != null\n        ? { reasoning_tokens: reasoningTokens }\n        : undefined),\n      ...(cachedInputTokens != null\n        ? { cached_input_tokens: cachedInputTokens }\n        : undefined),\n      duration_ms: durationMs,\n    });\n\n    return response;\n  };\n}\n\n/**\n * Creates a collector that accumulates sampling call data during tool execution.\n * Use with `wrapSamplingHandler` to capture all sampling calls for a tool invocation.\n *\n * @example\n * ```ts\n * const collector = createSamplingCollector();\n * const wrappedHandler = wrapSamplingHandler(handler, collector.collect);\n * // ... execute MCP tool ...\n * const calls = collector.getCalls(); // SamplingCallData[]\n * ```\n */\nexport function createSamplingCollector() {\n  const calls: SamplingCallData[] = [];\n  return {\n    collect: (data: SamplingCallData) => calls.push(data),\n    getCalls: () => [...calls],\n    reset: () => {\n      calls.length = 0;\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AA+DA,SAAgB,oBACd,SACA,gBACoB;CACpB,OAAO,OAAO,YAAY;EACxB,MAAM,YAAY,KAAK,IAAI;EAC3B,MAAM,WAAW,MAAM,QAAQ,OAAO;EACtC,MAAM,aAAa,KAAK,IAAI,IAAI;EAEhC,MAAM,UACJ,SAAS,SAAS,QAAQ,OAAO,kBAAkB,QAAQ,EAAE,EAAE;EAEjE,MAAM,cACJ,SAAS,OAAO,eAAe,SAAS,OAAO;EACjD,MAAM,eACJ,SAAS,OAAO,gBAAgB,SAAS,OAAO;EAClD,MAAM,kBAAkB,SAAS,OAAO;EACxC,MAAM,oBAAoB,SAAS,OAAO;EAE1C,eAAe;GACb,GAAI,UAAU,EAAE,UAAU,QAAQ,IAAI,KAAA;GACtC,GAAI,eAAe,OAAO,EAAE,cAAc,YAAY,IAAI,KAAA;GAC1D,GAAI,gBAAgB,OAAO,EAAE,eAAe,aAAa,IAAI,KAAA;GAC7D,GAAI,mBAAmB,OACnB,EAAE,kBAAkB,gBAAgB,IACpC,KAAA;GACJ,GAAI,qBAAqB,OACrB,EAAE,qBAAqB,kBAAkB,IACzC,KAAA;GACJ,aAAa;EACf,CAAC;EAED,OAAO;CACT;AACF;;;;;;;;;;;;;AAcA,SAAgB,0BAA0B;CACxC,MAAM,QAA4B,CAAC;CACnC,OAAO;EACL,UAAU,SAA2B,MAAM,KAAK,IAAI;EACpD,gBAAgB,CAAC,GAAG,KAAK;EACzB,aAAa;GACX,MAAM,SAAS;EACjB;CACF;AACF"}
+{"version":3,"file":"instrumentMcpSampling.js","names":[],"sources":["../src/instrumentMcpSampling.ts"],"sourcesContent":["/**\n * MCP sampling instrumentation utility.\n *\n * Wraps an MCP client's sampling handler to capture nested LLM calls\n * (sampling/createMessage requests) made during tool execution.\n * The captured data can be reported as child generation spans.\n */\n\nexport type SamplingCallData = {\n  model_id?: string;\n  input_tokens?: number;\n  output_tokens?: number;\n  reasoning_tokens?: number;\n  cached_input_tokens?: number;\n  duration_ms?: number;\n};\n\nexport type McpSamplingHandler = (\n  request: McpSamplingRequest,\n) => Promise<McpSamplingResponse>;\n\nexport type McpSamplingRequest = {\n  method: \"sampling/createMessage\";\n  params: {\n    messages: unknown[];\n    modelPreferences?: { hints?: { name?: string }[] };\n    maxTokens?: number;\n    [key: string]: unknown;\n  };\n};\n\nexport type McpSamplingResponse = {\n  model?: string;\n  content: unknown;\n  usage?: {\n    inputTokens?: number;\n    outputTokens?: number;\n    promptTokens?: number;\n    completionTokens?: number;\n    reasoningTokens?: number;\n    cachedInputTokens?: number;\n  };\n  [key: string]: unknown;\n};\n\nconst reportSamplingError = (error: unknown): void => {\n  try {\n    console.error(\n      \"[assistant-cloud] onSamplingCall callback threw an error\",\n      error,\n    );\n  } catch {}\n};\n\n/**\n * Wraps an MCP sampling handler to intercept and measure sampling calls.\n *\n * @param handler - The original sampling handler from the MCP client\n * @param onSamplingCall - Callback invoked with metrics for each sampling call\n * @returns A wrapped handler that transparently captures sampling metrics\n *\n * @example\n * ```ts\n * const samplingCalls: SamplingCallData[] = [];\n * const wrapped = wrapSamplingHandler(\n *   originalHandler,\n *   (data) => samplingCalls.push(data),\n * );\n * // Use `wrapped` as the MCP client's sampling handler\n * // After tool execution, `samplingCalls` contains metrics for all nested LLM calls\n * ```\n */\nexport function wrapSamplingHandler(\n  handler: McpSamplingHandler,\n  onSamplingCall: (data: SamplingCallData) => void,\n): McpSamplingHandler {\n  return async (request) => {\n    const startTime = Date.now();\n    const response = await handler(request);\n    const durationMs = Date.now() - startTime;\n\n    const modelId =\n      response.model ?? request.params.modelPreferences?.hints?.[0]?.name;\n\n    const inputTokens =\n      response.usage?.inputTokens ?? response.usage?.promptTokens;\n    const outputTokens =\n      response.usage?.outputTokens ?? response.usage?.completionTokens;\n    const reasoningTokens = response.usage?.reasoningTokens;\n    const cachedInputTokens = response.usage?.cachedInputTokens;\n\n    const data: SamplingCallData = {\n      ...(modelId ? { model_id: modelId } : undefined),\n      ...(inputTokens != null ? { input_tokens: inputTokens } : undefined),\n      ...(outputTokens != null ? { output_tokens: outputTokens } : undefined),\n      ...(reasoningTokens != null\n        ? { reasoning_tokens: reasoningTokens }\n        : undefined),\n      ...(cachedInputTokens != null\n        ? { cached_input_tokens: cachedInputTokens }\n        : undefined),\n      duration_ms: durationMs,\n    };\n    try {\n      void Promise.resolve(onSamplingCall(data)).catch(reportSamplingError);\n    } catch (error) {\n      reportSamplingError(error);\n    }\n\n    return response;\n  };\n}\n\n/**\n * Creates a collector that accumulates sampling call data during tool execution.\n * Use with `wrapSamplingHandler` to capture all sampling calls for a tool invocation.\n *\n * @example\n * ```ts\n * const collector = createSamplingCollector();\n * const wrappedHandler = wrapSamplingHandler(handler, collector.collect);\n * // ... execute MCP tool ...\n * const calls = collector.getCalls(); // SamplingCallData[]\n * ```\n */\nexport function createSamplingCollector() {\n  const calls: SamplingCallData[] = [];\n  return {\n    collect: (data: SamplingCallData) => calls.push(data),\n    getCalls: () => [...calls],\n    reset: () => {\n      calls.length = 0;\n    },\n  };\n}\n"],"mappings":";AA6CA,MAAM,uBAAuB,UAAyB;CACpD,IAAI;EACF,QAAQ,MACN,4DACA,KACF;CACF,QAAQ,CAAC;AACX;;;;;;;;;;;;;;;;;;;AAoBA,SAAgB,oBACd,SACA,gBACoB;CACpB,OAAO,OAAO,YAAY;EACxB,MAAM,YAAY,KAAK,IAAI;EAC3B,MAAM,WAAW,MAAM,QAAQ,OAAO;EACtC,MAAM,aAAa,KAAK,IAAI,IAAI;EAEhC,MAAM,UACJ,SAAS,SAAS,QAAQ,OAAO,kBAAkB,QAAQ,EAAE,EAAE;EAEjE,MAAM,cACJ,SAAS,OAAO,eAAe,SAAS,OAAO;EACjD,MAAM,eACJ,SAAS,OAAO,gBAAgB,SAAS,OAAO;EAClD,MAAM,kBAAkB,SAAS,OAAO;EACxC,MAAM,oBAAoB,SAAS,OAAO;EAE1C,MAAM,OAAyB;GAC7B,GAAI,UAAU,EAAE,UAAU,QAAQ,IAAI,KAAA;GACtC,GAAI,eAAe,OAAO,EAAE,cAAc,YAAY,IAAI,KAAA;GAC1D,GAAI,gBAAgB,OAAO,EAAE,eAAe,aAAa,IAAI,KAAA;GAC7D,GAAI,mBAAmB,OACnB,EAAE,kBAAkB,gBAAgB,IACpC,KAAA;GACJ,GAAI,qBAAqB,OACrB,EAAE,qBAAqB,kBAAkB,IACzC,KAAA;GACJ,aAAa;EACf;EACA,IAAI;GACF,QAAa,QAAQ,eAAe,IAAI,CAAC,CAAC,CAAC,MAAM,mBAAmB;EACtE,SAAS,OAAO;GACd,oBAAoB,KAAK;EAC3B;EAEA,OAAO;CACT;AACF;;;;;;;;;;;;;AAcA,SAAgB,0BAA0B;CACxC,MAAM,QAA4B,CAAC;CACnC,OAAO;EACL,UAAU,SAA2B,MAAM,KAAK,IAAI;EACpD,gBAAgB,CAAC,GAAG,KAAK;EACzB,aAAa;GACX,MAAM,SAAS;EACjB;CACF;AACF"}

package/dist/runTelemetry.d.ts

@@ -1,7 +1,7 @@
 import { SamplingCallData } from "./instrumentMcpSampling.js";
 import { AssistantCloudRunReport } from "./AssistantCloudRuns.js";
 //#region src/runTelemetry.d.ts
-type AssistantCloudRunReportToolCall = {
+export type AssistantCloudRunReportToolCall = {
   tool_name: string;
   tool_call_id: string;
   tool_args?: string;
@@ -11,12 +11,12 @@ type AssistantCloudRunReportToolCall = {
   end_ms?: number;
   sampling_calls?: SamplingCallData[];
 };
-type RunReportOutcome = "aborted" | "disconnected" | "length" | "content_filter";
+export type RunReportOutcome = "aborted" | "disconnected" | "length" | "content_filter";
 /**
  * Maps a finish event to the report status and outcome. `fallbackStatus`
  * applies when the event carries neither a finish reason nor a failure flag.
  */
-declare function deriveRunOutcome(input: {
+export declare function deriveRunOutcome(input: {
   finishReason?: string | undefined;
   isAbort?: boolean | undefined;
   isDisconnect?: boolean | undefined;
@@ -29,7 +29,7 @@ declare function deriveRunOutcome(input: {
  * Reads the message and code the runs endpoint stores for a failed run. The
  * code is the error's `code` when it has one, else its class name.
  */
-declare function describeRunError(error: unknown): {
+export declare function describeRunError(error: unknown): {
   error?: string;
   errorCode?: string;
 };
@@ -37,8 +37,8 @@ declare function describeRunError(error: unknown): {
  * Clamps a string to the size the runs endpoint accepts for a single span
  * field.
  */
-declare function truncateRunTelemetryText(value: string): string;
-type RunTelemetryToolCallInit = {
+export declare function truncateRunTelemetryText(value: string): string;
+export type RunTelemetryToolCallInit = {
   toolName: string;
   toolCallId: string;
   args?: unknown;
@@ -55,7 +55,7 @@ type RunTelemetryToolCallInit = {
  * source has its result summarized, because MCP content blocks carry inline
  * base64 image and audio payloads that would otherwise dominate the report.
  */
-declare function createRunTelemetryToolCall(init: RunTelemetryToolCallInit): AssistantCloudRunReportToolCall;
+export declare function createRunTelemetryToolCall(init: RunTelemetryToolCallInit): AssistantCloudRunReportToolCall;
 /**
  * Resolves the model ID a run reports, in the order an app can supply it: an
  * explicit `modelId`, the `custom` bag, then the per-step `response.modelId`
@@ -63,14 +63,14 @@ declare function createRunTelemetryToolCall(init: RunTelemetryToolCallInit): Ass
  * The AI SDK puts no model ID on a UI message part, so message metadata is the
  * only channel one arrives on.
  */
-declare function extractRunTelemetryModelId(metadata: Record<string, unknown> | undefined): string | undefined;
-type RunTelemetryUsage = {
+export declare function extractRunTelemetryModelId(metadata: Record<string, unknown> | undefined): string | undefined;
+export type RunTelemetryUsage = {
   inputTokens?: number | undefined;
   outputTokens?: number | undefined;
   reasoningTokens?: number | undefined;
   cachedInputTokens?: number | undefined;
 };
-type RunTelemetryUsageInit = RunTelemetryUsage & {
+export type RunTelemetryUsageInit = RunTelemetryUsage & {
   promptTokens?: number | undefined;
   completionTokens?: number | undefined;
   inputTokenDetails?: {
@@ -80,14 +80,30 @@ type RunTelemetryUsageInit = RunTelemetryUsage & {
     reasoningTokens?: number;
   };
 };
-type RunReportStepInit = {
+export type RunReportStepInit = {
   usage?: RunTelemetryUsageInit | undefined;
   toolCalls?: AssistantCloudRunReportToolCall[] | undefined;
   startMs?: number | undefined;
   endMs?: number | undefined;
   finishReason?: string | undefined;
 };
-type RunReportInit = {
+/**
+ * The run report fields read from the messages of one run, in whichever
+ * format they were stored: the status the messages imply, the tool calls, the
+ * steps, the text, the usage and the model.
+ */
+export type RunMessageTelemetry = {
+  assistantMessageId?: string;
+  status: "completed" | "incomplete";
+  toolCalls?: AssistantCloudRunReportToolCall[];
+  steps?: RunReportStepInit[];
+  totalSteps?: number;
+  outputText?: string;
+  usage?: RunTelemetryUsage;
+  modelId?: string;
+  metadata?: Record<string, unknown>;
+};
+export type RunReportInit = {
   threadId: string;
   status: AssistantCloudRunReport["status"];
   outcome?: RunReportOutcome | undefined;
@@ -111,14 +127,13 @@ type RunReportInit = {
     tags?: readonly string[] | undefined;
   };
 };
-declare function createRunReport(init: RunReportInit): AssistantCloudRunReport;
+export declare function createRunReport(init: RunReportInit): AssistantCloudRunReport;
 /**
  * Resolves the token counts a provider reports under any of the names the AI
  * SDK has used: the current top-level ones, the legacy prompt/completion pair,
  * and the v7 token detail objects. Returns undefined when no count is present,
  * so callers can tell an empty usage object from a zeroed one.
  */
-declare function normalizeRunTelemetryUsage(usage: RunTelemetryUsageInit): RunTelemetryUsage | undefined;
+export declare function normalizeRunTelemetryUsage(usage: RunTelemetryUsageInit): RunTelemetryUsage | undefined;
 //#endregion
-export { AssistantCloudRunReportToolCall, RunReportInit, RunReportOutcome, RunReportStepInit, RunTelemetryToolCallInit, RunTelemetryUsage, RunTelemetryUsageInit, createRunReport, createRunTelemetryToolCall, deriveRunOutcome, describeRunError, extractRunTelemetryModelId, normalizeRunTelemetryUsage, truncateRunTelemetryText };
 //# sourceMappingURL=runTelemetry.d.ts.map

package/dist/runTelemetry.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"runTelemetry.d.ts","names":[],"sources":["../src/runTelemetry.ts"],"mappings":";;;KAOY;EACV;EACA;EACA;EACA;EACA;EACA;EACA;EACA,iBAAiB;;KAGP;;;;;iBAUI,iBACd;EACE;EACA;EACA;EACA;GAEF;EAEA;EACA,UAAU;;;;;;iBA+BI,iBAAiB;EAC/B;EACA;;;;;;iBA+Bc,yBAAyB;KAwC7B;EACV;EACA;EACA;;;;;EAKA;EACA;EACA;;;;;;;iBAQc,2BACd,MAAM,2BACL;;;;;;;;iBAuBa,2BACd,UAAU;KAmBA;EACV;EACA;EACA;EACA;;KAGU,wBAAwB;EAClC;EACA;EACA;IAAsB;;EACtB;IAAuB;;;KAGb;EACV,QAAQ;EACR,YAAY;EACZ;EACA;EACA;;KAGU;EACV;EACA,QAAQ;EACR,UAAU;EACV;EACA;EACA;EACA;EACA;EACA;EACA,QAAQ;EACR,QAAQ;EACR;EACA,YAAY;EACZ;EACA;EACA;EACA,WAAW;EACX;IACE;IACA;IACA;;;iBAuEY,gBAAgB,MAAM,gBAAgB;;;;;;;iBAiDtC,2BACd,OAAO,wBACN"}
+{"version":3,"file":"runTelemetry.d.ts","names":[],"sources":["../src/runTelemetry.ts"],"mappings":";;;YAOY;EACV;EACA;EACA;EACA;EACA;EACA;EACA;EACA,iBAAiB;;YAGP;;;;;wBAUI,iBACd;EACE;EACA;EACA;EACA;GAEF;EAEA;EACA,UAAU;;;;;;wBA+BI,iBAAiB;EAC/B;EACA;;;;;;wBA+Bc,yBAAyB;YAwC7B;EACV;EACA;EACA;;;;;EAKA;EACA;EACA;;;;;;;wBAQc,2BACd,MAAM,2BACL;;;;;;;;wBAuBa,2BACd,UAAU;YAmBA;EACV;EACA;EACA;EACA;;YAGU,wBAAwB;EAClC;EACA;EACA;IAAsB;;EACtB;IAAuB;;;YAGb;EACV,QAAQ;EACR,YAAY;EACZ;EACA;EACA;;;;;;;YAQU;EACV;EACA;EACA,YAAY;EACZ,QAAQ;EACR;EACA;EACA,QAAQ;EACR;EACA,WAAW;;YAGD;EACV;EACA,QAAQ;EACR,UAAU;EACV;EACA;EACA;EACA;EACA;EACA;EACA,QAAQ;EACR,QAAQ;EACR;EACA,YAAY;EACZ;EACA;EACA;EACA,WAAW;EACX;IACE;IACA;IACA;;;wBAuEY,gBAAgB,MAAM,gBAAgB;;;;;;;wBAiDtC,2BACd,OAAO,wBACN"}

package/dist/runTelemetry.js.map

@@ -1 +1 @@
-{"version":3,"file":"runTelemetry.js","names":[],"sources":["../src/runTelemetry.ts"],"sourcesContent":["import type { SamplingCallData } from \"./instrumentMcpSampling\";\nimport type { AssistantCloudRunReport } from \"./AssistantCloudRuns\";\n\nconst MAX_TELEMETRY_TEXT_LENGTH = 50_000;\n\nconst BASE64_PATTERN = /^[A-Za-z0-9+/]{100,}={0,2}$/;\n\nexport type AssistantCloudRunReportToolCall = {\n  tool_name: string;\n  tool_call_id: string;\n  tool_args?: string;\n  tool_result?: string;\n  tool_source?: \"mcp\" | \"frontend\" | \"backend\";\n  start_ms?: number;\n  end_ms?: number;\n  sampling_calls?: SamplingCallData[];\n};\n\nexport type RunReportOutcome =\n  | \"aborted\"\n  | \"disconnected\"\n  | \"length\"\n  | \"content_filter\";\n\n/**\n * Maps a finish event to the report status and outcome. `fallbackStatus`\n * applies when the event carries neither a finish reason nor a failure flag.\n */\nexport function deriveRunOutcome(\n  input: {\n    finishReason?: string | undefined;\n    isAbort?: boolean | undefined;\n    isDisconnect?: boolean | undefined;\n    isError?: boolean | undefined;\n  },\n  fallbackStatus: \"completed\" | \"incomplete\" = \"completed\",\n): {\n  status: \"completed\" | \"incomplete\" | \"error\";\n  outcome?: RunReportOutcome;\n} {\n  if (input.isError) return { status: \"error\" };\n  if (input.isAbort) return { status: \"incomplete\", outcome: \"aborted\" };\n  if (input.isDisconnect) {\n    return { status: \"incomplete\", outcome: \"disconnected\" };\n  }\n  switch (input.finishReason) {\n    case \"length\":\n      return { status: \"incomplete\", outcome: \"length\" };\n    case \"content-filter\":\n    case \"content_filter\":\n      return { status: \"incomplete\", outcome: \"content_filter\" };\n    case \"cancelled\":\n      return { status: \"incomplete\", outcome: \"aborted\" };\n    case \"error\":\n      return { status: \"error\" };\n    default:\n      return {\n        status: input.finishReason === undefined ? fallbackStatus : \"completed\",\n      };\n  }\n}\n\nconst MAX_RUN_ERROR_CODE_LENGTH = 64;\nconst MAX_RUN_ERROR_LENGTH = 2048;\n\n/**\n * Reads the message and code the runs endpoint stores for a failed run. The\n * code is the error's `code` when it has one, else its class name.\n */\nexport function describeRunError(error: unknown): {\n  error?: string;\n  errorCode?: string;\n} {\n  if (error == null) return {};\n  const record =\n    typeof error === \"object\" ? (error as Record<string, unknown>) : undefined;\n  const message =\n    typeof error === \"string\"\n      ? error\n      : typeof record?.message === \"string\"\n        ? record.message\n        : undefined;\n  const code =\n    typeof record?.code === \"string\"\n      ? record.code\n      : typeof record?.name === \"string\" && record.name !== \"Error\"\n        ? record.name\n        : undefined;\n  return {\n    ...(message\n      ? { error: message.slice(0, MAX_RUN_ERROR_LENGTH) }\n      : undefined),\n    ...(code\n      ? { errorCode: code.slice(0, MAX_RUN_ERROR_CODE_LENGTH) }\n      : undefined),\n  };\n}\n\n/**\n * Clamps a string to the size the runs endpoint accepts for a single span\n * field.\n */\nexport function truncateRunTelemetryText(value: string): string {\n  if (value.length <= MAX_TELEMETRY_TEXT_LENGTH) return value;\n  return value.slice(0, MAX_TELEMETRY_TEXT_LENGTH);\n}\n\nfunction safeStringify(value: unknown): string | undefined {\n  if (value == null) return undefined;\n  try {\n    return truncateRunTelemetryText(JSON.stringify(value));\n  } catch {\n    return undefined;\n  }\n}\n\nfunction summarizeMcpResult(value: unknown): string | undefined {\n  if (value == null) return undefined;\n  try {\n    const parsed = typeof value === \"string\" ? JSON.parse(value) : value;\n    if (Array.isArray(parsed)) {\n      const summarized = parsed.map((item) => {\n        if (item && typeof item === \"object\" && item.type) {\n          if (\n            (item.type === \"image\" || item.type === \"audio\") &&\n            typeof item.data === \"string\" &&\n            BASE64_PATTERN.test(item.data.slice(0, 200))\n          ) {\n            const sizeKB = ((item.data.length * 3) / 4 / 1024).toFixed(1);\n            return { ...item, data: `[${item.type}: ${sizeKB}KB]` };\n          }\n        }\n        return item;\n      });\n      return truncateRunTelemetryText(JSON.stringify(summarized));\n    }\n  } catch {\n    // not JSON array, fall through\n  }\n  return safeStringify(value);\n}\n\nexport type RunTelemetryToolCallInit = {\n  toolName: string;\n  toolCallId: string;\n  args?: unknown;\n  /**\n   * Pre-serialized arguments, used in place of serializing `args`. Values over\n   * the span size are clamped before they are included in the report.\n   */\n  argsText?: string | undefined;\n  result?: unknown;\n  toolSource?: \"mcp\" | \"frontend\" | \"backend\" | undefined;\n};\n\n/**\n * Serializes one tool call into the shape the runs endpoint accepts. An `mcp`\n * source has its result summarized, because MCP content blocks carry inline\n * base64 image and audio payloads that would otherwise dominate the report.\n */\nexport function createRunTelemetryToolCall(\n  init: RunTelemetryToolCallInit,\n): AssistantCloudRunReportToolCall {\n  const { toolName, toolCallId, args, argsText, result, toolSource } = init;\n  const call: AssistantCloudRunReportToolCall = {\n    tool_name: toolName,\n    tool_call_id: toolCallId,\n  };\n  const toolArgs =\n    argsText != null ? truncateRunTelemetryText(argsText) : safeStringify(args);\n  if (toolArgs !== undefined) call.tool_args = toolArgs;\n  const toolResult =\n    toolSource === \"mcp\" ? summarizeMcpResult(result) : safeStringify(result);\n  if (toolResult !== undefined) call.tool_result = toolResult;\n  if (toolSource) call.tool_source = toolSource;\n  return call;\n}\n\n/**\n * Resolves the model ID a run reports, in the order an app can supply it: an\n * explicit `modelId`, the `custom` bag, then the per-step `response.modelId`\n * that a `messageMetadata` callback copies off the AI SDK's finish-step part.\n * The AI SDK puts no model ID on a UI message part, so message metadata is the\n * only channel one arrives on.\n */\nexport function extractRunTelemetryModelId(\n  metadata: Record<string, unknown> | undefined,\n): string | undefined {\n  if (!metadata) return undefined;\n  if (typeof metadata.modelId === \"string\") return metadata.modelId;\n  const custom = metadata.custom as Record<string, unknown> | undefined;\n  if (typeof custom?.modelId === \"string\") return custom.modelId;\n\n  const steps: unknown = metadata.steps;\n  if (!Array.isArray(steps)) return undefined;\n  for (const step of steps as unknown[]) {\n    if (!step || typeof step !== \"object\") continue;\n    const response = (step as Record<string, unknown>).response;\n    if (!response || typeof response !== \"object\") continue;\n    const modelId = (response as Record<string, unknown>).modelId;\n    if (typeof modelId === \"string\") return modelId;\n  }\n  return undefined;\n}\n\nexport type RunTelemetryUsage = {\n  inputTokens?: number | undefined;\n  outputTokens?: number | undefined;\n  reasoningTokens?: number | undefined;\n  cachedInputTokens?: number | undefined;\n};\n\nexport type RunTelemetryUsageInit = RunTelemetryUsage & {\n  promptTokens?: number | undefined;\n  completionTokens?: number | undefined;\n  inputTokenDetails?: { cacheReadTokens?: number };\n  outputTokenDetails?: { reasoningTokens?: number };\n};\n\nexport type RunReportStepInit = {\n  usage?: RunTelemetryUsageInit | undefined;\n  toolCalls?: AssistantCloudRunReportToolCall[] | undefined;\n  startMs?: number | undefined;\n  endMs?: number | undefined;\n  finishReason?: string | undefined;\n};\n\nexport type RunReportInit = {\n  threadId: string;\n  status: AssistantCloudRunReport[\"status\"];\n  outcome?: RunReportOutcome | undefined;\n  errorCode?: string | undefined;\n  error?: string | undefined;\n  messageId?: string | undefined;\n  traceId?: string | undefined;\n  modelId?: string | undefined;\n  provider?: string | undefined;\n  usage?: RunTelemetryUsageInit | undefined;\n  steps?: RunReportStepInit[] | undefined;\n  totalSteps?: number | undefined;\n  toolCalls?: AssistantCloudRunReportToolCall[] | undefined;\n  durationMs?: number | undefined;\n  firstTokenMs?: number | undefined;\n  outputText?: string | undefined;\n  metadata?: Record<string, unknown> | undefined;\n  telemetry?: {\n    environment?: string | undefined;\n    release?: string | undefined;\n    tags?: readonly string[] | undefined;\n  };\n};\n\nfunction assignUsage(\n  report: AssistantCloudRunReport,\n  usage: RunTelemetryUsageInit | undefined,\n): void {\n  if (!usage) return;\n  const normalized = normalizeRunTelemetryUsage(usage);\n  if (!normalized) return;\n  if (normalized.inputTokens !== undefined) {\n    report.input_tokens = normalized.inputTokens;\n  }\n  if (normalized.outputTokens !== undefined) {\n    report.output_tokens = normalized.outputTokens;\n  }\n  if (normalized.reasoningTokens !== undefined) {\n    report.reasoning_tokens = normalized.reasoningTokens;\n  }\n  if (normalized.cachedInputTokens !== undefined) {\n    report.cached_input_tokens = normalized.cachedInputTokens;\n  }\n}\n\nfunction createRunReportStep(\n  init: RunReportStepInit,\n): NonNullable<AssistantCloudRunReport[\"steps\"]>[number] {\n  const step: NonNullable<AssistantCloudRunReport[\"steps\"]>[number] = {};\n  const usage = init.usage ? normalizeRunTelemetryUsage(init.usage) : undefined;\n  if (usage?.inputTokens !== undefined) step.input_tokens = usage.inputTokens;\n  if (usage?.outputTokens !== undefined)\n    step.output_tokens = usage.outputTokens;\n  if (usage?.reasoningTokens !== undefined) {\n    step.reasoning_tokens = usage.reasoningTokens;\n  }\n  if (usage?.cachedInputTokens !== undefined) {\n    step.cached_input_tokens = usage.cachedInputTokens;\n  }\n  if (init.toolCalls !== undefined) step.tool_calls = init.toolCalls;\n  if (init.startMs !== undefined) step.start_ms = init.startMs;\n  if (init.endMs !== undefined) step.end_ms = init.endMs;\n  if (init.finishReason !== undefined) {\n    step.finish_reason = init.finishReason.slice(0, 32);\n  }\n  return step;\n}\n\nfunction normalizeRunReportTags(\n  tags: readonly string[] | undefined,\n): string[] | undefined {\n  if (!tags) return undefined;\n  const result: string[] = [];\n  const seen = new Set<string>();\n  for (const tag of tags) {\n    const normalized = tag.trim().slice(0, 64).trim();\n    if (!normalized || seen.has(normalized)) continue;\n    seen.add(normalized);\n    result.push(normalized);\n    if (result.length === 20) break;\n  }\n  return result.length > 0 ? result : undefined;\n}\n\nfunction normalizeRunReportMilliseconds(\n  value: number | undefined,\n): number | undefined {\n  if (value === undefined || !Number.isFinite(value)) return undefined;\n  return Math.max(0, Math.round(value));\n}\n\nexport function createRunReport(init: RunReportInit): AssistantCloudRunReport {\n  const report: AssistantCloudRunReport = {\n    thread_id: init.threadId,\n    status: init.status,\n  };\n  const traceId = init.traceId?.toLowerCase();\n  if (traceId && /^[0-9a-f]{32}$/.test(traceId)) report.trace_id = traceId;\n  if (init.outcome !== undefined) report.outcome_type = init.outcome;\n  if (init.errorCode !== undefined) report.error_code = init.errorCode;\n  if (init.error !== undefined) report.error = init.error;\n  if (init.messageId !== undefined) report.message_id = init.messageId;\n  if (init.modelId !== undefined) report.model_id = init.modelId;\n  if (init.provider !== undefined) {\n    report.provider = init.provider;\n    report.provider_type = init.provider;\n  }\n  assignUsage(report, init.usage);\n  if (init.steps !== undefined) {\n    report.steps = init.steps.map(createRunReportStep);\n    report.total_steps = init.steps.length;\n  } else if (init.totalSteps !== undefined) {\n    report.total_steps = init.totalSteps;\n  }\n  if (init.toolCalls !== undefined) report.tool_calls = init.toolCalls;\n  const durationMs = normalizeRunReportMilliseconds(init.durationMs);\n  if (durationMs !== undefined) report.duration_ms = durationMs;\n  const firstTokenMs = normalizeRunReportMilliseconds(init.firstTokenMs);\n  if (firstTokenMs !== undefined) report.first_token_ms = firstTokenMs;\n  if (init.outputText !== undefined) {\n    report.output_text = truncateRunTelemetryText(init.outputText);\n  }\n  if (init.metadata !== undefined) report.metadata = init.metadata;\n  if (init.telemetry?.environment !== undefined) {\n    report.environment = init.telemetry.environment;\n  }\n  if (init.telemetry?.release !== undefined) {\n    report.release = init.telemetry.release;\n  }\n  const tags = normalizeRunReportTags(init.telemetry?.tags);\n  if (tags !== undefined) report.tags = tags;\n  return report;\n}\n\n/**\n * Resolves the token counts a provider reports under any of the names the AI\n * SDK has used: the current top-level ones, the legacy prompt/completion pair,\n * and the v7 token detail objects. Returns undefined when no count is present,\n * so callers can tell an empty usage object from a zeroed one.\n */\nexport function normalizeRunTelemetryUsage(\n  usage: RunTelemetryUsageInit,\n): RunTelemetryUsage | undefined {\n  const inputTokens = usage.inputTokens ?? usage.promptTokens;\n  const outputTokens = usage.outputTokens ?? usage.completionTokens;\n  // AI SDK v7 moved these under token detail objects; v6 kept them top-level.\n  const reasoningTokens =\n    usage.reasoningTokens ?? usage.outputTokenDetails?.reasoningTokens;\n  const cachedInputTokens =\n    usage.cachedInputTokens ?? usage.inputTokenDetails?.cacheReadTokens;\n\n  if (\n    inputTokens == null &&\n    outputTokens == null &&\n    reasoningTokens == null &&\n    cachedInputTokens == null\n  ) {\n    return undefined;\n  }\n\n  return {\n    ...(inputTokens != null ? { inputTokens } : undefined),\n    ...(outputTokens != null ? { outputTokens } : undefined),\n    ...(reasoningTokens != null ? { reasoningTokens } : undefined),\n    ...(cachedInputTokens != null ? { cachedInputTokens } : undefined),\n  };\n}\n"],"mappings":";AAGA,MAAM,4BAA4B;AAElC,MAAM,iBAAiB;;;;;AAuBvB,SAAgB,iBACd,OAMA,iBAA6C,aAI7C;CACA,IAAI,MAAM,SAAS,OAAO,EAAE,QAAQ,QAAQ;CAC5C,IAAI,MAAM,SAAS,OAAO;EAAE,QAAQ;EAAc,SAAS;CAAU;CACrE,IAAI,MAAM,cACR,OAAO;EAAE,QAAQ;EAAc,SAAS;CAAe;CAEzD,QAAQ,MAAM,cAAd;EACE,KAAK,UACH,OAAO;GAAE,QAAQ;GAAc,SAAS;EAAS;EACnD,KAAK;EACL,KAAK,kBACH,OAAO;GAAE,QAAQ;GAAc,SAAS;EAAiB;EAC3D,KAAK,aACH,OAAO;GAAE,QAAQ;GAAc,SAAS;EAAU;EACpD,KAAK,SACH,OAAO,EAAE,QAAQ,QAAQ;EAC3B,SACE,OAAO,EACL,QAAQ,MAAM,iBAAiB,KAAA,IAAY,iBAAiB,YAC9D;CACJ;AACF;AAEA,MAAM,4BAA4B;AAClC,MAAM,uBAAuB;;;;;AAM7B,SAAgB,iBAAiB,OAG/B;CACA,IAAI,SAAS,MAAM,OAAO,CAAC;CAC3B,MAAM,SACJ,OAAO,UAAU,WAAY,QAAoC,KAAA;CACnE,MAAM,UACJ,OAAO,UAAU,WACb,QACA,OAAO,QAAQ,YAAY,WACzB,OAAO,UACP,KAAA;CACR,MAAM,OACJ,OAAO,QAAQ,SAAS,WACpB,OAAO,OACP,OAAO,QAAQ,SAAS,YAAY,OAAO,SAAS,UAClD,OAAO,OACP,KAAA;CACR,OAAO;EACL,GAAI,UACA,EAAE,OAAO,QAAQ,MAAM,GAAG,oBAAoB,EAAE,IAChD,KAAA;EACJ,GAAI,OACA,EAAE,WAAW,KAAK,MAAM,GAAG,yBAAyB,EAAE,IACtD,KAAA;CACN;AACF;;;;;AAMA,SAAgB,yBAAyB,OAAuB;CAC9D,IAAI,MAAM,UAAU,2BAA2B,OAAO;CACtD,OAAO,MAAM,MAAM,GAAG,yBAAyB;AACjD;AAEA,SAAS,cAAc,OAAoC;CACzD,IAAI,SAAS,MAAM,OAAO,KAAA;CAC1B,IAAI;EACF,OAAO,yBAAyB,KAAK,UAAU,KAAK,CAAC;CACvD,QAAQ;EACN;CACF;AACF;AAEA,SAAS,mBAAmB,OAAoC;CAC9D,IAAI,SAAS,MAAM,OAAO,KAAA;CAC1B,IAAI;EACF,MAAM,SAAS,OAAO,UAAU,WAAW,KAAK,MAAM,KAAK,IAAI;EAC/D,IAAI,MAAM,QAAQ,MAAM,GAAG;GACzB,MAAM,aAAa,OAAO,KAAK,SAAS;IACtC,IAAI,QAAQ,OAAO,SAAS,YAAY,KAAK,MAExC;UAAA,KAAK,SAAS,WAAW,KAAK,SAAS,YACxC,OAAO,KAAK,SAAS,YACrB,eAAe,KAAK,KAAK,KAAK,MAAM,GAAG,GAAG,CAAC,GAC3C;MACA,MAAM,UAAW,KAAK,KAAK,SAAS,IAAK,IAAI,KAAA,CAAM,QAAQ,CAAC;MAC5D,OAAO;OAAE,GAAG;OAAM,MAAM,IAAI,KAAK,KAAK,IAAI,OAAO;MAAK;KACxD;;IAEF,OAAO;GACT,CAAC;GACD,OAAO,yBAAyB,KAAK,UAAU,UAAU,CAAC;EAC5D;CACF,QAAQ,CAER;CACA,OAAO,cAAc,KAAK;AAC5B;;;;;;AAoBA,SAAgB,2BACd,MACiC;CACjC,MAAM,EAAE,UAAU,YAAY,MAAM,UAAU,QAAQ,eAAe;CACrE,MAAM,OAAwC;EAC5C,WAAW;EACX,cAAc;CAChB;CACA,MAAM,WACJ,YAAY,OAAO,yBAAyB,QAAQ,IAAI,cAAc,IAAI;CAC5E,IAAI,aAAa,KAAA,GAAW,KAAK,YAAY;CAC7C,MAAM,aACJ,eAAe,QAAQ,mBAAmB,MAAM,IAAI,cAAc,MAAM;CAC1E,IAAI,eAAe,KAAA,GAAW,KAAK,cAAc;CACjD,IAAI,YAAY,KAAK,cAAc;CACnC,OAAO;AACT;;;;;;;;AASA,SAAgB,2BACd,UACoB;CACpB,IAAI,CAAC,UAAU,OAAO,KAAA;CACtB,IAAI,OAAO,SAAS,YAAY,UAAU,OAAO,SAAS;CAC1D,MAAM,SAAS,SAAS;CACxB,IAAI,OAAO,QAAQ,YAAY,UAAU,OAAO,OAAO;CAEvD,MAAM,QAAiB,SAAS;CAChC,IAAI,CAAC,MAAM,QAAQ,KAAK,GAAG,OAAO,KAAA;CAClC,KAAK,MAAM,QAAQ,OAAoB;EACrC,IAAI,CAAC,QAAQ,OAAO,SAAS,UAAU;EACvC,MAAM,WAAY,KAAiC;EACnD,IAAI,CAAC,YAAY,OAAO,aAAa,UAAU;EAC/C,MAAM,UAAW,SAAqC;EACtD,IAAI,OAAO,YAAY,UAAU,OAAO;CAC1C;AAEF;AAiDA,SAAS,YACP,QACA,OACM;CACN,IAAI,CAAC,OAAO;CACZ,MAAM,aAAa,2BAA2B,KAAK;CACnD,IAAI,CAAC,YAAY;CACjB,IAAI,WAAW,gBAAgB,KAAA,GAC7B,OAAO,eAAe,WAAW;CAEnC,IAAI,WAAW,iBAAiB,KAAA,GAC9B,OAAO,gBAAgB,WAAW;CAEpC,IAAI,WAAW,oBAAoB,KAAA,GACjC,OAAO,mBAAmB,WAAW;CAEvC,IAAI,WAAW,sBAAsB,KAAA,GACnC,OAAO,sBAAsB,WAAW;AAE5C;AAEA,SAAS,oBACP,MACuD;CACvD,MAAM,OAA8D,CAAC;CACrE,MAAM,QAAQ,KAAK,QAAQ,2BAA2B,KAAK,KAAK,IAAI,KAAA;CACpE,IAAI,OAAO,gBAAgB,KAAA,GAAW,KAAK,eAAe,MAAM;CAChE,IAAI,OAAO,iBAAiB,KAAA,GAC1B,KAAK,gBAAgB,MAAM;CAC7B,IAAI,OAAO,oBAAoB,KAAA,GAC7B,KAAK,mBAAmB,MAAM;CAEhC,IAAI,OAAO,sBAAsB,KAAA,GAC/B,KAAK,sBAAsB,MAAM;CAEnC,IAAI,KAAK,cAAc,KAAA,GAAW,KAAK,aAAa,KAAK;CACzD,IAAI,KAAK,YAAY,KAAA,GAAW,KAAK,WAAW,KAAK;CACrD,IAAI,KAAK,UAAU,KAAA,GAAW,KAAK,SAAS,KAAK;CACjD,IAAI,KAAK,iBAAiB,KAAA,GACxB,KAAK,gBAAgB,KAAK,aAAa,MAAM,GAAG,EAAE;CAEpD,OAAO;AACT;AAEA,SAAS,uBACP,MACsB;CACtB,IAAI,CAAC,MAAM,OAAO,KAAA;CAClB,MAAM,SAAmB,CAAC;CAC1B,MAAM,uBAAO,IAAI,IAAY;CAC7B,KAAK,MAAM,OAAO,MAAM;EACtB,MAAM,aAAa,IAAI,KAAK,CAAC,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,KAAK;EAChD,IAAI,CAAC,cAAc,KAAK,IAAI,UAAU,GAAG;EACzC,KAAK,IAAI,UAAU;EACnB,OAAO,KAAK,UAAU;EACtB,IAAI,OAAO,WAAW,IAAI;CAC5B;CACA,OAAO,OAAO,SAAS,IAAI,SAAS,KAAA;AACtC;AAEA,SAAS,+BACP,OACoB;CACpB,IAAI,UAAU,KAAA,KAAa,CAAC,OAAO,SAAS,KAAK,GAAG,OAAO,KAAA;CAC3D,OAAO,KAAK,IAAI,GAAG,KAAK,MAAM,KAAK,CAAC;AACtC;AAEA,SAAgB,gBAAgB,MAA8C;CAC5E,MAAM,SAAkC;EACtC,WAAW,KAAK;EAChB,QAAQ,KAAK;CACf;CACA,MAAM,UAAU,KAAK,SAAS,YAAY;CAC1C,IAAI,WAAW,iBAAiB,KAAK,OAAO,GAAG,OAAO,WAAW;CACjE,IAAI,KAAK,YAAY,KAAA,GAAW,OAAO,eAAe,KAAK;CAC3D,IAAI,KAAK,cAAc,KAAA,GAAW,OAAO,aAAa,KAAK;CAC3D,IAAI,KAAK,UAAU,KAAA,GAAW,OAAO,QAAQ,KAAK;CAClD,IAAI,KAAK,cAAc,KAAA,GAAW,OAAO,aAAa,KAAK;CAC3D,IAAI,KAAK,YAAY,KAAA,GAAW,OAAO,WAAW,KAAK;CACvD,IAAI,KAAK,aAAa,KAAA,GAAW;EAC/B,OAAO,WAAW,KAAK;EACvB,OAAO,gBAAgB,KAAK;CAC9B;CACA,YAAY,QAAQ,KAAK,KAAK;CAC9B,IAAI,KAAK,UAAU,KAAA,GAAW;EAC5B,OAAO,QAAQ,KAAK,MAAM,IAAI,mBAAmB;EACjD,OAAO,cAAc,KAAK,MAAM;CAClC,OAAO,IAAI,KAAK,eAAe,KAAA,GAC7B,OAAO,cAAc,KAAK;CAE5B,IAAI,KAAK,cAAc,KAAA,GAAW,OAAO,aAAa,KAAK;CAC3D,MAAM,aAAa,+BAA+B,KAAK,UAAU;CACjE,IAAI,eAAe,KAAA,GAAW,OAAO,cAAc;CACnD,MAAM,eAAe,+BAA+B,KAAK,YAAY;CACrE,IAAI,iBAAiB,KAAA,GAAW,OAAO,iBAAiB;CACxD,IAAI,KAAK,eAAe,KAAA,GACtB,OAAO,cAAc,yBAAyB,KAAK,UAAU;CAE/D,IAAI,KAAK,aAAa,KAAA,GAAW,OAAO,WAAW,KAAK;CACxD,IAAI,KAAK,WAAW,gBAAgB,KAAA,GAClC,OAAO,cAAc,KAAK,UAAU;CAEtC,IAAI,KAAK,WAAW,YAAY,KAAA,GAC9B,OAAO,UAAU,KAAK,UAAU;CAElC,MAAM,OAAO,uBAAuB,KAAK,WAAW,IAAI;CACxD,IAAI,SAAS,KAAA,GAAW,OAAO,OAAO;CACtC,OAAO;AACT;;;;;;;AAQA,SAAgB,2BACd,OAC+B;CAC/B,MAAM,cAAc,MAAM,eAAe,MAAM;CAC/C,MAAM,eAAe,MAAM,gBAAgB,MAAM;CAEjD,MAAM,kBACJ,MAAM,mBAAmB,MAAM,oBAAoB;CACrD,MAAM,oBACJ,MAAM,qBAAqB,MAAM,mBAAmB;CAEtD,IACE,eAAe,QACf,gBAAgB,QAChB,mBAAmB,QACnB,qBAAqB,MAErB;CAGF,OAAO;EACL,GAAI,eAAe,OAAO,EAAE,YAAY,IAAI,KAAA;EAC5C,GAAI,gBAAgB,OAAO,EAAE,aAAa,IAAI,KAAA;EAC9C,GAAI,mBAAmB,OAAO,EAAE,gBAAgB,IAAI,KAAA;EACpD,GAAI,qBAAqB,OAAO,EAAE,kBAAkB,IAAI,KAAA;CAC1D;AACF"}
+{"version":3,"file":"runTelemetry.js","names":[],"sources":["../src/runTelemetry.ts"],"sourcesContent":["import type { SamplingCallData } from \"./instrumentMcpSampling\";\nimport type { AssistantCloudRunReport } from \"./AssistantCloudRuns\";\n\nconst MAX_TELEMETRY_TEXT_LENGTH = 50_000;\n\nconst BASE64_PATTERN = /^[A-Za-z0-9+/]{100,}={0,2}$/;\n\nexport type AssistantCloudRunReportToolCall = {\n  tool_name: string;\n  tool_call_id: string;\n  tool_args?: string;\n  tool_result?: string;\n  tool_source?: \"mcp\" | \"frontend\" | \"backend\";\n  start_ms?: number;\n  end_ms?: number;\n  sampling_calls?: SamplingCallData[];\n};\n\nexport type RunReportOutcome =\n  | \"aborted\"\n  | \"disconnected\"\n  | \"length\"\n  | \"content_filter\";\n\n/**\n * Maps a finish event to the report status and outcome. `fallbackStatus`\n * applies when the event carries neither a finish reason nor a failure flag.\n */\nexport function deriveRunOutcome(\n  input: {\n    finishReason?: string | undefined;\n    isAbort?: boolean | undefined;\n    isDisconnect?: boolean | undefined;\n    isError?: boolean | undefined;\n  },\n  fallbackStatus: \"completed\" | \"incomplete\" = \"completed\",\n): {\n  status: \"completed\" | \"incomplete\" | \"error\";\n  outcome?: RunReportOutcome;\n} {\n  if (input.isError) return { status: \"error\" };\n  if (input.isAbort) return { status: \"incomplete\", outcome: \"aborted\" };\n  if (input.isDisconnect) {\n    return { status: \"incomplete\", outcome: \"disconnected\" };\n  }\n  switch (input.finishReason) {\n    case \"length\":\n      return { status: \"incomplete\", outcome: \"length\" };\n    case \"content-filter\":\n    case \"content_filter\":\n      return { status: \"incomplete\", outcome: \"content_filter\" };\n    case \"cancelled\":\n      return { status: \"incomplete\", outcome: \"aborted\" };\n    case \"error\":\n      return { status: \"error\" };\n    default:\n      return {\n        status: input.finishReason === undefined ? fallbackStatus : \"completed\",\n      };\n  }\n}\n\nconst MAX_RUN_ERROR_CODE_LENGTH = 64;\nconst MAX_RUN_ERROR_LENGTH = 2048;\n\n/**\n * Reads the message and code the runs endpoint stores for a failed run. The\n * code is the error's `code` when it has one, else its class name.\n */\nexport function describeRunError(error: unknown): {\n  error?: string;\n  errorCode?: string;\n} {\n  if (error == null) return {};\n  const record =\n    typeof error === \"object\" ? (error as Record<string, unknown>) : undefined;\n  const message =\n    typeof error === \"string\"\n      ? error\n      : typeof record?.message === \"string\"\n        ? record.message\n        : undefined;\n  const code =\n    typeof record?.code === \"string\"\n      ? record.code\n      : typeof record?.name === \"string\" && record.name !== \"Error\"\n        ? record.name\n        : undefined;\n  return {\n    ...(message\n      ? { error: message.slice(0, MAX_RUN_ERROR_LENGTH) }\n      : undefined),\n    ...(code\n      ? { errorCode: code.slice(0, MAX_RUN_ERROR_CODE_LENGTH) }\n      : undefined),\n  };\n}\n\n/**\n * Clamps a string to the size the runs endpoint accepts for a single span\n * field.\n */\nexport function truncateRunTelemetryText(value: string): string {\n  if (value.length <= MAX_TELEMETRY_TEXT_LENGTH) return value;\n  return value.slice(0, MAX_TELEMETRY_TEXT_LENGTH);\n}\n\nfunction safeStringify(value: unknown): string | undefined {\n  if (value == null) return undefined;\n  try {\n    return truncateRunTelemetryText(JSON.stringify(value));\n  } catch {\n    return undefined;\n  }\n}\n\nfunction summarizeMcpResult(value: unknown): string | undefined {\n  if (value == null) return undefined;\n  try {\n    const parsed = typeof value === \"string\" ? JSON.parse(value) : value;\n    if (Array.isArray(parsed)) {\n      const summarized = parsed.map((item) => {\n        if (item && typeof item === \"object\" && item.type) {\n          if (\n            (item.type === \"image\" || item.type === \"audio\") &&\n            typeof item.data === \"string\" &&\n            BASE64_PATTERN.test(item.data.slice(0, 200))\n          ) {\n            const sizeKB = ((item.data.length * 3) / 4 / 1024).toFixed(1);\n            return { ...item, data: `[${item.type}: ${sizeKB}KB]` };\n          }\n        }\n        return item;\n      });\n      return truncateRunTelemetryText(JSON.stringify(summarized));\n    }\n  } catch {\n    // not JSON array, fall through\n  }\n  return safeStringify(value);\n}\n\nexport type RunTelemetryToolCallInit = {\n  toolName: string;\n  toolCallId: string;\n  args?: unknown;\n  /**\n   * Pre-serialized arguments, used in place of serializing `args`. Values over\n   * the span size are clamped before they are included in the report.\n   */\n  argsText?: string | undefined;\n  result?: unknown;\n  toolSource?: \"mcp\" | \"frontend\" | \"backend\" | undefined;\n};\n\n/**\n * Serializes one tool call into the shape the runs endpoint accepts. An `mcp`\n * source has its result summarized, because MCP content blocks carry inline\n * base64 image and audio payloads that would otherwise dominate the report.\n */\nexport function createRunTelemetryToolCall(\n  init: RunTelemetryToolCallInit,\n): AssistantCloudRunReportToolCall {\n  const { toolName, toolCallId, args, argsText, result, toolSource } = init;\n  const call: AssistantCloudRunReportToolCall = {\n    tool_name: toolName,\n    tool_call_id: toolCallId,\n  };\n  const toolArgs =\n    argsText != null ? truncateRunTelemetryText(argsText) : safeStringify(args);\n  if (toolArgs !== undefined) call.tool_args = toolArgs;\n  const toolResult =\n    toolSource === \"mcp\" ? summarizeMcpResult(result) : safeStringify(result);\n  if (toolResult !== undefined) call.tool_result = toolResult;\n  if (toolSource) call.tool_source = toolSource;\n  return call;\n}\n\n/**\n * Resolves the model ID a run reports, in the order an app can supply it: an\n * explicit `modelId`, the `custom` bag, then the per-step `response.modelId`\n * that a `messageMetadata` callback copies off the AI SDK's finish-step part.\n * The AI SDK puts no model ID on a UI message part, so message metadata is the\n * only channel one arrives on.\n */\nexport function extractRunTelemetryModelId(\n  metadata: Record<string, unknown> | undefined,\n): string | undefined {\n  if (!metadata) return undefined;\n  if (typeof metadata.modelId === \"string\") return metadata.modelId;\n  const custom = metadata.custom as Record<string, unknown> | undefined;\n  if (typeof custom?.modelId === \"string\") return custom.modelId;\n\n  const steps: unknown = metadata.steps;\n  if (!Array.isArray(steps)) return undefined;\n  for (const step of steps as unknown[]) {\n    if (!step || typeof step !== \"object\") continue;\n    const response = (step as Record<string, unknown>).response;\n    if (!response || typeof response !== \"object\") continue;\n    const modelId = (response as Record<string, unknown>).modelId;\n    if (typeof modelId === \"string\") return modelId;\n  }\n  return undefined;\n}\n\nexport type RunTelemetryUsage = {\n  inputTokens?: number | undefined;\n  outputTokens?: number | undefined;\n  reasoningTokens?: number | undefined;\n  cachedInputTokens?: number | undefined;\n};\n\nexport type RunTelemetryUsageInit = RunTelemetryUsage & {\n  promptTokens?: number | undefined;\n  completionTokens?: number | undefined;\n  inputTokenDetails?: { cacheReadTokens?: number };\n  outputTokenDetails?: { reasoningTokens?: number };\n};\n\nexport type RunReportStepInit = {\n  usage?: RunTelemetryUsageInit | undefined;\n  toolCalls?: AssistantCloudRunReportToolCall[] | undefined;\n  startMs?: number | undefined;\n  endMs?: number | undefined;\n  finishReason?: string | undefined;\n};\n\n/**\n * The run report fields read from the messages of one run, in whichever\n * format they were stored: the status the messages imply, the tool calls, the\n * steps, the text, the usage and the model.\n */\nexport type RunMessageTelemetry = {\n  assistantMessageId?: string;\n  status: \"completed\" | \"incomplete\";\n  toolCalls?: AssistantCloudRunReportToolCall[];\n  steps?: RunReportStepInit[];\n  totalSteps?: number;\n  outputText?: string;\n  usage?: RunTelemetryUsage;\n  modelId?: string;\n  metadata?: Record<string, unknown>;\n};\n\nexport type RunReportInit = {\n  threadId: string;\n  status: AssistantCloudRunReport[\"status\"];\n  outcome?: RunReportOutcome | undefined;\n  errorCode?: string | undefined;\n  error?: string | undefined;\n  messageId?: string | undefined;\n  traceId?: string | undefined;\n  modelId?: string | undefined;\n  provider?: string | undefined;\n  usage?: RunTelemetryUsageInit | undefined;\n  steps?: RunReportStepInit[] | undefined;\n  totalSteps?: number | undefined;\n  toolCalls?: AssistantCloudRunReportToolCall[] | undefined;\n  durationMs?: number | undefined;\n  firstTokenMs?: number | undefined;\n  outputText?: string | undefined;\n  metadata?: Record<string, unknown> | undefined;\n  telemetry?: {\n    environment?: string | undefined;\n    release?: string | undefined;\n    tags?: readonly string[] | undefined;\n  };\n};\n\nfunction assignUsage(\n  report: AssistantCloudRunReport,\n  usage: RunTelemetryUsageInit | undefined,\n): void {\n  if (!usage) return;\n  const normalized = normalizeRunTelemetryUsage(usage);\n  if (!normalized) return;\n  if (normalized.inputTokens !== undefined) {\n    report.input_tokens = normalized.inputTokens;\n  }\n  if (normalized.outputTokens !== undefined) {\n    report.output_tokens = normalized.outputTokens;\n  }\n  if (normalized.reasoningTokens !== undefined) {\n    report.reasoning_tokens = normalized.reasoningTokens;\n  }\n  if (normalized.cachedInputTokens !== undefined) {\n    report.cached_input_tokens = normalized.cachedInputTokens;\n  }\n}\n\nfunction createRunReportStep(\n  init: RunReportStepInit,\n): NonNullable<AssistantCloudRunReport[\"steps\"]>[number] {\n  const step: NonNullable<AssistantCloudRunReport[\"steps\"]>[number] = {};\n  const usage = init.usage ? normalizeRunTelemetryUsage(init.usage) : undefined;\n  if (usage?.inputTokens !== undefined) step.input_tokens = usage.inputTokens;\n  if (usage?.outputTokens !== undefined)\n    step.output_tokens = usage.outputTokens;\n  if (usage?.reasoningTokens !== undefined) {\n    step.reasoning_tokens = usage.reasoningTokens;\n  }\n  if (usage?.cachedInputTokens !== undefined) {\n    step.cached_input_tokens = usage.cachedInputTokens;\n  }\n  if (init.toolCalls !== undefined) step.tool_calls = init.toolCalls;\n  if (init.startMs !== undefined) step.start_ms = init.startMs;\n  if (init.endMs !== undefined) step.end_ms = init.endMs;\n  if (init.finishReason !== undefined) {\n    step.finish_reason = init.finishReason.slice(0, 32);\n  }\n  return step;\n}\n\nfunction normalizeRunReportTags(\n  tags: readonly string[] | undefined,\n): string[] | undefined {\n  if (!tags) return undefined;\n  const result: string[] = [];\n  const seen = new Set<string>();\n  for (const tag of tags) {\n    const normalized = tag.trim().slice(0, 64).trim();\n    if (!normalized || seen.has(normalized)) continue;\n    seen.add(normalized);\n    result.push(normalized);\n    if (result.length === 20) break;\n  }\n  return result.length > 0 ? result : undefined;\n}\n\nfunction normalizeRunReportMilliseconds(\n  value: number | undefined,\n): number | undefined {\n  if (value === undefined || !Number.isFinite(value)) return undefined;\n  return Math.max(0, Math.round(value));\n}\n\nexport function createRunReport(init: RunReportInit): AssistantCloudRunReport {\n  const report: AssistantCloudRunReport = {\n    thread_id: init.threadId,\n    status: init.status,\n  };\n  const traceId = init.traceId?.toLowerCase();\n  if (traceId && /^[0-9a-f]{32}$/.test(traceId)) report.trace_id = traceId;\n  if (init.outcome !== undefined) report.outcome_type = init.outcome;\n  if (init.errorCode !== undefined) report.error_code = init.errorCode;\n  if (init.error !== undefined) report.error = init.error;\n  if (init.messageId !== undefined) report.message_id = init.messageId;\n  if (init.modelId !== undefined) report.model_id = init.modelId;\n  if (init.provider !== undefined) {\n    report.provider = init.provider;\n    report.provider_type = init.provider;\n  }\n  assignUsage(report, init.usage);\n  if (init.steps !== undefined) {\n    report.steps = init.steps.map(createRunReportStep);\n    report.total_steps = init.steps.length;\n  } else if (init.totalSteps !== undefined) {\n    report.total_steps = init.totalSteps;\n  }\n  if (init.toolCalls !== undefined) report.tool_calls = init.toolCalls;\n  const durationMs = normalizeRunReportMilliseconds(init.durationMs);\n  if (durationMs !== undefined) report.duration_ms = durationMs;\n  const firstTokenMs = normalizeRunReportMilliseconds(init.firstTokenMs);\n  if (firstTokenMs !== undefined) report.first_token_ms = firstTokenMs;\n  if (init.outputText !== undefined) {\n    report.output_text = truncateRunTelemetryText(init.outputText);\n  }\n  if (init.metadata !== undefined) report.metadata = init.metadata;\n  if (init.telemetry?.environment !== undefined) {\n    report.environment = init.telemetry.environment;\n  }\n  if (init.telemetry?.release !== undefined) {\n    report.release = init.telemetry.release;\n  }\n  const tags = normalizeRunReportTags(init.telemetry?.tags);\n  if (tags !== undefined) report.tags = tags;\n  return report;\n}\n\n/**\n * Resolves the token counts a provider reports under any of the names the AI\n * SDK has used: the current top-level ones, the legacy prompt/completion pair,\n * and the v7 token detail objects. Returns undefined when no count is present,\n * so callers can tell an empty usage object from a zeroed one.\n */\nexport function normalizeRunTelemetryUsage(\n  usage: RunTelemetryUsageInit,\n): RunTelemetryUsage | undefined {\n  const inputTokens = usage.inputTokens ?? usage.promptTokens;\n  const outputTokens = usage.outputTokens ?? usage.completionTokens;\n  // AI SDK v7 moved these under token detail objects; v6 kept them top-level.\n  const reasoningTokens =\n    usage.reasoningTokens ?? usage.outputTokenDetails?.reasoningTokens;\n  const cachedInputTokens =\n    usage.cachedInputTokens ?? usage.inputTokenDetails?.cacheReadTokens;\n\n  if (\n    inputTokens == null &&\n    outputTokens == null &&\n    reasoningTokens == null &&\n    cachedInputTokens == null\n  ) {\n    return undefined;\n  }\n\n  return {\n    ...(inputTokens != null ? { inputTokens } : undefined),\n    ...(outputTokens != null ? { outputTokens } : undefined),\n    ...(reasoningTokens != null ? { reasoningTokens } : undefined),\n    ...(cachedInputTokens != null ? { cachedInputTokens } : undefined),\n  };\n}\n"],"mappings":";AAGA,MAAM,4BAA4B;AAElC,MAAM,iBAAiB;;;;;AAuBvB,SAAgB,iBACd,OAMA,iBAA6C,aAI7C;CACA,IAAI,MAAM,SAAS,OAAO,EAAE,QAAQ,QAAQ;CAC5C,IAAI,MAAM,SAAS,OAAO;EAAE,QAAQ;EAAc,SAAS;CAAU;CACrE,IAAI,MAAM,cACR,OAAO;EAAE,QAAQ;EAAc,SAAS;CAAe;CAEzD,QAAQ,MAAM,cAAd;EACE,KAAK,UACH,OAAO;GAAE,QAAQ;GAAc,SAAS;EAAS;EACnD,KAAK;EACL,KAAK,kBACH,OAAO;GAAE,QAAQ;GAAc,SAAS;EAAiB;EAC3D,KAAK,aACH,OAAO;GAAE,QAAQ;GAAc,SAAS;EAAU;EACpD,KAAK,SACH,OAAO,EAAE,QAAQ,QAAQ;EAC3B,SACE,OAAO,EACL,QAAQ,MAAM,iBAAiB,KAAA,IAAY,iBAAiB,YAC9D;CACJ;AACF;AAEA,MAAM,4BAA4B;AAClC,MAAM,uBAAuB;;;;;AAM7B,SAAgB,iBAAiB,OAG/B;CACA,IAAI,SAAS,MAAM,OAAO,CAAC;CAC3B,MAAM,SACJ,OAAO,UAAU,WAAY,QAAoC,KAAA;CACnE,MAAM,UACJ,OAAO,UAAU,WACb,QACA,OAAO,QAAQ,YAAY,WACzB,OAAO,UACP,KAAA;CACR,MAAM,OACJ,OAAO,QAAQ,SAAS,WACpB,OAAO,OACP,OAAO,QAAQ,SAAS,YAAY,OAAO,SAAS,UAClD,OAAO,OACP,KAAA;CACR,OAAO;EACL,GAAI,UACA,EAAE,OAAO,QAAQ,MAAM,GAAG,oBAAoB,EAAE,IAChD,KAAA;EACJ,GAAI,OACA,EAAE,WAAW,KAAK,MAAM,GAAG,yBAAyB,EAAE,IACtD,KAAA;CACN;AACF;;;;;AAMA,SAAgB,yBAAyB,OAAuB;CAC9D,IAAI,MAAM,UAAU,2BAA2B,OAAO;CACtD,OAAO,MAAM,MAAM,GAAG,yBAAyB;AACjD;AAEA,SAAS,cAAc,OAAoC;CACzD,IAAI,SAAS,MAAM,OAAO,KAAA;CAC1B,IAAI;EACF,OAAO,yBAAyB,KAAK,UAAU,KAAK,CAAC;CACvD,QAAQ;EACN;CACF;AACF;AAEA,SAAS,mBAAmB,OAAoC;CAC9D,IAAI,SAAS,MAAM,OAAO,KAAA;CAC1B,IAAI;EACF,MAAM,SAAS,OAAO,UAAU,WAAW,KAAK,MAAM,KAAK,IAAI;EAC/D,IAAI,MAAM,QAAQ,MAAM,GAAG;GACzB,MAAM,aAAa,OAAO,KAAK,SAAS;IACtC,IAAI,QAAQ,OAAO,SAAS,YAAY,KAAK,MAExC;UAAA,KAAK,SAAS,WAAW,KAAK,SAAS,YACxC,OAAO,KAAK,SAAS,YACrB,eAAe,KAAK,KAAK,KAAK,MAAM,GAAG,GAAG,CAAC,GAC3C;MACA,MAAM,UAAW,KAAK,KAAK,SAAS,IAAK,IAAI,KAAA,CAAM,QAAQ,CAAC;MAC5D,OAAO;OAAE,GAAG;OAAM,MAAM,IAAI,KAAK,KAAK,IAAI,OAAO;MAAK;KACxD;;IAEF,OAAO;GACT,CAAC;GACD,OAAO,yBAAyB,KAAK,UAAU,UAAU,CAAC;EAC5D;CACF,QAAQ,CAER;CACA,OAAO,cAAc,KAAK;AAC5B;;;;;;AAoBA,SAAgB,2BACd,MACiC;CACjC,MAAM,EAAE,UAAU,YAAY,MAAM,UAAU,QAAQ,eAAe;CACrE,MAAM,OAAwC;EAC5C,WAAW;EACX,cAAc;CAChB;CACA,MAAM,WACJ,YAAY,OAAO,yBAAyB,QAAQ,IAAI,cAAc,IAAI;CAC5E,IAAI,aAAa,KAAA,GAAW,KAAK,YAAY;CAC7C,MAAM,aACJ,eAAe,QAAQ,mBAAmB,MAAM,IAAI,cAAc,MAAM;CAC1E,IAAI,eAAe,KAAA,GAAW,KAAK,cAAc;CACjD,IAAI,YAAY,KAAK,cAAc;CACnC,OAAO;AACT;;;;;;;;AASA,SAAgB,2BACd,UACoB;CACpB,IAAI,CAAC,UAAU,OAAO,KAAA;CACtB,IAAI,OAAO,SAAS,YAAY,UAAU,OAAO,SAAS;CAC1D,MAAM,SAAS,SAAS;CACxB,IAAI,OAAO,QAAQ,YAAY,UAAU,OAAO,OAAO;CAEvD,MAAM,QAAiB,SAAS;CAChC,IAAI,CAAC,MAAM,QAAQ,KAAK,GAAG,OAAO,KAAA;CAClC,KAAK,MAAM,QAAQ,OAAoB;EACrC,IAAI,CAAC,QAAQ,OAAO,SAAS,UAAU;EACvC,MAAM,WAAY,KAAiC;EACnD,IAAI,CAAC,YAAY,OAAO,aAAa,UAAU;EAC/C,MAAM,UAAW,SAAqC;EACtD,IAAI,OAAO,YAAY,UAAU,OAAO;CAC1C;AAEF;AAkEA,SAAS,YACP,QACA,OACM;CACN,IAAI,CAAC,OAAO;CACZ,MAAM,aAAa,2BAA2B,KAAK;CACnD,IAAI,CAAC,YAAY;CACjB,IAAI,WAAW,gBAAgB,KAAA,GAC7B,OAAO,eAAe,WAAW;CAEnC,IAAI,WAAW,iBAAiB,KAAA,GAC9B,OAAO,gBAAgB,WAAW;CAEpC,IAAI,WAAW,oBAAoB,KAAA,GACjC,OAAO,mBAAmB,WAAW;CAEvC,IAAI,WAAW,sBAAsB,KAAA,GACnC,OAAO,sBAAsB,WAAW;AAE5C;AAEA,SAAS,oBACP,MACuD;CACvD,MAAM,OAA8D,CAAC;CACrE,MAAM,QAAQ,KAAK,QAAQ,2BAA2B,KAAK,KAAK,IAAI,KAAA;CACpE,IAAI,OAAO,gBAAgB,KAAA,GAAW,KAAK,eAAe,MAAM;CAChE,IAAI,OAAO,iBAAiB,KAAA,GAC1B,KAAK,gBAAgB,MAAM;CAC7B,IAAI,OAAO,oBAAoB,KAAA,GAC7B,KAAK,mBAAmB,MAAM;CAEhC,IAAI,OAAO,sBAAsB,KAAA,GAC/B,KAAK,sBAAsB,MAAM;CAEnC,IAAI,KAAK,cAAc,KAAA,GAAW,KAAK,aAAa,KAAK;CACzD,IAAI,KAAK,YAAY,KAAA,GAAW,KAAK,WAAW,KAAK;CACrD,IAAI,KAAK,UAAU,KAAA,GAAW,KAAK,SAAS,KAAK;CACjD,IAAI,KAAK,iBAAiB,KAAA,GACxB,KAAK,gBAAgB,KAAK,aAAa,MAAM,GAAG,EAAE;CAEpD,OAAO;AACT;AAEA,SAAS,uBACP,MACsB;CACtB,IAAI,CAAC,MAAM,OAAO,KAAA;CAClB,MAAM,SAAmB,CAAC;CAC1B,MAAM,uBAAO,IAAI,IAAY;CAC7B,KAAK,MAAM,OAAO,MAAM;EACtB,MAAM,aAAa,IAAI,KAAK,CAAC,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,KAAK;EAChD,IAAI,CAAC,cAAc,KAAK,IAAI,UAAU,GAAG;EACzC,KAAK,IAAI,UAAU;EACnB,OAAO,KAAK,UAAU;EACtB,IAAI,OAAO,WAAW,IAAI;CAC5B;CACA,OAAO,OAAO,SAAS,IAAI,SAAS,KAAA;AACtC;AAEA,SAAS,+BACP,OACoB;CACpB,IAAI,UAAU,KAAA,KAAa,CAAC,OAAO,SAAS,KAAK,GAAG,OAAO,KAAA;CAC3D,OAAO,KAAK,IAAI,GAAG,KAAK,MAAM,KAAK,CAAC;AACtC;AAEA,SAAgB,gBAAgB,MAA8C;CAC5E,MAAM,SAAkC;EACtC,WAAW,KAAK;EAChB,QAAQ,KAAK;CACf;CACA,MAAM,UAAU,KAAK,SAAS,YAAY;CAC1C,IAAI,WAAW,iBAAiB,KAAK,OAAO,GAAG,OAAO,WAAW;CACjE,IAAI,KAAK,YAAY,KAAA,GAAW,OAAO,eAAe,KAAK;CAC3D,IAAI,KAAK,cAAc,KAAA,GAAW,OAAO,aAAa,KAAK;CAC3D,IAAI,KAAK,UAAU,KAAA,GAAW,OAAO,QAAQ,KAAK;CAClD,IAAI,KAAK,cAAc,KAAA,GAAW,OAAO,aAAa,KAAK;CAC3D,IAAI,KAAK,YAAY,KAAA,GAAW,OAAO,WAAW,KAAK;CACvD,IAAI,KAAK,aAAa,KAAA,GAAW;EAC/B,OAAO,WAAW,KAAK;EACvB,OAAO,gBAAgB,KAAK;CAC9B;CACA,YAAY,QAAQ,KAAK,KAAK;CAC9B,IAAI,KAAK,UAAU,KAAA,GAAW;EAC5B,OAAO,QAAQ,KAAK,MAAM,IAAI,mBAAmB;EACjD,OAAO,cAAc,KAAK,MAAM;CAClC,OAAO,IAAI,KAAK,eAAe,KAAA,GAC7B,OAAO,cAAc,KAAK;CAE5B,IAAI,KAAK,cAAc,KAAA,GAAW,OAAO,aAAa,KAAK;CAC3D,MAAM,aAAa,+BAA+B,KAAK,UAAU;CACjE,IAAI,eAAe,KAAA,GAAW,OAAO,cAAc;CACnD,MAAM,eAAe,+BAA+B,KAAK,YAAY;CACrE,IAAI,iBAAiB,KAAA,GAAW,OAAO,iBAAiB;CACxD,IAAI,KAAK,eAAe,KAAA,GACtB,OAAO,cAAc,yBAAyB,KAAK,UAAU;CAE/D,IAAI,KAAK,aAAa,KAAA,GAAW,OAAO,WAAW,KAAK;CACxD,IAAI,KAAK,WAAW,gBAAgB,KAAA,GAClC,OAAO,cAAc,KAAK,UAAU;CAEtC,IAAI,KAAK,WAAW,YAAY,KAAA,GAC9B,OAAO,UAAU,KAAK,UAAU;CAElC,MAAM,OAAO,uBAAuB,KAAK,WAAW,IAAI;CACxD,IAAI,SAAS,KAAA,GAAW,OAAO,OAAO;CACtC,OAAO;AACT;;;;;;;AAQA,SAAgB,2BACd,OAC+B;CAC/B,MAAM,cAAc,MAAM,eAAe,MAAM;CAC/C,MAAM,eAAe,MAAM,gBAAgB,MAAM;CAEjD,MAAM,kBACJ,MAAM,mBAAmB,MAAM,oBAAoB;CACrD,MAAM,oBACJ,MAAM,qBAAqB,MAAM,mBAAmB;CAEtD,IACE,eAAe,QACf,gBAAgB,QAChB,mBAAmB,QACnB,qBAAqB,MAErB;CAGF,OAAO;EACL,GAAI,eAAe,OAAO,EAAE,YAAY,IAAI,KAAA;EAC5C,GAAI,gBAAgB,OAAO,EAAE,aAAa,IAAI,KAAA;EAC9C,GAAI,mBAAmB,OAAO,EAAE,gBAAgB,IAAI,KAAA;EACpD,GAAI,qBAAqB,OAAO,EAAE,kBAAkB,IAAI,KAAA;CAC1D;AACF"}

package/dist/telemetry/index.d.ts

@@ -1,24 +1,24 @@
 import { ReadableSpan, SpanExporter, SpanProcessor } from "@opentelemetry/sdk-trace-base";
 //#region src/telemetry/index.d.ts
-type AssistantCloudTraceExportOptions = {
+export type AssistantCloudTraceExportOptions = {
   apiKey: string;
   baseUrl?: string;
   headers?: Record<string, string>;
 };
-type AssistantCloudSpanProcessorOptions = {
+export type AssistantCloudSpanProcessorOptions = {
   filter?: (span: ReadableSpan) => boolean;
 };
-declare function assistantCloudTraceExportOptions({ apiKey, baseUrl, headers }: AssistantCloudTraceExportOptions): {
+export declare function assistantCloudTraceExportOptions({ apiKey, baseUrl, headers }: AssistantCloudTraceExportOptions): {
   url: string;
   headers: Record<string, string>;
 };
-declare function createAssistantCloudTraceExporter(options: AssistantCloudTraceExportOptions): SpanExporter;
-declare function isAssistantCloudSpan(span: ReadableSpan): boolean;
-declare function createAssistantCloudSpanProcessor(exporter: SpanExporter, options?: AssistantCloudSpanProcessorOptions): SpanProcessor;
-declare function assistantCloudTraceMetadata(): {
+export declare function createAssistantCloudTraceExporter(options: AssistantCloudTraceExportOptions): SpanExporter;
+export declare function isAssistantCloudSpan(span: ReadableSpan): boolean;
+export declare function createAssistantCloudSpanProcessor(exporter: SpanExporter, options?: AssistantCloudSpanProcessorOptions): SpanProcessor;
+export declare function assistantCloudTraceMetadata(): {
   traceId?: string;
 };
-declare function withAssistantCloudTraceMetadata<Part extends {
+export declare function withAssistantCloudTraceMetadata<Part extends {
   type: string;
 } = {
   type: string;
@@ -28,5 +28,4 @@ declare function withAssistantCloudTraceMetadata<Part extends {
   part: Part;
 }) => Record<string, unknown> | undefined;
 //#endregion
-export { AssistantCloudSpanProcessorOptions, AssistantCloudTraceExportOptions, assistantCloudTraceExportOptions, assistantCloudTraceMetadata, createAssistantCloudSpanProcessor, createAssistantCloudTraceExporter, isAssistantCloudSpan, withAssistantCloudTraceMetadata };
 //# sourceMappingURL=index.d.ts.map

package/dist/telemetry/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","names":[],"sources":["../../src/telemetry/index.ts"],"mappings":";;KAYY;EACV;EACA;EACA,UAAU;;KAGA;EACV,UAAU,MAAM;;iBAGF,mCACd,QACA,SACA,WACC;EACD;EACA,SAAS;;iBAmBK,kCACd,SAAS,mCACR;iBAIa,qBAAqB,MAAM;iBAO3B,kCACd,UAAU,cACV,UAAS,qCACR;iBAca;EAAiC;;iBAOjC,gCACd;EAAe;;EAAmB;GAElC,mBAAmB;EACjB,MAAM;MACF,uCACJ;EAAW,MAAM;MAAW"}
+{"version":3,"file":"index.d.ts","names":[],"sources":["../../src/telemetry/index.ts"],"mappings":";;YAYY;EACV;EACA;EACA,UAAU;;YAGA;EACV,UAAU,MAAM;;wBAGF,mCACd,QACA,SACA,WACC;EACD;EACA,SAAS;;wBAmBK,kCACd,SAAS,mCACR;wBAIa,qBAAqB,MAAM;wBAO3B,kCACd,UAAU,cACV,UAAS,qCACR;wBAca;EAAiC;;wBAOjC,gCACd;EAAe;;EAAmB;GAElC,mBAAmB;EACjB,MAAM;MACF,uCACJ;EAAW,MAAM;MAAW"}

package/dist/version.d.ts

@@ -0,0 +1,4 @@
+//#region src/version.d.ts
+export declare const ASSISTANT_CLOUD_VERSION: string;
+//#endregion
+//# sourceMappingURL=version.d.ts.map

package/dist/version.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"version.d.ts","names":[],"sources":["../src/version.ts"],"mappings":";qBAAa"}

package/dist/version.js

@@ -0,0 +1,6 @@
+//#region src/version.ts
+const ASSISTANT_CLOUD_VERSION = "0.2.2";
+//#endregion
+export { ASSISTANT_CLOUD_VERSION };
+
+//# sourceMappingURL=version.js.map

package/dist/version.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"version.js","names":[],"sources":["../src/version.ts"],"sourcesContent":["export const ASSISTANT_CLOUD_VERSION: string =\n  typeof __AUI_PACKAGE_VERSION__ === \"string\"\n    ? __AUI_PACKAGE_VERSION__\n    : \"0.0.0\";\n"],"mappings":";AAAA,MAAa,0BAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "assistant-cloud",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "description": "Cloud integration for assistant-ui",
   "keywords": [
     "assistant",
@@ -20,6 +20,10 @@
     "./telemetry": {
       "types": "./dist/telemetry/index.d.ts",
       "default": "./dist/telemetry/index.js"
+    },
+    "./ai-sdk": {
+      "types": "./dist/ai-sdk/index.d.ts",
+      "default": "./dist/ai-sdk/index.js"
     }
   },
   "main": "./dist/index.js",
@@ -31,12 +35,13 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "assistant-stream": "^0.3.42"
+    "assistant-stream": "^0.3.44"
   },
   "peerDependencies": {
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/exporter-trace-otlp-http": ">=0.200.0",
-    "@opentelemetry/sdk-trace-base": "^2.1.0"
+    "@opentelemetry/sdk-trace-base": "^2.1.0",
+    "ai": "^6.0.0 || ^7.0.0"
   },
   "peerDependenciesMeta": {
     "@opentelemetry/api": {
@@ -47,16 +52,21 @@
     },
     "@opentelemetry/sdk-trace-base": {
       "optional": true
+    },
+    "ai": {
+      "optional": true
     }
   },
   "devDependencies": {
-    "@assistant-ui/x-buildutils": "0.0.26",
-    "@opentelemetry/api": "^1.9.0",
-    "@opentelemetry/exporter-trace-otlp-http": "^0.205.0",
-    "@opentelemetry/sdk-trace-base": "^2.1.0",
-    "@types/node": "^26.4.1",
+    "@assistant-ui/x-buildutils": "0.0.28",
+    "@opentelemetry/api": "^1.9.1",
+    "@opentelemetry/exporter-trace-otlp-http": "^0.222.0",
+    "@opentelemetry/sdk-trace-base": "^2.11.0",
+    "@types/node": "^26.5.1",
+    "ai": "^7.0.101",
+    "ai-v6": "npm:ai@^6.0.277",
     "jsdom": "^30.0.1",
-    "vitest": "^4.1.11"
+    "vitest": "^5.0.1"
   },
   "publishConfig": {
     "access": "public",
@@ -73,7 +83,11 @@
   },
   "scripts": {
     "build": "aui-build",
-    "test": "vitest run",
+    "test": "pnpm run test:v7 && pnpm run test:peer-v6 && pnpm run test:types:peer-v6",
+    "typecheck": "tsc --noEmit",
+    "test:v7": "vitest run",
+    "test:peer-v6": "vitest run --config vitest.peer-v6.config.ts",
+    "test:types:peer-v6": "tsc --noEmit --project tsconfig.peer-v6.json",
     "test:watch": "vitest"
   }
 }

package/src/AssistantCloud.ts

@@ -1,6 +1,7 @@
 import {
   AssistantCloudAPI,
   type AssistantCloudConfig,
+  type SdkIdentity,
   type AssistantCloudTelemetryConfig,
 } from "./AssistantCloudAPI";
 import { AssistantCloudAuthTokens } from "./AssistantCloudAuthTokens";
@@ -20,9 +21,11 @@ export class AssistantCloud {
   public readonly events;
   public readonly scores;
   public readonly telemetry: AssistantCloudTelemetryConfig;
+  public readonly registerSdk: (sdk: SdkIdentity) => void;
 
   constructor(config: AssistantCloudConfig) {
     const api = new AssistantCloudAPI(config);
+    this.registerSdk = api.registerSdk;
     const t = config.telemetry;
     this.telemetry =
       t === false

package/src/AssistantCloudAPI.ts

@@ -6,6 +6,12 @@ import {
   normalizeBaseUrl,
 } from "./AssistantCloudAuthStrategy";
 import type { AssistantCloudRunReport } from "./AssistantCloudRuns";
+import { ASSISTANT_CLOUD_VERSION } from "./version";
+
+export type SdkIdentity = {
+  name: string;
+  version: string;
+};
 
 export type AssistantCloudTelemetryConfig = {
   /**
@@ -53,7 +59,10 @@ export type AssistantCloudConfig = (
    *
    * When enabled, the SDK automatically reports run metadata (status, step
    * count, tool calls, and token usage) to Assistant Cloud after each
-   * assistant message is saved. No message content is sent.
+   * assistant message is saved. Reports can also include assistant output,
+   * tool arguments and results, errors, and metadata, which may contain
+   * sensitive content. Use `beforeReport` to redact or drop reports, or
+   * `telemetry: false` to disable reporting.
    *
    * - `true` / `undefined` — enabled with defaults
    * - `false` — disabled
@@ -89,11 +98,31 @@ type MakeRequestOptions = {
   keepalive?: boolean | undefined;
 };
 
+const HEADER_TOKEN = /^[\x21-\x7e]+$/;
+
 export class AssistantCloudAPI {
   public _auth: AssistantCloudAuthStrategy;
   public _baseUrl;
+  public readonly registerSdk: (sdk: SdkIdentity) => void;
+  public readonly sdkHeader: () => string;
 
   constructor(config: AssistantCloudConfig) {
+    const sdks = new Map<string, SdkIdentity>();
+    this.registerSdk = (sdk) => {
+      const name = sdk.name.trim();
+      const version = sdk.version.trim();
+      if (!HEADER_TOKEN.test(name) || !HEADER_TOKEN.test(version)) return;
+      sdks.set(`${name}/${version}`, { name, version });
+    };
+    this.sdkHeader = () =>
+      [
+        `assistant-cloud/${ASSISTANT_CLOUD_VERSION}`,
+        ...Array.from(
+          sdks.values(),
+          ({ name, version }) => `${name}/${version}`,
+        ),
+      ].join(" ");
+
     if ("authToken" in config) {
       this._baseUrl = normalizeBaseUrl(config.baseUrl);
       this._auth = new AssistantCloudJWTAuthStrategy(config.authToken);
@@ -131,6 +160,7 @@ export class AssistantCloudAPI {
       ...authHeaders,
       ...options.headers,
       "Content-Type": "application/json",
+      "Aui-Sdk": this.sdkHeader(),
     };
 
     const queryParams = new URLSearchParams();

package/src/AssistantCloudAuthStrategy.ts

@@ -160,8 +160,9 @@ export class AssistantCloudJWTAuthStrategy implements AssistantCloudAuthStrategy
     const token = await this.#authTokenCallback();
     if (!token) return false;
 
+    const tokenExpiry = getJwtExpiry(token);
     this.cachedToken = token;
-    this.tokenExpiry = getJwtExpiry(token);
+    this.tokenExpiry = tokenExpiry;
 
     return { Authorization: `Bearer ${token}` };
   }
@@ -175,8 +176,9 @@ export class AssistantCloudJWTAuthStrategy implements AssistantCloudAuthStrategy
       throw new Error("Invalid auth header received");
     }
 
+    const tokenExpiry = getJwtExpiry(token);
     this.cachedToken = token;
-    this.tokenExpiry = getJwtExpiry(token);
+    this.tokenExpiry = tokenExpiry;
   }
 }
 

package/src/AssistantCloudEvents.test.ts

@@ -12,11 +12,11 @@ const event = (index: number): AssistantCloudEvent => ({
   thread_id: `thread-${index}`,
 });
 
-const createEvents = (enabled = true) => {
+const createEvents = (enabled: boolean | (() => boolean) = true) => {
   const makeRequest = vi.fn().mockResolvedValue({ accepted: 1 });
   const events = new AssistantCloudEvents(
     { makeRequest } as unknown as AssistantCloudAPI,
-    () => enabled,
+    typeof enabled === "function" ? enabled : () => enabled,
   );
   return { events, makeRequest };
 };
@@ -106,6 +106,31 @@ describe("AssistantCloudEvents", () => {
     expect(makeRequest).not.toHaveBeenCalled();
   });
 
+  it("does not send queued batches after telemetry is disabled", async () => {
+    let enabled = true;
+    let resolveFirstRequest!: (value: { accepted: number }) => void;
+    const { events, makeRequest } = createEvents(() => enabled);
+    makeRequest.mockImplementationOnce(
+      () =>
+        new Promise((resolve) => {
+          resolveFirstRequest = resolve;
+        }),
+    );
+
+    for (let index = 0; index < 51; index++) events.track(event(index));
+    await vi.waitFor(() => expect(makeRequest).toHaveBeenCalledOnce());
+
+    enabled = false;
+    resolveFirstRequest({ accepted: 50 });
+
+    await vi.waitFor(() =>
+      expect(makeRequest.mock.results[0]?.value).resolves.toEqual({
+        accepted: 50,
+      }),
+    );
+    expect(makeRequest).toHaveBeenCalledOnce();
+  });
+
   it("keeps every request below the server batch limit", async () => {
     const { events, makeRequest } = createEvents();
 
@@ -121,4 +146,45 @@ describe("AssistantCloudEvents", () => {
       expect(options.body.events.length).toBeLessThanOrEqual(50);
     }
   });
+
+  it.each([Number.NaN, Number.POSITIVE_INFINITY, Number.NEGATIVE_INFINITY])(
+    "drops props containing the non-finite number %s",
+    async (value) => {
+      const { events, makeRequest } = createEvents();
+
+      events.track({ kind: "message_sent", props: { value } });
+      events.dispose();
+
+      await vi.waitFor(() => expect(makeRequest).toHaveBeenCalledOnce());
+      expect(makeRequest).toHaveBeenCalledWith("/events", {
+        method: "POST",
+        body: { events: [{ kind: "message_sent" }] },
+        keepalive: true,
+      });
+    },
+  );
+
+  it("keeps finite numeric props", async () => {
+    const { events, makeRequest } = createEvents();
+
+    events.track({
+      kind: "message_sent",
+      props: { negative: -1.5, zero: 0, positive: 2.5 },
+    });
+    events.dispose();
+
+    await vi.waitFor(() => expect(makeRequest).toHaveBeenCalledOnce());
+    expect(makeRequest).toHaveBeenCalledWith("/events", {
+      method: "POST",
+      body: {
+        events: [
+          {
+            kind: "message_sent",
+            props: { negative: -1.5, zero: 0, positive: 2.5 },
+          },
+        ],
+      },
+      keepalive: true,
+    });
+  });
 });

package/src/AssistantCloudEvents.ts

@@ -118,6 +118,11 @@ export class AssistantCloudEvents {
 
   private async flushPending(): Promise<void> {
     while (this.buffer.length > 0) {
+      if (!this.isEnabled()) {
+        this.buffer = [];
+        return;
+      }
+
       const events = this.buffer.splice(0, MAX_BATCH_SIZE);
       try {
         await this.cloud.makeRequest("/events", {
@@ -171,6 +176,7 @@ const normalizeProps = (
     entries.some(
       ([, value]) =>
         (typeof value === "string" && value.length > 256) ||
+        (typeof value === "number" && !Number.isFinite(value)) ||
         (typeof value !== "string" &&
           typeof value !== "number" &&
           typeof value !== "boolean"),

package/src/AssistantCloudRuns.ts

@@ -68,6 +68,7 @@ export class AssistantCloudRuns {
         return {
           ...headers,
           Accept: "text/plain",
+          "Aui-Sdk": this.cloud.sdkHeader(),
         };
       },
       body: {