assistant-cloud 0.2.0 → 0.2.1

package/dist/ai-sdk/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","names":[],"sources":["../../src/ai-sdk/index.ts"],"sourcesContent":["import type { UIMessage } from \"ai\";\nimport type { MessageFormatAdapter } from \"../FormattedCloudPersistence\";\nimport type { SamplingCallData } from \"../instrumentMcpSampling\";\nimport {\n  type AssistantCloudRunReportToolCall,\n  createRunTelemetryToolCall,\n  extractRunTelemetryModelId,\n  normalizeRunTelemetryUsage,\n  type RunMessageTelemetry,\n  type RunReportStepInit,\n  type RunTelemetryUsage,\n  type RunTelemetryUsageInit,\n  truncateRunTelemetryText,\n} from \"../runTelemetry\";\n\nexport type AISDKStorageFormat = Omit<UIMessage, \"id\">;\n\n/** The stored form of an AI SDK message: the message without its id. */\nexport const aiSDKV6FormatAdapter: MessageFormatAdapter<\n  UIMessage,\n  AISDKStorageFormat\n> = {\n  format: \"ai-sdk/v6\",\n  encode: ({ message: { id: _id, ...message } }) => message,\n  decode: (stored) => ({\n    parentId: stored.parent_id,\n    message: { id: stored.id, ...stored.content } as UIMessage,\n  }),\n  getId: (message) => message.id,\n};\n\n/**\n * A message as an AI SDK integration holds it, or as the cloud stored it under\n * the ai-sdk/v6 format, which drops the id. Parts are typed by their `type`\n * alone, so this shape and the telemetry read from it need nothing from `ai`.\n */\nexport type AISDKMessageLike = {\n  id?: string | undefined;\n  role: string;\n  parts: readonly { type: string; [key: string]: unknown }[];\n  metadata?: unknown;\n};\n\ntype Part = Record<string, unknown> & { type: string };\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n  return value !== null && typeof value === \"object\";\n}\n\nfunction isPart(value: unknown): value is Part {\n  return isRecord(value) && typeof value.type === \"string\";\n}\n\n/**\n * The AI SDK's own tool part rules, kept here so the entry loads without the\n * `ai` runtime: a static tool part is `tool-<name>`, a dynamic one is\n * `dynamic-tool` with its name in `toolName`.\n */\nfunction toolCallOf(part: Part): AssistantCloudRunReportToolCall | undefined {\n  if (typeof part.toolCallId !== \"string\") return undefined;\n  const isStatic = part.type.startsWith(\"tool-\");\n  if (!isStatic && part.type !== \"dynamic-tool\") return undefined;\n  const toolName = isStatic\n    ? part.type.slice(\"tool-\".length)\n    : typeof part.toolName === \"string\"\n      ? part.toolName\n      : undefined;\n  if (!toolName) return undefined;\n  return createRunTelemetryToolCall({\n    toolName,\n    toolCallId: part.toolCallId,\n    args: part.input ?? part.args,\n    result: part.output ?? part.result,\n    toolSource: isStatic ? \"frontend\" : \"mcp\",\n  });\n}\n\nfunction attachSamplingCalls(\n  toolCalls: readonly AssistantCloudRunReportToolCall[],\n  metadata: Record<string, unknown> | undefined,\n): void {\n  const samplingCalls = isRecord(metadata?.samplingCalls)\n    ? (metadata.samplingCalls as Record<string, SamplingCallData[]>)\n    : undefined;\n  if (!samplingCalls) return;\n  for (const toolCall of toolCalls) {\n    const calls = samplingCalls[toolCall.tool_call_id];\n    if (Array.isArray(calls) && calls.length > 0) {\n      toolCall.sampling_calls = calls;\n    }\n  }\n}\n\nfunction stepUsages(\n  metadata: Record<string, unknown> | undefined,\n): (RunTelemetryUsageInit | undefined)[] {\n  const steps = metadata?.steps;\n  if (!Array.isArray(steps)) return [];\n  return steps.map((step) =>\n    isRecord(step) && isRecord(step.usage)\n      ? (step.usage as RunTelemetryUsageInit)\n      : undefined,\n  );\n}\n\nfunction sumUsage(usages: readonly RunTelemetryUsage[]): RunTelemetryUsage {\n  const total: RunTelemetryUsage = {};\n  for (const usage of usages) {\n    if (usage.inputTokens != null) {\n      total.inputTokens = (total.inputTokens ?? 0) + usage.inputTokens;\n    }\n    if (usage.outputTokens != null) {\n      total.outputTokens = (total.outputTokens ?? 0) + usage.outputTokens;\n    }\n    if (usage.reasoningTokens != null) {\n      total.reasoningTokens =\n        (total.reasoningTokens ?? 0) + usage.reasoningTokens;\n    }\n    if (usage.cachedInputTokens != null) {\n      total.cachedInputTokens =\n        (total.cachedInputTokens ?? 0) + usage.cachedInputTokens;\n    }\n  }\n  return total;\n}\n\n/**\n * The usage a message reports: `metadata.usage` when the integration copied\n * the run total there, else the sum over `metadata.steps[].usage`.\n */\nfunction messageUsage(\n  metadata: Record<string, unknown> | undefined,\n): RunTelemetryUsage | undefined {\n  const total = isRecord(metadata?.usage)\n    ? normalizeRunTelemetryUsage(metadata.usage as RunTelemetryUsageInit)\n    : undefined;\n  if (total) return total;\n  const perStep = stepUsages(metadata).flatMap((usage) => {\n    const normalized = usage ? normalizeRunTelemetryUsage(usage) : undefined;\n    return normalized ? [normalized] : [];\n  });\n  return perStep.length > 0 ? sumUsage(perStep) : undefined;\n}\n\n/**\n * Reads the run report fields out of the assistant messages of one run. A run\n * the AI SDK streamed as one message is one element; a run the cloud stored as\n * several assistant rows is aggregated, step by step, in order. Returns null\n * when no assistant message is present. Status reads completed when the run\n * produced text or tool calls, as a live finish with reason `tool-calls` does;\n * an integration that observed the finish event overrides it.\n */\nexport function extractAISDKRunTelemetry(\n  messages: readonly AISDKMessageLike[],\n): RunMessageTelemetry | null {\n  const textParts: string[] = [];\n  const toolCalls: AssistantCloudRunReportToolCall[] = [];\n  const steps: RunReportStepInit[] = [];\n  const usages: RunTelemetryUsage[] = [];\n  let assistant: AISDKMessageLike | undefined;\n\n  for (const message of messages) {\n    if (message.role !== \"assistant\") continue;\n    assistant = message;\n    const metadata = isRecord(message.metadata) ? message.metadata : undefined;\n    const usagePerStep = stepUsages(metadata);\n    const messageToolCalls: AssistantCloudRunReportToolCall[] = [];\n    let step: RunReportStepInit | undefined;\n    let stepIndex = -1;\n\n    for (const part of message.parts) {\n      if (!isPart(part)) continue;\n      if (part.type === \"step-start\") {\n        stepIndex += 1;\n        const usage = usagePerStep[stepIndex];\n        step = usage ? { usage } : {};\n        steps.push(step);\n        continue;\n      }\n      if (part.type === \"text\" && typeof part.text === \"string\" && part.text) {\n        textParts.push(part.text);\n        continue;\n      }\n      const toolCall = toolCallOf(part);\n      if (!toolCall) continue;\n      toolCalls.push(toolCall);\n      messageToolCalls.push(toolCall);\n      if (step) {\n        step.toolCalls = [...(step.toolCalls ?? []), toolCall];\n        step.finishReason = \"tool-calls\";\n      }\n    }\n\n    attachSamplingCalls(messageToolCalls, metadata);\n    const usage = messageUsage(metadata);\n    if (usage) usages.push(usage);\n  }\n\n  if (!assistant) return null;\n\n  const metadata = isRecord(assistant.metadata)\n    ? assistant.metadata\n    : undefined;\n\n  const usage = usages.length > 0 ? sumUsage(usages) : undefined;\n  const modelId = extractRunTelemetryModelId(metadata);\n  const completed = textParts.length > 0 || toolCalls.length > 0;\n  return {\n    ...(assistant.id !== undefined\n      ? { assistantMessageId: assistant.id }\n      : undefined),\n    status: completed ? \"completed\" : \"incomplete\",\n    ...(toolCalls.length > 0 ? { toolCalls } : undefined),\n    ...(steps.length > 0 ? { steps, totalSteps: steps.length } : undefined),\n    ...(textParts.length > 0\n      ? { outputText: truncateRunTelemetryText(textParts.join(\"\")) }\n      : undefined),\n    ...(usage ? { usage } : undefined),\n    ...(modelId ? { modelId } : undefined),\n    ...(metadata ? { metadata } : undefined),\n  };\n}\n"],"mappings":";;;AAkBA,MAAa,uBAGT;CACF,QAAQ;CACR,SAAS,EAAE,SAAS,EAAE,IAAI,KAAK,GAAG,gBAAgB;CAClD,SAAS,YAAY;EACnB,UAAU,OAAO;EACjB,SAAS;GAAE,IAAI,OAAO;GAAI,GAAG,OAAO;EAAQ;CAC9C;CACA,QAAQ,YAAY,QAAQ;AAC9B;AAgBA,SAAS,SAAS,OAAkD;CAClE,OAAO,UAAU,QAAQ,OAAO,UAAU;AAC5C;AAEA,SAAS,OAAO,OAA+B;CAC7C,OAAO,SAAS,KAAK,KAAK,OAAO,MAAM,SAAS;AAClD;;;;;;AAOA,SAAS,WAAW,MAAyD;CAC3E,IAAI,OAAO,KAAK,eAAe,UAAU,OAAO,KAAA;CAChD,MAAM,WAAW,KAAK,KAAK,WAAW,OAAO;CAC7C,IAAI,CAAC,YAAY,KAAK,SAAS,gBAAgB,OAAO,KAAA;CACtD,MAAM,WAAW,WACb,KAAK,KAAK,MAAM,CAAc,IAC9B,OAAO,KAAK,aAAa,WACvB,KAAK,WACL,KAAA;CACN,IAAI,CAAC,UAAU,OAAO,KAAA;CACtB,OAAO,2BAA2B;EAChC;EACA,YAAY,KAAK;EACjB,MAAM,KAAK,SAAS,KAAK;EACzB,QAAQ,KAAK,UAAU,KAAK;EAC5B,YAAY,WAAW,aAAa;CACtC,CAAC;AACH;AAEA,SAAS,oBACP,WACA,UACM;CACN,MAAM,gBAAgB,SAAS,UAAU,aAAa,IACjD,SAAS,gBACV,KAAA;CACJ,IAAI,CAAC,eAAe;CACpB,KAAK,MAAM,YAAY,WAAW;EAChC,MAAM,QAAQ,cAAc,SAAS;EACrC,IAAI,MAAM,QAAQ,KAAK,KAAK,MAAM,SAAS,GACzC,SAAS,iBAAiB;CAE9B;AACF;AAEA,SAAS,WACP,UACuC;CACvC,MAAM,QAAQ,UAAU;CACxB,IAAI,CAAC,MAAM,QAAQ,KAAK,GAAG,OAAO,CAAC;CACnC,OAAO,MAAM,KAAK,SAChB,SAAS,IAAI,KAAK,SAAS,KAAK,KAAK,IAChC,KAAK,QACN,KAAA,CACN;AACF;AAEA,SAAS,SAAS,QAAyD;CACzE,MAAM,QAA2B,CAAC;CAClC,KAAK,MAAM,SAAS,QAAQ;EAC1B,IAAI,MAAM,eAAe,MACvB,MAAM,eAAe,MAAM,eAAe,KAAK,MAAM;EAEvD,IAAI,MAAM,gBAAgB,MACxB,MAAM,gBAAgB,MAAM,gBAAgB,KAAK,MAAM;EAEzD,IAAI,MAAM,mBAAmB,MAC3B,MAAM,mBACH,MAAM,mBAAmB,KAAK,MAAM;EAEzC,IAAI,MAAM,qBAAqB,MAC7B,MAAM,qBACH,MAAM,qBAAqB,KAAK,MAAM;CAE7C;CACA,OAAO;AACT;;;;;AAMA,SAAS,aACP,UAC+B;CAC/B,MAAM,QAAQ,SAAS,UAAU,KAAK,IAClC,2BAA2B,SAAS,KAA8B,IAClE,KAAA;CACJ,IAAI,OAAO,OAAO;CAClB,MAAM,UAAU,WAAW,QAAQ,CAAC,CAAC,SAAS,UAAU;EACtD,MAAM,aAAa,QAAQ,2BAA2B,KAAK,IAAI,KAAA;EAC/D,OAAO,aAAa,CAAC,UAAU,IAAI,CAAC;CACtC,CAAC;CACD,OAAO,QAAQ,SAAS,IAAI,SAAS,OAAO,IAAI,KAAA;AAClD;;;;;;;;;AAUA,SAAgB,yBACd,UAC4B;CAC5B,MAAM,YAAsB,CAAC;CAC7B,MAAM,YAA+C,CAAC;CACtD,MAAM,QAA6B,CAAC;CACpC,MAAM,SAA8B,CAAC;CACrC,IAAI;CAEJ,KAAK,MAAM,WAAW,UAAU;EAC9B,IAAI,QAAQ,SAAS,aAAa;EAClC,YAAY;EACZ,MAAM,WAAW,SAAS,QAAQ,QAAQ,IAAI,QAAQ,WAAW,KAAA;EACjE,MAAM,eAAe,WAAW,QAAQ;EACxC,MAAM,mBAAsD,CAAC;EAC7D,IAAI;EACJ,IAAI,YAAY;EAEhB,KAAK,MAAM,QAAQ,QAAQ,OAAO;GAChC,IAAI,CAAC,OAAO,IAAI,GAAG;GACnB,IAAI,KAAK,SAAS,cAAc;IAC9B,aAAa;IACb,MAAM,QAAQ,aAAa;IAC3B,OAAO,QAAQ,EAAE,MAAM,IAAI,CAAC;IAC5B,MAAM,KAAK,IAAI;IACf;GACF;GACA,IAAI,KAAK,SAAS,UAAU,OAAO,KAAK,SAAS,YAAY,KAAK,MAAM;IACtE,UAAU,KAAK,KAAK,IAAI;IACxB;GACF;GACA,MAAM,WAAW,WAAW,IAAI;GAChC,IAAI,CAAC,UAAU;GACf,UAAU,KAAK,QAAQ;GACvB,iBAAiB,KAAK,QAAQ;GAC9B,IAAI,MAAM;IACR,KAAK,YAAY,CAAC,GAAI,KAAK,aAAa,CAAC,GAAI,QAAQ;IACrD,KAAK,eAAe;GACtB;EACF;EAEA,oBAAoB,kBAAkB,QAAQ;EAC9C,MAAM,QAAQ,aAAa,QAAQ;EACnC,IAAI,OAAO,OAAO,KAAK,KAAK;CAC9B;CAEA,IAAI,CAAC,WAAW,OAAO;CAEvB,MAAM,WAAW,SAAS,UAAU,QAAQ,IACxC,UAAU,WACV,KAAA;CAEJ,MAAM,QAAQ,OAAO,SAAS,IAAI,SAAS,MAAM,IAAI,KAAA;CACrD,MAAM,UAAU,2BAA2B,QAAQ;CACnD,MAAM,YAAY,UAAU,SAAS,KAAK,UAAU,SAAS;CAC7D,OAAO;EACL,GAAI,UAAU,OAAO,KAAA,IACjB,EAAE,oBAAoB,UAAU,GAAG,IACnC,KAAA;EACJ,QAAQ,YAAY,cAAc;EAClC,GAAI,UAAU,SAAS,IAAI,EAAE,UAAU,IAAI,KAAA;EAC3C,GAAI,MAAM,SAAS,IAAI;GAAE;GAAO,YAAY,MAAM;EAAO,IAAI,KAAA;EAC7D,GAAI,UAAU,SAAS,IACnB,EAAE,YAAY,yBAAyB,UAAU,KAAK,EAAE,CAAC,EAAE,IAC3D,KAAA;EACJ,GAAI,QAAQ,EAAE,MAAM,IAAI,KAAA;EACxB,GAAI,UAAU,EAAE,QAAQ,IAAI,KAAA;EAC5B,GAAI,WAAW,EAAE,SAAS,IAAI,KAAA;CAChC;AACF"}

