@traceai/google-genai-sdk 0.0.0-stage → 0.1.0

package/README.md

@@ -1,3 +1,73 @@
-# Temporary Holding Version
+# Google GenAI tracing for Node.js
 
-This version is a temporary placeholder for this package. An operational version to replace this has been submitted for review and is awaiting a staged release.
+Trace calls to the modern `@google/genai` SDK in Future AGI, including generation, chat, embeddings and Interactions.
+
+Use Node.js 20 or later and `@google/genai` 1.33–2.x. This package does not instrument the older `@google/generative-ai` SDK; keep `@traceai/google-genai` for applications using that legacy SDK. Keep your Future AGI secret key on your server, not in a browser bundle.
+
+## Install
+
+```bash
+npm install @google/genai@^2.32.0 @traceai/google-genai-sdk @traceai/fi-core
+```
+
+Set `GOOGLE_API_KEY`, `FI_API_KEY` and `FI_SECRET_KEY` in your server environment. Create your Future AGI keys in your workspace settings. The FI exporter sends OTLP traces to `https://api.futureagi.com/tracer/v1/traces`; it maps the FI environment variables to the `x-api-key` and `x-secret-key` collector headers. These credentials are not added to Google requests.
+
+## Trace your client
+
+Save this as `example.mjs` and run `node example.mjs`.
+
+```javascript
+import { GoogleGenAI } from '@google/genai';
+import { register, ProjectType } from '@traceai/fi-core';
+import { instrumentGoogleGenAI } from '@traceai/google-genai-sdk';
+
+const provider = register({
+  projectName: 'google-genai-app',
+  projectType: ProjectType.OBSERVE,
+  batch: true,
+});
+const ai = new GoogleGenAI({ apiKey: process.env.GOOGLE_API_KEY });
+const tracing = instrumentGoogleGenAI(ai, { tracerProvider: provider });
+
+try {
+  const response = await ai.models.generateContent({
+    model: 'gemini-2.5-flash',
+    contents: 'Suggest a name for a neighborhood bookstore.',
+  });
+  console.log(response.text);
+
+  for await (const chunk of await ai.models.generateContentStream({
+    model: 'gemini-2.5-flash',
+    contents: 'Write a one-sentence welcome for the bookstore.',
+  })) {
+    process.stdout.write(chunk.text ?? '');
+  }
+} finally {
+  tracing.dispose();
+  await provider.forceFlush();
+  await provider.shutdown();
+}
+```
+
+Open your `google-genai-app` project in Future AGI and inspect the generation spans. Each public SDK call is one logical span; automatic tool calling and continuation remain the SDK's responsibility, without additional synthetic tool-execution spans. Generation usage sums independent Google requests, counting each stream's cumulative usage only once. Tool definitions come from the declarations the SDK resolves, without invoking tools again. Existing OpenTelemetry parents and FI session/user context are retained.
+
+If a generation request omits usage, `google.genai.usage_capture=partial` marks the observed counts as incomplete. If per-request observation is unavailable, for example because SDK methods were frozen before setup, usage is omitted and marked `google.genai.usage_capture=unavailable` rather than reporting the final tool hop as a complete call total.
+
+## Supported calls
+
+- `models.generateContent` and `generateContentStream`, including chat calls through `chats.create().sendMessage` and `sendMessageStream`.
+- `models.embedContent`: text inputs, returned embedding count/dimensions and usage when exposed by the SDK. Vectors are never captured. Text-batch input counts are recorded only for unambiguous input shapes.
+- `interactions.create`, `get`, `cancel` and `delete` when the installed SDK exposes them. Older SDK promise helpers are preserved: `withResponse()` retains parsed data and the HTTP response; `asResponse()` leaves the body unread and records request-only timing (`google.genai.response_capture=request_only`). Background calls describe submission or an explicit poll, not remote execution. No extra polling is performed.
+- `models.generateImages`, `editImage` and `generateVideos` where the Google backend supports them. Video spans describe submission, not finished video generation. Media bytes and file URLs are never captured.
+
+Consume streams with `for await`, or use the Interactions stream's reader/legacy readable conversion. Finish consumption or close the iterator when you stop early so its span can end. Stream branching/piping retains the SDK's behavior but ends a request-only span, marked `google.genai.stream_capture=request_only`; use iteration or readers for full stream timing and content. Live WebSocket sessions, cache/file/batch administration and token counting are not traced by this package; their SDK methods remain available.
+
+## Privacy and lifecycle
+
+The shared FI controls apply when you instrument the client: `FI_HIDE_INPUTS`, `FI_HIDE_OUTPUTS`, `FI_HIDE_INPUT_MESSAGES`, `FI_HIDE_OUTPUT_MESSAGES`, `FI_HIDE_INPUT_TEXT`, `FI_HIDE_OUTPUT_TEXT`, `FI_HIDE_INPUT_IMAGES`, `FI_HIDE_EMBEDDING_VECTORS`, `FI_PII_REDACTION` and `FI_HIDE_ERROR_DETAILS`. Set them before calling `instrumentGoogleGenAI`, or pass the corresponding `traceConfig` options.
+
+Text-hiding options omit that side's entire content surface, including tool arguments and schemas. Raw request/response objects, transport headers, media bytes, URLs, embedding vectors and provider error messages/stacks are never recorded, even with content capture enabled. Error spans retain error status and a generic error type. `FI_BASE64_IMAGE_MAX_LENGTH` cannot enable media capture. Content capture is bounded to 16,384 characters per call; adjust `maxContentLength` between 0 and 65,536. Truncated captures carry `google.genai.capture_truncated`.
+
+`disable()` stops tracing new calls; `enable()` resumes it. `dispose()` stops tracing new calls immediately and restores the client's methods after in-flight calls drain. Repeated setup on a client returns the existing handle and retains its original options. To change providers or privacy settings, dispose and instrument again. Frozen methods cannot be instrumented; configure tracing before freezing your client. Always flush and shut down the provider when your process exits.
+
+If you already configured an OpenTelemetry provider, pass it as `tracerProvider` instead of registering another provider. Instrument each client once, and avoid combining this package with another instrumentor for the same SDK calls.

