@ai-sdk/otel 1.0.0-beta.59 → 1.0.0-beta.60

package/dist/index.js

@@ -1,8 +1,7 @@
 // src/open-telemetry.ts
 import {
-  context,
+  context as context2,
   SpanKind,
-  SpanStatusCode,
   trace
 } from "@opentelemetry/api";
 
@@ -58,7 +57,7 @@ function formatSystemInstructions(system) {
   return [{ type: "text", content: system.content }];
 }
 function convertMessagePartToSemConv(part) {
-  var _a, _b, _c, _d;
+  var _a, _b, _c, _d, _e;
   switch (part.type) {
     case "text":
       return { type: "text", content: part.text };
@@ -92,7 +91,22 @@ function convertMessagePartToSemConv(part) {
       };
     }
     case "file": {
-      const data = part.data;
+      const rawData = part.data;
+      const data = (() => {
+        if (typeof rawData === "object" && rawData !== null && !(rawData instanceof URL) && !(rawData instanceof Uint8Array) && !(rawData instanceof ArrayBuffer) && "type" in rawData) {
+          switch (rawData.type) {
+            case "data":
+              return rawData.data;
+            case "url":
+              return rawData.url;
+            case "text":
+              return rawData.text;
+            default:
+              return "";
+          }
+        }
+        return rawData;
+      })();
       let content;
       if (data instanceof Uint8Array) {
         content = convertDataContentToBase64String(data);
@@ -106,13 +120,20 @@ function convertMessagePartToSemConv(part) {
           };
         }
         content = data;
+      } else if (data instanceof URL) {
+        return {
+          type: "uri",
+          modality: getModality(part.mediaType),
+          mime_type: (_d = part.mediaType) != null ? _d : null,
+          uri: data.toString()
+        };
       } else {
         content = String(data);
       }
       return {
         type: "blob",
         modality: getModality(part.mediaType),
-        mime_type: (_d = part.mediaType) != null ? _d : null,
+        mime_type: (_e = part.mediaType) != null ? _e : null,
         content
       };
     }
