@amplitude/ai 0.1.0

package/dist/integrations/llamaindex.js

@@ -0,0 +1,144 @@
+import { getActiveContext } from "../context.js";
+import { calculateCost } from "../utils/costs.js";
+
+//#region src/integrations/llamaindex.ts
+var AmplitudeLlamaIndexHandler = class {
+	_ai;
+	_userId;
+	_sessionId;
+	_agentId;
+	_env;
+	_startTimes = /* @__PURE__ */ new Map();
+	constructor(options) {
+		this._ai = options.amplitudeAI;
+		this._userId = options.userId ?? null;
+		this._sessionId = options.sessionId ?? null;
+		this._agentId = options.agentId ?? null;
+		this._env = options.env ?? null;
+	}
+	_getContext() {
+		const ctx = getActiveContext();
+		return {
+			userId: this._userId ?? ctx?.userId ?? "unknown",
+			sessionId: this._sessionId ?? ctx?.sessionId ?? void 0,
+			agentId: this._agentId ?? ctx?.agentId ?? void 0,
+			env: this._env ?? ctx?.env ?? void 0,
+			traceId: ctx?.traceId ?? void 0
+		};
+	}
+	onLLMStart(eventId) {
+		this._startTimes.set(eventId, performance.now());
+	}
+	onLLMEnd(eventId, response) {
+		const startTime = this._startTimes.get(eventId) ?? performance.now();
+		this._startTimes.delete(eventId);
+		const latencyMs = performance.now() - startTime;
+		const ctx = this._getContext();
+		const normalized = _normalizeLlamaLlmResponse(response);
+		let costUsd;
+		if (normalized.model !== "unknown" && normalized.inputTokens != null && normalized.outputTokens != null) try {
+			const cost = calculateCost({
+				modelName: normalized.model,
+				inputTokens: normalized.inputTokens,
+				outputTokens: normalized.outputTokens
+			});
+			if (cost > 0) costUsd = cost;
+		} catch {}
+		this._ai.trackAiMessage({
+			userId: ctx.userId,
+			content: normalized.content,
+			sessionId: ctx.sessionId ?? "llamaindex-session",
+			model: normalized.model,
+			provider: "llamaindex",
+			latencyMs,
+			traceId: ctx.traceId,
+			agentId: ctx.agentId,
+			env: ctx.env,
+			inputTokens: normalized.inputTokens,
+			outputTokens: normalized.outputTokens,
+			totalCostUsd: costUsd
+		});
+	}
+	onEmbeddingStart(eventId) {
+		this._startTimes.set(eventId, performance.now());
+	}
+	onEmbeddingEnd(eventId, response) {
+		const startTime = this._startTimes.get(eventId) ?? performance.now();
+		this._startTimes.delete(eventId);
+		const latencyMs = performance.now() - startTime;
+		const ctx = this._getContext();
+		const normalized = _normalizeLlamaEmbeddingResponse(response);
+		this._ai.trackEmbedding({
+			userId: ctx.userId,
+			model: normalized.model,
+			provider: "llamaindex",
+			latencyMs,
+			sessionId: ctx.sessionId,
+			traceId: ctx.traceId,
+			agentId: ctx.agentId,
+			env: ctx.env,
+			inputTokens: normalized.inputTokens,
+			dimensions: normalized.dimensions
+		});
+	}
+	onToolStart(eventId) {
+		this._startTimes.set(eventId, performance.now());
+	}
+	onToolEnd(eventId, response) {
+		const startTime = this._startTimes.get(eventId) ?? performance.now();
+		this._startTimes.delete(eventId);
+		const latencyMs = performance.now() - startTime;
+		const ctx = this._getContext();
+		const normalized = _normalizeLlamaToolResponse(response);
+		this._ai.trackToolCall({
+			userId: ctx.userId,
+			toolName: normalized.toolName,
+			latencyMs,
+			success: normalized.success,
+			sessionId: ctx.sessionId,
+			traceId: ctx.traceId,
+			agentId: ctx.agentId,
+			env: ctx.env,
+			output: normalized.output
+		});
+	}
+};
+function createAmplitudeLlamaIndexHandler(options) {
+	return new AmplitudeLlamaIndexHandler(options);
+}
+function _normalizeLlamaLlmResponse(response) {
+	const resp = response != null && typeof response === "object" ? response : {};
+	const message = resp.message;
+	const content = (typeof resp.content === "string" ? resp.content : void 0) ?? (typeof message?.content === "string" ? message.content : "") ?? "";
+	const usage = resp.usage ?? {};
+	return {
+		content,
+		model: String(resp.model ?? message?.model ?? "unknown"),
+		inputTokens: _toNumber(resp.inputTokens ?? usage.input_tokens ?? usage.prompt_tokens),
+		outputTokens: _toNumber(resp.outputTokens ?? usage.output_tokens ?? usage.completion_tokens)
+	};
+}
+function _normalizeLlamaEmbeddingResponse(response) {
+	const resp = response != null && typeof response === "object" ? response : {};
+	const usage = resp.usage ?? {};
+	return {
+		model: String(resp.model ?? "unknown"),
+		inputTokens: _toNumber(resp.inputTokens ?? usage.input_tokens),
+		dimensions: _toNumber(resp.dimensions ?? resp.vectorSize ?? resp.embedding_dimensions)
+	};
+}
+function _normalizeLlamaToolResponse(response) {
+	const resp = response != null && typeof response === "object" ? response : {};
+	return {
+		toolName: String(resp.toolName ?? resp.name ?? "llamaindex-tool"),
+		success: Boolean(resp.success ?? resp.error == null),
+		output: resp.output ?? resp.result
+	};
+}
+function _toNumber(value) {
+	return typeof value === "number" ? value : void 0;
+}
+
+//#endregion
+export { AmplitudeLlamaIndexHandler, createAmplitudeLlamaIndexHandler };
+//# sourceMappingURL=llamaindex.js.map