package/dist/generateThreadTitle.js

@@ -16,6 +16,7 @@ async function generateThreadTitle(cloud, options) {
 	} finally {
 		reader.releaseLock();
 	}
+	title = title.trim();
 	if (title) await cloud.threads.update(options.threadId, { title });
 	return title || null;
 }

package/dist/generateThreadTitle.js.map

@@ -1 +1 @@
-{"version":3,"file":"generateThreadTitle.js","names":[],"sources":["../src/generateThreadTitle.ts"],"sourcesContent":["import type { AssistantCloud } from \"./AssistantCloud\";\n\nexport async function generateThreadTitle(\n  cloud: AssistantCloud,\n  options: {\n    threadId: string;\n    messages: readonly {\n      role: string;\n      content: readonly { type: \"text\"; text: string }[];\n    }[];\n  },\n): Promise<string | null> {\n  const stream = await cloud.runs.stream({\n    thread_id: options.threadId,\n    assistant_id: \"system/thread_title\",\n    messages: options.messages,\n  });\n\n  let title = \"\";\n  const reader = stream.getReader();\n  try {\n    while (true) {\n      const { done, value: chunk } = await reader.read();\n      if (done) break;\n      if (chunk.type === \"text-delta\") {\n        title += chunk.textDelta;\n      }\n    }\n  } finally {\n    reader.releaseLock();\n  }\n\n  if (title) {\n    await cloud.threads.update(options.threadId, { title });\n  }\n\n  return title || null;\n}\n"],"mappings":";AAEA,eAAsB,oBACpB,OACA,SAOwB;CACxB,MAAM,SAAS,MAAM,MAAM,KAAK,OAAO;EACrC,WAAW,QAAQ;EACnB,cAAc;EACd,UAAU,QAAQ;CACpB,CAAC;CAED,IAAI,QAAQ;CACZ,MAAM,SAAS,OAAO,UAAU;CAChC,IAAI;EACF,OAAO,MAAM;GACX,MAAM,EAAE,MAAM,OAAO,UAAU,MAAM,OAAO,KAAK;GACjD,IAAI,MAAM;GACV,IAAI,MAAM,SAAS,cACjB,SAAS,MAAM;EAEnB;CACF,UAAU;EACR,OAAO,YAAY;CACrB;CAEA,IAAI,OACF,MAAM,MAAM,QAAQ,OAAO,QAAQ,UAAU,EAAE,MAAM,CAAC;CAGxD,OAAO,SAAS;AAClB"}
+{"version":3,"file":"generateThreadTitle.js","names":[],"sources":["../src/generateThreadTitle.ts"],"sourcesContent":["import type { AssistantCloud } from \"./AssistantCloud\";\n\nexport async function generateThreadTitle(\n  cloud: AssistantCloud,\n  options: {\n    threadId: string;\n    messages: readonly {\n      role: string;\n      content: readonly { type: \"text\"; text: string }[];\n    }[];\n  },\n): Promise<string | null> {\n  const stream = await cloud.runs.stream({\n    thread_id: options.threadId,\n    assistant_id: \"system/thread_title\",\n    messages: options.messages,\n  });\n\n  let title = \"\";\n  const reader = stream.getReader();\n  try {\n    while (true) {\n      const { done, value: chunk } = await reader.read();\n      if (done) break;\n      if (chunk.type === \"text-delta\") {\n        title += chunk.textDelta;\n      }\n    }\n  } finally {\n    reader.releaseLock();\n  }\n\n  title = title.trim();\n  if (title) {\n    await cloud.threads.update(options.threadId, { title });\n  }\n\n  return title || null;\n}\n"],"mappings":";AAEA,eAAsB,oBACpB,OACA,SAOwB;CACxB,MAAM,SAAS,MAAM,MAAM,KAAK,OAAO;EACrC,WAAW,QAAQ;EACnB,cAAc;EACd,UAAU,QAAQ;CACpB,CAAC;CAED,IAAI,QAAQ;CACZ,MAAM,SAAS,OAAO,UAAU;CAChC,IAAI;EACF,OAAO,MAAM;GACX,MAAM,EAAE,MAAM,OAAO,UAAU,MAAM,OAAO,KAAK;GACjD,IAAI,MAAM;GACV,IAAI,MAAM,SAAS,cACjB,SAAS,MAAM;EAEnB;CACF,UAAU;EACR,OAAO,YAAY;CACrB;CAEA,QAAQ,MAAM,KAAK;CACnB,IAAI,OACF,MAAM,MAAM,QAAQ,OAAO,QAAQ,UAAU,EAAE,MAAM,CAAC;CAGxD,OAAO,SAAS;AAClB"}