package/dist/capture.d.ts

@@ -0,0 +1,44 @@
+import type { Span } from "@opentelemetry/api";
+import type { TraceConfig } from "@traceai/fi-core";
+type RecordValue = Record<string, unknown>;
+export declare function record(value: unknown): RecordValue;
+/** Capture is deliberately allowlisted: never serialize SDK options, HTTP responses or binary. */
+export declare class Capture {
+    private span;
+    private remaining;
+    private output;
+    private firstContent;
+    private truncated;
+    private generation;
+    private embedding;
+    private embeddingUsageSnapshot;
+    private hops;
+    private capturedTools;
+    private usageTotals;
+    private usageHops;
+    private started;
+    private allowInput;
+    private allowOutput;
+    constructor(span: Span, config: TraceConfig, limit: number);
+    private bounded;
+    private text;
+    private json;
+    private parts;
+    input(params: RecordValue, operation: string): void;
+    private tools;
+    /** A hop gets its own cumulative usage snapshot. Sum independent hops, not SSE chunks. */
+    generationHop(params: RecordValue): (value: unknown) => void;
+    string(key: string, value: unknown): void;
+    private usage;
+    /** Only numeric, allowlisted billing fields from the SDK-owned JSON parse. */
+    embeddingUsage(value: unknown): void;
+    /** Vertex exposes billing counts on the converted embeddings, not usageMetadata.
+     * Missing or invalid statistics must not turn a partial sum into a call total.
+     */
+    private embeddingTokens;
+    response(value: unknown): void;
+    private append;
+    finish(): void;
+}
+export {};
+//# sourceMappingURL=capture.d.ts.map

package/dist/capture.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"capture.d.ts","sourceRoot":"","sources":["../src/capture.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAGpD,KAAK,WAAW,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAC3C,wBAAgB,MAAM,CAAC,KAAK,EAAE,OAAO,GAAG,WAAW,CAIlD;AAID,kGAAkG;AAClG,qBAAa,OAAO;IAgBhB,OAAO,CAAC,IAAI;IAfd,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,sBAAsB,CAA6B;IAC3D,OAAO,CAAC,IAAI,CAAK;IACjB,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,WAAW,CAA6B;IAChD,OAAO,CAAC,SAAS,CAA6B;IAC9C,OAAO,CAAC,OAAO,CAAqB;IACpC,OAAO,CAAC,UAAU,CAAU;IAC5B,OAAO,CAAC,WAAW,CAAU;gBAEnB,IAAI,EAAE,IAAI,EAClB,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,MAAM;IAef,OAAO,CAAC,OAAO;IAKf,OAAO,CAAC,IAAI;IAQZ,OAAO,CAAC,IAAI;IA6BZ,OAAO,CAAC,KAAK;IAuBb,KAAK,CAAC,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,GAAG,IAAI;IAmEnD,OAAO,CAAC,KAAK;IAQb,0FAA0F;IAC1F,aAAa,CAAC,MAAM,EAAE,WAAW,GAAG,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI;IAiC5D,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,GAAG,IAAI;IAMzC,OAAO,CAAC,KAAK;IAsDb,8EAA8E;IAC9E,cAAc,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI;IAMpC;;OAEG;IACH,OAAO,CAAC,eAAe;IAWvB,QAAQ,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI;IAuH9B,OAAO,CAAC,MAAM;IAId,MAAM,IAAI,IAAI;CAiBf"}

package/dist/capture.js

