@depup/sentry__server-utils 10.75.1-depup.0 → 11.0.0-depup.0

package/build/cjs/integrations/vercel-ai/vercel-ai-dc-subscriber.js

@@ -0,0 +1,482 @@
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+
+const attributes = require('@sentry/conventions/attributes');
+const op = require('@sentry/conventions/op');
+const core = require('@sentry/core');
+const genAiAttributes = require('../../ai/core/gen-ai-attributes.js');
+const genAiRecordingMode = require('./gen-ai-recording-mode.js');
+const index = require('../../ai/vercel-ai/index.js');
+const constants = require('../../ai/workers-ai/constants.js');
+const tracingChannel = require('../../tracing-channel.js');
+const util = require('./util.js');
+const utils = require('../../ai/core/utils.js');
+
+const AI_SDK_TELEMETRY_TRACING_CHANNEL = "ai:telemetry";
+const ORIGIN = "auto.vercelai.channel";
+const GEN_AI_OPERATION_SPAN_OPS = {
+  embeddings: op.GEN_AI_EMBEDDINGS,
+  rerank: op.GEN_AI_RERANK,
+  invoke_agent: op.GEN_AI_INVOKE_AGENT,
+  execute_tool: op.GEN_AI_EXECUTE_TOOL,
+  // The model-call op matches the Vercel AI OTel integration (`gen_ai.generate_content`) rather than
+  // the generic `gen_ai.chat`, so v6 (OTel) and v7 (channel) produce the same spans.
+  generate_content: op.GEN_AI_GENERATE_CONTENT
+};
+const VERCEL_AI_OPERATION_ID_ATTRIBUTE = "vercel.ai.operationId";
+const VERCEL_AI_MODEL_PROVIDER_ATTRIBUTE = "vercel.ai.model.provider";
+const VERCEL_AI_SETTINGS_MAX_RETRIES_ATTRIBUTE = "vercel.ai.settings.maxRetries";
+const operationIdByCallId = /* @__PURE__ */ new Map();
+const toolDescriptionsByCallId = /* @__PURE__ */ new Map();
+const invokeAgentSpanByCallId = /* @__PURE__ */ new Map();
+const ROOT_OPERATION_TYPES = /* @__PURE__ */ new Set([
+  "generateText",
+  "streamText",
+  "generateObject",
+  "embed",
+  "embedMany",
+  "rerank"
+]);
+function clearOperationId(data) {
+  if (!ROOT_OPERATION_TYPES.has(data.type)) {
+    return;
+  }
+  const callId = util.asString(data.event.callId);
+  if (callId) {
+    clearOperationCallId(callId);
+  }
+}
+function clearOperationCallId(callId) {
+  operationIdByCallId.delete(callId);
+  toolDescriptionsByCallId.delete(callId);
+  invokeAgentSpanByCallId.delete(callId);
+}
+function dropLastStepOnlyUsage(providerAttributes, type) {
+  if (!ROOT_OPERATION_TYPES.has(type)) {
+    return;
+  }
+  const hasRecomputedOutput = attributes.GEN_AI_USAGE_OUTPUT_TOKENS in providerAttributes;
+  for (const key of index.LAST_STEP_ONLY_USAGE_KEYS) {
+    delete providerAttributes[key];
+  }
+  if (hasRecomputedOutput) {
+    delete providerAttributes[attributes.GEN_AI_USAGE_REASONING_OUTPUT_TOKENS];
+  }
+}
+function recordToolDescriptions(callId, tools) {
+  if (!callId || !Array.isArray(tools)) {
+    return;
+  }
+  let descriptions = toolDescriptionsByCallId.get(callId);
+  for (const tool of tools) {
+    if (core.isObjectLike(tool) && typeof tool.name === "string" && typeof tool.description === "string") {
+      descriptions = descriptions ?? /* @__PURE__ */ new Map();
+      if (!descriptions.has(tool.name)) {
+        descriptions.set(tool.name, tool.description);
+      }
+    }
+  }
+  if (descriptions) {
+    toolDescriptionsByCallId.set(callId, descriptions);
+  }
+}
+function resolveToolDescription(callId, toolName, tools) {
+  const fromMap = callId ? toolDescriptionsByCallId.get(callId)?.get(toolName) : void 0;
+  if (fromMap) {
+    return fromMap;
+  }
+  if (Array.isArray(tools)) {
+    const match = tools.find((tool) => core.isObjectLike(tool) && tool.name === toolName);
+    return core.isObjectLike(match) ? util.asString(match.description) : void 0;
+  }
+  if (core.isObjectLike(tools)) {
+    const tool = tools[toolName];
+    return core.isObjectLike(tool) ? util.asString(tool.description) : void 0;
+  }
+  return void 0;
+}
+function subscribeVercelAiTracingChannel(tracingChannel$1, options = {}) {
+  tracingChannel.bindTracingChannelToSpan(
+    tracingChannel$1(AI_SDK_TELEMETRY_TRACING_CHANNEL),
+    (data) => createSpanFromMessage(data, options),
+    {
+      // The helper ends the span; we enrich it from the settled result first (tokens, output messages,
+      // finish reasons, response model/id, provider metadata) and drop the per-operation `callId` maps.
+      beforeSpanEnd: (span, data) => {
+        enrichSpanOnEnd(span, data, options);
+        clearOperationId(data);
+      },
+      // A streamed model call resolves before its stream is drained, so we tap the stream, keep the
+      // span open, and end it (via `end`) once the final usage/finish/output chunks arrive.
+      deferSpanEnd: ({ data, end }) => deferStreamedModelCallEnd(data, options, end)
+    }
+  );
+}
+function deferStreamedModelCallEnd(data, options, end) {
+  if (data.type !== "languageModelCall" || !core.isObjectLike(data.result)) {
+    return false;
+  }
+  const result = data.result;
+  const stream = result.stream;
+  if (!utils.isReadableStream(stream)) {
+    return false;
+  }
+  const callId = util.asString(data.event.callId);
+  const { recordOutputs } = getRecordingOptions(data.event, options);
+  result.stream = util.tapModelCallStream(
+    stream,
+    (final) => {
+      data.result = { ...result, ...streamedResultToChannelResult(final) };
+      end();
+      enrichInvokeAgentFromStream(callId, final, recordOutputs);
+    },
+    (error) => end(error)
+  );
+  return true;
+}
+function streamedResultToChannelResult(final) {
+  const content = [];
+  if (final.text) {
+    content.push({ type: "text", text: final.text });
+  }
+  for (const toolCall of final.toolCalls) {
+    content.push({ type: "tool-call", ...toolCall });
+  }
+  return {
+    content,
+    ...final.usage !== void 0 ? { usage: final.usage } : {},
+    ...final.finishReason !== void 0 ? { finishReason: final.finishReason } : {},
+    ...final.providerMetadata !== void 0 ? { providerMetadata: final.providerMetadata } : {},
+    ...final.responseId || final.responseModel ? {
+      response: {
+        ...final.responseId ? { id: final.responseId } : {},
+        ...final.responseModel ? { modelId: final.responseModel } : {}
+      }
+    } : {}
+  };
+}
+function enrichInvokeAgentFromStream(callId, final, recordOutputs) {
+  const span = callId ? invokeAgentSpanByCallId.get(callId) : void 0;
+  if (!span) {
+    return;
+  }
+  const usage = core.isObjectLike(final.usage) ? final.usage : void 0;
+  if (usage) {
+    const input = tokenCount(usage.inputTokens) ?? tokenCount(usage.promptTokens) ?? tokenCount(usage.tokens);
+    const output = tokenCount(usage.outputTokens) ?? tokenCount(usage.completionTokens);
+    addTokensToSpan(span, attributes.GEN_AI_USAGE_INPUT_TOKENS, input);
+    addTokensToSpan(span, attributes.GEN_AI_USAGE_OUTPUT_TOKENS, output);
+    addTokensToSpan(span, attributes.GEN_AI_USAGE_TOTAL_TOKENS, tokenCount(usage.totalTokens) ?? util.sum(input, output));
+    const { cacheRead, cacheWrite } = cacheTokens(usage);
+    addTokensToSpan(span, attributes.GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS, cacheRead);
+    addTokensToSpan(span, attributes.GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS, cacheWrite);
+  }
+  if (recordOutputs) {
+    const parts = partsFromTextAndToolCalls(final.text, final.toolCalls);
+    const outputMessages = buildOutputMessages(parts, getFinishReason({ finishReason: final.finishReason }));
+    if (outputMessages) {
+      span.setAttribute(attributes.GEN_AI_OUTPUT_MESSAGES, outputMessages);
+    }
+  }
+}
+function addTokensToSpan(span, attribute, value) {
+  if (value === void 0) {
+    return;
+  }
+  const current = core.spanToJSON(span).attributes[attribute];
+  span.setAttribute(attribute, (typeof current === "number" ? current : 0) + value);
+}
+function createSpanFromMessage(data, channelOptions) {
+  const { type, event } = data;
+  if (type === "step" || !event || typeof event !== "object") {
+    return void 0;
+  }
+  const { recordInputs } = getRecordingOptions(event, channelOptions);
+  const provider = util.asString(event.provider);
+  const modelId = util.asString(event.modelId);
+  const callId = util.asString(event.callId);
+  const maxRetries = util.asNumber(event.maxRetries);
+  if (recordInputs) {
+    recordToolDescriptions(callId, event.tools);
+  }
+  const baseAttributes = {
+    [core.SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,
+    ...provider ? { [attributes.GEN_AI_PROVIDER_NAME]: provider, [VERCEL_AI_MODEL_PROVIDER_ATTRIBUTE]: provider } : {},
+    ...modelId ? { [attributes.GEN_AI_REQUEST_MODEL]: modelId } : {},
+    ...maxRetries !== void 0 ? { [VERCEL_AI_SETTINGS_MAX_RETRIES_ATTRIBUTE]: maxRetries } : {}
+  };
+  switch (type) {
+    case "generateText":
+    case "streamText":
+    case "generateObject":
+      return buildInvokeAgentSpan(event, baseAttributes, recordInputs, callId, type === "streamText");
+    case "languageModelCall":
+      core._INTERNAL_skipAiProviderWrapping([constants.WORKERS_AI_INTEGRATION_NAME]);
+      return buildModelCallSpan(event, baseAttributes, recordInputs, callId, modelId);
+    case "executeTool":
+      return buildToolSpan(event, recordInputs);
+    case "embed":
+    case "embedMany": {
+      const input = type === "embedMany" ? event.values : event.value;
+      return startGenAiSpan("embeddings", modelId, {
+        ...baseAttributes,
+        ...recordInputs && input !== void 0 ? { [attributes.GEN_AI_EMBEDDINGS_INPUT]: core.stringify(input) } : {}
+      });
+    }
+    case "rerank":
+      return startGenAiSpan("rerank", modelId, baseAttributes);
+    default:
+      return void 0;
+  }
+}
+function startGenAiSpan(operation, suffix, attributes$1) {
+  return core.startInactiveSpan({
+    name: suffix ? `${operation} ${suffix}` : operation,
+    attributes: {
+      [attributes.SENTRY_OP]: GEN_AI_OPERATION_SPAN_OPS[operation],
+      [attributes.GEN_AI_OPERATION_NAME]: operation,
+      ...attributes$1
+    }
+  });
+}
+function buildInvokeAgentSpan(event, baseAttributes, recordInputs, callId, isStream) {
+  const functionId = util.asString(event.functionId);
+  const operationId = util.asString(event.operationId) ?? (isStream ? "ai.streamText" : "ai.generateText");
+  if (callId) {
+    operationIdByCallId.set(callId, { operationId, isStream });
+  }
+  const span = startGenAiSpan("invoke_agent", functionId, {
+    ...baseAttributes,
+    [VERCEL_AI_OPERATION_ID_ATTRIBUTE]: operationId,
+    [attributes.GEN_AI_RESPONSE_STREAMING]: isStream,
+    ...functionId ? { [attributes.GEN_AI_FUNCTION_ID]: functionId } : {},
+    ...recordInputs ? buildInputMessageAttributes(event) : {}
+  });
+  if (isStream && callId) {
+    invokeAgentSpanByCallId.set(callId, span);
+  }
+  return span;
+}
+function buildModelCallSpan(event, baseAttributes, recordInputs, callId, modelId) {
+  const parent = callId ? operationIdByCallId.get(callId) : void 0;
+  const operationId = parent ? `${parent.operationId}.${parent.isStream ? "doStream" : "doGenerate"}` : "ai.generateText.doGenerate";
+  return startGenAiSpan("generate_content", modelId, {
+    ...baseAttributes,
+    [VERCEL_AI_OPERATION_ID_ATTRIBUTE]: operationId,
+    ...recordInputs ? buildInputMessageAttributes(event) : {},
+    ...recordInputs && Array.isArray(event.tools) ? { [attributes.GEN_AI_TOOL_DEFINITIONS]: core.stringify(event.tools) } : {}
+  });
+}
+function buildToolSpan(event, recordInputs) {
+  const toolCall = core.isObjectLike(event.toolCall) ? event.toolCall : {};
+  const toolName = util.asString(toolCall.toolName);
+  const toolCallId = util.asString(event.toolCallId) ?? util.asString(toolCall.toolCallId);
+  const toolInput = toolCall.input ?? toolCall.args;
+  const description = recordInputs && toolName ? resolveToolDescription(util.asString(event.callId), toolName, event.tools) : void 0;
+  return startGenAiSpan("execute_tool", toolName, {
+    [core.SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,
+    ...toolName ? { [attributes.GEN_AI_TOOL_NAME]: toolName } : {},
+    ...toolCallId ? { [genAiAttributes.GEN_AI_TOOL_CALL_ID_ATTRIBUTE]: toolCallId } : {},
+    ...description ? { [attributes.GEN_AI_TOOL_DESCRIPTION]: description } : {},
+    ...recordInputs && toolInput !== void 0 ? { [attributes.GEN_AI_TOOL_CALL_ARGUMENTS]: core.stringify(toolInput) } : {}
+  });
+}
+function enrichSpanOnEnd(span, data, channelOptions) {
+  const { type, result } = data;
+  if (!core.isObjectLike(result)) {
+    return;
+  }
+  const { recordOutputs } = getRecordingOptions(data.event, channelOptions);
+  if (type === "executeTool") {
+    if (recordOutputs) {
+      span.setAttribute(attributes.GEN_AI_TOOL_CALL_RESULT, core.stringify(result.output ?? result));
+    }
+    const output = core.isObjectLike(result.output) ? result.output : void 0;
+    if (output?.type === "tool-error") {
+      captureToolError(span, data, output.error);
+    }
+    return;
+  }
+  const usage = core.isObjectLike(result.usage) ? result.usage : void 0;
+  if (usage) {
+    const inputTokens = tokenCount(usage.inputTokens) ?? tokenCount(usage.promptTokens) ?? tokenCount(usage.tokens);
+    const outputTokens = tokenCount(usage.outputTokens) ?? tokenCount(usage.completionTokens);
+    const totalTokens = tokenCount(usage.totalTokens) ?? util.sum(inputTokens, outputTokens);
+    if (inputTokens !== void 0) {
+      span.setAttribute(attributes.GEN_AI_USAGE_INPUT_TOKENS, inputTokens);
+    }
+    if (outputTokens !== void 0) {
+      span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, outputTokens);
+    }
+    if (totalTokens !== void 0) {
+      span.setAttribute(attributes.GEN_AI_USAGE_TOTAL_TOKENS, totalTokens);
+    }
+    setCacheTokens(span, usage);
+  }
+  const finishReason = getFinishReason(result);
+  if (finishReason && type === "languageModelCall") {
+    span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, core.stringify([finishReason]));
+  }
+  const response = core.isObjectLike(result.response) ? result.response : void 0;
+  const responseId = util.asString(response?.id) ?? util.asString(result.responseId);
+  if (responseId) {
+    span.setAttribute(attributes.GEN_AI_RESPONSE_ID, responseId);
+  }
+  const responseModel = util.asString(response?.modelId) ?? util.asString(data.event.modelId);
+  if (responseModel) {
+    span.setAttribute(attributes.GEN_AI_RESPONSE_MODEL, responseModel);
+  }
+  const providerMetadata = result.providerMetadata;
+  const providerAttributes = index.getProviderMetadataAttributes(providerMetadata);
+  if (attributes.GEN_AI_CONVERSATION_ID in providerAttributes && core.spanToJSON(span).attributes[attributes.GEN_AI_CONVERSATION_ID]) {
+    delete providerAttributes[attributes.GEN_AI_CONVERSATION_ID];
+  }
+  dropLastStepOnlyUsage(providerAttributes, type);
+  span.setAttributes(providerAttributes);
+  if (recordOutputs) {
+    const parts = type === "languageModelCall" && Array.isArray(result.content) ? partsFromContent(result.content) : partsFromTextAndToolCalls(result.text, result.toolCalls);
+    const outputMessages = buildOutputMessages(parts, finishReason);
+    if (outputMessages) {
+      span.setAttribute(attributes.GEN_AI_OUTPUT_MESSAGES, outputMessages);
+    }
+  }
+}
+function normalizeFinishReason(finishReason) {
+  return finishReason === "tool-calls" ? "tool_call" : finishReason ?? "stop";
+}
+function getFinishReason(result) {
+  const finishReason = result.finishReason;
+  if (typeof finishReason === "string") {
+    return finishReason;
+  }
+  return core.isObjectLike(finishReason) ? util.asString(finishReason.unified) : void 0;
+}
+function tokenCount(value) {
+  return util.asNumber(value) ?? (core.isObjectLike(value) ? util.asNumber(value.total) : void 0);
+}
+function setCacheTokens(span, usage) {
+  const { cacheRead, cacheWrite } = cacheTokens(usage);
+  if (cacheRead !== void 0) {
+    span.setAttribute(attributes.GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS, cacheRead);
+  }
+  if (cacheWrite !== void 0) {
+    span.setAttribute(attributes.GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS, cacheWrite);
+  }
+}
+function cacheTokens(usage) {
+  const inputTokens = core.isObjectLike(usage.inputTokens) ? usage.inputTokens : void 0;
+  const inputTokenDetails = core.isObjectLike(usage.inputTokenDetails) ? usage.inputTokenDetails : void 0;
+  return {
+    cacheRead: util.asNumber(inputTokens?.cacheRead) ?? util.asNumber(inputTokenDetails?.cacheReadTokens) ?? util.asNumber(usage.cachedInputTokens),
+    cacheWrite: util.asNumber(inputTokens?.cacheWrite) ?? util.asNumber(inputTokenDetails?.cacheWriteTokens)
+  };
+}
+function buildOutputMessages(parts, finishReason) {
+  if (!parts.length) {
+    return void 0;
+  }
+  return core.stringify([{ role: "assistant", parts, finish_reason: normalizeFinishReason(finishReason) }]);
+}
+function toolCallPart(toolCall) {
+  const args = toolCall.input ?? toolCall.args;
+  return {
+    type: "tool_call",
+    id: util.asString(toolCall.toolCallId),
+    name: util.asString(toolCall.toolName),
+    arguments: typeof args === "string" ? args : core.stringify(args ?? {})
+  };
+}
+function partsFromContent(content) {
+  const parts = [];
+  for (const item of content) {
+    if (!core.isObjectLike(item)) {
+      continue;
+    }
+    if (item.type === "text" && typeof item.text === "string") {
+      parts.push({ type: "text", content: item.text });
+    } else if (item.type === "tool-call") {
+      parts.push(toolCallPart(item));
+    }
+  }
+  return parts;
+}
+function partsFromTextAndToolCalls(text, toolCalls) {
+  const parts = [];
+  if (typeof text === "string" && text.length) {
+    parts.push({ type: "text", content: text });
+  }
+  if (Array.isArray(toolCalls)) {
+    for (const toolCall of toolCalls) {
+      if (core.isObjectLike(toolCall)) {
+        parts.push(toolCallPart(toolCall));
+      }
+    }
+  }
+  return parts;
+}
+function captureToolError(span, data, error) {
+  span.setStatus({
+    code: core.SPAN_STATUS_ERROR,
+    message: error instanceof Error ? error.message : "tool_error"
+  });
+  const toolCall = core.isObjectLike(data.event.toolCall) ? data.event.toolCall : {};
+  const toolName = util.asString(toolCall.toolName);
+  const toolCallId = util.asString(data.event.toolCallId) ?? util.asString(toolCall.toolCallId);
+  core.withScope((scope) => {
+    scope.setContext("trace", core.spanToTraceContext(span));
+    if (toolName) {
+      scope.setTag("vercel.ai.tool.name", toolName);
+    }
+    if (toolCallId) {
+      scope.setTag("vercel.ai.tool.callId", toolCallId);
+    }
+    scope.setLevel("error");
+    core.captureException(
+      error instanceof Error ? error : new Error(typeof error === "string" ? error : "Tool execution failed"),
+      {
+        mechanism: { type: "auto.vercelai.channel", handled: false }
+      }
+    );
+  });
+}
+function getRecordingOptions(event, channelOptions) {
+  const client = core.getClient();
+  const genAI = client?.getDataCollectionOptions().genAI;
+  const eveMode = client ? genAiRecordingMode.isEveGenAiRecordingDefault(client) : false;
+  return {
+    recordInputs: resolveRecording(channelOptions.recordInputs, event.recordInputs, genAI?.inputs, eveMode),
+    recordOutputs: resolveRecording(channelOptions.recordOutputs, event.recordOutputs, genAI?.outputs, eveMode)
+  };
+}
+function resolveRecording(integrationOption, perCallOption, globalDefault, eveMode = false) {
+  if (typeof integrationOption === "boolean") {
+    return integrationOption;
+  }
+  if (eveMode) {
+    return typeof globalDefault === "boolean" ? globalDefault : true;
+  }
+  if (typeof perCallOption === "boolean") {
+    return perCallOption;
+  }
+  return globalDefault === true;
+}
+function buildInputMessageAttributes(event) {
+  const attributes$1 = {};
+  const instructions = util.asString(event.instructions);
+  if (instructions) {
+    attributes$1[attributes.GEN_AI_SYSTEM_INSTRUCTIONS] = core.stringify([{ type: "text", content: instructions }]);
+  }
+  const messages = event.messages ?? event.prompt;
+  if (messages !== void 0) {
+    attributes$1[attributes.GEN_AI_INPUT_MESSAGES] = core.stringify(messages);
+  }
+  return attributes$1;
+}
+
+exports.captureToolError = captureToolError;
+exports.clearOperationCallId = clearOperationCallId;
+exports.clearOperationId = clearOperationId;
+exports.createSpanFromMessage = createSpanFromMessage;
+exports.enrichSpanOnEnd = enrichSpanOnEnd;
+exports.streamedResultToChannelResult = streamedResultToChannelResult;
+exports.subscribeVercelAiTracingChannel = subscribeVercelAiTracingChannel;
+//# sourceMappingURL=vercel-ai-dc-subscriber.js.map