package/dist/index.d.ts

@@ -1,15 +1,17 @@
 import { readAnonymousRefreshToken } from "./AssistantCloudAuthStrategy.js";
 import { McpSamplingHandler, SamplingCallData, createSamplingCollector, wrapSamplingHandler } from "./instrumentMcpSampling.js";
-import { AssistantCloudRunReportToolCall, RunReportInit, RunReportOutcome, RunReportStepInit, RunTelemetryToolCallInit, RunTelemetryUsage, RunTelemetryUsageInit, createRunReport, createRunTelemetryToolCall, deriveRunOutcome, describeRunError, extractRunTelemetryModelId, normalizeRunTelemetryUsage, truncateRunTelemetryText } from "./runTelemetry.js";
+import { AssistantCloudRunReportToolCall, RunMessageTelemetry, RunReportInit, RunReportOutcome, RunReportStepInit, RunTelemetryToolCallInit, RunTelemetryUsage, RunTelemetryUsageInit, createRunReport, createRunTelemetryToolCall, deriveRunOutcome, describeRunError, extractRunTelemetryModelId, normalizeRunTelemetryUsage, truncateRunTelemetryText } from "./runTelemetry.js";
 import { AssistantCloudRunReport } from "./AssistantCloudRuns.js";
-import { AssistantCloudTelemetryConfig, CloudAPIError } from "./AssistantCloudAPI.js";
+import { AssistantCloudTelemetryConfig, CloudAPIError, SdkIdentity } from "./AssistantCloudAPI.js";
 import { AssistantCloudThreadMessageFeedbackBody, AssistantCloudThreadMessageFeedbackResponse, CloudMessage } from "./AssistantCloudThreadMessages.js";
 import { GeneratePresignedDownloadUrlResponse } from "./AssistantCloudFiles.js";
 import { AssistantCloudEvent, AssistantCloudEventKind, AssistantCloudEvents } from "./AssistantCloudEvents.js";
 import { AssistantCloudScoreBody, AssistantCloudScoreResponse, AssistantCloudScores } from "./AssistantCloudScores.js";
 import { AssistantCloud } from "./AssistantCloud.js";