@@ -0,0 +1,414 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Capture = void 0;
+exports.record = record;
+const fi_semantic_conventions_1 = require("@traceai/fi-semantic-conventions");
+function record(value) {
+    return value !== null && typeof value === "object"
+        ? value
+        : {};
+}
+const list = (value) => Array.isArray(value) ? value : value == null ? [] : [value];
+/** Capture is deliberately allowlisted: never serialize SDK options, HTTP responses or binary. */
+class Capture {
+    span;
+    remaining;
+    output = [];
+    firstContent = false;
+    truncated = false;
+    generation = false;
+    embedding = false;
+    embeddingUsageSnapshot = new Map();
+    hops = 0;
+    capturedTools = false;
+    usageTotals = new Map();
+    usageHops = new Map();
+    started = performance.now();
+    allowInput;
+    allowOutput;
+    constructor(span, config, limit) {
+        this.span = span;
+        this.remaining = limit;
+        // Suppress the whole content surface for text hiding, including tool arguments and schemas.
+        this.allowInput = !(config.hideInputs ||
+            config.hideInputMessages ||
+            config.hideInputText);
+        this.allowOutput = !(config.hideOutputs ||
+            config.hideOutputMessages ||
+            config.hideOutputText);
+    }
+    bounded(value, size) {
+        const items = list(value);
+        if (items.length > size)
+            this.truncated = true;
+        return items.slice(0, size);
+    }
+    text(value) {
+        if (typeof value !== "string")
+            return undefined;
+        const result = value.slice(0, Math.max(0, Math.min(this.remaining, 4096)));
+        if (result.length < value.length)
+            this.truncated = true;
+        if (this.remaining <= 0)
+            return undefined;
+        this.remaining -= result.length + 16;
+        return result;
+    }
+    json(value, depth = 0) {
+        if (typeof value === "function" || value === undefined)
+            return undefined;
+        if (this.remaining <= 0 || depth > 6) {
+            this.truncated = true;
+            return undefined;
+        }
+        this.remaining -= 16;
+        if (value === null)
+            return null;
+        if (typeof value === "string")
+            return this.text(value);
+        if (typeof value === "boolean" || typeof value === "number")
+            return value;
+        if (Array.isArray(value))
+            return this.bounded(value, 32).map((v) => this.json(v, depth + 1));
+        const result = {};
+        const keys = Object.keys(record(value));
+        if (keys.length > 32)
+            this.truncated = true;
+        for (const key of keys.slice(0, 32)) {
+            // Tool arguments and responses can carry credentials at any nesting level.
+            if (/data|bytes|base64|headers|api.?key|secret|token|uri|url|auth|cookie|password|passphrase|passwd|pwd|credential|private.?key|access.?key/i.test(key))
+                continue;
+            const boundedKey = this.text(key);
+            if (boundedKey === undefined)
+                break;
+            result[boundedKey] = this.json(record(value)[key], depth + 1);
+        }
+        return result;
+    }
+    parts(value) {
+        const result = [];
+        for (const item of this.bounded(value, 64)) {
+            if (this.remaining <= 0)
+                break;
+            const part = record(item);
+            const text = this.text(typeof item === "string" ? item : part.text);
+            if (text !== undefined)
+                result.push({ type: "text", text });
+            for (const key of ["functionCall", "functionResponse"]) {
+                if (part[key])
+                    result.push({ type: key, ...record(this.json(part[key])) });
+            }
+            if (part.type === "function_call" || part.type === "function_result")
+                result.push(this.json(part));
+            // Only modality is retained for media, not raw bytes, signed URLs, or file URIs.
+            for (const key of ["inlineData", "fileData", "image", "audio", "video"]) {
+                if (part[key]) {
+                    result.push({ type: key });
+                    this.remaining -= 32;
+                }
+            }
+        }
+        return result;
+    }
+    input(params, operation) {
+        this.generation =
+            operation === "generateContent" || operation === "generateContentStream";
+        this.embedding = operation === "embedContent";
+        this.string(fi_semantic_conventions_1.SemanticConventions.LLM_MODEL_NAME, params.model);
+        if (this.embedding) {
+            this.string(fi_semantic_conventions_1.SemanticConventions.EMBEDDING_MODEL_NAME, params.model);
+            // Text batch semantics. Multimodal embedding models group parts differently;
+            // omit a speculative input count for those shapes.
+            if (typeof params.contents === "string")
+                this.span.setAttribute("embedding.input_count", 1);
+            else if (Array.isArray(params.contents) &&
+                params.contents.every((v) => typeof v === "string") &&
+                !String(params.model).includes("gemini-embedding-2"))
+                this.span.setAttribute("embedding.input_count", params.contents.length);
+        }
+        this.string("google.genai.previous_interaction_id", params.previous_interaction_id);
+        if (typeof params.background === "boolean")
+            this.span.setAttribute("google.genai.background", params.background);
+        const config = record(params.config ?? params.generation_config);
+        const numeric = {};
+        for (const key of [
+            "temperature",
+            "topP",
+            "topK",
+            "maxOutputTokens",
+            "candidateCount",
+            "seed",
+            "top_p",
+            "top_k",
+            "max_output_tokens",
+        ]) {
+            if (typeof config[key] === "number")
+                numeric[key] = config[key];
+        }
+        this.span.setAttribute(fi_semantic_conventions_1.SemanticConventions.LLM_INVOCATION_PARAMETERS, JSON.stringify(numeric));
+        if (!this.allowInput)
+            return;
+        const messages = [];
+        const system = params.system_instruction ?? config.systemInstruction;
+        if (system)
+            messages.push({
+                role: "system",
+                content: this.parts(record(system).parts ?? system),
+            });
+        for (const item of this.bounded(params.contents ?? params.input ?? params.prompt, 64)) {
+            if (this.remaining <= 0)
+                break;
+            const content = record(item);
+            messages.push({
+                role: this.text(content.role) ?? "user",
+                content: this.parts(content.parts ?? content.content ?? item),
+            });
+        }
+        if (messages.length)
+            this.span.setAttribute(fi_semantic_conventions_1.SemanticConventions.LLM_INPUT_MESSAGES, JSON.stringify(messages));
+        // Generation tools are observed only after the SDK resolves CallableTools.
+        if (!this.generation)
+            this.tools(config.tools ?? params.tools);
+    }
+    tools(value) {
+        if (!this.allowInput || this.capturedTools || value == null)
+            return;
+        this.capturedTools = true;
+        this.span.setAttribute(fi_semantic_conventions_1.SemanticConventions.LLM_TOOLS, JSON.stringify(this.json(value)) ?? "[]");
+    }
+    /** A hop gets its own cumulative usage snapshot. Sum independent hops, not SSE chunks. */
+    generationHop(params) {
+        this.hops++;
+        this.tools(record(params.config).tools);
+        const previous = new Map();
+        let promptInput;
+        let toolInput;
+        return (value) => {
+            const usage = record(record(value).usageMetadata);
+            const counts = this.usage(usage);
+            const prompt = counts.get(fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_PROMPT);
+            if (prompt !== undefined)
+                promptInput = prompt;
+            const tool = usage.toolUsePromptTokenCount;
+            if (typeof tool === "number" && Number.isFinite(tool) && tool >= 0)
+                toolInput = tool;
+            // Built-in tool results are additional input, not generated output. Keep
+            // each counter's latest snapshot even when a later chunk omits one, then
+            // let the existing hop accounting deduplicate the combined input count.
+            if (toolInput !== undefined) {
+                const input = (promptInput ?? 0) + toolInput;
+                if ((promptInput ?? 0) >= 0 && Number.isFinite(input))
+                    counts.set(fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_PROMPT, input);
+            }
+            for (const [key, count] of counts) {
+                const total = (this.usageTotals.get(key) ?? 0) + count - (previous.get(key) ?? 0);
+                if (!previous.has(key))
+                    this.usageHops.set(key, (this.usageHops.get(key) ?? 0) + 1);
+                previous.set(key, count);
+                this.usageTotals.set(key, total);
+                this.span.setAttribute(key, total);
+            }
+        };
+    }
+    string(key, value) {
+        if (typeof value === "string") {
+            if (value.length > 256)
+                this.truncated = true;
+            this.span.setAttribute(key, value.slice(0, 256));
+        }
+    }
+    usage(usage) {
+        const result = new Map();
+        for (const [key, names] of [
+            [
+                fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_PROMPT,
+                ["promptTokenCount", "total_input_tokens", "input_tokens"],
+            ],
+            [
+                fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_COMPLETION,
+                ["candidatesTokenCount", "total_output_tokens", "output_tokens"],
+            ],
+            [fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_TOTAL, ["totalTokenCount", "total_tokens"]],
+            [
+                fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ,
+                ["cachedContentTokenCount", "total_cached_tokens"],
+            ],
+            [
+                fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_COMPLETION_DETAILS_REASONING,
+                [
+                    "thoughtsTokenCount",
+                    "total_thought_tokens",
+                    "total_reasoning_tokens",
+                ],
+            ],
+        ]) {
+            const count = names
+                .map((name) => usage[name])
+                .find((v) => typeof v === "number" && Number.isFinite(v));
+            if (typeof count === "number")
+                result.set(key, count);
+        }
+        // These are numeric cost inputs, not media content. Keep them even when all
+        // content is hidden. Fixed modality names avoid provider-controlled keys.
+        for (const [prefix, details] of [
+            // EmbeddingUsageMetadata uses singular Token; generation uses Tokens.
+            // These are aliases of one snapshot, not independent counts to add.
+            [fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_PROMPT, usage.promptTokensDetails ?? usage.promptTokenDetails ?? usage.input_tokens_by_modality],
+            [fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_COMPLETION, usage.candidatesTokensDetails ?? usage.output_tokens_by_modality],
+            [fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ, usage.cacheTokensDetails ?? usage.cached_tokens_by_modality],
+        ]) {
+            for (const value of list(details)) {
+                const detail = record(value);
+                const modality = typeof detail.modality === "string" ? detail.modality.toLowerCase() : "";
+                const count = detail.tokenCount ?? detail.tokens;
+                if (["text", "audio", "image", "video", "document"].includes(modality) &&
+                    typeof count === "number" && Number.isFinite(count) && count >= 0) {
+                    const key = `${prefix}.${modality}`;
+                    result.set(key, (result.get(key) ?? 0) + count);
+                }
+            }
+        }
+        return result;
+    }
+    /** Only numeric, allowlisted billing fields from the SDK-owned JSON parse. */
+    embeddingUsage(value) {
+        for (const [key, count] of this.usage(record(record(value).usageMetadata))) {
+            this.embeddingUsageSnapshot.set(key, count);
+            this.span.setAttribute(key, count);
+        }
+    }
+    /** Vertex exposes billing counts on the converted embeddings, not usageMetadata.
+     * Missing or invalid statistics must not turn a partial sum into a call total.
+     */
+    embeddingTokens(embeddings) {
+        if (!Array.isArray(embeddings) || embeddings.length === 0)
+            return undefined;
+        let total = 0;
+        for (const embedding of embeddings) {
+            const count = record(record(embedding).statistics).tokenCount;
+            if (typeof count !== "number" || !Number.isFinite(count) || count < 0)
+                return undefined;
+            total += count;
+        }
+        return Number.isFinite(total) ? total : undefined;
+    }
+    response(value) {
+        const response = record(value);
+        const interaction = record(response.interaction);
+        this.string(fi_semantic_conventions_1.SemanticConventions.GEN_AI_RESPONSE_ID, response.responseId ?? interaction.id ?? response.id);
+        this.string(fi_semantic_conventions_1.SemanticConventions.GEN_AI_RESPONSE_MODEL, response.modelVersion ?? interaction.model ?? response.model);
+        this.string("google.genai.status", interaction.status ?? response.status);
+        this.string("google.genai.operation_id", response.name);
+        if (typeof response.done === "boolean")
+            this.span.setAttribute("google.genai.operation_done", response.done);
+        if (Array.isArray(response.embeddings)) {
+            this.span.setAttribute("embedding.count", response.embeddings.length);
+            const values = record(response.embeddings[0]).values;
+            if (Array.isArray(values))
+                this.span.setAttribute("embedding.dimensions", values.length);
+        }
+        if (!this.generation) {
+            const usage = record(response.usageMetadata ??
+                interaction.usage ??
+                response.usage ??
+                record(response.metadata).total_usage);
+            const counts = this.usage(usage);
+            if (this.embedding) {
+                // The parse hook and public result describe the same response. Preserve
+                // aggregate fields stripped by SDK conversion, without adding snapshots.
+                for (const [key, count] of this.embeddingUsageSnapshot)
+                    if (!counts.has(key))
+                        counts.set(key, count);
+                if (!counts.has(fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_PROMPT)) {
+                    const input = this.embeddingTokens(response.embeddings);
+                    if (input !== undefined)
+                        counts.set(fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_PROMPT, input);
+                }
+                const input = counts.get(fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_PROMPT);
+                if (input !== undefined && !counts.has(fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_TOTAL))
+                    counts.set(fi_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_TOTAL, input);
+            }
+            for (const [key, count] of counts)
+                this.span.setAttribute(key, count);
+        }
+        const candidates = list(response.candidates);
+        const reasons = candidates
+            .map((c) => record(c).finishReason)
+            .filter((x) => typeof x === "string");
+        if (reasons.length > 16 || reasons.some((reason) => reason.length > 128))
+            this.truncated = true;
+        if (reasons.length)
+            this.span.setAttribute(fi_semantic_conventions_1.SemanticConventions.GEN_AI_RESPONSE_FINISH_REASONS, reasons.slice(0, 16).map((reason) => reason.slice(0, 128)));
+        if (!this.firstContent &&
+            (candidates
+                .slice(0, 16)
+                .some((candidate) => list(record(record(candidate).content).parts).length > 0) ||
+                response.event_type === "content.delta" ||
+                response.event_type === "step.delta" ||
+                list(response.steps ??
+                    interaction.steps ??
+                    response.outputs ??
+                    interaction.outputs).length > 0)) {
+            this.firstContent = true;
+            this.span.setAttribute("gen_ai.response.time_to_first_token", (performance.now() - this.started) / 1000);
+        }
+        if (!this.allowOutput || this.remaining <= 0)
+            return;
+        for (const candidate of this.bounded(candidates, 16)) {
+            const content = record(record(candidate).content);
+            this.append({
+                role: this.text(content.role) ?? "model",
+                content: this.parts(content.parts),
+            });
+        }
+        // Released Interactions use model_output steps; older releases use outputs.
+        const steps = response.steps ?? interaction.steps;
+        if (steps !== undefined) {
+            this.output = [];
+            for (const step of this.bounded(steps, 64)) {
+                if (record(step).type === "model_output")
+                    this.append({
+                        role: "model",
+                        content: this.parts(record(step).content),
+                    });
+            }
+        }
+        else if (response.outputs ?? interaction.outputs) {
+            this.output = [];
+            this.append({
+                role: "model",
+                content: this.parts(response.outputs ?? interaction.outputs),
+            });
+        }
+        else if (response.event_type === "content.delta" ||
+            response.event_type === "step.delta") {
+            this.append({ role: "model", content: this.parts(response.delta) });
+        }
+        else if (response.event_type === "step.start" &&
+            record(response.step).type === "model_output") {
+            this.append({
+                role: "model",
+                content: this.parts(record(response.step).content),
+            });
+        }
+    }
+    append(message) {
+        if (!list(record(message).content).length)
+            return;
+        this.output.push(message);
+    }
+    finish() {
+        if (this.output.length)
+            this.span.setAttribute(fi_semantic_conventions_1.SemanticConventions.LLM_OUTPUT_MESSAGES, JSON.stringify(this.output));
+        if (this.truncated || this.remaining <= 0)
+            this.span.setAttribute("google.genai.capture_truncated", true);
+        if (this.generation && !this.hops)
+            this.span.setAttribute("google.genai.usage_capture", "unavailable");
+        else if (this.generation &&
+            (!this.usageHops.size ||
+                [...this.usageHops.values()].some((count) => count < this.hops)))
+            this.span.setAttribute("google.genai.usage_capture", "partial");
+    }
+}
+exports.Capture = Capture;
+//# sourceMappingURL=capture.js.map