@@ -224,7 +245,22 @@ function convertModelMessageToSemConv(msg) {
           };
         }
         case "file": {
-          const data = part.data;
+          const rawData = part.data;
+          const data = (() => {
+            if (typeof rawData === "object" && rawData !== null && !(rawData instanceof URL) && !(rawData instanceof Uint8Array) && !(rawData instanceof ArrayBuffer) && "type" in rawData) {
+              switch (rawData.type) {
+                case "data":
+                  return rawData.data;
+                case "url":
+                  return rawData.url;
+                case "text":
+                  return rawData.text;
+                default:
+                  return "";
+              }
+            }
+            return rawData;
+          })();
           if (data instanceof URL) {
             return {
               type: "uri",
@@ -410,8 +446,12 @@ function mapFinishReason(reason) {
   return (_a = mapping[reason]) != null ? _a : reason;
 }
 
-// src/open-telemetry.ts
-function recordSpanError(span, error) {
+// src/record-span.ts
+import {
+  SpanStatusCode,
+  context
+} from "@opentelemetry/api";
+function recordErrorOnSpan(span, error) {
   if (error instanceof Error) {
     span.recordException({
       name: error.name,
@@ -426,6 +466,8 @@ function recordSpanError(span, error) {
     span.setStatus({ code: SpanStatusCode.ERROR });
   }
 }
+
+// src/select-attributes.ts
 function shouldRecord(telemetry) {
   return (telemetry == null ? void 0 : telemetry.isEnabled) !== false;
 }
@@ -452,11 +494,68 @@ function selectAttributes(telemetry, attributes) {
   }
   return result;
 }
+
+// src/supplemental-attributes.ts
+var disabledSupplementalAttributes = {
+  usage: false,
+  providerMetadata: false,
+  embedding: false,
+  reranking: false,
+  runtimeContext: false,
+  headers: false,
+  toolChoice: false,
+  schema: false
+};
+function normalizeSupplementalAttributes(options) {
+  var _a, _b, _c, _d, _e, _f, _g, _h;
+  return {
+    ...disabledSupplementalAttributes,
+    usage: (_a = options.usage) != null ? _a : false,
+    providerMetadata: (_b = options.providerMetadata) != null ? _b : false,
+    embedding: (_c = options.embedding) != null ? _c : false,
+    reranking: (_d = options.reranking) != null ? _d : false,
+    runtimeContext: (_e = options.runtimeContext) != null ? _e : false,
+    headers: (_f = options.headers) != null ? _f : false,
+    toolChoice: (_g = options.toolChoice) != null ? _g : false,
+    schema: (_h = options.schema) != null ? _h : false
+  };
+}
+function getRuntimeContextAttributes(context4) {
+  return Object.fromEntries(
+    Object.entries(context4 != null ? context4 : {}).filter(([, value]) => value != null).map(([key, value]) => [`ai.settings.context.${key}`, value])
+  );
+}
+function getHeaderAttributes(headers) {
+  return Object.fromEntries(
+    Object.entries(headers != null ? headers : {}).filter(([, value]) => value != null).map(([key, value]) => [`ai.request.headers.${key}`, value])
+  );
+}
+function getDetailedUsageAttributes(usage) {
+  var _a, _b, _c;
+  return {
+    "ai.usage.inputTokenDetails.noCacheTokens": (_a = usage.inputTokenDetails) == null ? void 0 : _a.noCacheTokens,
+    "ai.usage.outputTokenDetails.textTokens": (_b = usage.outputTokenDetails) == null ? void 0 : _b.textTokens,
+    "ai.usage.outputTokenDetails.reasoningTokens": (_c = usage.outputTokenDetails) == null ? void 0 : _c.reasoningTokens
+  };
+}
+function selectSupplementalAttributes(telemetry, enabledAttributes, attributes) {
+  const result = {};
+  for (const [key, value] of Object.entries(attributes)) {
+    if (!enabledAttributes[key] || value == null) {
+      continue;
+    }
+    Object.assign(result, selectAttributes(telemetry, value));
+  }
+  return result;
+}
+
+// src/open-telemetry.ts
 var OpenTelemetry = class {
   constructor(options = {}) {
     this.callStates = /* @__PURE__ */ new Map();
     var _a;
     this.tracer = (_a = options.tracer) != null ? _a : trace.getTracer("gen_ai");
+    this.supplementalAttributes = normalizeSupplementalAttributes(options);
   }
   getCallState(callId) {
     return this.callStates.get(callId);
@@ -474,7 +573,7 @@ var OpenTelemetry = class {
     if (toolSpanEntry == null) {
       return execute();
     }
-    return context.with(toolSpanEntry.context, execute);
+    return context2.with(toolSpanEntry.context, execute);
   }
   onStart(event) {
     if (event.operationId === "ai.embed" || event.operationId === "ai.embedMany") {
@@ -515,6 +614,16 @@ var OpenTelemetry = class {
     };
     const providerName = mapProviderName(event.provider);
     const operationName = mapOperationName(event.operationId);
+    const baseSupplementalAttributes = selectSupplementalAttributes(
+      telemetry,
+      this.supplementalAttributes,
+      {
+        runtimeContext: getRuntimeContextAttributes(
+          event.runtimeContext
+        ),
+        headers: getHeaderAttributes(event.headers)
+      }
+    );
     const attributes = selectAttributes(telemetry, {
       "gen_ai.operation.name": operationName,
       "gen_ai.provider.name": providerName,
@@ -538,14 +647,15 @@ var OpenTelemetry = class {
             messages: event.messages
           })
         )
-      }
+      },
+      ...baseSupplementalAttributes
     });
     const spanName = `${operationName} ${event.modelId}`;
     const rootSpan = this.tracer.startSpan(spanName, {
       attributes,
       kind: SpanKind.INTERNAL
     });
-    const rootContext = trace.setSpan(context.active(), rootSpan);
+    const rootContext = trace.setSpan(context2.active(), rootSpan);
     this.callStates.set(event.callId, {
       operationId: event.operationId,
       telemetry,
@@ -560,7 +670,8 @@ var OpenTelemetry = class {
       toolSpans: /* @__PURE__ */ new Map(),
       settings,
       provider: event.provider,
-      modelId: event.modelId
+      modelId: event.modelId,
+      baseSupplementalAttributes
     });
   }
   onObjectOperationStart(event) {
@@ -582,6 +693,13 @@ var OpenTelemetry = class {
     };
     const providerName = mapProviderName(event.provider);
     const operationName = mapOperationName(event.operationId);
+    const baseSupplementalAttributes = selectSupplementalAttributes(
+      telemetry,
+      this.supplementalAttributes,
+      {
+        headers: getHeaderAttributes(event.headers)
+      }
+    );
     const attributes = selectAttributes(telemetry, {
       "gen_ai.operation.name": operationName,
       "gen_ai.provider.name": providerName,
@@ -605,14 +723,23 @@ var OpenTelemetry = class {
             messages: event.messages
           })
         )
-      }
+      },
+      ...baseSupplementalAttributes,
+      ...selectSupplementalAttributes(telemetry, this.supplementalAttributes, {
+        schema: {
+          "ai.schema": event.schema ? { input: () => JSON.stringify(event.schema) } : void 0,
+          "ai.schema.name": event.schemaName,
+          "ai.schema.description": event.schemaDescription,
+          "ai.settings.output": event.output
+        }
+      })
     });
     const spanName = `${operationName} ${event.modelId}`;
     const rootSpan = this.tracer.startSpan(spanName, {
       attributes,
       kind: SpanKind.INTERNAL
     });
-    const rootContext = trace.setSpan(context.active(), rootSpan);
+    const rootContext = trace.setSpan(context2.active(), rootSpan);
     this.callStates.set(event.callId, {
       operationId: event.operationId,
       telemetry,
@@ -627,7 +754,8 @@ var OpenTelemetry = class {
       toolSpans: /* @__PURE__ */ new Map(),
       settings,
       provider: event.provider,
-      modelId: event.modelId
+      modelId: event.modelId,
+      baseSupplementalAttributes
     });
   }
   /** @deprecated */
@@ -650,7 +778,8 @@ var OpenTelemetry = class {
       "gen_ai.request.top_p": state.settings.topP,
       "gen_ai.input.messages": {
         input: () => event.promptMessages ? JSON.stringify(formatInputMessages(event.promptMessages)) : void 0
-      }
+      },
+      ...state.baseSupplementalAttributes
     });
     const spanName = `chat ${event.modelId}`;
     state.inferenceSpan = this.tracer.startSpan(
@@ -689,7 +818,17 @@ var OpenTelemetry = class {
               return event.objectText;
             }
           }
-        }
+        },
+        ...selectSupplementalAttributes(
+          telemetry,
+          this.supplementalAttributes,
+          {
+            providerMetadata: {
+              "ai.response.providerMetadata": event.providerMetadata ? JSON.stringify(event.providerMetadata) : void 0
+            },
+            usage: getDetailedUsageAttributes(event.usage)
+          }
+        )
       })
     );
     state.inferenceSpan.end();