+import { CloudEngagementReporter, EngagementEventIds, EngagementIdResolver } from "./CloudEngagementReporter.js";
 import { CloudMessagePersistence } from "./CloudMessagePersistence.js";
+import { CloudRunReportInit, CloudRunReporter } from "./CloudRunReporter.js";
 import { MessageFormatAdapter, createFormattedPersistence } from "./FormattedCloudPersistence.js";
 import { CloudResponseError } from "./cloudResponse.js";
 import { generateThreadTitle } from "./generateThreadTitle.js";
-export { AssistantCloud, type AssistantCloudEvent, type AssistantCloudEventKind, AssistantCloudEvents, type AssistantCloudRunReport, type AssistantCloudRunReportToolCall, type AssistantCloudScoreBody, type AssistantCloudScoreResponse, AssistantCloudScores, type AssistantCloudTelemetryConfig, type AssistantCloudThreadMessageFeedbackBody, type AssistantCloudThreadMessageFeedbackResponse, CloudAPIError, type CloudMessage, CloudMessagePersistence, CloudResponseError, type GeneratePresignedDownloadUrlResponse, type McpSamplingHandler, type MessageFormatAdapter, type RunReportInit, type RunReportOutcome, type RunReportStepInit, type RunTelemetryToolCallInit, type RunTelemetryUsage, type RunTelemetryUsageInit, type SamplingCallData, createFormattedPersistence, createRunReport, createRunTelemetryToolCall, createSamplingCollector, deriveRunOutcome, describeRunError, extractRunTelemetryModelId, generateThreadTitle, normalizeRunTelemetryUsage, readAnonymousRefreshToken, truncateRunTelemetryText, wrapSamplingHandler };
+export { AssistantCloud, type AssistantCloudEvent, type AssistantCloudEventKind, AssistantCloudEvents, type AssistantCloudRunReport, type AssistantCloudRunReportToolCall, type AssistantCloudScoreBody, type AssistantCloudScoreResponse, AssistantCloudScores, type AssistantCloudTelemetryConfig, type AssistantCloudThreadMessageFeedbackBody, type AssistantCloudThreadMessageFeedbackResponse, CloudAPIError, CloudEngagementReporter, type CloudMessage, CloudMessagePersistence, CloudResponseError, type CloudRunReportInit, CloudRunReporter, type EngagementEventIds, type EngagementIdResolver, type GeneratePresignedDownloadUrlResponse, type McpSamplingHandler, type MessageFormatAdapter, type RunMessageTelemetry, type RunReportInit, type RunReportOutcome, type RunReportStepInit, type RunTelemetryToolCallInit, type RunTelemetryUsage, type RunTelemetryUsageInit, type SamplingCallData, type SdkIdentity, createFormattedPersistence, createRunReport, createRunTelemetryToolCall, createSamplingCollector, deriveRunOutcome, describeRunError, extractRunTelemetryModelId, generateThreadTitle, normalizeRunTelemetryUsage, readAnonymousRefreshToken, truncateRunTelemetryText, wrapSamplingHandler };