package/dist/capture.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"capture.js","sourceRoot":"","sources":["../src/capture.ts"],"names":[],"mappings":";;;AAKA,wBAIC;AAPD,8EAA4E;AAG5E,SAAgB,MAAM,CAAC,KAAc;IACnC,OAAO,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ;QAChD,CAAC,CAAE,KAAqB;QACxB,CAAC,CAAC,EAAE,CAAC;AACT,CAAC;AACD,MAAM,IAAI,GAAG,CAAC,KAAc,EAAa,EAAE,CACzC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;AAE9D,kGAAkG;AAClG,MAAa,OAAO;IAgBR;IAfF,SAAS,CAAS;IAClB,MAAM,GAAc,EAAE,CAAC;IACvB,YAAY,GAAG,KAAK,CAAC;IACrB,SAAS,GAAG,KAAK,CAAC;IAClB,UAAU,GAAG,KAAK,CAAC;IACnB,SAAS,GAAG,KAAK,CAAC;IAClB,sBAAsB,GAAG,IAAI,GAAG,EAAkB,CAAC;IACnD,IAAI,GAAG,CAAC,CAAC;IACT,aAAa,GAAG,KAAK,CAAC;IACtB,WAAW,GAAG,IAAI,GAAG,EAAkB,CAAC;IACxC,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAC;IACtC,OAAO,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IAC5B,UAAU,CAAU;IACpB,WAAW,CAAU;IAC7B,YACU,IAAU,EAClB,MAAmB,EACnB,KAAa;QAFL,SAAI,GAAJ,IAAI,CAAM;QAIlB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,4FAA4F;QAC5F,IAAI,CAAC,UAAU,GAAG,CAAC,CACjB,MAAM,CAAC,UAAU;YACjB,MAAM,CAAC,iBAAiB;YACxB,MAAM,CAAC,aAAa,CACrB,CAAC;QACF,IAAI,CAAC,WAAW,GAAG,CAAC,CAClB,MAAM,CAAC,WAAW;YAClB,MAAM,CAAC,kBAAkB;YACzB,MAAM,CAAC,cAAc,CACtB,CAAC;IACJ,CAAC;IACO,OAAO,CAAC,KAAc,EAAE,IAAY;QAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,IAAI,KAAK,CAAC,MAAM,GAAG,IAAI;YAAE,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAC/C,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAC9B,CAAC;IACO,IAAI,CAAC,KAAc;QACzB,IAAI,OAAO,KAAK,KAAK,QAAQ;YAAE,OAAO,SAAS,CAAC;QAChD,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3E,IAAI,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM;YAAE,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACxD,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC;YAAE,OAAO,SAAS,CAAC;QAC1C,IAAI,CAAC,SAAS,IAAI,MAAM,CAAC,MAAM,GAAG,EAAE,CAAC;QACrC,OAAO,MAAM,CAAC;IAChB,CAAC;IACO,IAAI,CAAC,KAAc,EAAE,KAAK,GAAG,CAAC;QACpC,IAAI,OAAO,KAAK,KAAK,UAAU,IAAI,KAAK,KAAK,SAAS;YAAE,OAAO,SAAS,CAAC;QACzE,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YACrC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC;QACrB,IAAI,KAAK,KAAK,IAAI;YAAE,OAAO,IAAI,CAAC;QAChC,IAAI,OAAO,KAAK,KAAK,QAAQ;YAAE,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvD,IAAI,OAAO,KAAK,KAAK,SAAS,IAAI,OAAO,KAAK,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QAC1E,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;YACtB,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;QACrE,MAAM,MAAM,GAAgB,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QACxC,IAAI,IAAI,CAAC,MAAM,GAAG,EAAE;YAAE,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAC5C,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;YACpC,2EAA2E;YAC3E,IACE,yIAAyI,CAAC,IAAI,CAC5I,GAAG,CACJ;gBAED,SAAS;YACX,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAClC,IAAI,UAAU,KAAK,SAAS;gBAAE,MAAM;YACpC,MAAM,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;QAChE,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IACO,KAAK,CAAC,KAAc;QAC1B,MAAM,MAAM,GAAc,EAAE,CAAC;QAC7B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC;YAC3C,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC;gBAAE,MAAM;YAC/B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;YAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpE,IAAI,IAAI,KAAK,SAAS;gBAAE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5D,KAAK,MAAM,GAAG,IAAI,CAAC,cAAc,EAAE,kBAAkB,CAAU,EAAE,CAAC;gBAChE,IAAI,IAAI,CAAC,GAAG,CAAC;oBACX,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAChE,CAAC;YACD,IAAI,IAAI,CAAC,IAAI,KAAK,eAAe,IAAI,IAAI,CAAC,IAAI,KAAK,iBAAiB;gBAClE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAC/B,iFAAiF;YACjF,KAAK,MAAM,GAAG,IAAI,CAAC,YAAY,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,CAAC;gBACxE,IAAI,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;oBACd,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;oBAC3B,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC;gBACvB,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,KAAK,CAAC,MAAmB,EAAE,SAAiB;QAC1C,IAAI,CAAC,UAAU;YACb,SAAS,KAAK,iBAAiB,IAAI,SAAS,KAAK,uBAAuB,CAAC;QAC3E,IAAI,CAAC,SAAS,GAAG,SAAS,KAAK,cAAc,CAAC;QAC9C,IAAI,CAAC,MAAM,CAAC,6CAAC,CAAC,cAAc,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5C,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,CAAC,6CAAC,CAAC,oBAAoB,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;YAClD,6EAA6E;YAC7E,mDAAmD;YACnD,IAAI,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ;gBACrC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC;iBAChD,IACH,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC;gBAC9B,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC;gBACnD,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,oBAAoB,CAAC;gBAEpD,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,uBAAuB,EAAE,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC5E,CAAC;QACD,IAAI,CAAC,MAAM,CACT,sCAAsC,EACtC,MAAM,CAAC,uBAAuB,CAC/B,CAAC;QACF,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,SAAS;YACxC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,yBAAyB,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,iBAAiB,CAAC,CAAC;QACjE,MAAM,OAAO,GAAgB,EAAE,CAAC;QAChC,KAAK,MAAM,GAAG,IAAI;YAChB,aAAa;YACb,MAAM;YACN,MAAM;YACN,iBAAiB;YACjB,gBAAgB;YAChB,MAAM;YACN,OAAO;YACP,OAAO;YACP,mBAAmB;SACpB,EAAE,CAAC;YACF,IAAI,OAAO,MAAM,CAAC,GAAG,CAAC,KAAK,QAAQ;gBAAE,OAAO,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAClE,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,YAAY,CACpB,6CAAC,CAAC,yBAAyB,EAC3B,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CACxB,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,UAAU;YAAE,OAAO;QAC7B,MAAM,QAAQ,GAAc,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,MAAM,CAAC,kBAAkB,IAAI,MAAM,CAAC,iBAAiB,CAAC;QACrE,IAAI,MAAM;YACR,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,IAAI,MAAM,CAAC;aACpD,CAAC,CAAC;QACL,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,OAAO,CAC7B,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,MAAM,EAChD,EAAE,CACH,EAAE,CAAC;YACF,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC;gBAAE,MAAM;YAC/B,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;YAC7B,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,MAAM;gBACvC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC;aAC9D,CAAC,CAAC;QACL,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM;YACjB,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,6CAAC,CAAC,kBAAkB,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;QACzE,2EAA2E;QAC3E,IAAI,CAAC,IAAI,CAAC,UAAU;YAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC;IACjE,CAAC;IACO,KAAK,CAAC,KAAc;QAC1B,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,aAAa,IAAI,KAAK,IAAI,IAAI;YAAE,OAAO;QACpE,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAC1B,IAAI,CAAC,IAAI,CAAC,YAAY,CACpB,6CAAC,CAAC,SAAS,EACX,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,IAAI,CACzC,CAAC;IACJ,CAAC;IACD,0FAA0F;IAC1F,aAAa,CAAC,MAAmB;QAC/B,IAAI,CAAC,IAAI,EAAE,CAAC;QACZ,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC;QACxC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC3C,IAAI,WAA+B,CAAC;QACpC,IAAI,SAA6B,CAAC;QAClC,OAAO,CAAC,KAAK,EAAE,EAAE;YACf,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC,CAAC;YAClD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACjC,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,6CAAC,CAAC,sBAAsB,CAAC,CAAC;YACpD,IAAI,MAAM,KAAK,SAAS;gBAAE,WAAW,GAAG,MAAM,CAAC;YAC/C,MAAM,IAAI,GAAG,KAAK,CAAC,uBAAuB,CAAC;YAC3C,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC;gBAChE,SAAS,GAAG,IAAI,CAAC;YACnB,yEAAyE;YACzE,yEAAyE;YACzE,wEAAwE;YACxE,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,MAAM,KAAK,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,SAAS,CAAC;gBAC7C,IAAI,CAAC,WAAW,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC;oBACnD,MAAM,CAAC,GAAG,CAAC,6CAAC,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;YAChD,CAAC;YACD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,EAAE,CAAC;gBAClC,MAAM,KAAK,GACT,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;gBACtE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC;oBACpB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;gBAC9D,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBACzB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBACjC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YACrC,CAAC;QACH,CAAC,CAAC;IACJ,CAAC;IACD,MAAM,CAAC,GAAW,EAAE,KAAc;QAChC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,IAAI,KAAK,CAAC,MAAM,GAAG,GAAG;gBAAE,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YAC9C,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IACO,KAAK,CAAC,KAAkB;QAC9B,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAC;QACzC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI;YACzB;gBACE,6CAAC,CAAC,sBAAsB;gBACxB,CAAC,kBAAkB,EAAE,oBAAoB,EAAE,cAAc,CAAC;aAC3D;YACD;gBACE,6CAAC,CAAC,0BAA0B;gBAC5B,CAAC,sBAAsB,EAAE,qBAAqB,EAAE,eAAe,CAAC;aACjE;YACD,CAAC,6CAAC,CAAC,qBAAqB,EAAE,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC;YAC9D;gBACE,6CAAC,CAAC,yCAAyC;gBAC3C,CAAC,yBAAyB,EAAE,qBAAqB,CAAC;aACnD;YACD;gBACE,6CAAC,CAAC,4CAA4C;gBAC9C;oBACE,oBAAoB;oBACpB,sBAAsB;oBACtB,wBAAwB;iBACzB;aACF;SACO,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,KAAK;iBAChB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;iBAC1B,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5D,IAAI,OAAO,KAAK,KAAK,QAAQ;gBAAE,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACxD,CAAC;QACD,4EAA4E;QAC5E,0EAA0E;QAC1E,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI;YAC9B,sEAAsE;YACtE,oEAAoE;YACpE,CAAC,6CAAC,CAAC,sBAAsB,EAAE,KAAK,CAAC,mBAAmB,IAAI,KAAK,CAAC,kBAAkB,IAAI,KAAK,CAAC,wBAAwB,CAAC;YACnH,CAAC,6CAAC,CAAC,0BAA0B,EAAE,KAAK,CAAC,uBAAuB,IAAI,KAAK,CAAC,yBAAyB,CAAC;YAChG,CAAC,6CAAC,CAAC,yCAAyC,EAAE,KAAK,CAAC,kBAAkB,IAAI,KAAK,CAAC,yBAAyB,CAAC;SAClG,EAAE,CAAC;YACX,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;gBAClC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC7B,MAAM,QAAQ,GAAG,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC1F,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,MAAM,CAAC;gBACjD,IACE,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC;oBAClE,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,EACjE,CAAC;oBACD,MAAM,GAAG,GAAG,GAAG,MAAM,IAAI,QAAQ,EAAE,CAAC;oBACpC,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,8EAA8E;IAC9E,cAAc,CAAC,KAAc;QAC3B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC;YAC3E,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YAC5C,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IACD;;OAEG;IACK,eAAe,CAAC,UAAmB;QACzC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QAC5E,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,UAAU,CAAC,CAAC,UAAU,CAAC;YAC9D,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC;gBACnE,OAAO,SAAS,CAAC;YACnB,KAAK,IAAI,KAAK,CAAC;QACjB,CAAC;QACD,OAAO,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;IACpD,CAAC;IACD,QAAQ,CAAC,KAAc;QACrB,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QACjD,IAAI,CAAC,MAAM,CACT,6CAAC,CAAC,kBAAkB,EACpB,QAAQ,CAAC,UAAU,IAAI,WAAW,CAAC,EAAE,IAAI,QAAQ,CAAC,EAAE,CACrD,CAAC;QACF,IAAI,CAAC,MAAM,CACT,6CAAC,CAAC,qBAAqB,EACvB,QAAQ,CAAC,YAAY,IAAI,WAAW,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,CAC7D,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,qBAAqB,EAAE,WAAW,CAAC,MAAM,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC1E,IAAI,CAAC,MAAM,CAAC,2BAA2B,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QACxD,IAAI,OAAO,QAAQ,CAAC,IAAI,KAAK,SAAS;YACpC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,6BAA6B,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;QACvE,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YACvC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,EAAE,QAAQ,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YACtE,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YACrD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;gBACvB,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,sBAAsB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAClE,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,KAAK,GAAG,MAAM,CAClB,QAAQ,CAAC,aAAa;gBACpB,WAAW,CAAC,KAAK;gBACjB,QAAQ,CAAC,KAAK;gBACd,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,WAAW,CACxC,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACjC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnB,wEAAwE;gBACxE,yEAAyE;gBACzE,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,sBAAsB;oBACpD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC;wBAAE,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAC/C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,6CAAC,CAAC,sBAAsB,CAAC,EAAE,CAAC;oBAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;oBACxD,IAAI,KAAK,KAAK,SAAS;wBAAE,MAAM,CAAC,GAAG,CAAC,6CAAC,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;gBACvE,CAAC;gBACD,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,6CAAC,CAAC,sBAAsB,CAAC,CAAC;gBACnD,IAAI,KAAK,KAAK,SAAS,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,6CAAC,CAAC,qBAAqB,CAAC;oBAC7D,MAAM,CAAC,GAAG,CAAC,6CAAC,CAAC,qBAAqB,EAAE,KAAK,CAAC,CAAC;YAC/C,CAAC;YACD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM;gBAAE,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACxE,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC7C,MAAM,OAAO,GAAG,UAAU;aACvB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC;aAClC,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC;QACrD,IAAI,OAAO,CAAC,MAAM,GAAG,EAAE,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,GAAG,GAAG,CAAC;YACtE,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACxB,IAAI,OAAO,CAAC,MAAM;YAChB,IAAI,CAAC,IAAI,CAAC,YAAY,CACpB,6CAAC,CAAC,8BAA8B,EAChC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAC3D,CAAC;QACJ,IACE,CAAC,IAAI,CAAC,YAAY;YAClB,CAAC,UAAU;iBACR,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;iBACZ,IAAI,CACH,CAAC,SAAS,EAAE,EAAE,CACZ,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,CAC3D;gBACD,QAAQ,CAAC,UAAU,KAAK,eAAe;gBACvC,QAAQ,CAAC,UAAU,KAAK,YAAY;gBACpC,IAAI,CACF,QAAQ,CAAC,KAAK;oBACZ,WAAW,CAAC,KAAK;oBACjB,QAAQ,CAAC,OAAO;oBAChB,WAAW,CAAC,OAAO,CACtB,CAAC,MAAM,GAAG,CAAC,CAAC,EACf,CAAC;YACD,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,YAAY,CACpB,qCAAqC,EACrC,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,CAC1C,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC;YAAE,OAAO;QACrD,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC;YACrD,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC;YAClD,IAAI,CAAC,MAAM,CAAC;gBACV,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,OAAO;gBACxC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;aACnC,CAAC,CAAC;QACL,CAAC;QACD,4EAA4E;QAC5E,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,IAAI,WAAW,CAAC,KAAK,CAAC;QAClD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;YACjB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC;gBAC3C,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,cAAc;oBACtC,IAAI,CAAC,MAAM,CAAC;wBACV,IAAI,EAAE,OAAO;wBACb,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC;qBAC1C,CAAC,CAAC;YACP,CAAC;QACH,CAAC;aAAM,IAAI,QAAQ,CAAC,OAAO,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;YACnD,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;YACjB,IAAI,CAAC,MAAM,CAAC;gBACV,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,IAAI,WAAW,CAAC,OAAO,CAAC;aAC7D,CAAC,CAAC;QACL,CAAC;aAAM,IACL,QAAQ,CAAC,UAAU,KAAK,eAAe;YACvC,QAAQ,CAAC,UAAU,KAAK,YAAY,EACpC,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACtE,CAAC;aAAM,IACL,QAAQ,CAAC,UAAU,KAAK,YAAY;YACpC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,cAAc,EAC7C,CAAC;YACD,IAAI,CAAC,MAAM,CAAC;gBACV,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC;aACnD,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,OAAgB;QAC7B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM;YAAE,OAAO;QAClD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC5B,CAAC;IACD,MAAM;QACJ,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM;YACpB,IAAI,CAAC,IAAI,CAAC,YAAY,CACpB,6CAAC,CAAC,mBAAmB,EACrB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAC5B,CAAC;QACJ,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC;YACvC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,gCAAgC,EAAE,IAAI,CAAC,CAAC;QACjE,IAAI,IAAI,CAAC,UAAU,IAAI,CAAC,IAAI,CAAC,IAAI;YAC/B,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,4BAA4B,EAAE,aAAa,CAAC,CAAC;aACjE,IACH,IAAI,CAAC,UAAU;YACf,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI;gBACnB,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;YAElE,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,4BAA4B,EAAE,SAAS,CAAC,CAAC;IACpE,CAAC;CACF;AA5aD,0BA4aC"}

