@ai-sdk/otel 1.0.0-beta.5 → 1.0.0-beta.50

package/dist/index.js

@@ -1,31 +1,1070 @@
-"use strict";
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+// src/gen-ai-open-telemetry-integration.ts
+import {
+  context,
+  SpanKind,
+  SpanStatusCode,
+  trace
+} from "@opentelemetry/api";
+
+// src/gen-ai-format-messages.ts
+import { convertDataContentToBase64String } from "ai";
+function mapProviderName(provider) {
+  const lower = provider.toLowerCase();
+  const wellKnownPrefixes = [
+    ["google.vertex", "gcp.vertex_ai"],
+    ["google.generative-ai", "gcp.gemini"],
+    ["google-vertex", "gcp.vertex_ai"],
+    ["amazon-bedrock", "aws.bedrock"],
+    ["azure-openai", "azure.ai.openai"],
+    ["anthropic", "anthropic"],
+    ["openai", "openai"],
+    ["azure", "azure.ai.inference"],
+    ["google", "gcp.gemini"],
+    ["mistral", "mistral_ai"],
+    ["cohere", "cohere"],
+    ["bedrock", "aws.bedrock"],
+    ["groq", "groq"],
+    ["deepseek", "deepseek"],
+    ["perplexity", "perplexity"],
+    ["xai", "x_ai"]
+  ];
+  for (const [prefix, mapped] of wellKnownPrefixes) {
+    if (lower === prefix || lower.startsWith(prefix + ".") || lower.startsWith(prefix + "-")) {
+      return mapped;
+    }
+  }
+  return provider;
+}
+function mapOperationName(operationId) {
+  var _a;
+  const mapping = {
+    "ai.generateText": "invoke_agent",
+    "ai.streamText": "invoke_agent",
+    "ai.generateObject": "invoke_agent",
+    "ai.streamObject": "invoke_agent",
+    "ai.embed": "embeddings",
+    "ai.embedMany": "embeddings",
+    "ai.rerank": "rerank"
+  };
+  return (_a = mapping[operationId]) != null ? _a : operationId;
+}
+function formatSystemInstructions(system) {
+  if (typeof system === "string") {
+    return [{ type: "text", content: system }];
+  }
+  if (Array.isArray(system)) {
+    return system.map((msg) => ({ type: "text", content: msg.content }));
+  }
+  return [{ type: "text", content: system.content }];
+}
+function convertMessagePartToSemConv(part) {
+  var _a, _b, _c, _d;
+  switch (part.type) {
+    case "text":
+      return { type: "text", content: part.text };
+    case "reasoning":
+      return { type: "reasoning", content: part.text };
+    case "tool-call":
+      return {
+        type: "tool_call",
+        id: (_a = part.toolCallId) != null ? _a : null,
+        name: part.toolName,
+        arguments: part.input
+      };
+    case "tool-result": {
+      const output = part.output;
+      let response;
+      if (output) {
+        if (output.type === "text" || output.type === "error-text") {
+          response = output.value;
+        } else if (output.type === "json" || output.type === "error-json") {
+          response = output.value;
+        } else if (output.type === "execution-denied") {
+          response = { denied: true, reason: output.reason };
+        } else {
+          response = output;
+        }
+      }
+      return {
+        type: "tool_call_response",
+        id: (_b = part.toolCallId) != null ? _b : null,
+        response
+      };
+    }
+    case "file": {
+      const data = part.data;
+      let content;
+      if (data instanceof Uint8Array) {
+        content = convertDataContentToBase64String(data);
+      } else if (typeof data === "string") {
+        if (data.startsWith("http://") || data.startsWith("https://")) {
+          return {
+            type: "uri",
+            modality: getModality(part.mediaType),
+            mime_type: (_c = part.mediaType) != null ? _c : null,
+            uri: data
+          };
+        }
+        content = data;
+      } else {
+        content = String(data);
+      }
+      return {
+        type: "blob",
+        modality: getModality(part.mediaType),
+        mime_type: (_d = part.mediaType) != null ? _d : null,
+        content
+      };
+    }
+    case "tool-approval-response":
+      return {
+        type: "tool_approval_response",
+        approval_id: part.approvalId,
+        approved: part.approved,
+        reason: part.reason
+      };
+    case "custom":
+      return { type: "custom", kind: part.kind };
+    case "reasoning-file":
+      return { type: String(part.type) };
+    default: {
+      const _exhaustive = part;
+      return { type: String(_exhaustive.type) };
+    }
+  }
+}
+function getModality(mediaType) {
+  if (!mediaType) return "image";
+  if (mediaType.startsWith("image/")) return "image";
+  if (mediaType.startsWith("video/")) return "video";
+  if (mediaType.startsWith("audio/")) return "audio";
+  return "image";
+}
+function formatInputMessages(prompt) {
+  return prompt.filter((msg) => msg.role !== "system").map((message) => {
+    if (message.role === "system") {
+      return {
+        role: "system",
+        parts: [{ type: "text", content: message.content }]
+      };
+    }
+    const parts = message.content.map(convertMessagePartToSemConv);
+    return { role: message.role, parts };
+  });
+}
+function formatModelMessages({
+  prompt,
+  messages
+}) {
+  const result = [];
+  if (typeof prompt === "string") {
+    result.push({
+      role: "user",
+      parts: [{ type: "text", content: prompt }]
+    });
+  } else if (Array.isArray(prompt)) {
+    for (const msg of prompt) {
+      const converted = convertModelMessageToSemConv(msg);
+      if (converted) result.push(converted);
+    }
+  }
+  if (messages) {
+    for (const msg of messages) {
+      const converted = convertModelMessageToSemConv(msg);
+      if (converted) result.push(converted);
+    }
+  }
+  return result;
+}
+function convertModelMessageToSemConv(msg) {
+  if (msg.role === "system") return void 0;
+  if (msg.role === "user") {
+    if (typeof msg.content === "string") {
+      return {
+        role: "user",
+        parts: [{ type: "text", content: msg.content }]
+      };
+    }
+    const parts = msg.content.map((part) => {
+      var _a, _b, _c, _d, _e, _f, _g, _h;
+      switch (part.type) {
+        case "text":
+          return { type: "text", content: part.text };
+        case "image": {
+          const data = part.image;
+          if (data instanceof URL) {
+            return {
+              type: "uri",
+              modality: "image",
+              mime_type: (_a = part.mediaType) != null ? _a : null,
+              uri: data.toString()
+            };
+          }
+          if (typeof data === "string") {
+            if (data.startsWith("http://") || data.startsWith("https://")) {
+              return {
+                type: "uri",
+                modality: "image",
+                mime_type: (_b = part.mediaType) != null ? _b : null,
+                uri: data
+              };
+            }
+            return {
+              type: "blob",
+              modality: "image",
+              mime_type: (_c = part.mediaType) != null ? _c : null,
+              content: data
+            };
+          }
+          return {
+            type: "blob",
+            modality: "image",
+            mime_type: (_d = part.mediaType) != null ? _d : null,
+            content: convertDataContentToBase64String(data)
+          };
+        }
+        case "file": {
+          const data = part.data;
+          if (data instanceof URL) {
+            return {
+              type: "uri",
+              modality: getModality(part.mediaType),
+              mime_type: (_e = part.mediaType) != null ? _e : null,
+              uri: data.toString()
+            };
+          }
+          if (typeof data === "string") {
+            if (data.startsWith("http://") || data.startsWith("https://")) {
+              return {
+                type: "uri",
+                modality: getModality(part.mediaType),
+                mime_type: (_f = part.mediaType) != null ? _f : null,
+                uri: data
+              };
+            }
+            return {
+              type: "blob",
+              modality: getModality(part.mediaType),
+              mime_type: (_g = part.mediaType) != null ? _g : null,
+              content: data
+            };
+          }
+          return {
+            type: "blob",
+            modality: getModality(part.mediaType),
+            mime_type: (_h = part.mediaType) != null ? _h : null,
+            content: convertDataContentToBase64String(data)
+          };
+        }
+        default:
+          return { type: String(part.type) };
+      }
+    });
+    return { role: "user", parts };
+  }
+  if (msg.role === "assistant") {
+    if (typeof msg.content === "string") {
+      return {
+        role: "assistant",
+        parts: [{ type: "text", content: msg.content }]
+      };
+    }
+    const parts = msg.content.map((part) => {
+      var _a, _b;
+      switch (part.type) {
+        case "text":
+          return { type: "text", content: part.text };
+        case "reasoning":
+          return { type: "reasoning", content: part.text };
+        case "tool-call":
+          return {
+            type: "tool_call",
+            id: (_a = part.toolCallId) != null ? _a : null,
+            name: part.toolName,
+            arguments: part.input
+          };
+        case "tool-result": {
+          const output = part.output;
+          let response;
+          if (output) {
+            if (output.type === "text" || output.type === "error-text") {
+              response = output.value;
+            } else if (output.type === "json" || output.type === "error-json") {
+              response = output.value;
+            } else if (output.type === "execution-denied") {
+              response = { denied: true, reason: output.reason };
+            } else {
+              response = output;
+            }
+          }
+          return {
+            type: "tool_call_response",
+            id: (_b = part.toolCallId) != null ? _b : null,
+            response
+          };
+        }
+        default:
+          return { type: String(part.type) };
+      }
+    });
+    return { role: "assistant", parts };
+  }
+  if (msg.role === "tool") {
+    const parts = msg.content.map((part) => {
+      var _a;
+      if (part.type === "tool-result") {
+        const output = part.output;
+        let response;
+        if (output) {
+          if (output.type === "text" || output.type === "error-text") {
+            response = output.value;
+          } else if (output.type === "json" || output.type === "error-json") {
+            response = output.value;
+          } else if (output.type === "execution-denied") {
+            response = { denied: true, reason: output.reason };
+          } else {
+            response = output;
+          }
+        }
+        return {
+          type: "tool_call_response",
+          id: (_a = part.toolCallId) != null ? _a : null,
+          response
+        };
+      }
+      return { type: String(part.type) };
+    });
+    return { role: "tool", parts };
+  }
+  return void 0;
+}
+function formatOutputMessages({
+  text,
+  reasoning,
+  toolCalls,
+  files,
+  finishReason
+}) {
+  const parts = [];
+  if (reasoning) {
+    for (const r of reasoning) {
+      if ("text" in r && r.text) {
+        parts.push({ type: "reasoning", content: r.text });
+      }
+    }
+  }
+  if (text != null && text.length > 0) {
+    parts.push({ type: "text", content: text });
+  }
+  if (toolCalls) {
+    for (const tc of toolCalls) {
+      parts.push({
+        type: "tool_call",
+        id: tc.toolCallId,
+        name: tc.toolName,
+        arguments: tc.input
+      });
+    }
+  }
+  if (files) {
+    for (const file of files) {
+      parts.push({
+        type: "blob",
+        modality: getModality(file.mediaType),
+        mime_type: file.mediaType,
+        content: file.base64
+      });
+    }
+  }
+  return [
+    {
+      role: "assistant",
+      parts,
+      finish_reason: mapFinishReason(finishReason)
+    }
+  ];
+}
+function formatObjectOutputMessages({
+  objectText,
+  finishReason
+}) {
+  return [
+    {
+      role: "assistant",
+      parts: [{ type: "text", content: objectText }],
+      finish_reason: mapFinishReason(finishReason)
+    }
+  ];
+}
+function mapFinishReason(reason) {
+  var _a;
+  const mapping = {
+    stop: "stop",
+    length: "length",
+    "content-filter": "content_filter",
+    "tool-calls": "tool_call",
+    error: "error",
+    other: "stop",
+    unknown: "stop"
+  };
+  return (_a = mapping[reason]) != null ? _a : reason;
+}
+
+// src/gen-ai-open-telemetry-integration.ts
+function recordSpanError(span, error) {
+  if (error instanceof Error) {
+    span.recordException({
+      name: error.name,
+      message: error.message,
+      stack: error.stack
+    });
+    span.setStatus({
+      code: SpanStatusCode.ERROR,
+      message: error.message
+    });
+  } else {
+    span.setStatus({ code: SpanStatusCode.ERROR });
+  }
+}
+function shouldRecord(telemetry) {
+  return (telemetry == null ? void 0 : telemetry.isEnabled) !== false;
+}
+function selectAttributes(telemetry, attributes) {
+  if (!shouldRecord(telemetry)) {
+    return {};
+  }
+  const result = {};
+  for (const [key, value] of Object.entries(attributes)) {
+    if (value == null) continue;
+    if (typeof value === "object" && "input" in value && typeof value.input === "function") {
+      if ((telemetry == null ? void 0 : telemetry.recordInputs) === false) continue;
+      const resolved = value.input();
+      if (resolved != null) result[key] = resolved;
+      continue;
+    }
+    if (typeof value === "object" && "output" in value && typeof value.output === "function") {
+      if ((telemetry == null ? void 0 : telemetry.recordOutputs) === false) continue;
+      const resolved = value.output();
+      if (resolved != null) result[key] = resolved;
+      continue;
+    }
+    result[key] = value;
+  }
+  return result;
+}
+var GenAIOpenTelemetryIntegration = class {
+  constructor(options = {}) {
+    this.callStates = /* @__PURE__ */ new Map();
+    var _a;
+    this.tracer = (_a = options.tracer) != null ? _a : trace.getTracer("gen_ai");
+  }
+  getCallState(callId) {
+    return this.callStates.get(callId);
+  }
+  cleanupCallState(callId) {
+    this.callStates.delete(callId);
+  }
+  executeTool({
+    callId,
+    toolCallId,
+    execute
+  }) {
+    var _a;
+    const toolSpanEntry = (_a = this.getCallState(callId)) == null ? void 0 : _a.toolSpans.get(toolCallId);
+    if (toolSpanEntry == null) {
+      return execute();
+    }
+    return context.with(toolSpanEntry.context, execute);
+  }
+  onStart(event) {
+    if (event.isEnabled === false) return;
+    if (event.operationId === "ai.embed" || event.operationId === "ai.embedMany") {
+      this.onEmbedOperationStart(event);
+      return;
+    }
+    if (event.operationId === "ai.rerank") {
+      this.onRerankOperationStart(event);
+      return;
+    }
+    if (event.operationId === "ai.generateObject" || event.operationId === "ai.streamObject") {
+      this.onObjectOperationStart(event);
+      return;
+    }
+    this.onGenerateStart(event);
+  }
+  onGenerateStart(event) {
+    var _a;
+    const telemetry = {
+      isEnabled: event.isEnabled,
+      recordInputs: event.recordInputs,
+      recordOutputs: event.recordOutputs,
+      functionId: event.functionId
+    };
+    const settings = {
+      maxOutputTokens: event.maxOutputTokens,
+      temperature: event.temperature,
+      topP: event.topP,
+      topK: event.topK,
+      presencePenalty: event.presencePenalty,
+      frequencyPenalty: event.frequencyPenalty,
+      stopSequences: event.stopSequences,
+      seed: event.seed,
+      maxRetries: event.maxRetries
+    };
+    const providerName = mapProviderName(event.provider);
+    const operationName = mapOperationName(event.operationId);
+    const attributes = selectAttributes(telemetry, {
+      "gen_ai.operation.name": operationName,
+      "gen_ai.provider.name": providerName,
+      "gen_ai.request.model": event.modelId,
+      "gen_ai.agent.name": telemetry.functionId,
+      "gen_ai.request.frequency_penalty": event.frequencyPenalty,
+      "gen_ai.request.max_tokens": event.maxOutputTokens,
+      "gen_ai.request.presence_penalty": event.presencePenalty,
+      "gen_ai.request.temperature": (_a = event.temperature) != null ? _a : void 0,
+      "gen_ai.request.top_k": event.topK,
+      "gen_ai.request.top_p": event.topP,
+      "gen_ai.request.stop_sequences": event.stopSequences,
+      "gen_ai.request.seed": event.seed,
+      "gen_ai.system_instructions": event.system ? {
+        input: () => JSON.stringify(formatSystemInstructions(event.system))
+      } : void 0,
+      "gen_ai.input.messages": {
+        input: () => JSON.stringify(
+          formatModelMessages({
+            prompt: event.prompt,
+            messages: event.messages
+          })
+        )
+      }
+    });
+    const spanName = `${operationName} ${event.modelId}`;
+    const rootSpan = this.tracer.startSpan(spanName, {
+      attributes,
+      kind: SpanKind.INTERNAL
+    });
+    const rootContext = trace.setSpan(context.active(), rootSpan);
+    this.callStates.set(event.callId, {
+      operationId: event.operationId,
+      telemetry,
+      rootSpan,
+      rootContext,
+      stepSpan: void 0,
+      stepContext: void 0,
+      embedSpans: /* @__PURE__ */ new Map(),
+      rerankSpan: void 0,
+      toolSpans: /* @__PURE__ */ new Map(),
+      settings,
+      provider: event.provider,
+      modelId: event.modelId
+    });
+  }
+  onObjectOperationStart(event) {
+    var _a;
+    const telemetry = {
+      isEnabled: event.isEnabled,
+      recordInputs: event.recordInputs,
+      recordOutputs: event.recordOutputs,
+      functionId: event.functionId
+    };
+    const settings = {
+      maxOutputTokens: event.maxOutputTokens,
+      temperature: event.temperature,
+      topP: event.topP,
+      topK: event.topK,
+      presencePenalty: event.presencePenalty,
+      frequencyPenalty: event.frequencyPenalty,
+      seed: event.seed,
+      maxRetries: event.maxRetries
+    };
+    const providerName = mapProviderName(event.provider);
+    const operationName = mapOperationName(event.operationId);
+    const attributes = selectAttributes(telemetry, {
+      "gen_ai.operation.name": operationName,
+      "gen_ai.provider.name": providerName,
+      "gen_ai.request.model": event.modelId,
+      "gen_ai.agent.name": telemetry.functionId,
+      "gen_ai.output.type": "json",
+      "gen_ai.request.frequency_penalty": event.frequencyPenalty,
+      "gen_ai.request.max_tokens": event.maxOutputTokens,
+      "gen_ai.request.presence_penalty": event.presencePenalty,
+      "gen_ai.request.temperature": (_a = event.temperature) != null ? _a : void 0,
+      "gen_ai.request.top_k": event.topK,
+      "gen_ai.request.top_p": event.topP,
+      "gen_ai.request.seed": event.seed,
+      "gen_ai.system_instructions": event.system ? {
+        input: () => JSON.stringify(formatSystemInstructions(event.system))
+      } : void 0,
+      "gen_ai.input.messages": {
+        input: () => JSON.stringify(
+          formatModelMessages({
+            prompt: event.prompt,
+            messages: event.messages
+          })
+        )
+      }
+    });
+    const spanName = `${operationName} ${event.modelId}`;
+    const rootSpan = this.tracer.startSpan(spanName, {
+      attributes,
+      kind: SpanKind.INTERNAL
+    });
+    const rootContext = trace.setSpan(context.active(), rootSpan);
+    this.callStates.set(event.callId, {
+      operationId: event.operationId,
+      telemetry,
+      rootSpan,
+      rootContext,
+      stepSpan: void 0,
+      stepContext: void 0,
+      embedSpans: /* @__PURE__ */ new Map(),
+      rerankSpan: void 0,
+      toolSpans: /* @__PURE__ */ new Map(),
+      settings,
+      provider: event.provider,
+      modelId: event.modelId
+    });
+  }
+  /** @deprecated */
+  onObjectStepStart(event) {
+    var _a;
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.rootSpan) || !state.rootContext) return;
+    const { telemetry } = state;
+    const providerName = mapProviderName(event.provider);
+    const attributes = selectAttributes(telemetry, {
+      "gen_ai.operation.name": "chat",
+      "gen_ai.provider.name": providerName,
+      "gen_ai.request.model": event.modelId,
+      "gen_ai.output.type": "json",
+      "gen_ai.request.frequency_penalty": state.settings.frequencyPenalty,
+      "gen_ai.request.max_tokens": state.settings.maxOutputTokens,
+      "gen_ai.request.presence_penalty": state.settings.presencePenalty,
+      "gen_ai.request.temperature": (_a = state.settings.temperature) != null ? _a : void 0,
+      "gen_ai.request.top_k": state.settings.topK,
+      "gen_ai.request.top_p": state.settings.topP,
+      "gen_ai.input.messages": {
+        input: () => event.promptMessages ? JSON.stringify(formatInputMessages(event.promptMessages)) : void 0
+      }
+    });
+    const spanName = `chat ${event.modelId}`;
+    state.stepSpan = this.tracer.startSpan(
+      spanName,
+      { attributes, kind: SpanKind.CLIENT },
+      state.rootContext
+    );
+    state.stepContext = trace.setSpan(state.rootContext, state.stepSpan);
+  }
+  /** @deprecated */
+  onObjectStepFinish(event) {
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.stepSpan)) return;
+    const { telemetry } = state;
+    state.stepSpan.setAttributes(
+      selectAttributes(telemetry, {
+        "gen_ai.response.finish_reasons": [event.finishReason],
+        "gen_ai.response.id": event.response.id,
+        "gen_ai.response.model": event.response.modelId,
+        "gen_ai.usage.input_tokens": event.usage.inputTokens,
+        "gen_ai.usage.output_tokens": event.usage.outputTokens,
+        "gen_ai.usage.cache_read.input_tokens": event.usage.cachedInputTokens,
+        "gen_ai.output.messages": {
+          output: () => {
+            try {
+              return JSON.stringify(
+                formatObjectOutputMessages({
+                  objectText: event.objectText,
+                  finishReason: event.finishReason
+                })
+              );
+            } catch (e) {
+              return event.objectText;
+            }
+          }
+        }
+      })
+    );
+    state.stepSpan.end();
+    state.stepSpan = void 0;
+    state.stepContext = void 0;
+  }
+  onEmbedOperationStart(event) {
+    const telemetry = {
+      isEnabled: event.isEnabled,
+      recordInputs: event.recordInputs,
+      recordOutputs: event.recordOutputs,
+      functionId: event.functionId
+    };
+    const settings = {
+      maxRetries: event.maxRetries
+    };
+    const providerName = mapProviderName(event.provider);
+    const attributes = selectAttributes(telemetry, {
+      "gen_ai.operation.name": "embeddings",
+      "gen_ai.provider.name": providerName,
+      "gen_ai.request.model": event.modelId
+    });
+    const spanName = `embeddings ${event.modelId}`;
+    const rootSpan = this.tracer.startSpan(spanName, {
+      attributes,
+      kind: SpanKind.CLIENT
+    });
+    const rootContext = trace.setSpan(context.active(), rootSpan);
+    this.callStates.set(event.callId, {
+      operationId: event.operationId,
+      telemetry,
+      rootSpan,
+      rootContext,
+      stepSpan: void 0,
+      stepContext: void 0,
+      embedSpans: /* @__PURE__ */ new Map(),
+      rerankSpan: void 0,
+      toolSpans: /* @__PURE__ */ new Map(),
+      settings,
+      provider: event.provider,
+      modelId: event.modelId
+    });
+  }
+  onStepStart(event) {
+    var _a;
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.rootSpan) || !state.rootContext) return;
+    const { telemetry } = state;
+    const providerName = mapProviderName(event.provider);
+    const attributes = selectAttributes(telemetry, {
+      "gen_ai.operation.name": "chat",
+      "gen_ai.provider.name": providerName,
+      "gen_ai.request.model": event.modelId,
+      "gen_ai.request.frequency_penalty": state.settings.frequencyPenalty,
+      "gen_ai.request.max_tokens": state.settings.maxOutputTokens,
+      "gen_ai.request.presence_penalty": state.settings.presencePenalty,
+      "gen_ai.request.stop_sequences": state.settings.stopSequences,
+      "gen_ai.request.temperature": (_a = state.settings.temperature) != null ? _a : void 0,
+      "gen_ai.request.top_k": state.settings.topK,
+      "gen_ai.request.top_p": state.settings.topP,
+      "gen_ai.input.messages": {
+        input: () => event.promptMessages ? JSON.stringify(formatInputMessages(event.promptMessages)) : void 0
+      },
+      "gen_ai.tool.definitions": {
+        input: () => event.stepTools ? JSON.stringify(event.stepTools) : void 0
+      }
+    });
+    const spanName = `chat ${event.modelId}`;
+    state.stepSpan = this.tracer.startSpan(
+      spanName,
+      { attributes, kind: SpanKind.CLIENT },
+      state.rootContext
+    );
+    state.stepContext = trace.setSpan(state.rootContext, state.stepSpan);
+  }
+  onToolCallStart(event) {
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.stepContext)) return;
+    const { telemetry } = state;
+    const { toolCall } = event;
+    const attributes = selectAttributes(telemetry, {
+      "gen_ai.operation.name": "execute_tool",
+      "gen_ai.tool.name": toolCall.toolName,
+      "gen_ai.tool.call.id": toolCall.toolCallId,
+      "gen_ai.tool.type": "function",
+      "gen_ai.tool.call.arguments": {
+        input: () => JSON.stringify(toolCall.input)
+      }
+    });
+    const spanName = `execute_tool ${toolCall.toolName}`;
+    const toolSpan = this.tracer.startSpan(
+      spanName,
+      { attributes, kind: SpanKind.INTERNAL },
+      state.stepContext
+    );
+    const toolContext = trace.setSpan(state.stepContext, toolSpan);
+    state.toolSpans.set(toolCall.toolCallId, {
+      span: toolSpan,
+      context: toolContext
+    });
+  }
+  onToolCallFinish(event) {
+    const state = this.getCallState(event.callId);
+    if (!state) return;
+    const toolSpanEntry = state.toolSpans.get(event.toolCall.toolCallId);
+    if (!toolSpanEntry) return;
+    const { span } = toolSpanEntry;
+    const { telemetry } = state;
+    if (event.success) {
+      try {
+        span.setAttributes(
+          selectAttributes(telemetry, {
+            "gen_ai.tool.call.result": {
+              output: () => JSON.stringify(event.output)
+            }
+          })
+        );
+      } catch (e) {
+      }
+    } else {
+      recordSpanError(span, event.error);
+    }
+    span.end();
+    state.toolSpans.delete(event.toolCall.toolCallId);
+  }
+  onStepFinish(event) {
+    var _a, _b, _c;
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.stepSpan)) return;
+    const { telemetry } = state;
+    state.stepSpan.setAttributes(
+      selectAttributes(telemetry, {
+        "gen_ai.response.finish_reasons": [event.finishReason],
+        "gen_ai.response.id": event.response.id,
+        "gen_ai.response.model": event.response.modelId,
+        "gen_ai.usage.input_tokens": event.usage.inputTokens,
+        "gen_ai.usage.output_tokens": event.usage.outputTokens,
+        "gen_ai.usage.cache_read.input_tokens": (_b = (_a = event.usage.inputTokenDetails) == null ? void 0 : _a.cacheReadTokens) != null ? _b : event.usage.cachedInputTokens,
+        "gen_ai.usage.cache_creation.input_tokens": (_c = event.usage.inputTokenDetails) == null ? void 0 : _c.cacheWriteTokens,
+        "gen_ai.output.messages": {
+          output: () => {
+            var _a2;
+            return JSON.stringify(
+              formatOutputMessages({
+                text: (_a2 = event.text) != null ? _a2 : void 0,
+                reasoning: event.reasoning,
+                toolCalls: event.toolCalls,
+                files: event.files,
+                finishReason: event.finishReason
+              })
+            );
+          }
+        }
+      })
+    );
+    state.stepSpan.end();
+    state.stepSpan = void 0;
+    state.stepContext = void 0;
+  }
+  onFinish(event) {
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.rootSpan)) return;
+    if (state.operationId === "ai.embed" || state.operationId === "ai.embedMany") {
+      this.onEmbedOperationFinish(event);
+      return;
+    }
+    if (state.operationId === "ai.rerank") {
+      this.onRerankOperationFinish(event);
+      return;
+    }
+    if (state.operationId === "ai.generateObject" || state.operationId === "ai.streamObject") {
+      this.onObjectOperationFinish(event);
+      return;
+    }
+    this.onGenerateFinish(event);
+  }
+  onGenerateFinish(event) {
+    var _a, _b, _c;
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.rootSpan)) return;
+    const { telemetry } = state;
+    state.rootSpan.setAttributes(
+      selectAttributes(telemetry, {
+        "gen_ai.response.finish_reasons": [event.finishReason],
+        "gen_ai.usage.input_tokens": event.totalUsage.inputTokens,
+        "gen_ai.usage.output_tokens": event.totalUsage.outputTokens,
+        "gen_ai.usage.cache_read.input_tokens": (_b = (_a = event.totalUsage.inputTokenDetails) == null ? void 0 : _a.cacheReadTokens) != null ? _b : event.totalUsage.cachedInputTokens,
+        "gen_ai.usage.cache_creation.input_tokens": (_c = event.totalUsage.inputTokenDetails) == null ? void 0 : _c.cacheWriteTokens,
+        "gen_ai.output.messages": {
+          output: () => {
+            var _a2;
+            return JSON.stringify(
+              formatOutputMessages({
+                text: (_a2 = event.text) != null ? _a2 : void 0,
+                reasoning: event.reasoning,
+                toolCalls: event.toolCalls,
+                files: event.files,
+                finishReason: event.finishReason
+              })
+            );
+          }
+        }
+      })
+    );
+    state.rootSpan.end();
+    this.cleanupCallState(event.callId);
+  }
+  onObjectOperationFinish(event) {
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.rootSpan)) return;
+    const { telemetry } = state;
+    state.rootSpan.setAttributes(
+      selectAttributes(telemetry, {
+        "gen_ai.response.finish_reasons": [event.finishReason],
+        "gen_ai.usage.input_tokens": event.usage.inputTokens,
+        "gen_ai.usage.output_tokens": event.usage.outputTokens,
+        "gen_ai.usage.cache_read.input_tokens": event.usage.cachedInputTokens,
+        "gen_ai.output.messages": {
+          output: () => event.object != null ? JSON.stringify(
+            formatObjectOutputMessages({
+              objectText: JSON.stringify(event.object),
+              finishReason: event.finishReason
+            })
+          ) : void 0
+        }
+      })
+    );
+    state.rootSpan.end();
+    this.cleanupCallState(event.callId);
+  }
+  onEmbedOperationFinish(event) {
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.rootSpan)) return;
+    const { telemetry } = state;
+    state.rootSpan.setAttributes(
+      selectAttributes(telemetry, {
+        "gen_ai.usage.input_tokens": event.usage.tokens
+      })
+    );
+    state.rootSpan.end();
+    this.cleanupCallState(event.callId);
+  }
+  onEmbedStart(event) {
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.rootSpan) || !state.rootContext) return;
+    const { telemetry } = state;
+    const providerName = mapProviderName(state.provider);
+    const attributes = selectAttributes(telemetry, {
+      "gen_ai.operation.name": "embeddings",
+      "gen_ai.provider.name": providerName,
+      "gen_ai.request.model": state.modelId
+    });
+    const spanName = `embeddings ${state.modelId}`;
+    const embedSpan = this.tracer.startSpan(
+      spanName,
+      { attributes, kind: SpanKind.CLIENT },
+      state.rootContext
+    );
+    const embedContext = trace.setSpan(state.rootContext, embedSpan);
+    state.embedSpans.set(event.embedCallId, {
+      span: embedSpan,
+      context: embedContext
+    });
+  }
+  onEmbedFinish(event) {
+    const state = this.getCallState(event.callId);
+    if (!state) return;
+    const embedSpanEntry = state.embedSpans.get(event.embedCallId);
+    if (!embedSpanEntry) return;
+    const { span } = embedSpanEntry;
+    const { telemetry } = state;
+    span.setAttributes(
+      selectAttributes(telemetry, {
+        "gen_ai.usage.input_tokens": event.usage.tokens
+      })
+    );
+    span.end();
+    state.embedSpans.delete(event.embedCallId);
+  }
+  onRerankOperationStart(event) {
+    const telemetry = {
+      isEnabled: event.isEnabled,
+      recordInputs: event.recordInputs,
+      recordOutputs: event.recordOutputs,
+      functionId: event.functionId
+    };
+    const settings = {
+      maxRetries: event.maxRetries
+    };
+    const providerName = mapProviderName(event.provider);
+    const attributes = selectAttributes(telemetry, {
+      "gen_ai.operation.name": "rerank",
+      "gen_ai.provider.name": providerName,
+      "gen_ai.request.model": event.modelId
+    });
+    const spanName = `rerank ${event.modelId}`;
+    const rootSpan = this.tracer.startSpan(spanName, {
+      attributes,
+      kind: SpanKind.CLIENT
+    });
+    const rootContext = trace.setSpan(context.active(), rootSpan);
+    this.callStates.set(event.callId, {
+      operationId: event.operationId,
+      telemetry,
+      rootSpan,
+      rootContext,
+      stepSpan: void 0,
+      stepContext: void 0,
+      embedSpans: /* @__PURE__ */ new Map(),
+      rerankSpan: void 0,
+      toolSpans: /* @__PURE__ */ new Map(),
+      settings,
+      provider: event.provider,
+      modelId: event.modelId
+    });
+  }
+  onRerankOperationFinish(event) {
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.rootSpan)) return;
+    state.rootSpan.end();
+    this.cleanupCallState(event.callId);
+  }
+  onRerankStart(event) {
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.rootSpan) || !state.rootContext) return;
+    const { telemetry } = state;
+    const providerName = mapProviderName(state.provider);
+    const attributes = selectAttributes(telemetry, {
+      "gen_ai.operation.name": "rerank",
+      "gen_ai.provider.name": providerName,
+      "gen_ai.request.model": state.modelId
+    });
+    const spanName = `rerank ${state.modelId}`;
+    const rerankSpan = this.tracer.startSpan(
+      spanName,
+      { attributes, kind: SpanKind.CLIENT },
+      state.rootContext
+    );
+    const rerankContext = trace.setSpan(state.rootContext, rerankSpan);
+    state.rerankSpan = { span: rerankSpan, context: rerankContext };
+  }
+  onRerankFinish(event) {
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.rerankSpan)) return;
+    const { span } = state.rerankSpan;
+    span.end();
+    state.rerankSpan = void 0;
+  }
+  onChunk(_event) {
+  }
+  onError(error) {
+    var _a;
+    const event = error;
+    if (!(event == null ? void 0 : event.callId)) return;
+    const state = this.getCallState(event.callId);
+    if (!(state == null ? void 0 : state.rootSpan)) return;
+    const actualError = (_a = event.error) != null ? _a : error;
+    if (state.stepSpan) {
+      recordSpanError(state.stepSpan, actualError);
+      state.stepSpan.end();
+    }
+    for (const { span: embedSpan } of state.embedSpans.values()) {
+      recordSpanError(embedSpan, actualError);
+      embedSpan.end();
+    }
+    state.embedSpans.clear();
+    if (state.rerankSpan) {
+      recordSpanError(state.rerankSpan.span, actualError);
+      state.rerankSpan.span.end();
+      state.rerankSpan = void 0;
+    }
+    recordSpanError(state.rootSpan, actualError);
+    state.rootSpan.end();
+    this.cleanupCallState(event.callId);
   }