package/dist/index.js

@@ -4,9 +4,11 @@ import { CloudAPIError } from "./AssistantCloudAPI.js";
 import { AssistantCloudEvents } from "./AssistantCloudEvents.js";
 import { AssistantCloudScores } from "./AssistantCloudScores.js";
 import { AssistantCloud } from "./AssistantCloud.js";
+import { CloudEngagementReporter } from "./CloudEngagementReporter.js";
 import { CloudMessagePersistence } from "./CloudMessagePersistence.js";
+import { createRunReport, createRunTelemetryToolCall, deriveRunOutcome, describeRunError, extractRunTelemetryModelId, normalizeRunTelemetryUsage, truncateRunTelemetryText } from "./runTelemetry.js";
+import { CloudRunReporter } from "./CloudRunReporter.js";
 import { createFormattedPersistence } from "./FormattedCloudPersistence.js";
 import { generateThreadTitle } from "./generateThreadTitle.js";
-import { createRunReport, createRunTelemetryToolCall, deriveRunOutcome, describeRunError, extractRunTelemetryModelId, normalizeRunTelemetryUsage, truncateRunTelemetryText } from "./runTelemetry.js";
 import { createSamplingCollector, wrapSamplingHandler } from "./instrumentMcpSampling.js";
-export { AssistantCloud, AssistantCloudEvents, AssistantCloudScores, CloudAPIError, CloudMessagePersistence, CloudResponseError, createFormattedPersistence, createRunReport, createRunTelemetryToolCall, createSamplingCollector, deriveRunOutcome, describeRunError, extractRunTelemetryModelId, generateThreadTitle, normalizeRunTelemetryUsage, readAnonymousRefreshToken, truncateRunTelemetryText, wrapSamplingHandler };
+export { AssistantCloud, AssistantCloudEvents, AssistantCloudScores, CloudAPIError, CloudEngagementReporter, CloudMessagePersistence, CloudResponseError, CloudRunReporter, createFormattedPersistence, createRunReport, createRunTelemetryToolCall, createSamplingCollector, deriveRunOutcome, describeRunError, extractRunTelemetryModelId, generateThreadTitle, normalizeRunTelemetryUsage, readAnonymousRefreshToken, truncateRunTelemetryText, wrapSamplingHandler };