package/dist/index.d.ts

@@ -0,0 +1,24 @@
+import { type TracerProvider } from "@opentelemetry/api";
+import { type TraceConfigOptions } from "@traceai/fi-core";
+export { VERSION } from "./version.js";
+export interface GoogleGenAITracingOptions {
+    tracerProvider?: TracerProvider;
+    traceConfig?: TraceConfigOptions;
+    /** Total captured content budget per call, in characters. Default 16,384; maximum 65,536. */
+    maxContentLength?: number;
+}
+export interface GoogleGenAITracing {
+    /** Stops tracing new calls. In-flight calls finish normally. */
+    disable(): void;
+    enable(): void;
+    /** Stops new tracing; restores methods after in-flight calls drain. Does not shut down the provider. */
+    dispose(): void;
+}
+/** Instruments one modern @google/genai client in place, including existing chats.
+ * Duplicate setup returns the existing handle (its original options win).
+ * No global module patch, private chat-field rewrite, exporter, or hidden API call is installed.
+ */
+export declare function instrumentGoogleGenAI(client: {
+    readonly models: object;
+}, options?: GoogleGenAITracingOptions): GoogleGenAITracing;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAQL,KAAK,cAAc,EACpB,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EAIL,KAAK,kBAAkB,EACxB,MAAM,kBAAkB,CAAC;AAK1B,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC,MAAM,WAAW,yBAAyB;IACxC,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,WAAW,CAAC,EAAE,kBAAkB,CAAC;IACjC,6FAA6F;IAC7F,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AACD,MAAM,WAAW,kBAAkB;IACjC,gEAAgE;IAChE,OAAO,IAAI,IAAI,CAAC;IAChB,MAAM,IAAI,IAAI,CAAC;IACf,wGAAwG;IACxG,OAAO,IAAI,IAAI,CAAC;CACjB;AA2ND;;;GAGG;AACH,wBAAgB,qBAAqB,CACnC,MAAM,EAAE;IAAE,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;CAAE,EACnC,OAAO,GAAE,yBAA8B,GACtC,kBAAkB,CAsVpB"}