@introspection-sdk/introspection-node 0.4.3 → 0.4.4

package/dist/utils.js

@@ -1,10 +1,18 @@
 import { DiagLogLevel } from "@opentelemetry/api";
 import { HttpsProxyAgent } from "https-proxy-agent";
-function getHttpsProxyUrl() {
-    return process.env.HTTPS_PROXY || undefined;
-}
+import { resolveForwardProxyUrl } from "@introspection-sdk/introspection-proxy";
+/**
+ * Attach a forward-proxy agent to OTLP exporter options when a proxy is
+ * configured in the environment.
+ *
+ * Note: the OTLP proto exporters use Node's `http`/`https` stack, so this uses
+ * `HttpsProxyAgent` (an `http.Agent`) rather than the undici dispatcher from
+ * `./proxy.js` — a dispatcher only applies to `fetch`. The proxy *URL* is
+ * resolved by the shared {@link resolveForwardProxyUrl} so fetch and OTLP
+ * traffic agree on which proxy to use.
+ */
 export function withOtlpHttpsProxy(options) {
-    const proxyUrl = getHttpsProxyUrl();
+    const proxyUrl = resolveForwardProxyUrl();
     if (!proxyUrl)
         return options;
     return {

package/dist/utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAc,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpD,SAAS,gBAAgB;IACvB,OAAO,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,SAAS,CAAC;AAC9C,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAmB,OAAU;IAC7D,MAAM,QAAQ,GAAG,gBAAgB,EAAE,CAAC;IACpC,IAAI,CAAC,QAAQ;QAAE,OAAO,OAAO,CAAC;IAE9B,OAAO;QACL,GAAG,OAAO;QACV,gBAAgB,EAAE,GAAG,EAAE,CAAC,IAAI,eAAe,CAAC,QAAQ,CAAC;KACjD,CAAC;AACT,CAAC;AAED;;;GAGG;AACH,MAAM,mBAAmB;IACf,QAAQ,CAAe;IAE/B;QACE,MAAM,WAAW,GAAG,CAClB,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,MAAM,CAC9C,CAAC,WAAW,EAAE,CAAC;QAChB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC;IAClD,CAAC;IAEO,aAAa,CAAC,KAAa;QACjC,QAAQ,KAAK,EAAE,CAAC;YACd,KAAK,OAAO;gBACV,OAAO,YAAY,CAAC,KAAK,CAAC;YAC5B,KAAK,MAAM;gBACT,OAAO,YAAY,CAAC,IAAI,CAAC;YAC3B,KAAK,MAAM;gBACT,OAAO,YAAY,CAAC,IAAI,CAAC;YAC3B,KAAK,OAAO;gBACV,OAAO,YAAY,CAAC,KAAK,CAAC;YAC5B,KAAK,SAAS;gBACZ,OAAO,YAAY,CAAC,OAAO,CAAC;YAC9B;gBACE,OAAO,YAAY,CAAC,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,KAAmB;QACnC,OAAO,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,GAAG,IAAe;QACtB,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;YACvC,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,GAAG,IAAI,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,IAAI,CAAC,GAAG,IAAe;QACrB,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;YACtC,OAAO,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,IAAI,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IAED,IAAI,CAAC,GAAG,IAAe;QACrB,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;YACtC,OAAO,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,IAAI,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG,IAAe;QACtB,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;YACvC,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,GAAG,IAAI,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAG,IAAe;QACxB,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC;YACzC,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,GAAG,IAAI,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;CACF;AAED,MAAM,CAAC,MAAM,MAAM,GAAG,IAAI,mBAAmB,EAAE,CAAC"}
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAc,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,sBAAsB,EAAE,MAAM,wCAAwC,CAAC;AAEhF;;;;;;;;;GASG;AACH,MAAM,UAAU,kBAAkB,CAAmB,OAAU;IAC7D,MAAM,QAAQ,GAAG,sBAAsB,EAAE,CAAC;IAC1C,IAAI,CAAC,QAAQ;QAAE,OAAO,OAAO,CAAC;IAE9B,OAAO;QACL,GAAG,OAAO;QACV,gBAAgB,EAAE,GAAG,EAAE,CAAC,IAAI,eAAe,CAAC,QAAQ,CAAC;KACjD,CAAC;AACT,CAAC;AAED;;;GAGG;AACH,MAAM,mBAAmB;IACf,QAAQ,CAAe;IAE/B;QACE,MAAM,WAAW,GAAG,CAClB,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,MAAM,CAC9C,CAAC,WAAW,EAAE,CAAC;QAChB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC;IAClD,CAAC;IAEO,aAAa,CAAC,KAAa;QACjC,QAAQ,KAAK,EAAE,CAAC;YACd,KAAK,OAAO;gBACV,OAAO,YAAY,CAAC,KAAK,CAAC;YAC5B,KAAK,MAAM;gBACT,OAAO,YAAY,CAAC,IAAI,CAAC;YAC3B,KAAK,MAAM;gBACT,OAAO,YAAY,CAAC,IAAI,CAAC;YAC3B,KAAK,OAAO;gBACV,OAAO,YAAY,CAAC,KAAK,CAAC;YAC5B,KAAK,SAAS;gBACZ,OAAO,YAAY,CAAC,OAAO,CAAC;YAC9B;gBACE,OAAO,YAAY,CAAC,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,KAAmB;QACnC,OAAO,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,GAAG,IAAe;QACtB,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;YACvC,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,GAAG,IAAI,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,IAAI,CAAC,GAAG,IAAe;QACrB,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;YACtC,OAAO,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,IAAI,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IAED,IAAI,CAAC,GAAG,IAAe;QACrB,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;YACtC,OAAO,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,IAAI,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG,IAAe;QACtB,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;YACvC,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,GAAG,IAAI,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAG,IAAe;QACxB,IAAI,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC;YACzC,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,GAAG,IAAI,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;CACF;AAED,MAAM,CAAC,MAAM,MAAM,GAAG,IAAI,mBAAmB,EAAE,CAAC"}

package/package.json

@@ -1,8 +1,11 @@
 {
   "name": "@introspection-sdk/introspection-node",
-  "version": "0.4.3",
+  "version": "0.4.4",
   "license": "Apache-2.0",
   "description": "Introspection observability SDK for Node.js",
+  "engines": {
+    "node": ">=22"
+  },
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -15,6 +18,10 @@
       "types": "./dist/otel/index.d.ts",
       "default": "./dist/otel/index.js"
     },
+    "./otel/register": {
+      "types": "./dist/otel/register.d.ts",
+      "default": "./dist/otel/register.js"
+    },
     "./mastra": {
       "types": "./dist/otel/mastra-exporter.d.ts",
       "default": "./dist/otel/mastra-exporter.js"
@@ -43,10 +50,14 @@
     "@mastra/observability": "^1.14.0",
     "@mariozechner/pi-agent-core": "^0.73.1",
     "@mariozechner/pi-ai": "^0.73.1",
+    "@anthropic-ai/sdk": "^0.100.0",
+    "@anthropic-ai/claude-agent-sdk": "^0.3.154",
+    "@google/genai": "^2.6.0",
+    "ai": "^6.0.192",
     "openai": "^6.39.1",
     "typescript": "5.9.3",
-    "@repo/typescript-config": "0.0.0",
-    "@repo/eslint-config": "0.0.0"
+    "@repo/eslint-config": "0.0.0",
+    "@repo/typescript-config": "0.0.0"
   },
   "peerDependencies": {
     "@openai/agents": ">=0.0.1",
@@ -127,8 +138,9 @@
   "dependencies": {
     "@opentelemetry/api": "^1.9.1",
     "https-proxy-agent": "^7.0.6",
-    "@introspection-sdk/types": "0.4.3",
-    "@introspection-sdk/introspection-pi": "0.4.3"
+    "@introspection-sdk/types": "0.4.4",
+    "@introspection-sdk/introspection-pi": "0.4.4",
+    "@introspection-sdk/introspection-proxy": "0.1.0"
   },
   "scripts": {
     "lint": "eslint . --max-warnings 0",

package/src/otel/anthropic.ts

@@ -438,10 +438,92 @@ export async function tracedMessagesCreate(
  * Captures all content blocks including thinking (extended thinking) with
  * signatures. Supports both non-streaming and streaming calls.
  */
+/**
+ * Build the traced replacement for `messages.create`.
+ *
+ * Shared by {@link AnthropicInstrumentor.instrument} (patches a single client
+ * instance) and {@link AnthropicInstrumentor.instrumentClass} (patches
+ * `Anthropic.Messages.prototype.create` so every client is auto-traced). The
+ * returned function preserves the call-site `this`, so it works whether it is
+ * assigned to an instance property or onto the class prototype.
+ */
+function buildTracedCreate(
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  origCreate: (...args: any[]) => any,
+  tracer: Tracer,
+  defaultConversationId: string,
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+): (...args: any[]) => any {
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  return function tracedCreate(this: unknown, ...args: any[]) {
+    const kwargs = args[0] as Record<string, unknown>;
+    const isStream = kwargs?.stream === true;
+
+    // Baggage on the active context overrides the conversation ID bound at
+    // instrument() time, so callers can wrap a request in
+    // IntrospectionClient.withConversation() to scope a single call to a
+    // distinct conversation without re-instrumenting the client.
+    const baggage = propagation.getBaggage(otelContext.active());
+    const baggageConvId = baggage?.getEntry("gen_ai.conversation.id")?.value;
+    const baggageAgentName = baggage?.getEntry("gen_ai.agent.name")?.value;
+    const baggageAgentId = baggage?.getEntry("gen_ai.agent.id")?.value;
+
+    const { span } = startSpan(
+      tracer,
+      (kwargs?.model as string) || "unknown",
+      baggageConvId ?? defaultConversationId,
+    );
+    if (baggageAgentName) {
+      span.setAttribute("gen_ai.agent.name", baggageAgentName);
+    }
+    if (baggageAgentId) {
+      span.setAttribute("gen_ai.agent.id", baggageAgentId);
+    }
+    setRequestAttrs(span, kwargs || {});
+
+    if (isStream) {
+      const streamPromise = origCreate.apply(this, args) as Promise<
+        AsyncIterable<StreamEvent>
+      >;
+      return streamPromise.then(
+        (stream: AsyncIterable<StreamEvent>) => new TracedStream(stream, span),
+        (err: Error) => {
+          span.setStatus({ code: SpanStatusCode.ERROR, message: String(err) });
+          span.end();
+          throw err;
+        },
+      );
+    }
+
+    const resultPromise = origCreate.apply(this, args) as Promise<
+      Record<string, unknown>
+    >;
+    return resultPromise.then(
+      (response: Record<string, unknown>) => {
+        setResponseAttrs(span, response);
+        span.end();
+        return response;
+      },
+      (err: Error) => {
+        span.setStatus({ code: SpanStatusCode.ERROR, message: String(err) });
+        span.end();
+        throw err;
+      },
+    );
+  };
+}
+
+/**
+ * Marker placed on a patched `Messages.prototype.create` so repeated
+ * `instrumentClass()` calls (e.g. a second `introspection.init()`) are no-ops.
+ */
+const PATCHED = Symbol.for("introspection.anthropic.patched");
+
 export class AnthropicInstrumentor {
   private tracer: Tracer | null = null;
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
   private patchedClients: Array<{ client: any; originalCreate: any }> = [];
+  private protoRestores: Array<() => void> = [];
   private conversationId: string | undefined;
 
   instrument(opts: {
@@ -473,73 +555,54 @@ export class AnthropicInstrumentor {
 
     const origCreate = client.messages.create.bind(client.messages);
     this.patchedClients.push({ client, originalCreate: origCreate });
-    const tracer = this.tracer!;
-    const defaultConversationId = this.conversationId;
 
+    client.messages.create = buildTracedCreate(
+      origCreate,
+      this.tracer,
+      this.conversationId,
+    );
+  }
+
+  /**
+   * Patch `Anthropic.Messages.prototype.create` so **every** client constructed
+   * from the given module is traced — the one-liner path used by
+   * `introspection.init()`. Idempotent: a second call is a no-op.
+   *
+   * @param opts.anthropic The `@anthropic-ai/sdk` module's default export
+   *   (the `Anthropic` class). Its `.Messages.prototype.create` is patched.
+   */
+  instrumentClass(opts: {
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    client.messages.create = function (...args: any[]) {
-      const kwargs = args[0] as Record<string, unknown>;
-      const isStream = kwargs?.stream === true;
-
-      // Baggage on the active context overrides the conversation ID bound
-      // at instrument() time, so callers can wrap a request in
-      // IntrospectionClient.withConversation() to scope a single call to a
-      // distinct conversation without re-instrumenting the client.
-      const baggage = propagation.getBaggage(otelContext.active());
-      const baggageConvId = baggage?.getEntry("gen_ai.conversation.id")?.value;
-      const baggageAgentName = baggage?.getEntry("gen_ai.agent.name")?.value;
-      const baggageAgentId = baggage?.getEntry("gen_ai.agent.id")?.value;
-
-      const { span } = startSpan(
-        tracer,
-        (kwargs?.model as string) || "unknown",
-        baggageConvId ?? defaultConversationId,
+    anthropic: any;
+    tracerProvider?: BasicTracerProvider;
+    conversationId?: string;
+  }): void {
+    const Messages = opts.anthropic?.Messages;
+    const proto = Messages?.prototype;
+    if (typeof proto?.create !== "function") {
+      throw new Error(
+        "Invalid Anthropic module — expected `Anthropic.Messages.prototype.create`",
       );
-      if (baggageAgentName) {
-        span.setAttribute("gen_ai.agent.name", baggageAgentName);
-      }
-      if (baggageAgentId) {
-        span.setAttribute("gen_ai.agent.id", baggageAgentId);
-      }
-      setRequestAttrs(span, kwargs || {});
-
-      if (isStream) {
-        const streamPromise = origCreate(...args) as Promise<
-          AsyncIterable<StreamEvent>
-        >;
-        return streamPromise.then(
-          (stream: AsyncIterable<StreamEvent>) =>
-            new TracedStream(stream, span),
-          (err: Error) => {
-            span.setStatus({
-              code: SpanStatusCode.ERROR,
-              message: String(err),
-            });
-            span.end();
-            throw err;
-          },
-        );
-      }
+    }
+    if (proto.create[PATCHED]) return;
 
-      const resultPromise = origCreate(...args) as Promise<
-        Record<string, unknown>
-      >;
-      return resultPromise.then(
-        (response: Record<string, unknown>) => {
-          setResponseAttrs(span, response);
-          span.end();
-          return response;
-        },
-        (err: Error) => {
-          span.setStatus({
-            code: SpanStatusCode.ERROR,
-            message: String(err),
-          });
-          span.end();
-          throw err;
-        },
-      );
-    };
+    this.conversationId = opts.conversationId ?? crypto.randomUUID();
+    const provider = opts.tracerProvider ?? trace.getTracerProvider();
+    this.tracer = (provider as BasicTracerProvider).getTracer(
+      "introspection-anthropic",
+    );
+
+    const origCreate = proto.create;
+    const patched = buildTracedCreate(
+      origCreate,
+      this.tracer,
+      this.conversationId,
+    ) as ((...args: unknown[]) => unknown) & { [PATCHED]?: boolean };
+    patched[PATCHED] = true;
+    proto.create = patched;
+    this.protoRestores.push(() => {
+      proto.create = origCreate;
+    });
   }
 
   uninstrument(): void {
@@ -547,6 +610,8 @@ export class AnthropicInstrumentor {
       client.messages.create = originalCreate;
     }
     this.patchedClients = [];
+    for (const restore of this.protoRestores) restore();
+    this.protoRestores = [];
     this.tracer = null;
   }
 }

package/src/otel/gemini.ts

@@ -297,6 +297,79 @@ interface GeminiModelsApi {
  * Captures all content parts including thought summaries and signed
  * `thoughtSignature` payloads required for multi-turn replay.
  */
+/**
+ * Build the traced replacement for a non-streaming `generateContent`-style
+ * method. Shared by {@link GeminiInstrumentor.instrument} (patches a client's
+ * public `models.generateContent`) and {@link GeminiInstrumentor.instrumentClass}
+ * (patches `Models.prototype.generateContentInternal` so every client is
+ * auto-traced). Preserves the call-site `this`.
+ */
+function buildTracedGenerate(
+  origGenerate: (...args: unknown[]) => Promise<unknown>,
+  tracer: Tracer,
+  conversationId: string,
+): (...args: unknown[]) => Promise<unknown> {
+  return async function tracedGenerate(this: unknown, ...args: unknown[]) {
+    const kwargs = (args[0] as Record<string, unknown>) || {};
+    const span = startSpan(
+      tracer,
+      (kwargs.model as string) || "unknown",
+      conversationId,
+    );
+    setRequestAttrs(span, kwargs);
+    try {
+      const response = (await origGenerate.apply(this, args)) as Record<
+        string,
+        unknown
+      >;
+      setResponseAttrs(span, response);
+      return response;
+    } catch (err) {
+      span.setStatus({ code: SpanStatusCode.ERROR, message: String(err) });
+      throw err;
+    } finally {
+      span.end();
+    }
+  };
+}
+
+/** Build the traced replacement for a streaming `generateContentStream`-style method. */
+function buildTracedGenerateStream(
+  origStream: (...args: unknown[]) => Promise<unknown>,
+  tracer: Tracer,
+  conversationId: string,
+): (...args: unknown[]) => Promise<unknown> {
+  return async function tracedGenerateStream(
+    this: unknown,
+    ...args: unknown[]
+  ) {
+    const kwargs = (args[0] as Record<string, unknown>) || {};
+    const span = startSpan(
+      tracer,
+      (kwargs.model as string) || "unknown",
+      conversationId,
+    );
+    setRequestAttrs(span, kwargs);
+    try {
+      const stream = (await origStream.apply(
+        this,
+        args,
+      )) as AsyncIterable<StreamChunk>;
+      return new TracedStream(stream, span);
+    } catch (err) {
+      span.setStatus({ code: SpanStatusCode.ERROR, message: String(err) });
+      span.end();
+      throw err;
+    }
+  };
+}
+
+/**
+ * Marker placed on a patched `Models.prototype` method so repeated
+ * `instrumentClass()` calls (e.g. a second `introspection.init()`) are no-ops.
+ */
+const PATCHED = Symbol.for("introspection.gemini.patched");
+
 export class GeminiInstrumentor {
   private tracer: Tracer | null = null;
   private restores: Array<() => void> = [];
@@ -337,55 +410,76 @@ export class GeminiInstrumentor {
       if (origGenerateStream) api.generateContentStream = origGenerateStream;
     });
 
-    api.generateContent = async (...args: unknown[]) => {
-      const kwargs = (args[0] as Record<string, unknown>) || {};
-      const span = startSpan(
+    api.generateContent = buildTracedGenerate(
+      origGenerate,
+      tracer,
+      conversationId,
+    ) as GeminiModelsApi["generateContent"];
+
+    if (origGenerateStream) {
+      api.generateContentStream = buildTracedGenerateStream(
+        origGenerateStream,
         tracer,
-        (kwargs.model as string) || "unknown",
         conversationId,
+      ) as GeminiModelsApi["generateContentStream"];
+    }
+  }
+
+  /**
+   * Patch `Models.prototype.generateContentInternal` / `…StreamInternal` so
+   * **every** `GoogleGenAI` client is traced — the one-liner path used by
+   * `introspection.init()`. The public `generateContent` /
+   * `generateContentStream` are per-instance bound class fields that delegate
+   * to these prototype methods, so patching the prototype covers all clients.
+   * Idempotent: a second call is a no-op.
+   *
+   * @param opts.genai The `@google/genai` module (must expose `Models`).
+   */
+  instrumentClass(opts: {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    genai: any;
+    tracerProvider?: BasicTracerProvider;
+    conversationId?: string;
+  }): void {
+    const proto = opts.genai?.Models?.prototype;
+    if (typeof proto?.generateContentInternal !== "function") {
+      throw new Error(
+        "Invalid @google/genai module — expected `Models.prototype.generateContentInternal`",
       );
-      setRequestAttrs(span, kwargs);
-      try {
-        const response = (await origGenerate(...args)) as Record<
-          string,
-          unknown
-        >;
-        setResponseAttrs(span, response);
-        return response;
-      } catch (err) {
-        span.setStatus({
-          code: SpanStatusCode.ERROR,
-          message: String(err),
-        });
-        throw err;
-      } finally {
-        span.end();
-      }
-    };
+    }
+    if (proto.generateContentInternal[PATCHED]) return;
 
-    if (origGenerateStream) {
-      api.generateContentStream = async (...args: unknown[]) => {
-        const kwargs = (args[0] as Record<string, unknown>) || {};
-        const span = startSpan(
-          tracer,
-          (kwargs.model as string) || "unknown",
-          conversationId,
-        );
-        setRequestAttrs(span, kwargs);
-        try {
-          const stream = (await origGenerateStream(
-            ...args,
-          )) as AsyncIterable<StreamChunk>;
-          return new TracedStream(stream, span);
-        } catch (err) {
-          span.setStatus({
-            code: SpanStatusCode.ERROR,
-            message: String(err),
-          });
-          span.end();
-          throw err;
-        }
-      };
+    this.conversationId = opts.conversationId ?? crypto.randomUUID();
+    const provider = opts.tracerProvider ?? trace.getTracerProvider();
+    this.tracer = (provider as BasicTracerProvider).getTracer(
+      "introspection-gemini",
+    );
+    const conversationId = this.conversationId;
+
+    const origGenerate = proto.generateContentInternal;
+    const patchedGenerate = buildTracedGenerate(
+      origGenerate,
+      this.tracer,
+      conversationId,
+    ) as ((...args: unknown[]) => Promise<unknown>) & { [PATCHED]?: boolean };
+    patchedGenerate[PATCHED] = true;
+    proto.generateContentInternal = patchedGenerate;
+    this.restores.push(() => {
+      proto.generateContentInternal = origGenerate;
+    });
+
+    if (typeof proto.generateContentStreamInternal === "function") {
+      const origStream = proto.generateContentStreamInternal;
+      const patchedStream = buildTracedGenerateStream(
+        origStream,
+        this.tracer,
+        conversationId,
+      ) as ((...args: unknown[]) => Promise<unknown>) & { [PATCHED]?: boolean };
+      patchedStream[PATCHED] = true;
+      proto.generateContentStreamInternal = patchedStream;
+      this.restores.push(() => {
+        proto.generateContentStreamInternal = origStream;
+      });
     }
   }
 

package/src/otel/index.ts

@@ -78,3 +78,44 @@ export {
   addOpenInferenceAttributes,
   OpenInferenceSpanExporter,
 } from "../converters/openinference.js";
+
+// One-liner bootstrap (`introspection.init()`) + analytics proxies and the
+// per-framework handle accessors it binds. Auto-detects installed frameworks
+// and wires them into one shared provider.
+export {
+  init,
+  shutdown,
+  track,
+  feedback,
+  identify,
+  conversation,
+  withAgent,
+  withConversation,
+  withUserId,
+  withAnonymousId,
+  newConversationId,
+  getClient,
+  getTracerProvider,
+  getLangchainHandler,
+  getMastraExporter,
+  instrumentPi,
+  instrumentClaudeAgent,
+  _resetForTests,
+} from "./init.js";
+export type { InitOptions } from "./init.js";
+
+// Low-level OTel registration helper (also used by `init`).
+export { registerOTelGlobals } from "./setup.js";
+
+// Integration registry — for custom integrations passed to `init`.
+export {
+  DidNotEnable,
+  discoverIntegrations,
+  setupIntegrations,
+  resetInstalledForTests,
+} from "./integrations/index.js";
+export type {
+  Integration,
+  IntegrationSetupContext,
+  IntegrationHandles,
+} from "./integrations/index.js";