package/dist/integrations/llamaindex.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamaindex.js","names":["costUsd: number | undefined"],"sources":["../../src/integrations/llamaindex.ts"],"sourcesContent":["/**\n * LlamaIndex integration — AmplitudeLlamaIndexHandler.\n *\n * Tracks LLM calls, function calls, and embedding events\n * from LlamaIndex via its callback handler system.\n */\n\nimport type { AmplitudeAI } from '../client.js';\nimport { getActiveContext } from '../context.js';\nimport { calculateCost } from '../utils/costs.js';\n\nexport interface LlamaIndexHandlerOptions {\n  amplitudeAI: AmplitudeAI;\n  userId?: string;\n  sessionId?: string;\n  agentId?: string;\n  env?: string;\n}\n\nexport class AmplitudeLlamaIndexHandler {\n  private _ai: AmplitudeAI;\n  private _userId: string | null;\n  private _sessionId: string | null;\n  private _agentId: string | null;\n  private _env: string | null;\n  private _startTimes: Map<string, number> = new Map();\n\n  constructor(options: LlamaIndexHandlerOptions) {\n    this._ai = options.amplitudeAI;\n    this._userId = options.userId ?? null;\n    this._sessionId = options.sessionId ?? null;\n    this._agentId = options.agentId ?? null;\n    this._env = options.env ?? null;\n  }\n\n  private _getContext() {\n    const ctx = getActiveContext();\n    return {\n      userId: this._userId ?? ctx?.userId ?? 'unknown',\n      sessionId: this._sessionId ?? ctx?.sessionId ?? undefined,\n      agentId: this._agentId ?? ctx?.agentId ?? undefined,\n      env: this._env ?? ctx?.env ?? undefined,\n      traceId: ctx?.traceId ?? undefined,\n    };\n  }\n\n  onLLMStart(eventId: string): void {\n    this._startTimes.set(eventId, performance.now());\n  }\n\n  onLLMEnd(\n    eventId: string,\n    response: {\n      content?: string;\n      model?: string;\n      inputTokens?: number;\n      outputTokens?: number;\n    },\n  ): void {\n    const startTime = this._startTimes.get(eventId) ?? performance.now();\n    this._startTimes.delete(eventId);\n    const latencyMs = performance.now() - startTime;\n\n    const ctx = this._getContext();\n    const normalized = _normalizeLlamaLlmResponse(response as unknown);\n\n    let costUsd: number | undefined;\n    if (\n      normalized.model !== 'unknown' &&\n      normalized.inputTokens != null &&\n      normalized.outputTokens != null\n    ) {\n      try {\n        const cost = calculateCost({\n          modelName: normalized.model,\n          inputTokens: normalized.inputTokens,\n          outputTokens: normalized.outputTokens,\n        });\n        if (cost > 0) costUsd = cost;\n      } catch {\n        // cost calculation is best-effort\n      }\n    }\n\n    this._ai.trackAiMessage({\n      userId: ctx.userId,\n      content: normalized.content,\n      sessionId: ctx.sessionId ?? 'llamaindex-session',\n      model: normalized.model,\n      provider: 'llamaindex',\n      latencyMs,\n      traceId: ctx.traceId,\n      agentId: ctx.agentId,\n      env: ctx.env,\n      inputTokens: normalized.inputTokens,\n      outputTokens: normalized.outputTokens,\n      totalCostUsd: costUsd,\n    });\n  }\n\n  onEmbeddingStart(eventId: string): void {\n    this._startTimes.set(eventId, performance.now());\n  }\n\n  onEmbeddingEnd(\n    eventId: string,\n    response: { model?: string; inputTokens?: number; dimensions?: number },\n  ): void {\n    const startTime = this._startTimes.get(eventId) ?? performance.now();\n    this._startTimes.delete(eventId);\n    const latencyMs = performance.now() - startTime;\n\n    const ctx = this._getContext();\n    const normalized = _normalizeLlamaEmbeddingResponse(response as unknown);\n    this._ai.trackEmbedding({\n      userId: ctx.userId,\n      model: normalized.model,\n      provider: 'llamaindex',\n      latencyMs,\n      sessionId: ctx.sessionId,\n      traceId: ctx.traceId,\n      agentId: ctx.agentId,\n      env: ctx.env,\n      inputTokens: normalized.inputTokens,\n      dimensions: normalized.dimensions,\n    });\n  }\n\n  onToolStart(eventId: string): void {\n    this._startTimes.set(eventId, performance.now());\n  }\n\n  onToolEnd(\n    eventId: string,\n    response: { toolName?: string; output?: unknown; success?: boolean },\n  ): void {\n    const startTime = this._startTimes.get(eventId) ?? performance.now();\n    this._startTimes.delete(eventId);\n    const latencyMs = performance.now() - startTime;\n\n    const ctx = this._getContext();\n    const normalized = _normalizeLlamaToolResponse(response as unknown);\n    this._ai.trackToolCall({\n      userId: ctx.userId,\n      toolName: normalized.toolName,\n      latencyMs,\n      success: normalized.success,\n      sessionId: ctx.sessionId,\n      traceId: ctx.traceId,\n      agentId: ctx.agentId,\n      env: ctx.env,\n      output: normalized.output,\n    });\n  }\n}\n\nexport function createAmplitudeLlamaIndexHandler(\n  options: LlamaIndexHandlerOptions,\n): AmplitudeLlamaIndexHandler {\n  return new AmplitudeLlamaIndexHandler(options);\n}\n\nfunction _normalizeLlamaLlmResponse(response: unknown): {\n  content: string;\n  model: string;\n  inputTokens?: number;\n  outputTokens?: number;\n} {\n  const resp =\n    response != null && typeof response === 'object'\n      ? (response as Record<string, unknown>)\n      : {};\n  const message = resp.message as Record<string, unknown> | undefined;\n  const content =\n    (typeof resp.content === 'string' ? resp.content : undefined) ??\n    (typeof message?.content === 'string' ? (message.content as string) : '') ??\n    '';\n  const usage = (resp.usage as Record<string, unknown> | undefined) ?? {};\n  return {\n    content,\n    model: String(resp.model ?? message?.model ?? 'unknown'),\n    inputTokens: _toNumber(\n      resp.inputTokens ?? usage.input_tokens ?? usage.prompt_tokens,\n    ),\n    outputTokens: _toNumber(\n      resp.outputTokens ?? usage.output_tokens ?? usage.completion_tokens,\n    ),\n  };\n}\n\nfunction _normalizeLlamaEmbeddingResponse(response: unknown): {\n  model: string;\n  inputTokens?: number;\n  dimensions?: number;\n} {\n  const resp =\n    response != null && typeof response === 'object'\n      ? (response as Record<string, unknown>)\n      : {};\n  const usage = (resp.usage as Record<string, unknown> | undefined) ?? {};\n  return {\n    model: String(resp.model ?? 'unknown'),\n    inputTokens: _toNumber(resp.inputTokens ?? usage.input_tokens),\n    dimensions: _toNumber(\n      resp.dimensions ?? resp.vectorSize ?? resp.embedding_dimensions,\n    ),\n  };\n}\n\nfunction _normalizeLlamaToolResponse(response: unknown): {\n  toolName: string;\n  success: boolean;\n  output?: unknown;\n} {\n  const resp =\n    response != null && typeof response === 'object'\n      ? (response as Record<string, unknown>)\n      : {};\n  return {\n    toolName: String(resp.toolName ?? resp.name ?? 'llamaindex-tool'),\n    success: Boolean(resp.success ?? resp.error == null),\n    output: resp.output ?? resp.result,\n  };\n}\n\nfunction _toNumber(value: unknown): number | undefined {\n  return typeof value === 'number' ? value : undefined;\n}\n"],"mappings":";;;;AAmBA,IAAa,6BAAb,MAAwC;CACtC,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ,8BAAmC,IAAI,KAAK;CAEpD,YAAY,SAAmC;AAC7C,OAAK,MAAM,QAAQ;AACnB,OAAK,UAAU,QAAQ,UAAU;AACjC,OAAK,aAAa,QAAQ,aAAa;AACvC,OAAK,WAAW,QAAQ,WAAW;AACnC,OAAK,OAAO,QAAQ,OAAO;;CAG7B,AAAQ,cAAc;EACpB,MAAM,MAAM,kBAAkB;AAC9B,SAAO;GACL,QAAQ,KAAK,WAAW,KAAK,UAAU;GACvC,WAAW,KAAK,cAAc,KAAK,aAAa;GAChD,SAAS,KAAK,YAAY,KAAK,WAAW;GAC1C,KAAK,KAAK,QAAQ,KAAK,OAAO;GAC9B,SAAS,KAAK,WAAW;GAC1B;;CAGH,WAAW,SAAuB;AAChC,OAAK,YAAY,IAAI,SAAS,YAAY,KAAK,CAAC;;CAGlD,SACE,SACA,UAMM;EACN,MAAM,YAAY,KAAK,YAAY,IAAI,QAAQ,IAAI,YAAY,KAAK;AACpE,OAAK,YAAY,OAAO,QAAQ;EAChC,MAAM,YAAY,YAAY,KAAK,GAAG;EAEtC,MAAM,MAAM,KAAK,aAAa;EAC9B,MAAM,aAAa,2BAA2B,SAAoB;EAElE,IAAIA;AACJ,MACE,WAAW,UAAU,aACrB,WAAW,eAAe,QAC1B,WAAW,gBAAgB,KAE3B,KAAI;GACF,MAAM,OAAO,cAAc;IACzB,WAAW,WAAW;IACtB,aAAa,WAAW;IACxB,cAAc,WAAW;IAC1B,CAAC;AACF,OAAI,OAAO,EAAG,WAAU;UAClB;AAKV,OAAK,IAAI,eAAe;GACtB,QAAQ,IAAI;GACZ,SAAS,WAAW;GACpB,WAAW,IAAI,aAAa;GAC5B,OAAO,WAAW;GAClB,UAAU;GACV;GACA,SAAS,IAAI;GACb,SAAS,IAAI;GACb,KAAK,IAAI;GACT,aAAa,WAAW;GACxB,cAAc,WAAW;GACzB,cAAc;GACf,CAAC;;CAGJ,iBAAiB,SAAuB;AACtC,OAAK,YAAY,IAAI,SAAS,YAAY,KAAK,CAAC;;CAGlD,eACE,SACA,UACM;EACN,MAAM,YAAY,KAAK,YAAY,IAAI,QAAQ,IAAI,YAAY,KAAK;AACpE,OAAK,YAAY,OAAO,QAAQ;EAChC,MAAM,YAAY,YAAY,KAAK,GAAG;EAEtC,MAAM,MAAM,KAAK,aAAa;EAC9B,MAAM,aAAa,iCAAiC,SAAoB;AACxE,OAAK,IAAI,eAAe;GACtB,QAAQ,IAAI;GACZ,OAAO,WAAW;GAClB,UAAU;GACV;GACA,WAAW,IAAI;GACf,SAAS,IAAI;GACb,SAAS,IAAI;GACb,KAAK,IAAI;GACT,aAAa,WAAW;GACxB,YAAY,WAAW;GACxB,CAAC;;CAGJ,YAAY,SAAuB;AACjC,OAAK,YAAY,IAAI,SAAS,YAAY,KAAK,CAAC;;CAGlD,UACE,SACA,UACM;EACN,MAAM,YAAY,KAAK,YAAY,IAAI,QAAQ,IAAI,YAAY,KAAK;AACpE,OAAK,YAAY,OAAO,QAAQ;EAChC,MAAM,YAAY,YAAY,KAAK,GAAG;EAEtC,MAAM,MAAM,KAAK,aAAa;EAC9B,MAAM,aAAa,4BAA4B,SAAoB;AACnE,OAAK,IAAI,cAAc;GACrB,QAAQ,IAAI;GACZ,UAAU,WAAW;GACrB;GACA,SAAS,WAAW;GACpB,WAAW,IAAI;GACf,SAAS,IAAI;GACb,SAAS,IAAI;GACb,KAAK,IAAI;GACT,QAAQ,WAAW;GACpB,CAAC;;;AAIN,SAAgB,iCACd,SAC4B;AAC5B,QAAO,IAAI,2BAA2B,QAAQ;;AAGhD,SAAS,2BAA2B,UAKlC;CACA,MAAM,OACJ,YAAY,QAAQ,OAAO,aAAa,WACnC,WACD,EAAE;CACR,MAAM,UAAU,KAAK;CACrB,MAAM,WACH,OAAO,KAAK,YAAY,WAAW,KAAK,UAAU,YAClD,OAAO,SAAS,YAAY,WAAY,QAAQ,UAAqB,OACtE;CACF,MAAM,QAAS,KAAK,SAAiD,EAAE;AACvE,QAAO;EACL;EACA,OAAO,OAAO,KAAK,SAAS,SAAS,SAAS,UAAU;EACxD,aAAa,UACX,KAAK,eAAe,MAAM,gBAAgB,MAAM,cACjD;EACD,cAAc,UACZ,KAAK,gBAAgB,MAAM,iBAAiB,MAAM,kBACnD;EACF;;AAGH,SAAS,iCAAiC,UAIxC;CACA,MAAM,OACJ,YAAY,QAAQ,OAAO,aAAa,WACnC,WACD,EAAE;CACR,MAAM,QAAS,KAAK,SAAiD,EAAE;AACvE,QAAO;EACL,OAAO,OAAO,KAAK,SAAS,UAAU;EACtC,aAAa,UAAU,KAAK,eAAe,MAAM,aAAa;EAC9D,YAAY,UACV,KAAK,cAAc,KAAK,cAAc,KAAK,qBAC5C;EACF;;AAGH,SAAS,4BAA4B,UAInC;CACA,MAAM,OACJ,YAAY,QAAQ,OAAO,aAAa,WACnC,WACD,EAAE;AACR,QAAO;EACL,UAAU,OAAO,KAAK,YAAY,KAAK,QAAQ,kBAAkB;EACjE,SAAS,QAAQ,KAAK,WAAW,KAAK,SAAS,KAAK;EACpD,QAAQ,KAAK,UAAU,KAAK;EAC7B;;AAGH,SAAS,UAAU,OAAoC;AACrD,QAAO,OAAO,UAAU,WAAW,QAAQ"}

package/dist/integrations/openai-agents.d.ts

@@ -0,0 +1,35 @@
+import { AmplitudeAI } from "../client.js";
+
+//#region src/integrations/openai-agents.d.ts
+interface TracingProcessorOptions {
+  amplitudeAI: AmplitudeAI;
+  userId?: string;
+  sessionId?: string;
+  traceId?: string;
+  agentId?: string;
+  env?: string;
+}
+declare class AmplitudeTracingProcessor {
+  private _ai;
+  private _defaults;
+  private _turnId;
+  constructor(options: TracingProcessorOptions);
+  onSpanStart(_span: Record<string, unknown>): void;
+  onTraceStart(_trace: Record<string, unknown>): void;
+  onTraceEnd(_trace: Record<string, unknown>): void;
+  onSpanEnd(span: Record<string, unknown>): void;
+  shutdown(): void;
+  private _handleGeneration;
+  private _handleFunction;
+  private _inferKind;
+  private _getSpanData;
+  private _getTraceId;
+  private _getLatencyMs;
+  private _safeNumber;
+  private _normalizeMessagesArray;
+  private _extractAssistantText;
+  private _extractToolCalls;
+}
+//#endregion
+export { AmplitudeTracingProcessor, TracingProcessorOptions };
+//# sourceMappingURL=openai-agents.d.ts.map

package/dist/integrations/openai-agents.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-agents.d.ts","names":[],"sources":["../../src/integrations/openai-agents.ts"],"sourcesContent":[],"mappings":";;;UAKiB,uBAAA;eACF;EADE,MAAA,CAAA,EAAA,MAAA;EASJ,SAAA,CAAA,EAAA,MAAA;EAWU,OAAA,CAAA,EAAA,MAAA;EAYF,OAAA,CAAA,EAAA,MAAA;EAIE,GAAA,CAAA,EAAA,MAAA;;AAQL,cAnCL,yBAAA,CAmCK;EAAM,QAAA,GAAA;;;uBAxBD;qBAYF;uBAIE;qBAIF;kBAIH"}

package/dist/integrations/openai-agents.js

@@ -0,0 +1,248 @@
+import { getActiveContext } from "../context.js";
+import { calculateCost, inferProvider } from "../utils/costs.js";
+import { randomUUID } from "node:crypto";
+
+//#region src/integrations/openai-agents.ts
+var AmplitudeTracingProcessor = class {
+	_ai;
+	_defaults;
+	_turnId = 1;
+	constructor(options) {
+		const ctx = getActiveContext();
+		this._ai = options.amplitudeAI;
+		this._defaults = {
+			userId: options.userId ?? ctx?.userId ?? "openai-agents-user",
+			sessionId: options.sessionId ?? ctx?.sessionId ?? randomUUID(),
+			traceId: options.traceId ?? ctx?.traceId ?? randomUUID(),
+			agentId: options.agentId ?? ctx?.agentId ?? null,
+			env: options.env ?? ctx?.env ?? null
+		};
+	}
+	onSpanStart(_span) {}
+	onTraceStart(_trace) {}
+	onTraceEnd(_trace) {}
+	onSpanEnd(span) {
+		const spanData = this._getSpanData(span);
+		if (spanData == null) return;
+		const traceId = this._getTraceId(span);
+		const latencyMs = this._getLatencyMs(span);
+		const kind = this._inferKind(spanData);
+		if (kind === "generation") {
+			this._handleGeneration(spanData, traceId, latencyMs);
+			return;
+		}
+		if (kind === "function") {
+			this._handleFunction(spanData, traceId, latencyMs);
+			return;
+		}
+		if (kind === "handoff") {
+			const fromAgent = String(spanData.from_agent ?? "unknown");
+			const toAgent = String(spanData.to_agent ?? "unknown");
+			this._ai.trackSpan({
+				userId: this._defaults.userId,
+				spanName: `handoff:${fromAgent}->${toAgent}`,
+				traceId,
+				latencyMs,
+				sessionId: this._defaults.sessionId,
+				agentId: this._defaults.agentId,
+				env: this._defaults.env,
+				inputState: { from_agent: fromAgent },
+				outputState: { to_agent: toAgent }
+			});
+			return;
+		}
+		if (kind === "guardrail") {
+			const guardrail = String(spanData.name ?? "guardrail");
+			const triggered = Boolean(spanData.triggered);
+			this._ai.trackSpan({
+				userId: this._defaults.userId,
+				spanName: `guardrail:${guardrail}`,
+				traceId,
+				latencyMs,
+				sessionId: this._defaults.sessionId,
+				agentId: this._defaults.agentId,
+				env: this._defaults.env,
+				outputState: { triggered },
+				isError: triggered
+			});
+			return;
+		}
+		const name = String(spanData.name ?? "agent");
+		this._ai.trackSpan({
+			userId: this._defaults.userId,
+			spanName: `agent:${name}`,
+			traceId,
+			latencyMs,
+			sessionId: this._defaults.sessionId,
+			agentId: this._defaults.agentId ?? name,
+			env: this._defaults.env,
+			outputState: spanData.output != null ? { output: String(spanData.output) } : void 0
+		});
+	}
+	shutdown() {
+		this._ai.flush();
+	}
+	_handleGeneration(data, traceId, latencyMs) {
+		const input = this._normalizeMessagesArray(data.input);
+		for (const message of input) {
+			const role = message?.role;
+			const content = message?.content;
+			if (role !== "user" || typeof content !== "string" || content.length === 0) continue;
+			this._ai.trackUserMessage({
+				userId: this._defaults.userId,
+				content,
+				sessionId: this._defaults.sessionId,
+				traceId,
+				turnId: this._turnId,
+				agentId: this._defaults.agentId,
+				env: this._defaults.env
+			});
+			this._turnId += 1;
+		}
+		const output = this._normalizeMessagesArray(data.output);
+		const responseText = this._extractAssistantText(output);
+		const toolCalls = this._extractToolCalls(output);
+		const model = String(data.model ?? "unknown");
+		const usage = data.usage ?? {};
+		const inputTokens = this._safeNumber(usage.input_tokens ?? usage.prompt_tokens ?? null);
+		const outputTokens = this._safeNumber(usage.output_tokens ?? usage.completion_tokens ?? null);
+		const totalTokens = this._safeNumber(usage.total_tokens ?? null);
+		let costUsd;
+		if (model !== "unknown" && inputTokens != null && outputTokens != null) try {
+			const cost = calculateCost({
+				modelName: model,
+				inputTokens,
+				outputTokens
+			});
+			if (cost > 0) costUsd = cost;
+		} catch {}
+		this._ai.trackAiMessage({
+			userId: this._defaults.userId,
+			content: responseText,
+			sessionId: this._defaults.sessionId,
+			traceId,
+			turnId: this._turnId,
+			model,
+			provider: inferProvider(model),
+			latencyMs,
+			inputTokens,
+			outputTokens,
+			totalTokens,
+			totalCostUsd: costUsd,
+			toolCalls: toolCalls.length > 0 ? toolCalls : void 0,
+			agentId: this._defaults.agentId,
+			env: this._defaults.env
+		});
+		this._turnId += 1;
+	}
+	_handleFunction(data, traceId, latencyMs) {
+		const toolName = String(data.name ?? "unknown");
+		const errorMessage = data.error == null ? void 0 : String(data.error ?? "unknown error");
+		const toolInput = data.input != null && typeof data.input === "object" ? data.input : data.input != null ? { raw: String(data.input) } : void 0;
+		const toolOutput = data.output == null ? void 0 : String(data.output ?? void 0);
+		this._ai.trackToolCall({
+			userId: this._defaults.userId,
+			toolName,
+			success: errorMessage == null,
+			latencyMs,
+			sessionId: this._defaults.sessionId,
+			traceId,
+			turnId: this._turnId,
+			input: toolInput,
+			output: toolOutput,
+			errorMessage,
+			agentId: this._defaults.agentId,
+			env: this._defaults.env
+		});
+	}
+	_inferKind(data) {
+		if ("from_agent" in data || "to_agent" in data) return "handoff";
+		if ("triggered" in data) return "guardrail";
+		if ("model" in data && ("input" in data || "output" in data)) return "generation";
+		if ("name" in data && ("input" in data || "output" in data || "error" in data)) return "function";
+		return "agent";
+	}
+	_getSpanData(span) {
+		const data = span.span_data ?? span.data ?? span;
+		if (data == null || typeof data !== "object") return null;
+		return data;
+	}
+	_getTraceId(span) {
+		const traceId = span.trace_id ?? span.traceId ?? this._defaults.traceId;
+		return traceId == null ? randomUUID() : String(traceId);
+	}
+	_getLatencyMs(span) {
+		if (typeof span.latency_ms === "number") return span.latency_ms;
+		if (typeof span.latencyMs === "number") return span.latencyMs;
+		const start = span.start_time_ms ?? span.startTimeMs;
+		const end = span.end_time_ms ?? span.endTimeMs;
+		if (typeof start === "number" && typeof end === "number" && end >= start) return end - start;
+		return 0;
+	}
+	_safeNumber(value) {
+		if (typeof value === "number") return value;
+		if (typeof value === "string") {
+			const parsed = Number(value);
+			return Number.isNaN(parsed) ? void 0 : parsed;
+		}
+	}
+	_normalizeMessagesArray(value) {
+		if (Array.isArray(value)) return value;
+		if (value == null) return [];
+		return [value];
+	}
+	_extractAssistantText(output) {
+		let responseText = "";
+		for (const block of output) {
+			if (block == null || typeof block !== "object") continue;
+			const item = block;
+			if (!(item.role === "assistant" || item.type === "message" || item.type === "output_text")) continue;
+			if (typeof item.content === "string") {
+				responseText += item.content;
+				continue;
+			}
+			if (Array.isArray(item.content)) for (const part of item.content) {
+				const p = part;
+				if (p.type === "text" && typeof p.text === "string") responseText += p.text;
+				if (typeof p.value === "string") responseText += p.value;
+			}
+			if (item.type === "output_text" && typeof item.text === "string") responseText += item.text;
+		}
+		return responseText;
+	}
+	_extractToolCalls(output) {
+		const result = [];
+		for (const block of output) {
+			if (block == null || typeof block !== "object") continue;
+			const item = block;
+			if (item.type === "function_call") {
+				result.push({
+					type: "function",
+					id: String(item.id ?? ""),
+					function: {
+						name: String(item.name ?? ""),
+						arguments: String(item.arguments ?? "")
+					}
+				});
+				continue;
+			}
+			if (!Array.isArray(item.tool_calls)) continue;
+			for (const rawCall of item.tool_calls) {
+				const call = rawCall;
+				result.push({
+					type: "function",
+					id: String(call.id ?? ""),
+					function: {
+						name: String(call.function?.name ?? ""),
+						arguments: String(call.function?.arguments ?? "")
+					}
+				});
+			}
+		}
+		return result;
+	}
+};
+
+//#endregion
+export { AmplitudeTracingProcessor };
+//# sourceMappingURL=openai-agents.js.map

package/dist/integrations/openai-agents.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-agents.js","names":["costUsd: number | undefined","result: Array<Record<string, unknown>>"],"sources":["../../src/integrations/openai-agents.ts"],"sourcesContent":["import { randomUUID } from 'node:crypto';\nimport type { AmplitudeAI } from '../client.js';\nimport { getActiveContext } from '../context.js';\nimport { calculateCost, inferProvider } from '../utils/costs.js';\n\nexport interface TracingProcessorOptions {\n  amplitudeAI: AmplitudeAI;\n  userId?: string;\n  sessionId?: string;\n  traceId?: string;\n  agentId?: string;\n  env?: string;\n}\n\nexport class AmplitudeTracingProcessor {\n  private _ai: AmplitudeAI;\n  private _defaults: {\n    userId: string;\n    sessionId: string;\n    traceId: string;\n    agentId: string | null;\n    env: string | null;\n  };\n  private _turnId = 1;\n\n  constructor(options: TracingProcessorOptions) {\n    const ctx = getActiveContext();\n    this._ai = options.amplitudeAI;\n    this._defaults = {\n      userId: options.userId ?? ctx?.userId ?? 'openai-agents-user',\n      sessionId: options.sessionId ?? ctx?.sessionId ?? randomUUID(),\n      traceId: options.traceId ?? ctx?.traceId ?? randomUUID(),\n      agentId: options.agentId ?? ctx?.agentId ?? null,\n      env: options.env ?? ctx?.env ?? null,\n    };\n  }\n\n  onSpanStart(_span: Record<string, unknown>): void {\n    // compatibility hook with tracing processors\n  }\n\n  onTraceStart(_trace: Record<string, unknown>): void {\n    // compatibility hook with tracing processors\n  }\n\n  onTraceEnd(_trace: Record<string, unknown>): void {\n    // compatibility hook with tracing processors\n  }\n\n  onSpanEnd(span: Record<string, unknown>): void {\n    const spanData = this._getSpanData(span);\n    if (spanData == null) return;\n\n    const traceId = this._getTraceId(span);\n    const latencyMs = this._getLatencyMs(span);\n    const kind = this._inferKind(spanData);\n    if (kind === 'generation') {\n      this._handleGeneration(spanData, traceId, latencyMs);\n      return;\n    }\n    if (kind === 'function') {\n      this._handleFunction(spanData, traceId, latencyMs);\n      return;\n    }\n    if (kind === 'handoff') {\n      const fromAgent = String(spanData.from_agent ?? 'unknown');\n      const toAgent = String(spanData.to_agent ?? 'unknown');\n      this._ai.trackSpan({\n        userId: this._defaults.userId,\n        spanName: `handoff:${fromAgent}->${toAgent}`,\n        traceId,\n        latencyMs,\n        sessionId: this._defaults.sessionId,\n        agentId: this._defaults.agentId,\n        env: this._defaults.env,\n        inputState: { from_agent: fromAgent },\n        outputState: { to_agent: toAgent },\n      });\n      return;\n    }\n    if (kind === 'guardrail') {\n      const guardrail = String(spanData.name ?? 'guardrail');\n      const triggered = Boolean(spanData.triggered);\n      this._ai.trackSpan({\n        userId: this._defaults.userId,\n        spanName: `guardrail:${guardrail}`,\n        traceId,\n        latencyMs,\n        sessionId: this._defaults.sessionId,\n        agentId: this._defaults.agentId,\n        env: this._defaults.env,\n        outputState: { triggered },\n        isError: triggered,\n      });\n      return;\n    }\n\n    const name = String(spanData.name ?? 'agent');\n    this._ai.trackSpan({\n      userId: this._defaults.userId,\n      spanName: `agent:${name}`,\n      traceId,\n      latencyMs,\n      sessionId: this._defaults.sessionId,\n      agentId: this._defaults.agentId ?? name,\n      env: this._defaults.env,\n      outputState:\n        spanData.output != null\n          ? { output: String(spanData.output) }\n          : undefined,\n    });\n  }\n\n  shutdown(): void {\n    this._ai.flush();\n  }\n\n  private _handleGeneration(\n    data: Record<string, unknown>,\n    traceId: string,\n    latencyMs: number,\n  ): void {\n    const input = this._normalizeMessagesArray(data.input);\n    for (const message of input) {\n      const role = (message as Record<string, unknown>)?.role;\n      const content = (message as Record<string, unknown>)?.content;\n      if (\n        role !== 'user' ||\n        typeof content !== 'string' ||\n        content.length === 0\n      )\n        continue;\n      this._ai.trackUserMessage({\n        userId: this._defaults.userId,\n        content,\n        sessionId: this._defaults.sessionId,\n        traceId,\n        turnId: this._turnId,\n        agentId: this._defaults.agentId,\n        env: this._defaults.env,\n      });\n      this._turnId += 1;\n    }\n\n    const output = this._normalizeMessagesArray(data.output);\n    const responseText = this._extractAssistantText(output);\n    const toolCalls = this._extractToolCalls(output);\n    const model = String(data.model ?? 'unknown');\n    const usage = (data.usage ?? {}) as Record<string, unknown>;\n    const inputTokens = this._safeNumber(\n      usage.input_tokens ?? usage.prompt_tokens ?? null,\n    );\n    const outputTokens = this._safeNumber(\n      usage.output_tokens ?? usage.completion_tokens ?? null,\n    );\n    const totalTokens = this._safeNumber(usage.total_tokens ?? null);\n    let costUsd: number | undefined;\n    if (model !== 'unknown' && inputTokens != null && outputTokens != null) {\n      try {\n        const cost = calculateCost({\n          modelName: model,\n          inputTokens,\n          outputTokens,\n        });\n        if (cost > 0) costUsd = cost;\n      } catch {\n        // cost calculation is best-effort\n      }\n    }\n\n    this._ai.trackAiMessage({\n      userId: this._defaults.userId,\n      content: responseText,\n      sessionId: this._defaults.sessionId,\n      traceId,\n      turnId: this._turnId,\n      model,\n      provider: inferProvider(model),\n      latencyMs,\n      inputTokens,\n      outputTokens,\n      totalTokens,\n      totalCostUsd: costUsd,\n      toolCalls: toolCalls.length > 0 ? toolCalls : undefined,\n      agentId: this._defaults.agentId,\n      env: this._defaults.env,\n    });\n    this._turnId += 1;\n  }\n\n  private _handleFunction(\n    data: Record<string, unknown>,\n    traceId: string,\n    latencyMs: number,\n  ): void {\n    const toolName = String(data.name ?? 'unknown');\n    const errorMessage =\n      data.error == null ? undefined : String(data.error ?? 'unknown error');\n    const toolInput =\n      data.input != null && typeof data.input === 'object'\n        ? (data.input as Record<string, unknown>)\n        : data.input != null\n          ? { raw: String(data.input) }\n          : undefined;\n    const toolOutput =\n      data.output == null ? undefined : String(data.output ?? undefined);\n\n    this._ai.trackToolCall({\n      userId: this._defaults.userId,\n      toolName,\n      success: errorMessage == null,\n      latencyMs,\n      sessionId: this._defaults.sessionId,\n      traceId,\n      turnId: this._turnId,\n      input: toolInput,\n      output: toolOutput,\n      errorMessage,\n      agentId: this._defaults.agentId,\n      env: this._defaults.env,\n    });\n  }\n\n  private _inferKind(\n    data: Record<string, unknown>,\n  ): 'generation' | 'function' | 'agent' | 'handoff' | 'guardrail' {\n    if ('from_agent' in data || 'to_agent' in data) return 'handoff';\n    if ('triggered' in data) return 'guardrail';\n    if ('model' in data && ('input' in data || 'output' in data))\n      return 'generation';\n    if (\n      'name' in data &&\n      ('input' in data || 'output' in data || 'error' in data)\n    )\n      return 'function';\n    return 'agent';\n  }\n\n  private _getSpanData(\n    span: Record<string, unknown>,\n  ): Record<string, unknown> | null {\n    const data = (span.span_data ?? span.data ?? span) as unknown;\n    if (data == null || typeof data !== 'object') return null;\n    return data as Record<string, unknown>;\n  }\n\n  private _getTraceId(span: Record<string, unknown>): string {\n    const traceId = span.trace_id ?? span.traceId ?? this._defaults.traceId;\n    return traceId == null ? randomUUID() : String(traceId);\n  }\n\n  private _getLatencyMs(span: Record<string, unknown>): number {\n    if (typeof span.latency_ms === 'number') return span.latency_ms;\n    if (typeof span.latencyMs === 'number') return span.latencyMs;\n    const start = span.start_time_ms ?? span.startTimeMs;\n    const end = span.end_time_ms ?? span.endTimeMs;\n    if (typeof start === 'number' && typeof end === 'number' && end >= start)\n      return end - start;\n    return 0;\n  }\n\n  private _safeNumber(value: unknown): number | undefined {\n    if (typeof value === 'number') return value;\n    if (typeof value === 'string') {\n      const parsed = Number(value);\n      return Number.isNaN(parsed) ? undefined : parsed;\n    }\n    return undefined;\n  }\n\n  private _normalizeMessagesArray(value: unknown): unknown[] {\n    if (Array.isArray(value)) return value;\n    if (value == null) return [];\n    return [value];\n  }\n\n  private _extractAssistantText(output: unknown[]): string {\n    let responseText = '';\n    for (const block of output) {\n      if (block == null || typeof block !== 'object') continue;\n      const item = block as Record<string, unknown>;\n      const isAssistantLike =\n        item.role === 'assistant' ||\n        item.type === 'message' ||\n        item.type === 'output_text';\n      if (!isAssistantLike) continue;\n      if (typeof item.content === 'string') {\n        responseText += item.content;\n        continue;\n      }\n      if (Array.isArray(item.content)) {\n        for (const part of item.content) {\n          const p = part as Record<string, unknown>;\n          if (p.type === 'text' && typeof p.text === 'string')\n            responseText += p.text;\n          if (typeof p.value === 'string') responseText += p.value;\n        }\n      }\n      if (item.type === 'output_text' && typeof item.text === 'string') {\n        responseText += item.text;\n      }\n    }\n    return responseText;\n  }\n\n  private _extractToolCalls(output: unknown[]): Array<Record<string, unknown>> {\n    const result: Array<Record<string, unknown>> = [];\n    for (const block of output) {\n      if (block == null || typeof block !== 'object') continue;\n      const item = block as Record<string, unknown>;\n      if (item.type === 'function_call') {\n        result.push({\n          type: 'function',\n          id: String(item.id ?? ''),\n          function: {\n            name: String(item.name ?? ''),\n            arguments: String(item.arguments ?? ''),\n          },\n        });\n        continue;\n      }\n      if (!Array.isArray(item.tool_calls)) continue;\n      for (const rawCall of item.tool_calls) {\n        const call = rawCall as Record<string, unknown>;\n        result.push({\n          type: 'function',\n          id: String(call.id ?? ''),\n          function: {\n            name: String(\n              (call.function as Record<string, unknown> | undefined)?.name ??\n                '',\n            ),\n            arguments: String(\n              (call.function as Record<string, unknown> | undefined)\n                ?.arguments ?? '',\n            ),\n          },\n        });\n      }\n    }\n    return result;\n  }\n}\n"],"mappings":";;;;;AAcA,IAAa,4BAAb,MAAuC;CACrC,AAAQ;CACR,AAAQ;CAOR,AAAQ,UAAU;CAElB,YAAY,SAAkC;EAC5C,MAAM,MAAM,kBAAkB;AAC9B,OAAK,MAAM,QAAQ;AACnB,OAAK,YAAY;GACf,QAAQ,QAAQ,UAAU,KAAK,UAAU;GACzC,WAAW,QAAQ,aAAa,KAAK,aAAa,YAAY;GAC9D,SAAS,QAAQ,WAAW,KAAK,WAAW,YAAY;GACxD,SAAS,QAAQ,WAAW,KAAK,WAAW;GAC5C,KAAK,QAAQ,OAAO,KAAK,OAAO;GACjC;;CAGH,YAAY,OAAsC;CAIlD,aAAa,QAAuC;CAIpD,WAAW,QAAuC;CAIlD,UAAU,MAAqC;EAC7C,MAAM,WAAW,KAAK,aAAa,KAAK;AACxC,MAAI,YAAY,KAAM;EAEtB,MAAM,UAAU,KAAK,YAAY,KAAK;EACtC,MAAM,YAAY,KAAK,cAAc,KAAK;EAC1C,MAAM,OAAO,KAAK,WAAW,SAAS;AACtC,MAAI,SAAS,cAAc;AACzB,QAAK,kBAAkB,UAAU,SAAS,UAAU;AACpD;;AAEF,MAAI,SAAS,YAAY;AACvB,QAAK,gBAAgB,UAAU,SAAS,UAAU;AAClD;;AAEF,MAAI,SAAS,WAAW;GACtB,MAAM,YAAY,OAAO,SAAS,cAAc,UAAU;GAC1D,MAAM,UAAU,OAAO,SAAS,YAAY,UAAU;AACtD,QAAK,IAAI,UAAU;IACjB,QAAQ,KAAK,UAAU;IACvB,UAAU,WAAW,UAAU,IAAI;IACnC;IACA;IACA,WAAW,KAAK,UAAU;IAC1B,SAAS,KAAK,UAAU;IACxB,KAAK,KAAK,UAAU;IACpB,YAAY,EAAE,YAAY,WAAW;IACrC,aAAa,EAAE,UAAU,SAAS;IACnC,CAAC;AACF;;AAEF,MAAI,SAAS,aAAa;GACxB,MAAM,YAAY,OAAO,SAAS,QAAQ,YAAY;GACtD,MAAM,YAAY,QAAQ,SAAS,UAAU;AAC7C,QAAK,IAAI,UAAU;IACjB,QAAQ,KAAK,UAAU;IACvB,UAAU,aAAa;IACvB;IACA;IACA,WAAW,KAAK,UAAU;IAC1B,SAAS,KAAK,UAAU;IACxB,KAAK,KAAK,UAAU;IACpB,aAAa,EAAE,WAAW;IAC1B,SAAS;IACV,CAAC;AACF;;EAGF,MAAM,OAAO,OAAO,SAAS,QAAQ,QAAQ;AAC7C,OAAK,IAAI,UAAU;GACjB,QAAQ,KAAK,UAAU;GACvB,UAAU,SAAS;GACnB;GACA;GACA,WAAW,KAAK,UAAU;GAC1B,SAAS,KAAK,UAAU,WAAW;GACnC,KAAK,KAAK,UAAU;GACpB,aACE,SAAS,UAAU,OACf,EAAE,QAAQ,OAAO,SAAS,OAAO,EAAE,GACnC;GACP,CAAC;;CAGJ,WAAiB;AACf,OAAK,IAAI,OAAO;;CAGlB,AAAQ,kBACN,MACA,SACA,WACM;EACN,MAAM,QAAQ,KAAK,wBAAwB,KAAK,MAAM;AACtD,OAAK,MAAM,WAAW,OAAO;GAC3B,MAAM,OAAQ,SAAqC;GACnD,MAAM,UAAW,SAAqC;AACtD,OACE,SAAS,UACT,OAAO,YAAY,YACnB,QAAQ,WAAW,EAEnB;AACF,QAAK,IAAI,iBAAiB;IACxB,QAAQ,KAAK,UAAU;IACvB;IACA,WAAW,KAAK,UAAU;IAC1B;IACA,QAAQ,KAAK;IACb,SAAS,KAAK,UAAU;IACxB,KAAK,KAAK,UAAU;IACrB,CAAC;AACF,QAAK,WAAW;;EAGlB,MAAM,SAAS,KAAK,wBAAwB,KAAK,OAAO;EACxD,MAAM,eAAe,KAAK,sBAAsB,OAAO;EACvD,MAAM,YAAY,KAAK,kBAAkB,OAAO;EAChD,MAAM,QAAQ,OAAO,KAAK,SAAS,UAAU;EAC7C,MAAM,QAAS,KAAK,SAAS,EAAE;EAC/B,MAAM,cAAc,KAAK,YACvB,MAAM,gBAAgB,MAAM,iBAAiB,KAC9C;EACD,MAAM,eAAe,KAAK,YACxB,MAAM,iBAAiB,MAAM,qBAAqB,KACnD;EACD,MAAM,cAAc,KAAK,YAAY,MAAM,gBAAgB,KAAK;EAChE,IAAIA;AACJ,MAAI,UAAU,aAAa,eAAe,QAAQ,gBAAgB,KAChE,KAAI;GACF,MAAM,OAAO,cAAc;IACzB,WAAW;IACX;IACA;IACD,CAAC;AACF,OAAI,OAAO,EAAG,WAAU;UAClB;AAKV,OAAK,IAAI,eAAe;GACtB,QAAQ,KAAK,UAAU;GACvB,SAAS;GACT,WAAW,KAAK,UAAU;GAC1B;GACA,QAAQ,KAAK;GACb;GACA,UAAU,cAAc,MAAM;GAC9B;GACA;GACA;GACA;GACA,cAAc;GACd,WAAW,UAAU,SAAS,IAAI,YAAY;GAC9C,SAAS,KAAK,UAAU;GACxB,KAAK,KAAK,UAAU;GACrB,CAAC;AACF,OAAK,WAAW;;CAGlB,AAAQ,gBACN,MACA,SACA,WACM;EACN,MAAM,WAAW,OAAO,KAAK,QAAQ,UAAU;EAC/C,MAAM,eACJ,KAAK,SAAS,OAAO,SAAY,OAAO,KAAK,SAAS,gBAAgB;EACxE,MAAM,YACJ,KAAK,SAAS,QAAQ,OAAO,KAAK,UAAU,WACvC,KAAK,QACN,KAAK,SAAS,OACZ,EAAE,KAAK,OAAO,KAAK,MAAM,EAAE,GAC3B;EACR,MAAM,aACJ,KAAK,UAAU,OAAO,SAAY,OAAO,KAAK,UAAU,OAAU;AAEpE,OAAK,IAAI,cAAc;GACrB,QAAQ,KAAK,UAAU;GACvB;GACA,SAAS,gBAAgB;GACzB;GACA,WAAW,KAAK,UAAU;GAC1B;GACA,QAAQ,KAAK;GACb,OAAO;GACP,QAAQ;GACR;GACA,SAAS,KAAK,UAAU;GACxB,KAAK,KAAK,UAAU;GACrB,CAAC;;CAGJ,AAAQ,WACN,MAC+D;AAC/D,MAAI,gBAAgB,QAAQ,cAAc,KAAM,QAAO;AACvD,MAAI,eAAe,KAAM,QAAO;AAChC,MAAI,WAAW,SAAS,WAAW,QAAQ,YAAY,MACrD,QAAO;AACT,MACE,UAAU,SACT,WAAW,QAAQ,YAAY,QAAQ,WAAW,MAEnD,QAAO;AACT,SAAO;;CAGT,AAAQ,aACN,MACgC;EAChC,MAAM,OAAQ,KAAK,aAAa,KAAK,QAAQ;AAC7C,MAAI,QAAQ,QAAQ,OAAO,SAAS,SAAU,QAAO;AACrD,SAAO;;CAGT,AAAQ,YAAY,MAAuC;EACzD,MAAM,UAAU,KAAK,YAAY,KAAK,WAAW,KAAK,UAAU;AAChE,SAAO,WAAW,OAAO,YAAY,GAAG,OAAO,QAAQ;;CAGzD,AAAQ,cAAc,MAAuC;AAC3D,MAAI,OAAO,KAAK,eAAe,SAAU,QAAO,KAAK;AACrD,MAAI,OAAO,KAAK,cAAc,SAAU,QAAO,KAAK;EACpD,MAAM,QAAQ,KAAK,iBAAiB,KAAK;EACzC,MAAM,MAAM,KAAK,eAAe,KAAK;AACrC,MAAI,OAAO,UAAU,YAAY,OAAO,QAAQ,YAAY,OAAO,MACjE,QAAO,MAAM;AACf,SAAO;;CAGT,AAAQ,YAAY,OAAoC;AACtD,MAAI,OAAO,UAAU,SAAU,QAAO;AACtC,MAAI,OAAO,UAAU,UAAU;GAC7B,MAAM,SAAS,OAAO,MAAM;AAC5B,UAAO,OAAO,MAAM,OAAO,GAAG,SAAY;;;CAK9C,AAAQ,wBAAwB,OAA2B;AACzD,MAAI,MAAM,QAAQ,MAAM,CAAE,QAAO;AACjC,MAAI,SAAS,KAAM,QAAO,EAAE;AAC5B,SAAO,CAAC,MAAM;;CAGhB,AAAQ,sBAAsB,QAA2B;EACvD,IAAI,eAAe;AACnB,OAAK,MAAM,SAAS,QAAQ;AAC1B,OAAI,SAAS,QAAQ,OAAO,UAAU,SAAU;GAChD,MAAM,OAAO;AAKb,OAAI,EAHF,KAAK,SAAS,eACd,KAAK,SAAS,aACd,KAAK,SAAS,eACM;AACtB,OAAI,OAAO,KAAK,YAAY,UAAU;AACpC,oBAAgB,KAAK;AACrB;;AAEF,OAAI,MAAM,QAAQ,KAAK,QAAQ,CAC7B,MAAK,MAAM,QAAQ,KAAK,SAAS;IAC/B,MAAM,IAAI;AACV,QAAI,EAAE,SAAS,UAAU,OAAO,EAAE,SAAS,SACzC,iBAAgB,EAAE;AACpB,QAAI,OAAO,EAAE,UAAU,SAAU,iBAAgB,EAAE;;AAGvD,OAAI,KAAK,SAAS,iBAAiB,OAAO,KAAK,SAAS,SACtD,iBAAgB,KAAK;;AAGzB,SAAO;;CAGT,AAAQ,kBAAkB,QAAmD;EAC3E,MAAMC,SAAyC,EAAE;AACjD,OAAK,MAAM,SAAS,QAAQ;AAC1B,OAAI,SAAS,QAAQ,OAAO,UAAU,SAAU;GAChD,MAAM,OAAO;AACb,OAAI,KAAK,SAAS,iBAAiB;AACjC,WAAO,KAAK;KACV,MAAM;KACN,IAAI,OAAO,KAAK,MAAM,GAAG;KACzB,UAAU;MACR,MAAM,OAAO,KAAK,QAAQ,GAAG;MAC7B,WAAW,OAAO,KAAK,aAAa,GAAG;MACxC;KACF,CAAC;AACF;;AAEF,OAAI,CAAC,MAAM,QAAQ,KAAK,WAAW,CAAE;AACrC,QAAK,MAAM,WAAW,KAAK,YAAY;IACrC,MAAM,OAAO;AACb,WAAO,KAAK;KACV,MAAM;KACN,IAAI,OAAO,KAAK,MAAM,GAAG;KACzB,UAAU;MACR,MAAM,OACH,KAAK,UAAkD,QACtD,GACH;MACD,WAAW,OACR,KAAK,UACF,aAAa,GAClB;MACF;KACF,CAAC;;;AAGN,SAAO"}

package/dist/integrations/opentelemetry.d.ts

@@ -0,0 +1,40 @@
+import { AmplitudeAI } from "../client.js";
+
+//#region src/integrations/opentelemetry.d.ts
+
+interface ExporterOptions {
+  amplitudeAI: AmplitudeAI;
+  defaultUserId?: string;
+}
+interface OTELSpan {
+  name: string;
+  kind?: number;
+  attributes?: Record<string, unknown>;
+  startTimeUnixNano?: bigint | number;
+  endTimeUnixNano?: bigint | number;
+  status?: {
+    code?: number;
+    message?: string;
+  };
+  parentSpanId?: string;
+  spanContext?: () => {
+    traceId: string;
+    spanId: string;
+  };
+}
+declare class AmplitudeAgentExporter {
+  private _ai;
+  private _defaultUserId;
+  constructor(options: ExporterOptions);
+  export(spans: OTELSpan[], resultCallback: (result: {
+    code: number;
+  }) => void): void;
+  shutdown(): Promise<void>;
+  forceFlush(): Promise<void>;
+  private _processSpan;
+}
+declare const AmplitudeGenAIExporter: typeof AmplitudeAgentExporter;
+declare const AmplitudeSpanExporter: typeof AmplitudeAgentExporter;
+//#endregion
+export { AmplitudeAgentExporter, AmplitudeGenAIExporter, AmplitudeSpanExporter, ExporterOptions };
+//# sourceMappingURL=opentelemetry.d.ts.map

package/dist/integrations/opentelemetry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"opentelemetry.d.ts","names":[],"sources":["../../src/integrations/opentelemetry.ts"],"sourcesContent":[],"mappings":";;;;AAkDc,UAvCG,eAAA,CAuCH;EAIE,WAAA,EA1CD,WA0CC;EAAO,aAAA,CAAA,EAAA,MAAA;AAgKvB;AACA,UAvMU,QAAA,CAuMG;;;eApME;;;;;;;;;;;;;cAQF,sBAAA;;;uBAIU;gBAMZ;;;cAaG;gBAIE;;;cAgKH,+BAAsB;cACtB,8BAAqB"}