package/dist/instrumentMcpSampling.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"instrumentMcpSampling.d.ts","names":[],"sources":["../src/instrumentMcpSampling.ts"],"mappings":";;;;;;;;KAQY;EACV;EACA;EACA;EACA;EACA;EACA;;KAGU,sBACV,SAAS,uBACN,QAAQ;KAED;EACV;EACA;IACE;IACA;MAAqB;QAAU;;;IAC/B;KACC;;;KAIO;EACV;EACA;EACA;IACE;IACA;IACA;IACA;IACA;IACA;;GAED;;;;;;;;;;;;;;;;;;;;iBAqBa,oBACd,SAAS,oBACT,iBAAiB,MAAM,4BACtB;;;;;;;;;;;;;iBA6Ca;EAGI,UAAA,MAAA;kBAAA"}
+{"version":3,"file":"instrumentMcpSampling.d.ts","names":[],"sources":["../src/instrumentMcpSampling.ts"],"mappings":";;;;;;;;KAQY;EACV;EACA;EACA;EACA;EACA;EACA;;KAGU,sBACV,SAAS,uBACN,QAAQ;KAED;EACV;EACA;IACE;IACA;MAAqB;QAAU;;;IAC/B;KACC;;;KAIO;EACV;EACA;EACA;IACE;IACA;IACA;IACA;IACA;IACA;;GAED;;;;;;;;;;;;;;;;;;;;iBA8Ba,oBACd,SAAS,oBACT,iBAAiB,MAAM,4BACtB;;;;;;;;;;;;;iBAkDa;EAGI,UAAA,MAAA;kBAAA"}