@@ -702,21 +841,39 @@ var OpenTelemetry = class {
       recordOutputs: event.recordOutputs,
       functionId: event.functionId
     };
-    const settings = {
-      maxRetries: event.maxRetries
-    };
     const providerName = mapProviderName(event.provider);
+    const baseSupplementalAttributes = selectSupplementalAttributes(
+      telemetry,
+      this.supplementalAttributes,
+      {
+        headers: getHeaderAttributes(event.headers)
+      }
+    );
+    const value = event.value;
+    const isMany = event.operationId === "ai.embedMany";
     const attributes = selectAttributes(telemetry, {
       "gen_ai.operation.name": "embeddings",
       "gen_ai.provider.name": providerName,
-      "gen_ai.request.model": event.modelId
+      "gen_ai.request.model": event.modelId,
+      ...baseSupplementalAttributes,
+      ...selectSupplementalAttributes(telemetry, this.supplementalAttributes, {
+        embedding: isMany ? {
+          "ai.values": {
+            input: () => value.map((v) => JSON.stringify(v))
+          }
+        } : {
+          "ai.value": {
+            input: () => JSON.stringify(value)
+          }
+        }
+      })
     });
     const spanName = `embeddings ${event.modelId}`;
     const rootSpan = this.tracer.startSpan(spanName, {
       attributes,
       kind: SpanKind.CLIENT
     });
