raindrop-ai 0.3.1 → 0.4.0

package/dist/tracing/index.js

@@ -36,7 +36,7 @@ __export(tracing_exports, {
 module.exports = __toCommonJS(tracing_exports);
 
 // src/tracing/tracer-core.ts
-var import_api5 = require("@opentelemetry/api");
+var import_api6 = require("@opentelemetry/api");
 var import_exporter_trace_otlp_http = require("@opentelemetry/exporter-trace-otlp-http");
 
 // ../core/dist/chunk-3V4MIA75.js
@@ -1097,11 +1097,404 @@ var import_instrumentation_bedrock = require("@traceloop/instrumentation-bedrock
 var import_instrumentation_chromadb = require("@traceloop/instrumentation-chromadb");
 var import_instrumentation_cohere = require("@traceloop/instrumentation-cohere");
 var import_instrumentation_openai = require("@traceloop/instrumentation-openai");
+var import_instrumentation_langchain = require("@traceloop/instrumentation-langchain");
+var import_instrumentation_llamaindex = require("@traceloop/instrumentation-llamaindex");
 var import_instrumentation_pinecone = require("@traceloop/instrumentation-pinecone");
 var import_instrumentation_qdrant = require("@traceloop/instrumentation-qdrant");
 var import_instrumentation_together = require("@traceloop/instrumentation-together");
 var import_instrumentation_vertexai = require("@traceloop/instrumentation-vertexai");
-var traceloop4 = __toESM(require("@traceloop/node-server-sdk"));
+
+// src/tracing/otel-core/index.ts
+var import_api = require("@opentelemetry/api");
+var import_core = require("@opentelemetry/core");
+var import_exporter_trace_otlp_proto = require("@opentelemetry/exporter-trace-otlp-proto");
+var import_instrumentation = require("@opentelemetry/instrumentation");
+var resourceModule = __toESM(require("@opentelemetry/resources"));
+var import_sdk_trace_base = require("@opentelemetry/sdk-trace-base");
+var import_sdk_trace_node = require("@opentelemetry/sdk-trace-node");
+var import_ai_semantic_conventions = require("@traceloop/ai-semantic-conventions");
+var ALL_INSTRUMENTATION_LIBRARIES = "all";
+var TRACER_NAME = "@traceloop/node-server-sdk";
+var WORKFLOW_NAME_KEY = /* @__PURE__ */ Symbol.for("raindrop.otelCore.workflowName");
+var ENTITY_NAME_KEY = /* @__PURE__ */ Symbol.for("raindrop.otelCore.entityName");
+var ASSOCIATION_PROPERTIES_KEY = /* @__PURE__ */ Symbol.for("raindrop.otelCore.associationProperties");
+var INSTRUMENTATION_LIBRARIES = [
+  "ai",
+  "@traceloop/node-server-sdk",
+  "@traceloop/instrumentation-openai",
+  "@traceloop/instrumentation-langchain",
+  "@traceloop/instrumentation-chroma",
+  "@traceloop/instrumentation-chromadb",
+  "@traceloop/instrumentation-anthropic",
+  "@traceloop/instrumentation-llamaindex",
+  "@traceloop/instrumentation-vertexai",
+  "@traceloop/instrumentation-bedrock",
+  "@traceloop/instrumentation-cohere",
+  "@traceloop/instrumentation-pinecone",
+  "@traceloop/instrumentation-qdrant",
+  "@traceloop/instrumentation-together",
+  TRACER_NAME,
+  "traceloop.tracer"
+];
+var AI_GENERATE_TEXT = "ai.generateText.doGenerate";
+var AI_STREAM_TEXT = "ai.streamText.doStream";
+var AI_RESPONSE_TEXT = "ai.response.text";
+var AI_PROMPT_MESSAGES = "ai.prompt.messages";
+var AI_PROMPT_TOKENS = "ai.usage.promptTokens";
+var AI_COMPLETION_TOKENS = "ai.usage.completionTokens";
+var AI_MODEL_PROVIDER = "ai.model.provider";
+var STATE_KEY = /* @__PURE__ */ Symbol.for("raindrop.otelCore.state");
+function getState() {
+  var _a;
+  const holder = globalThis;
+  (_a = holder[STATE_KEY]) != null ? _a : holder[STATE_KEY] = {};
+  return holder[STATE_KEY];
+}
+function parseHeaders(value) {
+  const headers = {};
+  if (!value) return headers;
+  for (const pair of value.split(",")) {
+    const separator = pair.indexOf("=");
+    if (separator <= 0) continue;
+    try {
+      const key = decodeURIComponent(pair.slice(0, separator).trim());
+      const val = decodeURIComponent(pair.slice(separator + 1).trim());
+      if (key && val) headers[key] = val;
+    } catch (e) {
+    }
+  }
+  return headers;
+}
+function safeStringify(value) {
+  try {
+    const result = JSON.stringify(value);
+    return result === void 0 ? String(value) : result;
+  } catch (e) {
+    return "[unserializable]";
+  }
+}
+function createResource(attributes) {
+  if ("resourceFromAttributes" in resourceModule) {
+    const factory = resourceModule.resourceFromAttributes;
+    if (typeof factory === "function") return factory(attributes);
+  }
+  return new resourceModule.Resource(attributes);
+}
+function detectRuntimeResource() {
+  try {
+    if ("detectResourcesSync" in resourceModule && typeof resourceModule.detectResourcesSync === "function" && "envDetectorSync" in resourceModule) {
+      return resourceModule.detectResourcesSync({
+        detectors: [
+          resourceModule.envDetectorSync,
+          resourceModule.processDetectorSync,
+          resourceModule.hostDetectorSync
+        ]
+      });
+    }
+    if ("detectResources" in resourceModule && typeof resourceModule.detectResources === "function" && "envDetector" in resourceModule) {
+      const detected = resourceModule.detectResources({
+        detectors: [
+          resourceModule.envDetector,
+          resourceModule.processDetector,
+          resourceModule.hostDetector
+        ]
+      });
+      if (!(detected instanceof Promise)) {
+        return detected;
+      }
+    }
+  } catch (e) {
+  }
+  return void 0;
+}
+function logLevel(level) {
+  switch (level) {
+    case "debug":
+      return import_api.DiagLogLevel.DEBUG;
+    case "info":
+      return import_api.DiagLogLevel.INFO;
+    case "warn":
+      return import_api.DiagLogLevel.WARN;
+    default:
+      return import_api.DiagLogLevel.ERROR;
+  }
+}
+function transformAiSdkSpan(span) {
+  const names = {
+    [AI_GENERATE_TEXT]: "ai.generateText.generate",
+    [AI_STREAM_TEXT]: "ai.streamText.stream"
+  };
+  if (!(span.name in names)) return span;
+  const attributes = span.attributes;
+  if (AI_RESPONSE_TEXT in attributes) {
+    const response = attributes[AI_RESPONSE_TEXT];
+    if (typeof response === "string") {
+      attributes[`${import_ai_semantic_conventions.SpanAttributes.LLM_COMPLETIONS}.0.content`] = response;
+    }
+    attributes[`${import_ai_semantic_conventions.SpanAttributes.LLM_COMPLETIONS}.0.role`] = "assistant";
+    delete attributes[AI_RESPONSE_TEXT];
+  }
+  if (AI_PROMPT_MESSAGES in attributes) {
+    try {
+      const parsed = JSON.parse(String(attributes[AI_PROMPT_MESSAGES]));
+      if (Array.isArray(parsed)) {
+        parsed.forEach((message, index) => {
+          if (!message || typeof message !== "object") return;
+          const content = "content" in message ? message.content : void 0;
+          const role = "role" in message ? message.role : void 0;
+          attributes[`${import_ai_semantic_conventions.SpanAttributes.LLM_PROMPTS}.${index}.content`] = typeof content === "string" ? content : safeStringify(content);
+          if (typeof role === "string") {
+            attributes[`${import_ai_semantic_conventions.SpanAttributes.LLM_PROMPTS}.${index}.role`] = role;
+          }
+        });
+      }
+      delete attributes[AI_PROMPT_MESSAGES];
+    } catch (e) {
+    }
+  }
+  if (AI_PROMPT_TOKENS in attributes) {
+    attributes[import_ai_semantic_conventions.SpanAttributes.LLM_USAGE_PROMPT_TOKENS] = Number(attributes[AI_PROMPT_TOKENS]);
+    delete attributes[AI_PROMPT_TOKENS];
+  }
+  if (AI_COMPLETION_TOKENS in attributes) {
+    attributes[import_ai_semantic_conventions.SpanAttributes.LLM_USAGE_COMPLETION_TOKENS] = Number(attributes[AI_COMPLETION_TOKENS]);
+    delete attributes[AI_COMPLETION_TOKENS];
+  }
+  const prompt = attributes[import_ai_semantic_conventions.SpanAttributes.LLM_USAGE_PROMPT_TOKENS];
+  const completion = attributes[import_ai_semantic_conventions.SpanAttributes.LLM_USAGE_COMPLETION_TOKENS];
+  if (prompt !== void 0 && completion !== void 0) {
+    attributes[import_ai_semantic_conventions.SpanAttributes.LLM_USAGE_TOTAL_TOKENS] = Number(prompt) + Number(completion);
+  }
+  if (AI_MODEL_PROVIDER in attributes) {
+    const vendor = String(attributes[AI_MODEL_PROVIDER]);
+    attributes[import_ai_semantic_conventions.SpanAttributes.LLM_SYSTEM] = vendor.startsWith("openai") ? "OpenAI" : vendor;
+    delete attributes[AI_MODEL_PROVIDER];
+  }
+  Object.defineProperty(span, "name", {
+    configurable: true,
+    value: names[span.name],
+    writable: true
+  });
+  return span;
+}
+function scopeName(span) {
+  var _a;
+  if ("instrumentationScope" in span) {
+    const scope = span.instrumentationScope;
+    if (scope && typeof scope === "object" && "name" in scope && typeof scope.name === "string") {
+      return scope.name;
+    }
+  }
+  return (_a = span.instrumentationLibrary) == null ? void 0 : _a.name;
+}
+function contentCaptureDisabled(options) {
+  return options.traceContent === false || (process.env.TRACELOOP_TRACE_CONTENT || "true").toLowerCase() === "false";
+}
+function shouldSendTraceContent() {
+  var _a;
+  const contextual = import_api.context.active().getValue(import_ai_semantic_conventions.CONTEXT_KEY_ALLOW_TRACE_CONTENT);
+  if (typeof contextual === "boolean") return contextual;
+  if (((_a = getState().configuration) == null ? void 0 : _a.traceContent) === false) return false;
+  return (process.env.TRACELOOP_TRACE_CONTENT || "true").toLowerCase() !== "false";
+}
+function enrichStart(span) {
+  const active = import_api.context.active();
+  const workflow = active.getValue(WORKFLOW_NAME_KEY);
+  if (typeof workflow === "string") span.setAttribute(import_ai_semantic_conventions.SpanAttributes.TRACELOOP_WORKFLOW_NAME, workflow);
+  const entity = active.getValue(ENTITY_NAME_KEY);
+  if (typeof entity === "string") span.setAttribute(import_ai_semantic_conventions.SpanAttributes.TRACELOOP_ENTITY_PATH, entity);
+  const properties = active.getValue(ASSOCIATION_PROPERTIES_KEY);
+  if (properties && typeof properties === "object") {
+    for (const [key, value] of Object.entries(properties)) {
+      if (typeof value === "string" || typeof value === "number" || typeof value === "boolean") {
+        span.setAttribute(`${import_ai_semantic_conventions.SpanAttributes.TRACELOOP_ASSOCIATION_PROPERTIES}.${key}`, value);
+      }
+    }
+  }
+}
+function createSpanProcessor(options) {
+  var _a, _b;
+  const baseUrl = options.baseUrl || process.env.TRACELOOP_BASE_URL || "https://api.traceloop.com";
+  const apiKey = options.apiKey || process.env.TRACELOOP_API_KEY;
+  const resolvedHeaders = (_a = options.headers) != null ? _a : process.env.TRACELOOP_HEADERS ? parseHeaders(process.env.TRACELOOP_HEADERS) : { Authorization: `Bearer ${apiKey}` };
+  const exporter = (_b = options.exporter) != null ? _b : new import_exporter_trace_otlp_proto.OTLPTraceExporter({ url: `${baseUrl}/v1/traces`, headers: resolvedHeaders });
+  const processor = options.disableBatch ? new import_sdk_trace_base.SimpleSpanProcessor(exporter) : new import_sdk_trace_base.BatchSpanProcessor(exporter);
+  const originalOnStart = processor.onStart.bind(processor);
+  const originalOnEnd = processor.onEnd.bind(processor);
+  processor.onStart = (span, parentContext) => {
+    try {
+      enrichStart(span);
+      originalOnStart(span, parentContext);
+    } catch (e) {
+    }
+  };
+  processor.onEnd = (span) => {
+    var _a2;
+    let outgoing = span;
+    try {
+      const allowed = options.allowedInstrumentationLibraries;
+      if (allowed !== ALL_INSTRUMENTATION_LIBRARIES && ![...INSTRUMENTATION_LIBRARIES, ...allowed != null ? allowed : []].includes((_a2 = scopeName(span)) != null ? _a2 : "")) {
+        return;
+      }
+      outgoing = transformAiSdkSpan(span);
+    } catch (e) {
+    }
+    try {
+      originalOnEnd(outgoing);
+    } catch (e) {
+    }
+  };
+  return processor;
+}
+function initialize(options = {}) {
+  var _a;
+  const state = getState();
+  if (state.configuration) return;
+  state.configuration = Object.freeze({ ...options });
+  if (options.logLevel) import_api.diag.setLogger(new import_api.DiagConsoleLogger(), logLevel(options.logLevel));
+  if (options.tracingEnabled === false) return;
+  const baseUrl = options.baseUrl || process.env.TRACELOOP_BASE_URL || "https://api.traceloop.com";
+  state.defaultProcessor = createSpanProcessor({
+    ...options,
+    baseUrl,
+    allowedInstrumentationLibraries: ALL_INSTRUMENTATION_LIBRARIES
+  });
+  const serviceResource = createResource({
+    "service.name": options.appName || process.env.npm_package_name || "unknown_service"
+  });
+  const detectedResource = detectRuntimeResource();
+  state.provider = new import_sdk_trace_node.NodeTracerProvider({
+    // Detected attributes win over the configured resource, matching
+    // sdk-node's `this._resource.merge(detectResourcesSync(...))` so
+    // OTEL_SERVICE_NAME / OTEL_RESOURCE_ATTRIBUTES can still override appName.
+    resource: detectedResource ? serviceResource.merge(detectedResource) : serviceResource,
+    spanProcessors: [state.defaultProcessor, ...options.processor ? [options.processor] : []]
+  });
+  state.provider.register({
+    contextManager: options.contextManager,
+    propagator: options.propagator
+  });
+  const instrumentations = (_a = options.instrumentations) != null ? _a : [];
+  if (contentCaptureDisabled(options)) {
+    for (const instrumentation of instrumentations) {
+      try {
+        const config = {
+          ...instrumentation.getConfig(),
+          traceContent: false
+        };
+        instrumentation.setConfig(config);
+      } catch (e) {
+      }
+    }
+  }
+  (0, import_instrumentation.registerInstrumentations)({ instrumentations });
+}
+async function forceFlush() {
+  var _a, _b;
+  const state = getState();
+  try {
+    await ((_a = state.provider) == null ? void 0 : _a.forceFlush());
+    await ((_b = state.defaultProcessor) == null ? void 0 : _b.forceFlush());
+  } catch (e) {
+  }
+}
+function withEntity(type, config, fn, args) {
+  let entityContext = import_api.context.active();
+  if (type === import_ai_semantic_conventions.TraceloopSpanKindValues.WORKFLOW) {
+    entityContext = entityContext.setValue(WORKFLOW_NAME_KEY, config.name);
+  }
+  const parentPath = entityContext.getValue(ENTITY_NAME_KEY);
+  if (type === import_ai_semantic_conventions.TraceloopSpanKindValues.TASK || type === import_ai_semantic_conventions.TraceloopSpanKindValues.TOOL) {
+    entityContext = entityContext.setValue(
+      ENTITY_NAME_KEY,
+      `${typeof parentPath === "string" ? `${parentPath}.` : ""}${config.name}`
+    );
+  }
+  if (config.traceContent !== void 0) {
+    entityContext = entityContext.setValue(import_ai_semantic_conventions.CONTEXT_KEY_ALLOW_TRACE_CONTENT, config.traceContent);
+  }
+  if (config.associationProperties) {
+    entityContext = entityContext.setValue(ASSOCIATION_PROPERTIES_KEY, config.associationProperties);
+  }
+  if (config.suppressTracing) {
+    entityContext = (0, import_core.suppressTracing)(entityContext);
+  }
+  let callbackStarted = false;
+  const run = (span) => {
+    var _a;
+    try {
+      span.setAttribute(import_ai_semantic_conventions.SpanAttributes.TRACELOOP_ENTITY_NAME, config.name);
+      span.setAttribute(import_ai_semantic_conventions.SpanAttributes.TRACELOOP_ENTITY_PATH, typeof parentPath === "string" ? parentPath : "");
+      span.setAttribute(import_ai_semantic_conventions.SpanAttributes.TRACELOOP_SPAN_KIND, type);
+      if (type === import_ai_semantic_conventions.TraceloopSpanKindValues.WORKFLOW) {
+        span.setAttribute(import_ai_semantic_conventions.SpanAttributes.TRACELOOP_WORKFLOW_NAME, config.name);
+      }
+      if (config.version !== void 0) span.setAttribute(import_ai_semantic_conventions.SpanAttributes.TRACELOOP_ENTITY_VERSION, config.version);
+      const input = (_a = config.inputParameters) != null ? _a : args;
+      if (shouldSendTraceContent()) {
+        span.setAttribute(
+          import_ai_semantic_conventions.SpanAttributes.TRACELOOP_ENTITY_INPUT,
+          safeStringify(input.length === 1 && input[0] !== null && typeof input[0] === "object" && !(input[0] instanceof Map) ? { args: [], kwargs: input[0] } : { args: input, kwargs: {} })
+        );
+      }
+    } catch (e) {
+    }
+    let result;
+    try {
+      callbackStarted = true;
+      result = fn(...args);
+    } catch (error) {
+      try {
+        span.end();
+      } catch (e) {
+      }
+      throw error;
+    }
+    if (result instanceof Promise) {
+      return result.then(
+        (value) => {
+          try {
+            if (shouldSendTraceContent()) span.setAttribute(import_ai_semantic_conventions.SpanAttributes.TRACELOOP_ENTITY_OUTPUT, safeStringify(value));
+            span.end();
+          } catch (e) {
+          }
+          return value;
+        },
+        (error) => {
+          try {
+            span.end();
+          } catch (e) {
+          }
+          throw error;
+        }
+      );
+    }
+    try {
+      if (shouldSendTraceContent()) span.setAttribute(import_ai_semantic_conventions.SpanAttributes.TRACELOOP_ENTITY_OUTPUT, safeStringify(result));
+      span.end();
+    } catch (e) {
+    }
+    return result;
+  };
+  try {
+    const tracer = import_api.trace.getTracer(TRACER_NAME);
+    return import_api.context.with(
+      entityContext,
+      () => tracer.startActiveSpan(`${config.name}.${type}`, {}, entityContext, run)
+    );
+  } catch (error) {
+    if (callbackStarted) throw error;
+    return fn(...args);
+  }
+}
+function withTask(config, fn, ...args) {
+  return withEntity(import_ai_semantic_conventions.TraceloopSpanKindValues.TASK, config, fn, args);
+}
+function withTool(config, fn, ...args) {
+  return withEntity(import_ai_semantic_conventions.TraceloopSpanKindValues.TOOL, config, fn, args);
+}
+
+// src/tracing/tracer-core.ts
 var import_weakref = require("weakref");
 
 // src/lib/inject-als.ts
@@ -1266,7 +1659,7 @@ function base64ToHex2(base64) {
 // package.json
 var package_default = {
   name: "raindrop-ai",
-  version: "0.3.1",
+  version: "0.4.0",
   main: "dist/index.js",
   module: "dist/index.mjs",
   types: "dist/index.d.ts",
@@ -1320,7 +1713,6 @@ var package_default = {
     "@raindrop-ai/redact-pii": "workspace:*",
     "@raindrop-ai/schemas": "workspace:*",
     ai: "^6.0.0",
-    "@opentelemetry/resources": "^2.0.1",
     "@types/node": "^20.11.17",
     msw: "^2.12.8",
     tsup: "^8.4.0",
@@ -1331,9 +1723,25 @@ var package_default = {
   },
   dependencies: {
     "@opentelemetry/api": "^1.9.0",
+    "@opentelemetry/core": "^1.30.1",
     "@opentelemetry/exporter-trace-otlp-http": "^0.57.2",
     "@opentelemetry/exporter-trace-otlp-proto": "^0.57.2",
-    "@traceloop/node-server-sdk": "0.19.0-otel-v1",
+    "@opentelemetry/instrumentation": "^0.57.2",
+    "@opentelemetry/resources": "^1.30.1",
+    "@opentelemetry/sdk-trace-base": "^1.30.1",
+    "@opentelemetry/sdk-trace-node": "^1.30.1",
+    "@traceloop/ai-semantic-conventions": "0.19.0-otel-v1",
+    "@traceloop/instrumentation-anthropic": "0.19.0-otel-v1",
+    "@traceloop/instrumentation-openai": "0.19.0-otel-v1",
+    "@traceloop/instrumentation-cohere": "0.19.0-otel-v1",
+    "@traceloop/instrumentation-bedrock": "0.19.0-otel-v1",
+    "@traceloop/instrumentation-vertexai": "0.19.0-otel-v1",
+    "@traceloop/instrumentation-pinecone": "0.19.0-otel-v1",
+    "@traceloop/instrumentation-chromadb": "0.19.0-otel-v1",
+    "@traceloop/instrumentation-qdrant": "0.19.0-otel-v1",
+    "@traceloop/instrumentation-together": "0.19.0-otel-v1",
+    "@traceloop/instrumentation-langchain": "0.19.0-otel-v1",
+    "@traceloop/instrumentation-llamaindex": "0.19.0-otel-v1",
     weakref: "^0.2.1",
     zod: "^3.23.8"
   },
@@ -1352,7 +1760,6 @@ var package_default = {
       "esm"
     ],
     external: [
-      "@traceloop/node-server-sdk",
       "@traceloop/instrumentation-anthropic",
       "@traceloop/instrumentation-openai",
       "@traceloop/instrumentation-cohere",
@@ -1362,10 +1769,17 @@ var package_default = {
       "@traceloop/instrumentation-chromadb",
       "@traceloop/instrumentation-qdrant",
       "@traceloop/instrumentation-together",
+      "@traceloop/instrumentation-langchain",
+      "@traceloop/instrumentation-llamaindex",
+      "@traceloop/ai-semantic-conventions",
+      "@opentelemetry/api",
+      "@opentelemetry/core",
       "@opentelemetry/instrumentation",
       "@opentelemetry/exporter-trace-otlp-http",
       "@opentelemetry/exporter-trace-otlp-proto",
-      "@opentelemetry/sdk-trace-base"
+      "@opentelemetry/resources",
+      "@opentelemetry/sdk-trace-base",
+      "@opentelemetry/sdk-trace-node"
     ],
     noExternal: [
       "@raindrop-ai/core",
@@ -1424,7 +1838,7 @@ function buildExportTraceServiceRequest2(spans) {
 }
 
 // src/tracing/direct/context.ts
-var import_api = require("@opentelemetry/api");
+var import_api2 = require("@opentelemetry/api");
 var INVALID_TRACE_ID = "00000000000000000000000000000000";
 var INVALID_SPAN_ID = "0000000000000000";
 function hexToBase64(hex) {
@@ -1435,7 +1849,7 @@ function hexToBase64(hex) {
   return base64Encode2(bytes);
 }
 function getExplicitTraceContext(traceId, parentSpanId) {
-  if (!traceId || !parentSpanId || !(0, import_api.isValidTraceId)(traceId) || !(0, import_api.isValidSpanId)(parentSpanId)) {
+  if (!traceId || !parentSpanId || !(0, import_api2.isValidTraceId)(traceId) || !(0, import_api2.isValidSpanId)(parentSpanId)) {
     return void 0;
   }
   return {
@@ -1444,7 +1858,7 @@ function getExplicitTraceContext(traceId, parentSpanId) {
   };
 }
 function getActiveTraceContext() {
-  const activeSpan = import_api.trace.getSpan(import_api.context.active());
+  const activeSpan = import_api2.trace.getSpan(import_api2.context.active());
   if (!activeSpan) return {};
   const spanContext = activeSpan.spanContext();
   const traceIdHex = spanContext.traceId;
@@ -1458,7 +1872,7 @@ function getActiveTraceContext() {
 
 // src/tracing/multi-project.ts
 var import_node_crypto = require("crypto");
-var import_exporter_trace_otlp_proto = require("@opentelemetry/exporter-trace-otlp-proto");
+var import_exporter_trace_otlp_proto2 = require("@opentelemetry/exporter-trace-otlp-proto");
 var PROJECT_ID_SPAN_ATTRIBUTE = "raindrop.project_id";
 var AUTH_HINT_SPAN_ATTRIBUTE = "raindrop.auth_hint";
 var EXPORT_SUCCESS = 0;
@@ -1468,10 +1882,10 @@ function authHintForKey(writeKey) {
   if (!writeKey) return void 0;
   return (0, import_node_crypto.createHash)("sha256").update(writeKey, "utf8").digest("hex").slice(0, 8);
 }
-var STATE_KEY = /* @__PURE__ */ Symbol.for("raindrop.tracing.multiProjectState");
-function getState() {
+var STATE_KEY2 = /* @__PURE__ */ Symbol.for("raindrop.tracing.multiProjectState");
+function getState2() {
   const holder = globalThis;
-  let state = holder[STATE_KEY];
+  let state = holder[STATE_KEY2];
   if (!state) {
     state = {
       foreignClientSeen: false,
@@ -1483,12 +1897,12 @@ function getState() {
       multiKeyWarned: false,
       pendingMultiKeyWarning: false
     };
-    holder[STATE_KEY] = state;
+    holder[STATE_KEY2] = state;
   }
   return state;
 }
 function emitMultiKeyWarningOnce() {
-  const state = getState();
+  const state = getState2();
   if (state.multiKeyWarned) return;
   state.multiKeyWarned = true;
   state.pendingMultiKeyWarning = false;
@@ -1498,7 +1912,7 @@ function emitMultiKeyWarningOnce() {
 }
 function registerClientKey(authHint, tracingEnabled = false) {
   if (!authHint) return;
-  const state = getState();
+  const state = getState2();
   state.seenClientHints.add(authHint);
   if (state.seenClientHints.size > 1) {
     state.foreignClientSeen = true;
@@ -1510,10 +1924,10 @@ function registerClientKey(authHint, tracingEnabled = false) {
   }
 }
 function markForeignClient() {
-  getState().foreignClientSeen = true;
+  getState2().foreignClientSeen = true;
 }
 function foreignClientSeen() {
-  return getState().foreignClientSeen;
+  return getState2().foreignClientSeen;
 }
 function stampSpan(span, projectId, authHint) {
   try {
@@ -1619,17 +2033,17 @@ var GuardedSpanExporter = class {
   }
 };
 function buildGuardedExporter(baseUrl, headers, ownerAuthHint) {
-  const inner = new import_exporter_trace_otlp_proto.OTLPTraceExporter({
+  const inner = new import_exporter_trace_otlp_proto2.OTLPTraceExporter({
     url: `${baseUrl}/v1/traces`,
     headers
   });
   return new GuardedSpanExporter(inner, ownerAuthHint);
 }
 function pipelineOwnerHint() {
-  return getState().ownerAuthHint;
+  return getState2().ownerAuthHint;
 }
 function claimPipeline(authHint, projectId) {
-  const state = getState();
+  const state = getState2();
   if (state.pipelineClaimed) {
     return false;
   }
@@ -1642,16 +2056,16 @@ function claimPipeline(authHint, projectId) {
   return true;
 }
 function releasePipelineClaim() {
-  const state = getState();
+  const state = getState2();
   state.pipelineClaimed = false;
   state.ownerAuthHint = void 0;
   state.ownerProjectId = void 0;
 }
 function markTraceloopConfiguredWithoutClaim() {
-  getState().traceloopConfiguredWithoutClaim = true;
+  getState2().traceloopConfiguredWithoutClaim = true;
 }
 function traceloopConfiguredWithoutClaim() {
-  return getState().traceloopConfiguredWithoutClaim;
+  return getState2().traceloopConfiguredWithoutClaim;
 }
 
 // src/tracing/direct/shipper.ts
@@ -1888,18 +2302,17 @@ var DirectTraceShipper = class {
 };
 
 // src/tracing/exit-flush.ts
-var traceloop = __toESM(require("@traceloop/node-server-sdk"));
-var STATE_KEY2 = /* @__PURE__ */ Symbol.for("raindrop.tracing.exitFlushState");
-function getState2() {
+var STATE_KEY3 = /* @__PURE__ */ Symbol.for("raindrop.tracing.exitFlushState");
+function getState3() {
   const holder = globalThis;
-  let state = holder[STATE_KEY2];
+  let state = holder[STATE_KEY3];
   if (!state) {
     state = {
       clientFlushers: /* @__PURE__ */ new Set(),
       pipelineFlushNeeded: false,
       handlerRegistered: false
     };
-    holder[STATE_KEY2] = state;
+    holder[STATE_KEY3] = state;
   }
   return state;
 }
@@ -1912,7 +2325,7 @@ function ensureExitHandler(state) {
   });
 }
 function registerClientFlusher(flush) {
-  const state = getState2();
+  const state = getState3();
   state.clientFlushers.add(flush);
   ensureExitHandler(state);
   return () => {
@@ -1920,12 +2333,12 @@ function registerClientFlusher(flush) {
   };
 }
 function markPipelineForExitFlush() {
-  const state = getState2();
+  const state = getState3();
   state.pipelineFlushNeeded = true;
   ensureExitHandler(state);
 }
 async function flushAll() {
-  const state = getState2();
+  const state = getState3();
   const tasks = [];
   for (const flush of state.clientFlushers) {
     tasks.push(Promise.resolve().then(flush).catch(() => {
@@ -1933,7 +2346,7 @@ async function flushAll() {
   }
   if (state.pipelineFlushNeeded) {
     tasks.push(
-      Promise.resolve().then(() => traceloop.forceFlush()).catch(() => {
+      Promise.resolve().then(() => forceFlush()).catch(() => {
       })
     );
   }
@@ -1941,11 +2354,10 @@ async function flushAll() {
 }
 
 // src/tracing/liveInteraction.ts
-var import_api3 = require("@opentelemetry/api");
-var traceloop2 = __toESM(require("@traceloop/node-server-sdk"));
+var import_api4 = require("@opentelemetry/api");
 
 // src/tracing/subagents.ts
-var import_api2 = require("@opentelemetry/api");
+var import_api3 = require("@opentelemetry/api");
 var DEFAULT_SUBAGENT_NAME = "subagent";
 var DISPATCH_ACCEPTED = "accepted";
 function reasonText(reason) {
@@ -1961,7 +2373,7 @@ function reasonText(reason) {
 }
 function stampCurrentSpan(attributes) {
   try {
-    const span = import_api2.trace.getSpan(import_api2.context.active());
+    const span = import_api3.trace.getSpan(import_api3.context.active());
     if (!(span == null ? void 0 : span.isRecording())) return;
     for (const [key, value] of Object.entries(attributes)) {
       span.setAttribute(key, value);
@@ -1972,7 +2384,7 @@ function stampCurrentSpan(attributes) {
 function stampOwnSpan(attributes, frame) {
   if (!frame) return;
   try {
-    const span = import_api2.trace.getSpan(import_api2.context.active());
+    const span = import_api3.trace.getSpan(import_api3.context.active());
     if (!(span == null ? void 0 : span.isRecording()) || !frame.ownsSpan(span)) return;
     for (const [key, value] of Object.entries(attributes)) {
       span.setAttribute(key, value);
@@ -1982,11 +2394,11 @@ function stampOwnSpan(attributes, frame) {
 }
 function failOwnSpan(reason, frame) {
   try {
-    const span = import_api2.trace.getSpan(import_api2.context.active());
+    const span = import_api3.trace.getSpan(import_api3.context.active());
     if (!(span == null ? void 0 : span.isRecording())) return false;
     if (!(frame == null ? void 0 : frame.ownsSpan(span))) return false;
     const message = reasonText(reason);
-    span.setStatus({ code: import_api2.SpanStatusCode.ERROR, message });
+    span.setStatus({ code: import_api3.SpanStatusCode.ERROR, message });
     span.recordException(reason instanceof Error ? reason : new Error(message));
     return true;
   } catch (e) {
@@ -2230,16 +2642,16 @@ var LiveSubagentRun = class {
 };
 
 // src/tracing/liveInteraction.ts
-function getPropertiesFromContext(context6) {
+function getPropertiesFromContext(context7) {
   const properties = {
-    ...context6.userId && { user_id: context6.userId },
-    ...context6.convoId && { convo_id: context6.convoId },
-    ...context6.eventId && { event_id: context6.eventId },
-    ...context6.event && { event_name: context6.event },
-    ...context6.properties || {}
+    ...context7.userId && { user_id: context7.userId },
+    ...context7.convoId && { convo_id: context7.convoId },
+    ...context7.eventId && { event_id: context7.eventId },
+    ...context7.event && { event_name: context7.event },
+    ...context7.properties || {}
   };
-  if (context6.attachments && context6.attachments.length > 0) {
-    properties.attachments = stringifyBounded(context6.attachments);
+  if (context7.attachments && context7.attachments.length > 0) {
+    properties.attachments = stringifyBounded(context7.attachments);
   }
   return properties;
 }
@@ -2259,12 +2671,12 @@ function serializeAssociationProperties(properties) {
 function serializeSpanValue(value) {
   return stringifyBounded(value);
 }
-function getLocalDebuggerMetadata(context6) {
+function getLocalDebuggerMetadata(context7) {
   return {
-    ...context6.eventId ? { eventId: context6.eventId } : {},
-    ...context6.event ? { eventName: context6.event } : {},
-    ...context6.userId ? { userId: context6.userId } : {},
-    ...context6.convoId ? { convoId: context6.convoId } : {}
+    ...context7.eventId ? { eventId: context7.eventId } : {},
+    ...context7.event ? { eventName: context7.event } : {},
+    ...context7.userId ? { userId: context7.userId } : {},
+    ...context7.convoId ? { convoId: context7.convoId } : {}
   };
 }
 var LiveToolSpan = class {
@@ -2276,7 +2688,7 @@ var LiveToolSpan = class {
   }
   spanIdHex() {
     const ctx = this.span.spanContext();
-    return (0, import_api3.isValidSpanId)(ctx.spanId) ? ctx.spanId : void 0;
+    return (0, import_api4.isValidSpanId)(ctx.spanId) ? ctx.spanId : void 0;
   }
   setInput(input) {
     this.span.setAttribute("traceloop.entity.input", serializeSpanValue(input));
@@ -2288,13 +2700,13 @@ var LiveToolSpan = class {
   setError(error) {
     this.hasError = true;
     this.errorMessage = error instanceof Error ? error.message : String(error);
-    this.span.setStatus({ code: import_api3.SpanStatusCode.ERROR, message: this.errorMessage });
+    this.span.setStatus({ code: import_api4.SpanStatusCode.ERROR, message: this.errorMessage });
     this.span.recordException(error instanceof Error ? error : new Error(this.errorMessage));
   }
   end() {
     var _a;
     if (!this.hasError) {
-      this.span.setStatus({ code: import_api3.SpanStatusCode.OK });
+      this.span.setStatus({ code: import_api4.SpanStatusCode.OK });
     }
     this.span.end();
     (_a = this.emitLiveEvent) == null ? void 0 : _a.call(this, {
@@ -2365,11 +2777,11 @@ var DirectToolSpan = class {
   }
 };
 var LiveInteraction = class {
-  constructor(context6, traceId, analytics, directShipper, routing) {
+  constructor(context7, traceId, analytics, directShipper, routing) {
     this.usageEmitted = false;
-    this.context = context6;
+    this.context = context7;
     this.analytics = analytics;
-    this.tracer = import_api3.trace.getTracer("traceloop.tracer");
+    this.tracer = import_api4.trace.getTracer("traceloop.tracer");
     this.traceId = traceId;
     this.directShipper = directShipper;
     this.routing = routing;
@@ -2404,7 +2816,7 @@ var LiveInteraction = class {
       this.traceId = explicitTraceId;
       return explicitTraceId;
     }
-    const activeSpan = import_api3.trace.getSpan(import_api3.context.active());
+    const activeSpan = import_api4.trace.getSpan(import_api4.context.active());
     const activeTraceId = activeSpan == null ? void 0 : activeSpan.spanContext().traceId;
     if (activeTraceId) {
       this.traceId = activeTraceId;
@@ -2449,12 +2861,12 @@ var LiveInteraction = class {
       const spanContext = {
         traceId: base64ToHex2(traceIdB64),
         spanId: base64ToHex2(spanIdB64),
-        traceFlags: import_api3.TraceFlags.SAMPLED
+        traceFlags: import_api4.TraceFlags.SAMPLED
       };
-      const wrappedContext = import_api3.trace.setSpan(import_api3.context.active(), import_api3.trace.wrapSpanContext(spanContext));
+      const wrappedContext = import_api4.trace.setSpan(import_api4.context.active(), import_api4.trace.wrapSpanContext(spanContext));
       const invoke = () => this.routing ? withRoutingContext(this.routing, () => fn.apply(thisArg, args)) : fn.apply(thisArg, args);
       try {
-        const result = await import_api3.context.with(wrappedContext, invoke);
+        const result = await import_api4.context.with(wrappedContext, invoke);
         const endTimeMs = Date.now();
         this.directShipper.sendSpan({
           name: taskName,
@@ -2489,7 +2901,7 @@ var LiveInteraction = class {
       }
     }
     const wrappedFn = (...fnArgs) => {
-      const activeSpan = import_api3.trace.getSpan(import_api3.context.active());
+      const activeSpan = import_api4.trace.getSpan(import_api4.context.active());
       const activeTraceId = activeSpan == null ? void 0 : activeSpan.spanContext().traceId;
       if (activeTraceId) {
         this.traceId = activeTraceId;
@@ -2502,7 +2914,7 @@ var LiveInteraction = class {
       }
       return fn.apply(thisArg, fnArgs);
     };
-    return traceloop2.withTask(
+    return withTask(
       {
         name: taskName,
         associationProperties: serializeAssociationProperties(properties),
@@ -2541,14 +2953,14 @@ var LiveInteraction = class {
       const spanContext = {
         traceId: base64ToHex2(traceIdB64),
         spanId: base64ToHex2(spanIdB64),
-        traceFlags: import_api3.TraceFlags.SAMPLED
+        traceFlags: import_api4.TraceFlags.SAMPLED
       };
-      const wrappedContext = import_api3.trace.setSpan(import_api3.context.active(), import_api3.trace.wrapSpanContext(spanContext));
+      const wrappedContext = import_api4.trace.setSpan(import_api4.context.active(), import_api4.trace.wrapSpanContext(spanContext));
       const invoke = () => this.routing ? withRoutingContext(this.routing, () => fn.apply(thisArg, args)) : fn.apply(thisArg, args);
       let result;
       let error;
       try {
-        result = await import_api3.context.with(wrappedContext, invoke);
+        result = await import_api4.context.with(wrappedContext, invoke);
       } catch (err) {
         error = err instanceof Error ? err.message : String(err);
         const endTimeMs2 = Date.now();
@@ -2613,9 +3025,9 @@ var LiveInteraction = class {
     const inputParams = params.inputParameters ? [params.inputParameters] : void 0;
     const emitLive = (event) => this.emitLiveEvent(event);
     const wrapped = async (...wrappedArgs) => {
-      const activeSpan = import_api3.trace.getSpan(import_api3.context.active());
+      const activeSpan = import_api4.trace.getSpan(import_api4.context.active());
       const ctx = activeSpan == null ? void 0 : activeSpan.spanContext();
-      const liveSpanId = ctx && (0, import_api3.isValidSpanId)(ctx.spanId) ? ctx.spanId : void 0;
+      const liveSpanId = ctx && (0, import_api4.isValidSpanId)(ctx.spanId) ? ctx.spanId : void 0;
       if (activeSpan) {
         this.stampRouting(activeSpan);
       }
@@ -2653,7 +3065,7 @@ var LiveInteraction = class {
         throw error;
       }
     };
-    return traceloop2.withTool(
+    return withTool(
       {
         name: toolName,
         associationProperties: serializeAssociationProperties(properties),
@@ -2723,7 +3135,7 @@ var LiveInteraction = class {
       console.log(`[raindrop] startToolSpan: started tool span "${name}"`);
     }
     const startSpanCtx = span.spanContext();
-    const startSpanIdHex = (0, import_api3.isValidSpanId)(startSpanCtx.spanId) ? startSpanCtx.spanId : void 0;
+    const startSpanIdHex = (0, import_api4.isValidSpanId)(startSpanCtx.spanId) ? startSpanCtx.spanId : void 0;
     this.emitLiveEvent({
       type: "tool_start",
       spanId: startSpanIdHex,
@@ -2894,19 +3306,19 @@ var LiveInteraction = class {
       span.setAttribute("traceloop.entity.input", input);
     }
     span.setAttribute("traceloop.entity.output", output);
-    span.setStatus({ code: import_api3.SpanStatusCode.OK });
+    span.setStatus({ code: import_api4.SpanStatusCode.OK });
     const spanContext = span.spanContext();
     span.end();
     this.emitLiveEvent({
-      spanId: (0, import_api3.isValidSpanId)(spanContext.spanId) ? spanContext.spanId : void 0,
+      spanId: (0, import_api4.isValidSpanId)(spanContext.spanId) ? spanContext.spanId : void 0,
       type: "tool_result",
       content: params.name,
       timestamp: Date.now(),
       metadata: { result: params.output }
     });
     return {
-      traceIdHex: (0, import_api3.isValidTraceId)(spanContext.traceId) ? spanContext.traceId : void 0,
-      spanIdHex: (0, import_api3.isValidSpanId)(spanContext.spanId) ? spanContext.spanId : void 0
+      traceIdHex: (0, import_api4.isValidTraceId)(spanContext.traceId) ? spanContext.traceId : void 0,
+      spanIdHex: (0, import_api4.isValidSpanId)(spanContext.spanId) ? spanContext.spanId : void 0
     };
   }
   /**
@@ -2951,7 +3363,7 @@ var LiveInteraction = class {
     for (const [key, value] of Object.entries(attributes)) {
       span.setAttribute(key, value);
     }
-    span.setStatus({ code: import_api3.SpanStatusCode.ERROR, message });
+    span.setStatus({ code: import_api4.SpanStatusCode.ERROR, message });
     span.recordException(reason instanceof Error ? reason : new Error(message));
     span.end();
   }
@@ -3078,10 +3490,10 @@ var LiveInteraction = class {
     }
     if (error) {
       const errorMessage = error instanceof Error ? error.message : String(error);
-      span.setStatus({ code: import_api3.SpanStatusCode.ERROR, message: errorMessage });
+      span.setStatus({ code: import_api4.SpanStatusCode.ERROR, message: errorMessage });
       span.recordException(error instanceof Error ? error : new Error(errorMessage));
     } else {
-      span.setStatus({ code: import_api3.SpanStatusCode.OK });
+      span.setStatus({ code: import_api4.SpanStatusCode.OK });
     }
     span.end(endTimeMs);
     if ((_c = this.analytics) == null ? void 0 : _c.debugLogs) {
@@ -3093,7 +3505,7 @@ var LiveInteraction = class {
       });
     }
     const trackToolCtx = span.spanContext();
-    const trackToolSpanIdHex = (0, import_api3.isValidSpanId)(trackToolCtx.spanId) ? trackToolCtx.spanId : void 0;
+    const trackToolSpanIdHex = (0, import_api4.isValidSpanId)(trackToolCtx.spanId) ? trackToolCtx.spanId : void 0;
     this.emitLiveEvent({
       type: "tool_result",
       spanId: trackToolSpanIdHex,
@@ -3109,11 +3521,10 @@ var LiveInteraction = class {
 };
 
 // src/tracing/liveTracer.ts
-var import_api4 = require("@opentelemetry/api");
-var traceloop3 = __toESM(require("@traceloop/node-server-sdk"));
+var import_api5 = require("@opentelemetry/api");
 var LiveTracer = class {
   constructor(globalProperties, directShipper, routing) {
-    this.tracer = import_api4.trace.getTracer("traceloop.tracer");
+    this.tracer = import_api5.trace.getTracer("traceloop.tracer");
     this.globalProperties = globalProperties || {};
     this.directShipper = directShipper;
     this.routing = routing;
@@ -3125,7 +3536,7 @@ var LiveTracer = class {
   }
   emitLiveEvent(event) {
     var _a, _b;
-    const activeSpan = import_api4.trace.getSpan(import_api4.context.active());
+    const activeSpan = import_api5.trace.getSpan(import_api5.context.active());
     const activeTraceId = activeSpan == null ? void 0 : activeSpan.spanContext().traceId;
     const active = getActiveTraceContext();
     const traceId = (_b = (_a = event.traceId) != null ? _a : activeTraceId) != null ? _b : active.traceIdB64;
@@ -3145,7 +3556,7 @@ var LiveTracer = class {
     };
     const inputParameters = typeof params === "string" ? void 0 : params.inputParameters;
     const wrappedFn = (...fnArgs) => {
-      const activeSpan = import_api4.trace.getSpan(import_api4.context.active());
+      const activeSpan = import_api5.trace.getSpan(import_api5.context.active());
       if (activeSpan) {
         this.stampRouting(activeSpan);
       }
@@ -3154,7 +3565,7 @@ var LiveTracer = class {
       }
       return fn.apply(thisArg, fnArgs);
     };
-    return traceloop3.withTask(
+    return withTask(
       {
         name: taskName,
         associationProperties: properties,
@@ -3240,15 +3651,15 @@ var LiveTracer = class {
     }
     if (error) {
       const errorMessage = error instanceof Error ? error.message : String(error);
-      span.setStatus({ code: import_api4.SpanStatusCode.ERROR, message: errorMessage });
+      span.setStatus({ code: import_api5.SpanStatusCode.ERROR, message: errorMessage });
       span.recordException(error instanceof Error ? error : new Error(errorMessage));
     } else {
-      span.setStatus({ code: import_api4.SpanStatusCode.OK });
+      span.setStatus({ code: import_api5.SpanStatusCode.OK });
     }
     span.end(endTimeMs);
     const ctx = span.spanContext();
-    const traceId = ctx.traceId || ((_b = import_api4.trace.getSpan(import_api4.context.active())) == null ? void 0 : _b.spanContext().traceId);
-    const spanIdHex = (0, import_api4.isValidSpanId)(ctx.spanId) ? ctx.spanId : void 0;
+    const traceId = ctx.traceId || ((_b = import_api5.trace.getSpan(import_api5.context.active())) == null ? void 0 : _b.spanContext().traceId);
+    const spanIdHex = (0, import_api5.isValidSpanId)(ctx.spanId) ? ctx.spanId : void 0;
     if (traceId) {
       this.emitLiveEvent({
         traceId,
@@ -3407,7 +3818,7 @@ function createLocalDebuggerSpanProcessor(options) {
     headers: {}
   });
   try {
-    return traceloop4.createSpanProcessor({
+    return createSpanProcessor({
       apiKey: options.writeKey,
       baseUrl: localDebuggerUrl,
       disableBatch: options.disableBatching,
@@ -3487,7 +3898,7 @@ try {
             e instanceof Error ? e.message : e
           );
           span.setStatus({
-            code: import_api5.SpanStatusCode.ERROR,
+            code: import_api6.SpanStatusCode.ERROR,
             message: e instanceof Error ? e.message : String(e)
           });
           span.end();
@@ -3519,7 +3930,7 @@ try {
                     );
                     if (!spanEnded) {
                       span.setStatus({
-                        code: import_api5.SpanStatusCode.ERROR,
+                        code: import_api6.SpanStatusCode.ERROR,
                         message: e instanceof Error ? e.message : String(e)
                       });
                       span.end();
@@ -3533,7 +3944,7 @@ try {
               } catch (error) {
                 if (!spanEnded) {
                   const msg = error instanceof Error ? error.message : String(error);
-                  span.setStatus({ code: import_api5.SpanStatusCode.ERROR, message: msg });
+                  span.setStatus({ code: import_api6.SpanStatusCode.ERROR, message: msg });
                   span.recordException(error);
                   span.end();
                   spanEnded = true;
@@ -3556,7 +3967,7 @@ try {
       return result;
     }).catch((error) => {
       span.setStatus({
-        code: import_api5.SpanStatusCode.ERROR,
+        code: import_api6.SpanStatusCode.ERROR,
         message: error.message
       });
       span.recordException(error);
@@ -3619,7 +4030,7 @@ function patchBedrockStreamingBug(inst) {
               } catch (error) {
                 if (!spanEnded) {
                   const msg = error instanceof Error ? error.message : String(error);
-                  span.setStatus({ code: import_api5.SpanStatusCode.ERROR, message: msg });
+                  span.setStatus({ code: import_api6.SpanStatusCode.ERROR, message: msg });
                   span.recordException(error);
                   span.end();
                   spanEnded = true;
@@ -3642,7 +4053,7 @@ function patchBedrockStreamingBug(inst) {
       return result;
     }).catch((error) => {
       span.setStatus({
-        code: import_api5.SpanStatusCode.ERROR,
+        code: import_api6.SpanStatusCode.ERROR,
         message: error.message
       });
       span.recordException(error);
@@ -3655,7 +4066,9 @@ function createDefaultInstrumentations() {
   const bedrockInst = new import_instrumentation_bedrock.BedrockInstrumentation();
   patchBedrockStreamingBug(bedrockInst);
   return [
-    new import_instrumentation_openai.OpenAIInstrumentation(),
+    new import_instrumentation_openai.OpenAIInstrumentation({
+      enrichTokens: (process.env.TRACELOOP_ENRICH_TOKENS || "true").toLowerCase() === "true"
+    }),
     new import_instrumentation_anthropic.AnthropicInstrumentation(),
     new import_instrumentation_cohere.CohereInstrumentation(),
     bedrockInst,
@@ -3664,7 +4077,9 @@ function createDefaultInstrumentations() {
     new import_instrumentation_pinecone.PineconeInstrumentation(),
     new import_instrumentation_together.TogetherInstrumentation(),
     new import_instrumentation_chromadb.ChromaDBInstrumentation(),
-    new import_instrumentation_qdrant.QdrantInstrumentation()
+    new import_instrumentation_qdrant.QdrantInstrumentation(),
+    new import_instrumentation_langchain.LangChainInstrumentation(),
+    new import_instrumentation_llamaindex.LlamaIndexInstrumentation()
   ];
 }
 function createManualInstrumentations(instrumentModules) {
@@ -3720,17 +4135,29 @@ function createManualInstrumentations(instrumentModules) {
     inst.manuallyInstrument(instrumentModules.qdrant);
     instrumentations.push(inst);
   }
+  instrumentations.push(new import_instrumentation_langchain.LangChainInstrumentation());
+  if (instrumentModules.llamaIndex) {
+    const inst = new import_instrumentation_llamaindex.LlamaIndexInstrumentation();
+    inst.manuallyInstrument(instrumentModules.llamaIndex);
+    instrumentations.push(inst);
+  }
   return instrumentations;
 }
+function createConfiguredInstrumentations(options) {
+  if (options.instrumentModules && Object.keys(options.instrumentModules).length > 0) {
+    return createManualInstrumentations(options.instrumentModules);
+  }
+  return createDefaultInstrumentations();
+}
 function getCurrentTraceId() {
   var _a;
-  return (_a = import_api5.trace.getSpan(import_api5.context.active())) == null ? void 0 : _a.spanContext().traceId;
+  return (_a = import_api6.trace.getSpan(import_api6.context.active())) == null ? void 0 : _a.spanContext().traceId;
 }
 var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
   var _a;
   process.env.TRACELOOP_TELEMETRY = "false";
   const {
-    logLevel,
+    logLevel: logLevel2,
     useExternalOtel,
     tracingEnabled,
     disableBatching,
@@ -3777,10 +4204,10 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
   const activeInteractionsByEventId = new import_weakref.WeakValueMap();
   try {
     if (useExternalOtel) {
-      traceloop4.initialize({
+      initialize({
         baseUrl: apiUrl,
         apiKey: writeKey,
-        logLevel,
+        logLevel: logLevel2,
         // Traceloop's init banner ("Traceloop exporting traces to <baseUrl>") is
         // misleading inside our wrapper: in external-OTEL mode we create no
         // exporter at all (tracingEnabled: false). Suppress it.
@@ -3806,28 +4233,29 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
       const localDebuggerProcessor = createLocalDebuggerSpanProcessor({
         writeKey,
         disableBatching,
-        allowedInstrumentationLibraries: traceloop4.ALL_INSTRUMENTATION_LIBRARIES,
+        allowedInstrumentationLibraries: ALL_INSTRUMENTATION_LIBRARIES,
         localDebuggerUrlOpt: localDebuggerUrl
       });
       if (!cloudEnabledResolved) {
         if (localDebuggerProcessor) {
-          traceloop4.initialize({
+          initialize({
             baseUrl: apiUrl,
             apiKey: writeKey,
             ...otherOptions,
-            logLevel,
+            logLevel: logLevel2,
             silenceInitializationMessage: true,
             tracingEnabled: tracingEnabled !== false,
             disableBatch: disableBatching,
-            processor: localDebuggerProcessor
+            processor: localDebuggerProcessor,
+            instrumentations: createConfiguredInstrumentations(otherOptions)
           });
           traceloopInitialized = true;
           markTraceloopConfiguredWithoutClaim();
         } else {
-          traceloop4.initialize({
+          initialize({
             baseUrl: apiUrl,
             apiKey: writeKey,
-            logLevel,
+            logLevel: logLevel2,
             silenceInitializationMessage: true,
             tracingEnabled: false
           });
@@ -3873,18 +4301,19 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
               localDebuggerProcessor ? [localDebuggerProcessor] : []
             );
             try {
-              traceloop4.initialize({
+              initialize({
                 baseUrl: apiUrl,
                 apiKey: writeKey,
                 ...otherOptions,
-                logLevel,
+                logLevel: logLevel2,
                 silenceInitializationMessage: true,
                 // This branch only runs when tracing is enabled (the disabled
                 // case is handled above), so tracing is always on here.
                 tracingEnabled: true,
                 disableBatch: disableBatching,
                 processor,
-                exporter: guardedExporter
+                exporter: guardedExporter,
+                instrumentations: createConfiguredInstrumentations(otherOptions)
               });
               traceloopInitialized = true;
             } catch (initError) {
@@ -4015,7 +4444,7 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
         });
         return;
       }
-      const span = import_api5.trace.getTracer("traceloop.tracer").startSpan("ai.model.usage");
+      const span = import_api6.trace.getTracer("traceloop.tracer").startSpan("ai.model.usage");
       span.setAttribute("traceloop.span.kind", "llm");
       span.setAttribute("traceloop.association.properties.event_id", params.eventId);
       stampSpan(span, projectId, authHint);
@@ -4123,7 +4552,7 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
         );
       }
       const headers = withProjectIdHeaders(processorOptions == null ? void 0 : processorOptions.headers);
-      const productionProcessor = traceloop4.createSpanProcessor({
+      const productionProcessor = createSpanProcessor({
         baseUrl: apiUrl,
         apiKey: writeKey,
         disableBatch: disableBatching,
@@ -4189,7 +4618,7 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
       }
       if (traceloopInitialized) {
         try {
-          await traceloop4.forceFlush();
+          await forceFlush();
         } catch (e) {
         }
       }
@@ -4201,7 +4630,7 @@ var tracing = (analytics, apiUrl, writeKey, options, isDebug = false) => {
       await (directShipper == null ? void 0 : directShipper.shutdown());
       if (traceloopInitialized) {
         try {
-          await traceloop4.forceFlush();
+          await forceFlush();
         } catch (e) {
         }
       }