package/dist/instrumentMcpSampling.js

@@ -1,4 +1,9 @@
 //#region src/instrumentMcpSampling.ts
+const reportSamplingError = (error) => {
+	try {
+		console.error("[assistant-cloud] onSamplingCall callback threw an error", error);
+	} catch {}
+};
 /**
 * Wraps an MCP sampling handler to intercept and measure sampling calls.
 *
@@ -27,14 +32,19 @@ function wrapSamplingHandler(handler, onSamplingCall) {
 		const outputTokens = response.usage?.outputTokens ?? response.usage?.completionTokens;
 		const reasoningTokens = response.usage?.reasoningTokens;
 		const cachedInputTokens = response.usage?.cachedInputTokens;
-		onSamplingCall({
+		const data = {
 			...modelId ? { model_id: modelId } : void 0,
 			...inputTokens != null ? { input_tokens: inputTokens } : void 0,
 			...outputTokens != null ? { output_tokens: outputTokens } : void 0,
 			...reasoningTokens != null ? { reasoning_tokens: reasoningTokens } : void 0,
 			...cachedInputTokens != null ? { cached_input_tokens: cachedInputTokens } : void 0,
 			duration_ms: durationMs
-		});
+		};
+		try {
+			Promise.resolve(onSamplingCall(data)).catch(reportSamplingError);
+		} catch (error) {
+			reportSamplingError(error);
+		}
 		return response;
 	};
 }

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

@@ -87,6 +87,22 @@ type RunReportStepInit = {
   endMs?: number | undefined;
   finishReason?: string | undefined;
 };
+/**
+ * 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.
+ */
+type RunMessageTelemetry = {
+  assistantMessageId?: string;
+  status: "completed" | "incomplete";
+  toolCalls?: AssistantCloudRunReportToolCall[];
+  steps?: RunReportStepInit[];
+  totalSteps?: number;
+  outputText?: string;
+  usage?: RunTelemetryUsage;
+  modelId?: string;
+  metadata?: Record<string, unknown>;
+};
 type RunReportInit = {
   threadId: string;
   status: AssistantCloudRunReport["status"];
@@ -120,5 +136,5 @@ declare function createRunReport(init: RunReportInit): AssistantCloudRunReport;
  */
 declare function normalizeRunTelemetryUsage(usage: RunTelemetryUsageInit): RunTelemetryUsage | undefined;
 //#endregion
-export { AssistantCloudRunReportToolCall, RunReportInit, RunReportOutcome, RunReportStepInit, RunTelemetryToolCallInit, RunTelemetryUsage, RunTelemetryUsageInit, createRunReport, createRunTelemetryToolCall, deriveRunOutcome, describeRunError, extractRunTelemetryModelId, normalizeRunTelemetryUsage, truncateRunTelemetryText };
+export { AssistantCloudRunReportToolCall, RunMessageTelemetry, 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":";;;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;;;;;;;KAQU;EACV;EACA;EACA,YAAY;EACZ,QAAQ;EACR;EACA;EACA,QAAQ;EACR;EACA,WAAW;;KAGD;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"}

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/version.d.ts

@@ -0,0 +1,5 @@
+//#region src/version.d.ts
+declare const ASSISTANT_CLOUD_VERSION: string;
+//#endregion
+export { ASSISTANT_CLOUD_VERSION };
+//# 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":";cAAa"}

package/dist/version.js

@@ -0,0 +1,6 @@
+//#region src/version.ts
+const ASSISTANT_CLOUD_VERSION = "0.2.1";
+//#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.1",
   "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.43"
   },
   "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",
+    "@assistant-ui/x-buildutils": "0.0.27",
     "@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",
+    "ai": "^7.0.93",
+    "ai-v6": "npm:ai@^6.0.277",
     "jsdom": "^30.0.1",
-    "vitest": "^4.1.11"
+    "vitest": "^5.0.0"
   },
   "publishConfig": {
     "access": "public",
@@ -73,7 +83,10 @@
   },
   "scripts": {
     "build": "aui-build",
-    "test": "vitest run",
+    "test": "pnpm run test:v7 && pnpm run test:peer-v6 && pnpm run test:types:peer-v6",
+    "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();
 

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", {

package/src/AssistantCloudRuns.ts

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