-    const rootContext = trace.setSpan(context.active(), rootSpan);
+    const rootContext = trace.setSpan(context2.active(), rootSpan);
     this.callStates.set(event.callId, {
       operationId: event.operationId,
       telemetry,
@@ -729,9 +886,10 @@ var OpenTelemetry = class {
       embedSpans: /* @__PURE__ */ new Map(),
       rerankSpan: void 0,
       toolSpans: /* @__PURE__ */ new Map(),
-      settings,
+      settings: { maxRetries: event.maxRetries },
       provider: event.provider,
-      modelId: event.modelId
+      modelId: event.modelId,
+      baseSupplementalAttributes
     });
   }
   onStepStart(event) {
@@ -739,7 +897,15 @@ var OpenTelemetry = class {
     if (!(state == null ? void 0 : state.rootSpan) || !state.rootContext) return;
     const { telemetry } = state;
     const stepAttributes = selectAttributes(telemetry, {
-      "gen_ai.operation.name": "agent_step"
+      "gen_ai.operation.name": "agent_step",
+      ...state.baseSupplementalAttributes,
+      ...selectSupplementalAttributes(telemetry, this.supplementalAttributes, {
+        toolChoice: {
+          "ai.prompt.toolChoice": {
+            input: () => event.stepToolChoice != null ? JSON.stringify(event.stepToolChoice) : void 0
+          }
+        }
+      })
     });
     state.stepSpan = this.tracer.startSpan(
       `step ${event.steps.length + 1}`,
@@ -811,7 +977,14 @@ var OpenTelemetry = class {
               finishReason: event.finishReason
             })
           )
-        }
+        },
+        ...selectSupplementalAttributes(
+          telemetry,
+          this.supplementalAttributes,
+          {
+            usage: getDetailedUsageAttributes(event.usage)
+          }
+        )
       })
     );
     state.inferenceSpan.end();
@@ -864,7 +1037,7 @@ var OpenTelemetry = class {
       } catch (e) {
       }
     } else {
-      recordSpanError(span, toolOutput.error);
+      recordErrorOnSpan(span, toolOutput.error);
     }
     span.end();
     state.toolSpans.delete(event.toolCall.toolCallId);
@@ -872,6 +1045,15 @@ var OpenTelemetry = class {
   onStepFinish(event) {
     const state = this.getCallState(event.callId);
     if (!(state == null ? void 0 : state.stepSpan)) return;
+    const { telemetry } = state;
+    state.stepSpan.setAttributes(
+      selectSupplementalAttributes(telemetry, this.supplementalAttributes, {
+        providerMetadata: {
+          "ai.response.providerMetadata": event.providerMetadata ? JSON.stringify(event.providerMetadata) : void 0
+        },
+        usage: getDetailedUsageAttributes(event.usage)
+      })
+    );
     state.stepSpan.end();
     state.stepSpan = void 0;
     state.stepContext = void 0;
@@ -918,7 +1100,17 @@ var OpenTelemetry = class {
               })
             );
           }
-        }
+        },
+        ...selectSupplementalAttributes(
+          telemetry,
+          this.supplementalAttributes,
+          {
+            providerMetadata: {
+              "ai.response.providerMetadata": event.providerMetadata ? JSON.stringify(event.providerMetadata) : void 0
+            },
+            usage: getDetailedUsageAttributes(event.totalUsage)
+          }
+        )
       })
     );
     state.rootSpan.end();
@@ -941,7 +1133,17 @@ var OpenTelemetry = class {
               finishReason: event.finishReason
             })
           ) : void 0
-        }
+        },
+        ...selectSupplementalAttributes(
+          telemetry,
+          this.supplementalAttributes,
+          {
+            providerMetadata: {
+              "ai.response.providerMetadata": event.providerMetadata ? JSON.stringify(event.providerMetadata) : void 0
+            },
+            usage: getDetailedUsageAttributes(event.usage)
+          }
+        )
       })
     );
     state.rootSpan.end();