-  return to;
 };
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-
-// src/index.ts
-var index_exports = {};
-__export(index_exports, {
-  OpenTelemetryIntegration: () => OpenTelemetryIntegration
-});
-module.exports = __toCommonJS(index_exports);
 
 // src/open-telemetry-integration.ts
-var import_api = require("@opentelemetry/api");
+import {
+  context as context2,
+  SpanStatusCode as SpanStatusCode2,
+  trace as trace2
+} from "@opentelemetry/api";
 
 // src/assemble-operation-name.ts
 function assembleOperationName({
@@ -46,10 +1085,9 @@ function assembleOperationName({
 function getBaseTelemetryAttributes({
   model,
   settings,
-  telemetry,
-  headers
+  headers,
+  context: context3
 }) {
-  var _a;
   return {
     "ai.model.provider": model.provider,
     "ai.model.id": model.modelId,
@@ -58,16 +1096,13 @@ function getBaseTelemetryAttributes({
       attributes[`ai.settings.${key}`] = value;
       return attributes;
     }, {}),
-    // add metadata as attributes:
-    ...Object.entries((_a = telemetry == null ? void 0 : telemetry.metadata) != null ? _a : {}).reduce(
-      (attributes, [key, value]) => {
-        if (value != void 0) {
-          attributes[`ai.telemetry.metadata.${key}`] = value;
-        }
-        return attributes;
-      },
-      {}
-    ),
+    // add context as attributes:
+    ...Object.entries(context3 != null ? context3 : {}).reduce((attributes, [key, value]) => {
+      if (value != void 0) {
+        attributes[`ai.settings.context.${key}`] = value;
+      }
+      return attributes;
+    }, {}),
     // request headers
     ...Object.entries(headers != null ? headers : {}).reduce((attributes, [key, value]) => {
       if (value !== void 0) {
@@ -79,7 +1114,7 @@ function getBaseTelemetryAttributes({
 }
 
 // src/stringify-for-telemetry.ts
-var import_ai = require("ai");
+import { convertDataContentToBase64String as convertDataContentToBase64String2 } from "ai";
 function stringifyForTelemetry(prompt) {
   return JSON.stringify(
     prompt.map((message) => ({
@@ -87,7 +1122,7 @@ function stringifyForTelemetry(prompt) {
       content: typeof message.content === "string" ? message.content : message.content.map(
         (part) => part.type === "file" ? {
           ...part,
-          data: part.data instanceof Uint8Array ? (0, import_ai.convertDataContentToBase64String)(part.data) : part.data
+          data: part.data instanceof Uint8Array ? convertDataContentToBase64String2(part.data) : part.data
         } : part
       )
     }))
@@ -95,7 +1130,7 @@ function stringifyForTelemetry(prompt) {
 }
 
 // src/open-telemetry-integration.ts
-function recordSpanError(span, error) {
+function recordSpanError2(span, error) {
   if (error instanceof Error) {
     span.recordException({
       name: error.name,
@@ -103,18 +1138,18 @@ function recordSpanError(span, error) {
       stack: error.stack
     });
     span.setStatus({
-      code: import_api.SpanStatusCode.ERROR,
+      code: SpanStatusCode2.ERROR,
       message: error.message
     });
   } else {
-    span.setStatus({ code: import_api.SpanStatusCode.ERROR });
+    span.setStatus({ code: SpanStatusCode2.ERROR });
   }
 }
-function shouldRecord(telemetry) {
-  return (telemetry == null ? void 0 : telemetry.isEnabled) === true;
+function shouldRecord2(telemetry) {
+  return (telemetry == null ? void 0 : telemetry.isEnabled) !== false;
 }
-function selectAttributes(telemetry, attributes) {
-  if (!shouldRecord(telemetry)) {
+function selectAttributes2(telemetry, attributes) {
+  if (!shouldRecord2(telemetry)) {
     return {};
   }
   const result = {};
@@ -140,7 +1175,7 @@ var OpenTelemetryIntegration = class {
   constructor(options = {}) {
     this.callStates = /* @__PURE__ */ new Map();
     var _a;
-    this.tracer = (_a = options.tracer) != null ? _a : import_api.trace.getTracer("ai");
+    this.tracer = (_a = options.tracer) != null ? _a : trace2.getTracer("ai");
   }
   getCallState(callId) {
     return this.callStates.get(callId);
@@ -158,10 +1193,10 @@ var OpenTelemetryIntegration = class {
     if (toolSpanEntry == null) {
       return execute();
     }
-    return import_api.context.with(toolSpanEntry.context, execute);
+    return context2.with(toolSpanEntry.context, execute);
   }
   onStart(event) {
-    if (event.isEnabled !== true) return;
+    if (event.isEnabled === false) return;
     if (event.operationId === "ai.embed" || event.operationId === "ai.embedMany") {
       this.onEmbedOperationStart(event);
       return;
@@ -181,8 +1216,7 @@ var OpenTelemetryIntegration = class {
       isEnabled: event.isEnabled,
       recordInputs: event.recordInputs,
       recordOutputs: event.recordOutputs,
-      functionId: event.functionId,
-      metadata: event.metadata
+      functionId: event.functionId
     };
     const settings = {
       maxOutputTokens: event.maxOutputTokens,
@@ -197,11 +1231,11 @@ var OpenTelemetryIntegration = class {
     };
     const baseTelemetryAttributes = getBaseTelemetryAttributes({
       model: { provider: event.provider, modelId: event.modelId },
-      telemetry,
       headers: event.headers,
-      settings
+      settings,
+      context: event.runtimeContext
     });
-    const attributes = selectAttributes(telemetry, {
+    const attributes = selectAttributes2(telemetry, {
       ...assembleOperationName({
         operationId: event.operationId,
         telemetry
@@ -218,7 +1252,7 @@ var OpenTelemetryIntegration = class {
       }
     });
     const rootSpan = this.tracer.startSpan(event.operationId, { attributes });
-    const rootContext = import_api.trace.setSpan(import_api.context.active(), rootSpan);
+    const rootContext = trace2.setSpan(context2.active(), rootSpan);
     this.callStates.set(event.callId, {
       operationId: event.operationId,
       telemetry,
@@ -238,8 +1272,7 @@ var OpenTelemetryIntegration = class {
       isEnabled: event.isEnabled,
       recordInputs: event.recordInputs,
       recordOutputs: event.recordOutputs,
-      functionId: event.functionId,
-      metadata: event.metadata
+      functionId: event.functionId
     };
     const settings = {
       maxOutputTokens: event.maxOutputTokens,
@@ -253,11 +1286,11 @@ var OpenTelemetryIntegration = class {
     };
     const baseTelemetryAttributes = getBaseTelemetryAttributes({
       model: { provider: event.provider, modelId: event.modelId },
-      telemetry,
       headers: event.headers,
-      settings
+      settings,
+      context: void 0
     });
-    const attributes = selectAttributes(telemetry, {
+    const attributes = selectAttributes2(telemetry, {
       ...assembleOperationName({
         operationId: event.operationId,
         telemetry
@@ -276,7 +1309,7 @@ var OpenTelemetryIntegration = class {
       "ai.settings.output": event.output
     });
     const rootSpan = this.tracer.startSpan(event.operationId, { attributes });
-    const rootContext = import_api.trace.setSpan(import_api.context.active(), rootSpan);
+    const rootContext = trace2.setSpan(context2.active(), rootSpan);
     this.callStates.set(event.callId, {
       operationId: event.operationId,
       telemetry,
@@ -298,7 +1331,7 @@ var OpenTelemetryIntegration = class {
     if (!(state == null ? void 0 : state.rootSpan) || !state.rootContext) return;
     const { telemetry } = state;
     const stepOperationId = state.operationId === "ai.streamObject" ? "ai.streamObject.doStream" : "ai.generateObject.doGenerate";
-    const attributes = selectAttributes(telemetry, {
+    const attributes = selectAttributes2(telemetry, {
       ...assembleOperationName({
         operationId: stepOperationId,
         telemetry
@@ -321,7 +1354,7 @@ var OpenTelemetryIntegration = class {
       { attributes },
       state.rootContext
     );
-    state.stepContext = import_api.trace.setSpan(state.rootContext, state.stepSpan);
+    state.stepContext = trace2.setSpan(state.rootContext, state.stepSpan);
   }
   /** @deprecated */
   onObjectStepFinish(event) {
@@ -329,7 +1362,7 @@ var OpenTelemetryIntegration = class {
     if (!(state == null ? void 0 : state.stepSpan)) return;
     const { telemetry } = state;
     state.stepSpan.setAttributes(
-      selectAttributes(telemetry, {
+      selectAttributes2(telemetry, {
         "ai.response.finishReason": event.finishReason,
         "ai.response.object": {
           output: () => {
@@ -373,21 +1406,20 @@ var OpenTelemetryIntegration = class {
       isEnabled: event.isEnabled,
       recordInputs: event.recordInputs,
       recordOutputs: event.recordOutputs,
-      functionId: event.functionId,
-      metadata: event.metadata
+      functionId: event.functionId
     };
     const settings = {
       maxRetries: event.maxRetries
     };
     const baseTelemetryAttributes = getBaseTelemetryAttributes({
       model: { provider: event.provider, modelId: event.modelId },
-      telemetry,
       headers: event.headers,
-      settings
+      settings,
+      context: void 0
     });
     const value = event.value;
     const isMany = event.operationId === "ai.embedMany";
-    const attributes = selectAttributes(telemetry, {
+    const attributes = selectAttributes2(telemetry, {
       ...assembleOperationName({
         operationId: event.operationId,
         telemetry
@@ -404,7 +1436,7 @@ var OpenTelemetryIntegration = class {
       }
     });
     const rootSpan = this.tracer.startSpan(event.operationId, { attributes });
-    const rootContext = import_api.trace.setSpan(import_api.context.active(), rootSpan);
+    const rootContext = trace2.setSpan(context2.active(), rootSpan);
     this.callStates.set(event.callId, {
       operationId: event.operationId,
       telemetry,
@@ -425,7 +1457,7 @@ var OpenTelemetryIntegration = class {
     if (!(state == null ? void 0 : state.rootSpan) || !state.rootContext) return;
     const { telemetry } = state;
     const stepOperationId = state.operationId === "ai.streamText" ? "ai.streamText.doStream" : "ai.generateText.doGenerate";
-    const attributes = selectAttributes(telemetry, {
+    const attributes = selectAttributes2(telemetry, {
       ...assembleOperationName({
         operationId: stepOperationId,
         telemetry
@@ -460,14 +1492,14 @@ var OpenTelemetryIntegration = class {
       { attributes },
       state.rootContext
     );
-    state.stepContext = import_api.trace.setSpan(state.rootContext, state.stepSpan);
+    state.stepContext = trace2.setSpan(state.rootContext, state.stepSpan);
   }
   onToolCallStart(event) {
     const state = this.getCallState(event.callId);
     if (!(state == null ? void 0 : state.stepContext)) return;
     const { telemetry } = state;
     const { toolCall } = event;
-    const attributes = selectAttributes(telemetry, {
+    const attributes = selectAttributes2(telemetry, {
       ...assembleOperationName({
         operationId: "ai.toolCall",
         telemetry
@@ -483,7 +1515,7 @@ var OpenTelemetryIntegration = class {
       { attributes },
       state.stepContext
     );
-    const toolContext = import_api.trace.setSpan(state.stepContext, toolSpan);
+    const toolContext = trace2.setSpan(state.stepContext, toolSpan);
     state.toolSpans.set(toolCall.toolCallId, {
       span: toolSpan,
       context: toolContext
@@ -499,16 +1531,16 @@ var OpenTelemetryIntegration = class {
     if (event.success) {
       try {
         span.setAttributes(
-          selectAttributes(telemetry, {
+          selectAttributes2(telemetry, {
             "ai.toolCall.result": {
               output: () => JSON.stringify(event.output)
             }
           })
         );
-      } catch (_ignored) {
+      } catch (e) {
       }
     } else {
-      recordSpanError(span, event.error);
+      recordSpanError2(span, event.error);
     }
     span.end();
     state.toolSpans.delete(event.toolCall.toolCallId);
@@ -519,7 +1551,7 @@ var OpenTelemetryIntegration = class {
     if (!(state == null ? void 0 : state.stepSpan)) return;
     const { telemetry } = state;
     state.stepSpan.setAttributes(
-      selectAttributes(telemetry, {
+      selectAttributes2(telemetry, {
         "ai.response.finishReason": event.finishReason,
         "ai.response.text": {
           output: () => {
@@ -596,7 +1628,7 @@ var OpenTelemetryIntegration = class {
     if (!(state == null ? void 0 : state.rootSpan)) return;
     const { telemetry } = state;
     state.rootSpan.setAttributes(
-      selectAttributes(telemetry, {
+      selectAttributes2(telemetry, {
         "ai.response.finishReason": event.finishReason,
         "ai.response.text": {
           output: () => {
@@ -646,7 +1678,7 @@ var OpenTelemetryIntegration = class {
     if (!(state == null ? void 0 : state.rootSpan)) return;
     const { telemetry } = state;
     state.rootSpan.setAttributes(
-      selectAttributes(telemetry, {
+      selectAttributes2(telemetry, {
         "ai.response.finishReason": event.finishReason,
         "ai.response.object": {
           output: () => event.object != null ? JSON.stringify(event.object) : void 0
@@ -668,7 +1700,7 @@ var OpenTelemetryIntegration = class {
     const { telemetry } = state;
     const isMany = state.operationId === "ai.embedMany";
     state.rootSpan.setAttributes(
-      selectAttributes(telemetry, {
+      selectAttributes2(telemetry, {
         ...isMany ? {
           "ai.embeddings": {
             output: () => event.embedding.map((e) => JSON.stringify(e))
@@ -688,7 +1720,7 @@ var OpenTelemetryIntegration = class {
     const state = this.getCallState(event.callId);
     if (!(state == null ? void 0 : state.rootSpan) || !state.rootContext) return;
     const { telemetry } = state;
-    const attributes = selectAttributes(telemetry, {
+    const attributes = selectAttributes2(telemetry, {
       ...assembleOperationName({
         operationId: event.operationId,
         telemetry
@@ -703,7 +1735,7 @@ var OpenTelemetryIntegration = class {
       { attributes },
       state.rootContext
     );
-    const embedContext = import_api.trace.setSpan(state.rootContext, embedSpan);
+    const embedContext = trace2.setSpan(state.rootContext, embedSpan);
     state.embedSpans.set(event.embedCallId, {
       span: embedSpan,
       context: embedContext
@@ -717,7 +1749,7 @@ var OpenTelemetryIntegration = class {
     const { span } = embedSpanEntry;
     const { telemetry } = state;
     span.setAttributes(
-      selectAttributes(telemetry, {
+      selectAttributes2(telemetry, {
         "ai.embeddings": {
           output: () => event.embeddings.map((embedding) => JSON.stringify(embedding))
         },
@@ -732,19 +1764,18 @@ var OpenTelemetryIntegration = class {
       isEnabled: event.isEnabled,
       recordInputs: event.recordInputs,
       recordOutputs: event.recordOutputs,
-      functionId: event.functionId,
-      metadata: event.metadata
+      functionId: event.functionId
     };
     const settings = {
       maxRetries: event.maxRetries
     };
     const baseTelemetryAttributes = getBaseTelemetryAttributes({
       model: { provider: event.provider, modelId: event.modelId },
-      telemetry,
       headers: event.headers,
-      settings
+      settings,
+      context: void 0
     });
-    const attributes = selectAttributes(telemetry, {
+    const attributes = selectAttributes2(telemetry, {
       ...assembleOperationName({
         operationId: event.operationId,
         telemetry
@@ -755,7 +1786,7 @@ var OpenTelemetryIntegration = class {
       }
     });
     const rootSpan = this.tracer.startSpan(event.operationId, { attributes });
-    const rootContext = import_api.trace.setSpan(import_api.context.active(), rootSpan);
+    const rootContext = trace2.setSpan(context2.active(), rootSpan);
     this.callStates.set(event.callId, {
       operationId: event.operationId,
       telemetry,
@@ -780,7 +1811,7 @@ var OpenTelemetryIntegration = class {
     const state = this.getCallState(event.callId);
     if (!(state == null ? void 0 : state.rootSpan) || !state.rootContext) return;
     const { telemetry } = state;
-    const attributes = selectAttributes(telemetry, {
+    const attributes = selectAttributes2(telemetry, {
       ...assembleOperationName({
         operationId: event.operationId,
         telemetry
@@ -795,7 +1826,7 @@ var OpenTelemetryIntegration = class {
       { attributes },
       state.rootContext
     );
-    const rerankContext = import_api.trace.setSpan(state.rootContext, rerankSpan);
+    const rerankContext = trace2.setSpan(state.rootContext, rerankSpan);
     state.rerankSpan = { span: rerankSpan, context: rerankContext };
   }
   onRerankFinish(event) {
@@ -804,7 +1835,7 @@ var OpenTelemetryIntegration = class {
     const { span } = state.rerankSpan;
     const { telemetry } = state;
     span.setAttributes(
-      selectAttributes(telemetry, {
+      selectAttributes2(telemetry, {
         "ai.ranking.type": event.documentsType,
         "ai.ranking": {
           output: () => event.ranking.map((r) => JSON.stringify(r))
@@ -843,26 +1874,26 @@ var OpenTelemetryIntegration = class {
     if (!(state == null ? void 0 : state.rootSpan)) return;
     const actualError = (_a = event.error) != null ? _a : error;
     if (state.stepSpan) {
-      recordSpanError(state.stepSpan, actualError);
+      recordSpanError2(state.stepSpan, actualError);
       state.stepSpan.end();
     }
     for (const { span: embedSpan } of state.embedSpans.values()) {
-      recordSpanError(embedSpan, actualError);
+      recordSpanError2(embedSpan, actualError);
       embedSpan.end();
     }
     state.embedSpans.clear();
     if (state.rerankSpan) {
-      recordSpanError(state.rerankSpan.span, actualError);
+      recordSpanError2(state.rerankSpan.span, actualError);
       state.rerankSpan.span.end();
       state.rerankSpan = void 0;
     }
-    recordSpanError(state.rootSpan, actualError);
+    recordSpanError2(state.rootSpan, actualError);
     state.rootSpan.end();
     this.cleanupCallState(event.callId);
   }
 };
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
+export {
+  GenAIOpenTelemetryIntegration,
   OpenTelemetryIntegration
-});
+};
 //# sourceMappingURL=index.js.map