@@ -951,9 +1153,27 @@ var OpenTelemetry = class {
     const state = this.getCallState(event.callId);
     if (!(state == null ? void 0 : state.rootSpan)) return;
     const { telemetry } = state;
+    const isMany = state.operationId === "ai.embedMany";
     state.rootSpan.setAttributes(
       selectAttributes(telemetry, {
-        "gen_ai.usage.input_tokens": event.usage.tokens
+        "gen_ai.usage.input_tokens": event.usage.tokens,
+        ...selectSupplementalAttributes(
+          telemetry,
+          this.supplementalAttributes,
+          {
+            embedding: isMany ? {
+              "ai.embeddings": {
+                output: () => event.embedding.map(
+                  (e) => JSON.stringify(e)
+                )
+              }
+            } : {
+              "ai.embedding": {
+                output: () => JSON.stringify(event.embedding)
+              }
+            }
+          }
+        )
       })
     );
     state.rootSpan.end();
@@ -967,7 +1187,15 @@ var OpenTelemetry = class {
     const attributes = selectAttributes(telemetry, {
       "gen_ai.operation.name": "embeddings",
       "gen_ai.provider.name": providerName,
-      "gen_ai.request.model": state.modelId
+      "gen_ai.request.model": state.modelId,
+      ...state.baseSupplementalAttributes,
+      ...selectSupplementalAttributes(telemetry, this.supplementalAttributes, {
+        embedding: {
+          "ai.values": {
+            input: () => event.values.map((v) => JSON.stringify(v))
+          }
+        }
+      })
     });
     const spanName = `embeddings ${state.modelId}`;
     const embedSpan = this.tracer.startSpan(
@@ -990,7 +1218,18 @@ var OpenTelemetry = class {
     const { telemetry } = state;
     span.setAttributes(
       selectAttributes(telemetry, {
-        "gen_ai.usage.input_tokens": event.usage.tokens
+        "gen_ai.usage.input_tokens": event.usage.tokens,
+        ...selectSupplementalAttributes(
+          telemetry,
+          this.supplementalAttributes,
+          {
+            embedding: {
+              "ai.embeddings": {
+                output: () => event.embeddings.map((embedding) => JSON.stringify(embedding))
+              }
+            }
+          }
+        )
       })
     );
     span.end();
@@ -1002,21 +1241,33 @@ var OpenTelemetry = class {
       recordOutputs: event.recordOutputs,
       functionId: event.functionId
     };
-    const settings = {
-      maxRetries: event.maxRetries
-    };
     const providerName = mapProviderName(event.provider);
+    const baseSupplementalAttributes = selectSupplementalAttributes(
+      telemetry,
+      this.supplementalAttributes,
+      {
+        headers: getHeaderAttributes(event.headers)
+      }
+    );
     const attributes = selectAttributes(telemetry, {
       "gen_ai.operation.name": "rerank",
       "gen_ai.provider.name": providerName,
-      "gen_ai.request.model": event.modelId
+      "gen_ai.request.model": event.modelId,
+      ...baseSupplementalAttributes,
+      ...selectSupplementalAttributes(telemetry, this.supplementalAttributes, {
+        reranking: {
+          "ai.documents": {
+            input: () => event.documents.map((d) => JSON.stringify(d))
+          }
+        }
+      })
     });
     const spanName = `rerank ${event.modelId}`;
     const rootSpan = this.tracer.startSpan(spanName, {
       attributes,
       kind: SpanKind.CLIENT
     });
-    const rootContext = trace.setSpan(context.active(), rootSpan);
+    const rootContext = trace.setSpan(context2.active(), rootSpan);
     this.callStates.set(event.callId, {
       operationId: event.operationId,
       telemetry,
@@ -1029,9 +1280,10 @@ var OpenTelemetry = class {
       embedSpans: /* @__PURE__ */ new Map(),
       rerankSpan: void 0,
       toolSpans: /* @__PURE__ */ new Map(),
-      settings,
+      settings: { maxRetries: event.maxRetries },
       provider: event.provider,
-      modelId: event.modelId
+      modelId: event.modelId,
+      baseSupplementalAttributes
     });
   }
   onRerankOperationFinish(event) {
@@ -1048,7 +1300,15 @@ var OpenTelemetry = class {
     const attributes = selectAttributes(telemetry, {
       "gen_ai.operation.name": "rerank",
       "gen_ai.provider.name": providerName,
-      "gen_ai.request.model": state.modelId
+      "gen_ai.request.model": state.modelId,
+      ...state.baseSupplementalAttributes,
+      ...selectSupplementalAttributes(telemetry, this.supplementalAttributes, {
+        reranking: {
+          "ai.documents": {
+            input: () => event.documents.map((d) => JSON.stringify(d))
+          }
+        }
+      })
     });
     const spanName = `rerank ${state.modelId}`;
     const rerankSpan = this.tracer.startSpan(
@@ -1063,6 +1323,17 @@ var OpenTelemetry = class {
     const state = this.getCallState(event.callId);
     if (!(state == null ? void 0 : state.rerankSpan)) return;
     const { span } = state.rerankSpan;
+    const { telemetry } = state;
+    span.setAttributes(
+      selectSupplementalAttributes(telemetry, this.supplementalAttributes, {
+        reranking: {
+          "ai.ranking.type": event.documentsType,
+          "ai.ranking": {
+            output: () => event.ranking.map((r) => JSON.stringify(r))
+          }
+        }
+      })
+    );
     span.end();
     state.rerankSpan = void 0;
   }
@@ -1076,33 +1347,33 @@ var OpenTelemetry = class {
     if (!(state == null ? void 0 : state.rootSpan)) return;
     const actualError = (_a = event.error) != null ? _a : error;
     for (const { span: toolSpan } of state.toolSpans.values()) {
-      recordSpanError(toolSpan, actualError);
+      recordErrorOnSpan(toolSpan, actualError);
       toolSpan.end();
     }
     state.toolSpans.clear();
     if (state.inferenceSpan) {
-      recordSpanError(state.inferenceSpan, actualError);
+      recordErrorOnSpan(state.inferenceSpan, actualError);
       state.inferenceSpan.end();
       state.inferenceSpan = void 0;
       state.inferenceContext = void 0;
     }
     if (state.stepSpan) {
-      recordSpanError(state.stepSpan, actualError);
+      recordErrorOnSpan(state.stepSpan, actualError);
       state.stepSpan.end();
       state.stepSpan = void 0;
       state.stepContext = void 0;
     }
     for (const { span: embedSpan } of state.embedSpans.values()) {
-      recordSpanError(embedSpan, actualError);
+      recordErrorOnSpan(embedSpan, actualError);
       embedSpan.end();
     }
     state.embedSpans.clear();
     if (state.rerankSpan) {
-      recordSpanError(state.rerankSpan.span, actualError);
+      recordErrorOnSpan(state.rerankSpan.span, actualError);
       state.rerankSpan.span.end();
       state.rerankSpan = void 0;
     }
-    recordSpanError(state.rootSpan, actualError);
+    recordErrorOnSpan(state.rootSpan, actualError);
     state.rootSpan.end();
     this.cleanupCallState(event.callId);
   }
@@ -1110,7 +1381,7 @@ var OpenTelemetry = class {
 
 // src/legacy-open-telemetry.ts
 import {
-  context as context2,
+  context as context3,
   SpanStatusCode as SpanStatusCode2,
   trace as trace2
 } from "@opentelemetry/api";
@@ -1135,7 +1406,7 @@ function getBaseTelemetryAttributes({
   model,
   settings,
   headers,
-  context: context3
+  context: context4
 }) {
   return {
     "ai.model.provider": model.provider,
@@ -1146,7 +1417,7 @@ function getBaseTelemetryAttributes({
       return attributes;
     }, {}),
     // add context as attributes:
-    ...Object.entries(context3 != null ? context3 : {}).reduce((attributes, [key, value]) => {
+    ...Object.entries(context4 != null ? context4 : {}).reduce((attributes, [key, value]) => {
       if (value != void 0) {
         attributes[`ai.settings.context.${key}`] = value;
       }
@@ -1171,15 +1442,27 @@ function stringifyForTelemetry(prompt) {
       content: typeof message.content === "string" ? message.content : message.content.map(
         (part) => part.type === "file" ? {
           ...part,
-          data: part.data instanceof Uint8Array ? convertDataContentToBase64String2(part.data) : part.data
+          data: serializeFileData(part.data)
         } : part
       )
     }))
   );
 }
+function serializeFileData(data) {
+  switch (data.type) {
+    case "data":
+      return data.data instanceof Uint8Array ? convertDataContentToBase64String2(data.data) : data.data;
+    case "url":
+      return data.url.toString();
+    case "reference":
+      return data.reference;
+    case "text":
+      return data.text;
+  }
+}
 
 // src/legacy-open-telemetry.ts
-function recordSpanError2(span, error) {
+function recordSpanError(span, error) {
   if (error instanceof Error) {
     span.recordException({
       name: error.name,
@@ -1242,7 +1525,7 @@ var LegacyOpenTelemetry = class {
     if (toolSpanEntry == null) {
       return execute();
     }
-    return context2.with(toolSpanEntry.context, execute);
+    return context3.with(toolSpanEntry.context, execute);
   }
   onStart(event) {
     if (event.operationId === "ai.embed" || event.operationId === "ai.embedMany") {
@@ -1302,7 +1585,7 @@ var LegacyOpenTelemetry = class {
       }
     });
     const rootSpan = this.tracer.startSpan(event.operationId, { attributes });
-    const rootContext = trace2.setSpan(context2.active(), rootSpan);
+    const rootContext = trace2.setSpan(context3.active(), rootSpan);
     this.callStates.set(event.callId, {
       operationId: event.operationId,
       telemetry,
@@ -1358,7 +1641,7 @@ var LegacyOpenTelemetry = class {
       "ai.settings.output": event.output
     });
     const rootSpan = this.tracer.startSpan(event.operationId, { attributes });
-    const rootContext = trace2.setSpan(context2.active(), rootSpan);
+    const rootContext = trace2.setSpan(context3.active(), rootSpan);
     this.callStates.set(event.callId, {
       operationId: event.operationId,
       telemetry,
@@ -1484,7 +1767,7 @@ var LegacyOpenTelemetry = class {
       }
     });
     const rootSpan = this.tracer.startSpan(event.operationId, { attributes });
-    const rootContext = trace2.setSpan(context2.active(), rootSpan);
+    const rootContext = trace2.setSpan(context3.active(), rootSpan);
     this.callStates.set(event.callId, {
       operationId: event.operationId,
       telemetry,
@@ -1589,7 +1872,7 @@ var LegacyOpenTelemetry = class {
       } catch (e) {
       }
     } else {
-      recordSpanError2(span, toolOutput.error);
+      recordSpanError(span, toolOutput.error);
     }
     span.end();
     state.toolSpans.delete(event.toolCall.toolCallId);
@@ -1834,7 +2117,7 @@ var LegacyOpenTelemetry = class {
       }
     });
     const rootSpan = this.tracer.startSpan(event.operationId, { attributes });
-    const rootContext = trace2.setSpan(context2.active(), rootSpan);
+    const rootContext = trace2.setSpan(context3.active(), rootSpan);
     this.callStates.set(event.callId, {
       operationId: event.operationId,
       telemetry,
@@ -1922,20 +2205,20 @@ var LegacyOpenTelemetry = class {
     if (!(state == null ? void 0 : state.rootSpan)) return;
     const actualError = (_a = event.error) != null ? _a : error;
     if (state.stepSpan) {
-      recordSpanError2(state.stepSpan, actualError);
+      recordSpanError(state.stepSpan, actualError);
       state.stepSpan.end();
     }
     for (const { span: embedSpan } of state.embedSpans.values()) {
-      recordSpanError2(embedSpan, actualError);
+      recordSpanError(embedSpan, actualError);
       embedSpan.end();
     }
     state.embedSpans.clear();
     if (state.rerankSpan) {
-      recordSpanError2(state.rerankSpan.span, actualError);
+      recordSpanError(state.rerankSpan.span, actualError);
       state.rerankSpan.span.end();
       state.rerankSpan = void 0;
     }
-    recordSpanError2(state.rootSpan, actualError);
+    recordSpanError(state.rootSpan, actualError);
     state.rootSpan.end();
     this.cleanupCallState(event.callId);
   }