@amplitude/ai 0.1.0

package/dist/decorators.d.ts

@@ -0,0 +1,109 @@
+import { PrivacyConfig } from "./core/privacy.js";
+import { AmplitudeLike } from "./types.js";
+
+//#region src/decorators.d.ts
+
+declare class ToolCallTracker {
+  static _amplitude: AmplitudeLike | null;
+  static _userId: string | null;
+  static _sessionId: string | null;
+  static _traceId: string | null;
+  static _turnId: number;
+  static _env: string | null;
+  static _agentId: string | null;
+  static _parentAgentId: string | null;
+  static _customerOrgId: string | null;
+  static _agentVersion: string | null;
+  static _context: Record<string, unknown> | null;
+  static _privacyConfig: PrivacyConfig | null;
+  static _eventProperties: Record<string, unknown> | null;
+  static _userProperties: Record<string, unknown> | null;
+  static _groups: Record<string, unknown> | null;
+  static setAmplitude(amplitude: AmplitudeLike, userId: string, opts?: {
+    sessionId?: string | null;
+    traceId?: string | null;
+    turnId?: number;
+    env?: string | null;
+    agentId?: string | null;
+    parentAgentId?: string | null;
+    customerOrgId?: string | null;
+    agentVersion?: string | null;
+    context?: Record<string, unknown> | null;
+    privacyConfig?: PrivacyConfig | null;
+    eventProperties?: Record<string, unknown> | null;
+    userProperties?: Record<string, unknown> | null;
+    groups?: Record<string, unknown> | null;
+  }): void;
+  static clear(): void;
+  static isConfigured(): boolean;
+}
+interface ToolOptions {
+  name?: string;
+  amplitude?: AmplitudeLike | null;
+  userId?: string | null;
+  sessionId?: string | null;
+  traceId?: string | null;
+  turnId?: number | null;
+  env?: string | null;
+  agentId?: string | null;
+  parentAgentId?: string | null;
+  customerOrgId?: string | null;
+  agentVersion?: string | null;
+  context?: Record<string, unknown> | null;
+  privacyConfig?: PrivacyConfig | null;
+  eventProperties?: Record<string, unknown> | null;
+  userProperties?: Record<string, unknown> | null;
+  groups?: Record<string, unknown> | null;
+  inputSchema?: Record<string, unknown> | null;
+  timeoutMs?: number | null;
+  onError?: ((error: Error, toolName: string) => void) | null;
+}
+type AnyFn = (...args: unknown[]) => unknown;
+/**
+ * Converts a function's return type to Promise-wrapped if not already.
+ * Reflects the fact that tool() always returns an async wrapper.
+ */
+type ToolWrapped<T extends AnyFn> = T extends ((...args: infer A) => infer R) ? R extends Promise<unknown> ? (...args: A) => R : (...args: A) => Promise<R> : T;
+/**
+ * Wraps a function and emits one `[Agent] Tool Call` event per invocation.
+ *
+ * Supports both forms:
+ * - `tool(fn, opts?)`
+ * - `tool(opts)(fn)`
+ *
+ * Resolution order for tracking context is:
+ * runtime overrides > decorator opts > active session context > `ToolCallTracker`.
+ * If tracking is not configured (`amplitude`/`userId` missing), the wrapped function
+ * still executes and no tracking event is emitted.
+ */
+declare function tool<T extends AnyFn>(fn: T, opts?: ToolOptions): ToolWrapped<T>;
+declare function tool(opts: ToolOptions): <T extends AnyFn>(fn: T) => ToolWrapped<T>;
+interface ObserveOptions {
+  name?: string;
+  amplitude?: AmplitudeLike | null;
+  userId?: string | null;
+  agentId?: string | null;
+  env?: string | null;
+  privacyConfig?: PrivacyConfig | null;
+}
+/**
+ * Converts a function's return type to Promise-wrapped if not already.
+ * Reflects the fact that observe() always returns an async wrapper.
+ */
+type ObserveWrapped<T extends AnyFn> = T extends ((...args: infer A) => infer R) ? R extends Promise<unknown> ? (...args: A) => R : (...args: A) => Promise<R> : T;
+/**
+ * Wraps a function and emits one `[Agent] Span` event per invocation.
+ *
+ * Supports both forms:
+ * - `observe(fn, opts?)`
+ * - `observe(opts)(fn)`
+ *
+ * If called inside an active session context, spans attach to that session.
+ * If no context is active, observe creates a short-lived session boundary and
+ * emits a session-end event when the wrapped function completes.
+ */
+declare function observe<T extends AnyFn>(fn: T, opts?: ObserveOptions): ObserveWrapped<T>;
+declare function observe(opts: ObserveOptions): <T extends AnyFn>(fn: T) => ObserveWrapped<T>;
+//#endregion
+export { ObserveWrapped, ToolCallTracker, ToolWrapped, observe, tool };
+//# sourceMappingURL=decorators.d.ts.map

package/dist/decorators.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"decorators.d.ts","names":[],"sources":["../src/decorators.ts"],"sourcesContent":[],"mappings":";;;;;AAsC0B,cAdb,eAAA,CAca;EACR,OAAA,UAAA,EAdG,aAcH,GAAA,IAAA;EAGH,OAAA,OAAA,EAAA,MAAA,GAAA,IAAA;EAWC,OAAA,UAAA,EAAA,MAAA,GAAA,IAAA;EACM,OAAA,QAAA,EAAA,MAAA,GAAA,IAAA;EACE,OAAA,OAAA,EAAA,MAAA;EACD,OAAA,IAAA,EAAA,MAAA,GAAA,IAAA;EACR,OAAA,QAAA,EAAA,MAAA,GAAA,IAAA;EAAM,OAAA,cAAA,EAAA,MAAA,GAAA,IAAA;EAiDX,OAAA,cAAW,EAAA,MAAA,GAAA,IAAA;EAEP,OAAA,aAAA,EAAA,MAAA,GAAA,IAAA;EAUF,OAAA,QAAA,EAnFO,MAmFP,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EACM,OAAA,cAAA,EAnFO,aAmFP,GAAA,IAAA;EACE,OAAA,gBAAA,EAnFO,MAmFP,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EACD,OAAA,eAAA,EAnFO,MAmFP,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EACR,OAAA,OAAA,EAnFO,MAmFP,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EACK,OAAA,YAAA,CAAA,SAAA,EAjFD,aAiFC,EAAA,MAAA,EAAA,MAAA,EAAA,IAiHX,CAjHW,EAAA;IAEK,SAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAAK,OAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAkGrB,MAAK,CAAA,EAAA,MAAA;IAOE,GAAA,CAAA,EAAA,MAAW,GAAA,IAAA;IAAW,OAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAAS,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAG7B,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IACE,YAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAAM,OAAA,CAAA,EArLN,MAqLM,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;IACN,aAAA,CAAA,EArLM,aAqLN,GAAA,IAAA;IAAc,eAAA,CAAA,EApLN,MAoLM,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;IAAR,cAAA,CAAA,EAnLC,MAmLD,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;IAClB,MAAA,CAAA,EAnLW,MAmLX,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EAAC,CAAA,CAAA,EAAA,IAAA;EAcW,OAAI,KAAA,CAAA,CAAA,EAAA,IAAA;EAAW,OAAA,YAAA,CAAA,CAAA,EAAA,OAAA;;UAhJrB,WAAA,CAkJD;EACM,IAAA,CAAA,EAAA,MAAA;EAAZ,SAAA,CAAA,EAjJW,aAiJX,GAAA,IAAA;EAAW,MAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EACE,SAAI,CAAA,EAAA,MAAA,GAAA,IAAA;EACZ,OAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EACM,MAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAAW,GAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAAkB,OAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAAZ,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAAW,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAgJhC,YAAA,CAAA,EAAA,MAAc,GAAA,IAAA;EAaZ,OAAA,CAAA,EAvSA,MAuSA,CAAc,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EAAW,aAAA,CAAA,EAtSnB,aAsSmB,GAAA,IAAA;EAAS,eAAA,CAAA,EArS1B,MAqS0B,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EAGhC,cAAA,CAAA,EAvSK,MAuSL,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EACE,MAAA,CAAA,EAvSL,MAuSK,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EAAM,WAAA,CAAA,EAtSN,MAsSM,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EACN,SAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAAc,OAAA,CAAA,EAAA,CAAA,CAAA,KAAA,EArST,KAqSS,EAAA,QAAA,EAAA,MAAA,EAAA,GAAA,IAAA,CAAA,GAAA,IAAA;;KAnMzB,KAAA,GAoMD,CAAA,GAAA,IAAA,EAAA,OAAA,EAAA,EAAA,GAAA,OAAA;;AAaJ;;;AAES,KA5MG,WA4MH,CAAA,UA5MyB,KA4MzB,CAAA,GA5MkC,CA4MlC,UAAA,CAAA,GAAA,IAAA,EAAA,KAAA,EAAA,EAAA,GAAA,KAAA,EAAA,IAAA,CAAA,SAzMK,OAyML,CAAA,OAAA,CAAA,GAAA,CAAA,GAAA,IAAA,EAxMO,CAwMP,EAAA,GAxMa,CAwMb,GAAA,CAAA,GAAA,IAAA,EAvMO,CAuMP,EAAA,GAvMa,OAuMb,CAvMqB,CAuMrB,CAAA,GAtML,CAsMK;;;;AAET;;;;;;;;;iBA1LgB,eAAe,WACzB,UACG,cACN,YAAY;iBACC,IAAA,OACR,yBACM,WAAW,MAAM,YAAY;UAgJjC,cAAA;;cAEI;;;;kBAII;;;;;;KAON,yBAAyB,SAAS,sDAGhC,6BACE,MAAM,cACN,MAAM,QAAQ,KAC1B;;;;;;;;;;;;iBAaY,kBAAkB,WAC5B,UACG,iBACN,eAAe;iBACF,OAAA,OACR,4BACM,WAAW,MAAM,eAAe"}

package/dist/decorators.js

@@ -0,0 +1,281 @@
+import { getLogger } from "./utils/logger.js";
+import { trackSessionEnd, trackSpan, trackToolCall } from "./core/tracking.js";
+import { SessionContext, _sessionStorage, getActiveContext } from "./context.js";
+import { randomUUID } from "node:crypto";
+
+//#region src/decorators.ts
+/**
+* Higher-order functions for automatic tracking of function calls as tool and span events.
+*
+* TypeScript equivalent of Python's @tool and @observe decorators.
+* Since TS/JS does not have native decorators with the same power as Python,
+* these are implemented as HOFs (higher-order functions) that wrap callables.
+*/
+var ToolCallTracker = class ToolCallTracker {
+	static _amplitude = null;
+	static _userId = null;
+	static _sessionId = null;
+	static _traceId = null;
+	static _turnId = 1;
+	static _env = null;
+	static _agentId = null;
+	static _parentAgentId = null;
+	static _customerOrgId = null;
+	static _agentVersion = null;
+	static _context = null;
+	static _privacyConfig = null;
+	static _eventProperties = null;
+	static _userProperties = null;
+	static _groups = null;
+	static setAmplitude(amplitude, userId, opts = {}) {
+		ToolCallTracker._amplitude = amplitude;
+		ToolCallTracker._userId = userId;
+		ToolCallTracker._sessionId = opts.sessionId ?? null;
+		ToolCallTracker._traceId = opts.traceId ?? null;
+		ToolCallTracker._turnId = opts.turnId ?? 1;
+		ToolCallTracker._env = opts.env ?? null;
+		ToolCallTracker._agentId = opts.agentId ?? null;
+		ToolCallTracker._parentAgentId = opts.parentAgentId ?? null;
+		ToolCallTracker._customerOrgId = opts.customerOrgId ?? null;
+		ToolCallTracker._agentVersion = opts.agentVersion ?? null;
+		ToolCallTracker._context = opts.context ?? null;
+		ToolCallTracker._privacyConfig = opts.privacyConfig ?? null;
+		ToolCallTracker._eventProperties = opts.eventProperties ?? null;
+		ToolCallTracker._userProperties = opts.userProperties ?? null;
+		ToolCallTracker._groups = opts.groups ?? null;
+	}
+	static clear() {
+		ToolCallTracker._amplitude = null;
+		ToolCallTracker._userId = null;
+		ToolCallTracker._sessionId = null;
+		ToolCallTracker._traceId = null;
+		ToolCallTracker._turnId = 1;
+		ToolCallTracker._env = null;
+		ToolCallTracker._agentId = null;
+		ToolCallTracker._parentAgentId = null;
+		ToolCallTracker._customerOrgId = null;
+		ToolCallTracker._agentVersion = null;
+		ToolCallTracker._context = null;
+		ToolCallTracker._privacyConfig = null;
+		ToolCallTracker._eventProperties = null;
+		ToolCallTracker._userProperties = null;
+		ToolCallTracker._groups = null;
+	}
+	static isConfigured() {
+		return ToolCallTracker._amplitude != null && ToolCallTracker._userId != null;
+	}
+};
+function _resolveContextFields(opts) {
+	const ctx = getActiveContext();
+	function resolve(decoratorVal, trackerVal, ctxVal = null) {
+		if (decoratorVal !== void 0 && decoratorVal !== null) return decoratorVal;
+		if (ctxVal !== void 0 && ctxVal !== null) return ctxVal;
+		if (trackerVal !== void 0 && trackerVal !== null) return trackerVal;
+		return null;
+	}
+	return {
+		amplitude: opts.amplitude ?? ToolCallTracker._amplitude ?? null,
+		userId: resolve(opts.userId, ToolCallTracker._userId, ctx?.userId),
+		sessionId: resolve(opts.sessionId, ToolCallTracker._sessionId, ctx?.sessionId),
+		traceId: resolve(opts.traceId, ToolCallTracker._traceId, ctx?.traceId),
+		turnId: opts.turnId ?? ToolCallTracker._turnId,
+		env: resolve(opts.env, ToolCallTracker._env, ctx?.env),
+		agentId: resolve(opts.agentId, ToolCallTracker._agentId, ctx?.agentId),
+		parentAgentId: resolve(opts.parentAgentId, ToolCallTracker._parentAgentId, ctx?.parentAgentId),
+		customerOrgId: resolve(opts.customerOrgId, ToolCallTracker._customerOrgId, ctx?.customerOrgId),
+		agentVersion: resolve(opts.agentVersion, ToolCallTracker._agentVersion, ctx?.agentVersion),
+		context: resolve(opts.context, ToolCallTracker._context, ctx?.context),
+		privacyConfig: resolve(opts.privacyConfig, ToolCallTracker._privacyConfig),
+		eventProperties: resolve(opts.eventProperties, ToolCallTracker._eventProperties),
+		userProperties: resolve(opts.userProperties, ToolCallTracker._userProperties),
+		groups: resolve(opts.groups, ToolCallTracker._groups, ctx?.groups)
+	};
+}
+function tool(fnOrOpts, maybeOpts) {
+	if (typeof fnOrOpts === "function") return _wrapTool(fnOrOpts, maybeOpts ?? {});
+	return (fn) => _wrapTool(fn, fnOrOpts);
+}
+function _wrapTool(fn, opts) {
+	const toolName = opts.name || fn.name || "anonymous";
+	const schemaJson = opts.inputSchema ? JSON.stringify(opts.inputSchema) : null;
+	const wrapper = async function(...args) {
+		const r = _resolveContextFields(opts);
+		if (r.amplitude == null || r.userId == null) {
+			getLogger().warn(`Tool '${toolName}' called but tracking not configured. Call ToolCallTracker.setAmplitude() or pass amplitude/userId to tool().`);
+			return fn.apply(this, args);
+		}
+		const toolInput = args.length === 1 && typeof args[0] === "object" && args[0] != null ? args[0] : args.length > 0 ? { args } : void 0;
+		const startTime = performance.now();
+		let success = true;
+		let errorMsg = null;
+		let result = void 0;
+		try {
+			if (opts.timeoutMs != null) result = await _runWithTimeout(fn.bind(this, ...args), opts.timeoutMs, toolName);
+			else result = await fn.apply(this, args);
+			return result;
+		} catch (e) {
+			success = false;
+			errorMsg = e instanceof Error ? e.message : String(e);
+			if (opts.onError != null) try {
+				opts.onError(e instanceof Error ? e : new Error(String(e)), toolName);
+			} catch {}
+			getLogger().error(`Tool '${toolName}' failed: ${errorMsg}`);
+			throw e;
+		} finally {
+			const latencyMs = performance.now() - startTime;
+			const extraProps = {};
+			if (schemaJson != null) extraProps["[Agent] Tool Input Schema"] = schemaJson;
+			const mergedEvtProps = {
+				...r.eventProperties ?? {},
+				...extraProps
+			};
+			try {
+				trackToolCall({
+					amplitude: r.amplitude,
+					userId: r.userId,
+					toolName,
+					success,
+					latencyMs,
+					sessionId: r.sessionId,
+					traceId: r.traceId,
+					turnId: r.turnId ?? void 0,
+					toolInput,
+					toolOutput: success ? result : void 0,
+					errorMessage: errorMsg,
+					env: r.env,
+					agentId: r.agentId,
+					parentAgentId: r.parentAgentId,
+					customerOrgId: r.customerOrgId,
+					agentVersion: r.agentVersion,
+					context: r.context,
+					eventProperties: Object.keys(extraProps).length > 0 ? mergedEvtProps : r.eventProperties,
+					userProperties: r.userProperties,
+					groups: r.groups,
+					privacyConfig: r.privacyConfig
+				});
+			} catch (trackError) {
+				getLogger().error(`Failed to track tool call '${toolName}': ${trackError}`);
+			}
+		}
+	};
+	Object.defineProperty(wrapper, "name", {
+		value: fn.name,
+		configurable: true
+	});
+	return wrapper;
+}
+async function _runWithTimeout(fn, timeoutMs, toolName) {
+	let timer;
+	const timeoutPromise = new Promise((_resolve, reject) => {
+		timer = setTimeout(() => reject(/* @__PURE__ */ new Error(`Tool '${toolName}' exceeded ${timeoutMs}ms timeout`)), timeoutMs);
+	});
+	try {
+		return await Promise.race([fn(), timeoutPromise]);
+	} finally {
+		clearTimeout(timer);
+	}
+}
+function observe(fnOrOpts, maybeOpts) {
+	if (typeof fnOrOpts === "function") return _wrapObserve(fnOrOpts, maybeOpts ?? {});
+	return (fn) => _wrapObserve(fn, fnOrOpts);
+}
+function _resolveObserveParams(opts) {
+	const ctx = getActiveContext();
+	return {
+		amplitude: opts.amplitude ?? ToolCallTracker._amplitude ?? null,
+		userId: opts.userId ?? ctx?.userId ?? ToolCallTracker._userId ?? "",
+		agentId: opts.agentId ?? ctx?.agentId ?? ToolCallTracker._agentId ?? null,
+		env: opts.env ?? ctx?.env ?? ToolCallTracker._env ?? null,
+		privacyConfig: opts.privacyConfig ?? ToolCallTracker._privacyConfig ?? null,
+		context: ctx
+	};
+}
+function _serializeState(value, pc) {
+	if (value == null) return null;
+	if (pc != null) {
+		const mode = pc.contentMode;
+		if (mode === "metadata_only" || mode == null && pc.privacyMode) return null;
+	}
+	if (typeof value === "object" && !Array.isArray(value)) return value;
+	return { value: String(value) };
+}
+function _wrapObserve(fn, opts) {
+	const spanName = opts.name || fn.name || "anonymous";
+	const wrapper = async function(...args) {
+		const params = _resolveObserveParams(opts);
+		const ctx = getActiveContext();
+		const ownsSession = ctx == null;
+		const sessionId = ctx?.sessionId ?? randomUUID();
+		const runFn = async () => {
+			const inputState = _serializeState(args.length === 1 ? args[0] : args.length > 0 ? { args } : null, params.privacyConfig);
+			const start = performance.now();
+			let isError = false;
+			let errorMessage = null;
+			let result = void 0;
+			try {
+				result = await fn.apply(this, args);
+				return result;
+			} catch (exc) {
+				isError = true;
+				errorMessage = exc instanceof Error ? exc.message : String(exc);
+				throw exc;
+			} finally {
+				const latencyMs = performance.now() - start;
+				const outputState = _serializeState(result, params.privacyConfig);
+				const activeCtx = getActiveContext();
+				if (params.amplitude != null) {
+					try {
+						trackSpan({
+							amplitude: params.amplitude,
+							userId: params.userId,
+							spanName,
+							traceId: activeCtx?.traceId ?? randomUUID(),
+							latencyMs,
+							inputState,
+							outputState,
+							isError,
+							errorMessage,
+							sessionId,
+							agentId: params.agentId || spanName,
+							env: params.env,
+							privacyConfig: params.privacyConfig
+						});
+					} catch (e) {
+						getLogger().error(`Failed to track @observe span '${spanName}': ${e}`);
+					}
+					if (ownsSession) try {
+						trackSessionEnd({
+							amplitude: params.amplitude,
+							userId: params.userId,
+							sessionId,
+							env: params.env,
+							agentId: params.agentId || spanName,
+							privacyConfig: params.privacyConfig
+						});
+					} catch (e) {
+						getLogger().error(`Failed to end @observe session '${spanName}': ${e}`);
+					}
+				}
+			}
+		};
+		if (ownsSession) {
+			const newCtx = new SessionContext({
+				sessionId,
+				userId: params.userId,
+				agentId: params.agentId || spanName,
+				env: params.env
+			});
+			return _sessionStorage.run(newCtx, runFn);
+		}
+		return runFn();
+	};
+	Object.defineProperty(wrapper, "name", {
+		value: fn.name,
+		configurable: true
+	});
+	return wrapper;
+}
+
+//#endregion
+export { ToolCallTracker, observe, tool };
+//# sourceMappingURL=decorators.js.map

package/dist/decorators.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"decorators.js","names":["errorMsg: string | null","result: unknown","extraProps: Record<string, unknown>","timer: ReturnType<typeof setTimeout> | undefined","errorMessage: string | null"],"sources":["../src/decorators.ts"],"sourcesContent":["/**\n * Higher-order functions for automatic tracking of function calls as tool and span events.\n *\n * TypeScript equivalent of Python's @tool and @observe decorators.\n * Since TS/JS does not have native decorators with the same power as Python,\n * these are implemented as HOFs (higher-order functions) that wrap callables.\n */\n\nimport { randomUUID } from 'node:crypto';\nimport {\n  _sessionStorage,\n  getActiveContext,\n  SessionContext,\n} from './context.js';\nimport type { PrivacyConfig } from './core/privacy.js';\nimport { trackSessionEnd, trackSpan, trackToolCall } from './core/tracking.js';\nimport type { AmplitudeLike } from './types.js';\nimport { getLogger } from './utils/logger.js';\n\n// ---------------------------------------------------------------------------\n// ToolCallTracker — global config singleton (parity with Python)\n// ---------------------------------------------------------------------------\n\n// biome-ignore lint/complexity/noStaticOnlyClass: mirrors Python SDK's ToolCallTracker API\nexport class ToolCallTracker {\n  static _amplitude: AmplitudeLike | null = null;\n  static _userId: string | null = null;\n  static _sessionId: string | null = null;\n  static _traceId: string | null = null;\n  static _turnId = 1;\n  static _env: string | null = null;\n  static _agentId: string | null = null;\n  static _parentAgentId: string | null = null;\n  static _customerOrgId: string | null = null;\n  static _agentVersion: string | null = null;\n  static _context: Record<string, unknown> | null = null;\n  static _privacyConfig: PrivacyConfig | null = null;\n  static _eventProperties: Record<string, unknown> | null = null;\n  static _userProperties: Record<string, unknown> | null = null;\n  static _groups: Record<string, unknown> | null = null;\n\n  static setAmplitude(\n    amplitude: AmplitudeLike,\n    userId: string,\n    opts: {\n      sessionId?: string | null;\n      traceId?: string | null;\n      turnId?: number;\n      env?: string | null;\n      agentId?: string | null;\n      parentAgentId?: string | null;\n      customerOrgId?: string | null;\n      agentVersion?: string | null;\n      context?: Record<string, unknown> | null;\n      privacyConfig?: PrivacyConfig | null;\n      eventProperties?: Record<string, unknown> | null;\n      userProperties?: Record<string, unknown> | null;\n      groups?: Record<string, unknown> | null;\n    } = {},\n  ): void {\n    ToolCallTracker._amplitude = amplitude;\n    ToolCallTracker._userId = userId;\n    ToolCallTracker._sessionId = opts.sessionId ?? null;\n    ToolCallTracker._traceId = opts.traceId ?? null;\n    ToolCallTracker._turnId = opts.turnId ?? 1;\n    ToolCallTracker._env = opts.env ?? null;\n    ToolCallTracker._agentId = opts.agentId ?? null;\n    ToolCallTracker._parentAgentId = opts.parentAgentId ?? null;\n    ToolCallTracker._customerOrgId = opts.customerOrgId ?? null;\n    ToolCallTracker._agentVersion = opts.agentVersion ?? null;\n    ToolCallTracker._context = opts.context ?? null;\n    ToolCallTracker._privacyConfig = opts.privacyConfig ?? null;\n    ToolCallTracker._eventProperties = opts.eventProperties ?? null;\n    ToolCallTracker._userProperties = opts.userProperties ?? null;\n    ToolCallTracker._groups = opts.groups ?? null;\n  }\n\n  static clear(): void {\n    ToolCallTracker._amplitude = null;\n    ToolCallTracker._userId = null;\n    ToolCallTracker._sessionId = null;\n    ToolCallTracker._traceId = null;\n    ToolCallTracker._turnId = 1;\n    ToolCallTracker._env = null;\n    ToolCallTracker._agentId = null;\n    ToolCallTracker._parentAgentId = null;\n    ToolCallTracker._customerOrgId = null;\n    ToolCallTracker._agentVersion = null;\n    ToolCallTracker._context = null;\n    ToolCallTracker._privacyConfig = null;\n    ToolCallTracker._eventProperties = null;\n    ToolCallTracker._userProperties = null;\n    ToolCallTracker._groups = null;\n  }\n\n  static isConfigured(): boolean {\n    return (\n      ToolCallTracker._amplitude != null && ToolCallTracker._userId != null\n    );\n  }\n}\n\n// ---------------------------------------------------------------------------\n// Context resolution (merges runtime overrides > explicit opts > session ctx > global)\n// ---------------------------------------------------------------------------\n\ninterface ToolOptions {\n  name?: string;\n  amplitude?: AmplitudeLike | null;\n  userId?: string | null;\n  sessionId?: string | null;\n  traceId?: string | null;\n  turnId?: number | null;\n  env?: string | null;\n  agentId?: string | null;\n  parentAgentId?: string | null;\n  customerOrgId?: string | null;\n  agentVersion?: string | null;\n  context?: Record<string, unknown> | null;\n  privacyConfig?: PrivacyConfig | null;\n  eventProperties?: Record<string, unknown> | null;\n  userProperties?: Record<string, unknown> | null;\n  groups?: Record<string, unknown> | null;\n  inputSchema?: Record<string, unknown> | null;\n  timeoutMs?: number | null;\n  onError?: ((error: Error, toolName: string) => void) | null;\n}\n\ninterface ResolvedContext {\n  amplitude: AmplitudeLike | null;\n  userId: string | null;\n  sessionId: string | null;\n  traceId: string | null;\n  turnId: number;\n  env: string | null;\n  agentId: string | null;\n  parentAgentId: string | null;\n  customerOrgId: string | null;\n  agentVersion: string | null;\n  context: Record<string, unknown> | null;\n  privacyConfig: PrivacyConfig | null;\n  eventProperties: Record<string, unknown> | null;\n  userProperties: Record<string, unknown> | null;\n  groups: Record<string, unknown> | null;\n}\n\nfunction _resolveContextFields(opts: ToolOptions): ResolvedContext {\n  const ctx = getActiveContext();\n\n  function resolve(\n    decoratorVal: unknown,\n    trackerVal: unknown,\n    ctxVal: unknown = null,\n  ): unknown {\n    if (decoratorVal !== undefined && decoratorVal !== null)\n      return decoratorVal;\n    if (ctxVal !== undefined && ctxVal !== null) return ctxVal;\n    if (trackerVal !== undefined && trackerVal !== null) return trackerVal;\n    return null;\n  }\n\n  return {\n    amplitude: opts.amplitude ?? ToolCallTracker._amplitude ?? null,\n    userId: resolve(opts.userId, ToolCallTracker._userId, ctx?.userId) as\n      | string\n      | null,\n    sessionId: resolve(\n      opts.sessionId,\n      ToolCallTracker._sessionId,\n      ctx?.sessionId,\n    ) as string | null,\n    traceId: resolve(opts.traceId, ToolCallTracker._traceId, ctx?.traceId) as\n      | string\n      | null,\n    turnId: opts.turnId ?? ToolCallTracker._turnId,\n    env: resolve(opts.env, ToolCallTracker._env, ctx?.env) as string | null,\n    agentId: resolve(opts.agentId, ToolCallTracker._agentId, ctx?.agentId) as\n      | string\n      | null,\n    parentAgentId: resolve(\n      opts.parentAgentId,\n      ToolCallTracker._parentAgentId,\n      ctx?.parentAgentId,\n    ) as string | null,\n    customerOrgId: resolve(\n      opts.customerOrgId,\n      ToolCallTracker._customerOrgId,\n      ctx?.customerOrgId,\n    ) as string | null,\n    agentVersion: resolve(\n      opts.agentVersion,\n      ToolCallTracker._agentVersion,\n      ctx?.agentVersion,\n    ) as string | null,\n    context: resolve(\n      opts.context,\n      ToolCallTracker._context,\n      ctx?.context,\n    ) as Record<string, unknown> | null,\n    privacyConfig: resolve(\n      opts.privacyConfig,\n      ToolCallTracker._privacyConfig,\n    ) as PrivacyConfig | null,\n    eventProperties: resolve(\n      opts.eventProperties,\n      ToolCallTracker._eventProperties,\n    ) as Record<string, unknown> | null,\n    userProperties: resolve(\n      opts.userProperties,\n      ToolCallTracker._userProperties,\n    ) as Record<string, unknown> | null,\n    groups: resolve(\n      opts.groups,\n      ToolCallTracker._groups,\n      ctx?.groups,\n    ) as Record<string, unknown> | null,\n  };\n}\n\n// ---------------------------------------------------------------------------\n// tool() HOF — wraps a function to auto-track as [Agent] Tool Call\n// ---------------------------------------------------------------------------\n\ntype AnyFn = (...args: unknown[]) => unknown;\ntype AsyncFn = (...args: unknown[]) => Promise<unknown>;\n\n/**\n * Converts a function's return type to Promise-wrapped if not already.\n * Reflects the fact that tool() always returns an async wrapper.\n */\nexport type ToolWrapped<T extends AnyFn> = T extends (\n  ...args: infer A\n) => infer R\n  ? R extends Promise<unknown>\n    ? (...args: A) => R\n    : (...args: A) => Promise<R>\n  : T;\n\n/**\n * Wraps a function and emits one `[Agent] Tool Call` event per invocation.\n *\n * Supports both forms:\n * - `tool(fn, opts?)`\n * - `tool(opts)(fn)`\n *\n * Resolution order for tracking context is:\n * runtime overrides > decorator opts > active session context > `ToolCallTracker`.\n * If tracking is not configured (`amplitude`/`userId` missing), the wrapped function\n * still executes and no tracking event is emitted.\n */\nexport function tool<T extends AnyFn>(\n  fn: T,\n  opts?: ToolOptions,\n): ToolWrapped<T>;\nexport function tool(\n  opts: ToolOptions,\n): <T extends AnyFn>(fn: T) => ToolWrapped<T>;\nexport function tool<T extends AnyFn>(\n  fnOrOpts: T | ToolOptions,\n  maybeOpts?: ToolOptions,\n): ToolWrapped<T> | (<U extends AnyFn>(fn: U) => ToolWrapped<U>) {\n  if (typeof fnOrOpts === 'function') {\n    return _wrapTool(fnOrOpts as T, maybeOpts ?? {}) as ToolWrapped<T>;\n  }\n  return <U extends AnyFn>(fn: U): ToolWrapped<U> =>\n    _wrapTool(fn, fnOrOpts) as ToolWrapped<U>;\n}\n\nfunction _wrapTool<T extends AnyFn>(fn: T, opts: ToolOptions): ToolWrapped<T> {\n  const toolName = opts.name || fn.name || 'anonymous';\n  const schemaJson = opts.inputSchema ? JSON.stringify(opts.inputSchema) : null;\n\n  const wrapper = async function (\n    this: unknown,\n    ...args: unknown[]\n  ): Promise<unknown> {\n    const r = _resolveContextFields(opts);\n\n    if (r.amplitude == null || r.userId == null) {\n      getLogger().warn(\n        `Tool '${toolName}' called but tracking not configured. Call ToolCallTracker.setAmplitude() or pass amplitude/userId to tool().`,\n      );\n      return fn.apply(this, args);\n    }\n\n    const toolInput =\n      args.length === 1 && typeof args[0] === 'object' && args[0] != null\n        ? args[0]\n        : args.length > 0\n          ? { args }\n          : undefined;\n\n    const startTime = performance.now();\n    let success = true;\n    let errorMsg: string | null = null;\n    let result: unknown = undefined;\n\n    try {\n      if (opts.timeoutMs != null) {\n        result = await _runWithTimeout(\n          fn.bind(this, ...args) as AsyncFn,\n          opts.timeoutMs,\n          toolName,\n        );\n      } else {\n        result = await fn.apply(this, args);\n      }\n      return result;\n    } catch (e) {\n      success = false;\n      errorMsg = e instanceof Error ? e.message : String(e);\n      if (opts.onError != null) {\n        try {\n          opts.onError(e instanceof Error ? e : new Error(String(e)), toolName);\n        } catch {\n          // swallow callback errors\n        }\n      }\n      getLogger().error(`Tool '${toolName}' failed: ${errorMsg}`);\n      throw e;\n    } finally {\n      const latencyMs = performance.now() - startTime;\n\n      const extraProps: Record<string, unknown> = {};\n      if (schemaJson != null) {\n        extraProps['[Agent] Tool Input Schema'] = schemaJson;\n      }\n\n      const mergedEvtProps = {\n        ...(r.eventProperties ?? {}),\n        ...extraProps,\n      };\n\n      try {\n        trackToolCall({\n          amplitude: r.amplitude,\n          userId: r.userId,\n          toolName,\n          success,\n          latencyMs,\n          sessionId: r.sessionId,\n          traceId: r.traceId,\n          turnId: r.turnId ?? undefined,\n          toolInput,\n          toolOutput: success ? result : undefined,\n          errorMessage: errorMsg,\n          env: r.env,\n          agentId: r.agentId,\n          parentAgentId: r.parentAgentId,\n          customerOrgId: r.customerOrgId,\n          agentVersion: r.agentVersion,\n          context: r.context,\n          eventProperties:\n            Object.keys(extraProps).length > 0\n              ? mergedEvtProps\n              : r.eventProperties,\n          userProperties: r.userProperties,\n          groups: r.groups,\n          privacyConfig: r.privacyConfig,\n        });\n      } catch (trackError) {\n        getLogger().error(\n          `Failed to track tool call '${toolName}': ${trackError}`,\n        );\n      }\n    }\n  } as unknown as ToolWrapped<T>;\n\n  Object.defineProperty(wrapper, 'name', {\n    value: fn.name,\n    configurable: true,\n  });\n  return wrapper;\n}\n\nasync function _runWithTimeout(\n  fn: AsyncFn,\n  timeoutMs: number,\n  toolName: string,\n): Promise<unknown> {\n  let timer: ReturnType<typeof setTimeout> | undefined;\n  const timeoutPromise = new Promise<never>((_resolve, reject) => {\n    timer = setTimeout(\n      () =>\n        reject(new Error(`Tool '${toolName}' exceeded ${timeoutMs}ms timeout`)),\n      timeoutMs,\n    );\n  });\n\n  try {\n    return await Promise.race([fn(), timeoutPromise]);\n  } finally {\n    clearTimeout(timer);\n  }\n}\n\n// ---------------------------------------------------------------------------\n// observe() HOF — wraps a function to auto-track as [Agent] Span\n// ---------------------------------------------------------------------------\n\ninterface ObserveOptions {\n  name?: string;\n  amplitude?: AmplitudeLike | null;\n  userId?: string | null;\n  agentId?: string | null;\n  env?: string | null;\n  privacyConfig?: PrivacyConfig | null;\n}\n\n/**\n * Converts a function's return type to Promise-wrapped if not already.\n * Reflects the fact that observe() always returns an async wrapper.\n */\nexport type ObserveWrapped<T extends AnyFn> = T extends (\n  ...args: infer A\n) => infer R\n  ? R extends Promise<unknown>\n    ? (...args: A) => R\n    : (...args: A) => Promise<R>\n  : T;\n\n/**\n * Wraps a function and emits one `[Agent] Span` event per invocation.\n *\n * Supports both forms:\n * - `observe(fn, opts?)`\n * - `observe(opts)(fn)`\n *\n * If called inside an active session context, spans attach to that session.\n * If no context is active, observe creates a short-lived session boundary and\n * emits a session-end event when the wrapped function completes.\n */\nexport function observe<T extends AnyFn>(\n  fn: T,\n  opts?: ObserveOptions,\n): ObserveWrapped<T>;\nexport function observe(\n  opts: ObserveOptions,\n): <T extends AnyFn>(fn: T) => ObserveWrapped<T>;\nexport function observe<T extends AnyFn>(\n  fnOrOpts: T | ObserveOptions,\n  maybeOpts?: ObserveOptions,\n): ObserveWrapped<T> | (<U extends AnyFn>(fn: U) => ObserveWrapped<U>) {\n  if (typeof fnOrOpts === 'function') {\n    return _wrapObserve(fnOrOpts as T, maybeOpts ?? {}) as ObserveWrapped<T>;\n  }\n  return <U extends AnyFn>(fn: U): ObserveWrapped<U> =>\n    _wrapObserve(fn, fnOrOpts) as ObserveWrapped<U>;\n}\n\ninterface ResolvedObserveParams {\n  amplitude: AmplitudeLike | null;\n  userId: string;\n  agentId: string | null;\n  env: string | null;\n  privacyConfig: PrivacyConfig | null;\n  context: SessionContext | null;\n}\n\nfunction _resolveObserveParams(opts: ObserveOptions): ResolvedObserveParams {\n  const ctx = getActiveContext();\n  return {\n    amplitude: opts.amplitude ?? ToolCallTracker._amplitude ?? null,\n    userId: opts.userId ?? ctx?.userId ?? ToolCallTracker._userId ?? '',\n    agentId: opts.agentId ?? ctx?.agentId ?? ToolCallTracker._agentId ?? null,\n    env: opts.env ?? ctx?.env ?? ToolCallTracker._env ?? null,\n    privacyConfig: opts.privacyConfig ?? ToolCallTracker._privacyConfig ?? null,\n    context: ctx,\n  };\n}\n\nfunction _serializeState(\n  value: unknown,\n  pc: PrivacyConfig | null,\n): Record<string, unknown> | null {\n  if (value == null) return null;\n  if (pc != null) {\n    const mode = pc.contentMode;\n    if (mode === 'metadata_only' || (mode == null && pc.privacyMode))\n      return null;\n  }\n  if (typeof value === 'object' && !Array.isArray(value))\n    return value as Record<string, unknown>;\n  return { value: String(value) };\n}\n\nfunction _wrapObserve<T extends AnyFn>(fn: T, opts: ObserveOptions): T {\n  const spanName = opts.name || fn.name || 'anonymous';\n\n  const wrapper = async function (\n    this: unknown,\n    ...args: unknown[]\n  ): Promise<unknown> {\n    const params = _resolveObserveParams(opts);\n    const ctx = getActiveContext();\n    const ownsSession = ctx == null;\n    const sessionId = ctx?.sessionId ?? randomUUID();\n\n    const runFn = async (): Promise<unknown> => {\n      const inputState = _serializeState(\n        args.length === 1 ? args[0] : args.length > 0 ? { args } : null,\n        params.privacyConfig,\n      );\n\n      const start = performance.now();\n      let isError = false;\n      let errorMessage: string | null = null;\n      let result: unknown = undefined;\n\n      try {\n        result = await fn.apply(this, args);\n        return result;\n      } catch (exc) {\n        isError = true;\n        errorMessage = exc instanceof Error ? exc.message : String(exc);\n        throw exc;\n      } finally {\n        const latencyMs = performance.now() - start;\n        const outputState = _serializeState(result, params.privacyConfig);\n        const activeCtx = getActiveContext();\n\n        if (params.amplitude != null) {\n          try {\n            trackSpan({\n              amplitude: params.amplitude,\n              userId: params.userId,\n              spanName,\n              traceId: activeCtx?.traceId ?? randomUUID(),\n              latencyMs,\n              inputState,\n              outputState,\n              isError,\n              errorMessage,\n              sessionId,\n              agentId: params.agentId || spanName,\n              env: params.env,\n              privacyConfig: params.privacyConfig,\n            });\n          } catch (e) {\n            getLogger().error(\n              `Failed to track @observe span '${spanName}': ${e}`,\n            );\n          }\n\n          if (ownsSession) {\n            try {\n              trackSessionEnd({\n                amplitude: params.amplitude,\n                userId: params.userId,\n                sessionId,\n                env: params.env,\n                agentId: params.agentId || spanName,\n                privacyConfig: params.privacyConfig,\n              });\n            } catch (e) {\n              getLogger().error(\n                `Failed to end @observe session '${spanName}': ${e}`,\n              );\n            }\n          }\n        }\n      }\n    };\n\n    if (ownsSession) {\n      const newCtx = new SessionContext({\n        sessionId,\n        userId: params.userId,\n        agentId: params.agentId || spanName,\n        env: params.env,\n      });\n      return _sessionStorage.run(newCtx, runFn);\n    }\n    return runFn();\n  } as unknown as T;\n\n  Object.defineProperty(wrapper, 'name', {\n    value: fn.name,\n    configurable: true,\n  });\n  return wrapper;\n}\n"],"mappings":";;;;;;;;;;;;;AAwBA,IAAa,kBAAb,MAAa,gBAAgB;CAC3B,OAAO,aAAmC;CAC1C,OAAO,UAAyB;CAChC,OAAO,aAA4B;CACnC,OAAO,WAA0B;CACjC,OAAO,UAAU;CACjB,OAAO,OAAsB;CAC7B,OAAO,WAA0B;CACjC,OAAO,iBAAgC;CACvC,OAAO,iBAAgC;CACvC,OAAO,gBAA+B;CACtC,OAAO,WAA2C;CAClD,OAAO,iBAAuC;CAC9C,OAAO,mBAAmD;CAC1D,OAAO,kBAAkD;CACzD,OAAO,UAA0C;CAEjD,OAAO,aACL,WACA,QACA,OAcI,EAAE,EACA;AACN,kBAAgB,aAAa;AAC7B,kBAAgB,UAAU;AAC1B,kBAAgB,aAAa,KAAK,aAAa;AAC/C,kBAAgB,WAAW,KAAK,WAAW;AAC3C,kBAAgB,UAAU,KAAK,UAAU;AACzC,kBAAgB,OAAO,KAAK,OAAO;AACnC,kBAAgB,WAAW,KAAK,WAAW;AAC3C,kBAAgB,iBAAiB,KAAK,iBAAiB;AACvD,kBAAgB,iBAAiB,KAAK,iBAAiB;AACvD,kBAAgB,gBAAgB,KAAK,gBAAgB;AACrD,kBAAgB,WAAW,KAAK,WAAW;AAC3C,kBAAgB,iBAAiB,KAAK,iBAAiB;AACvD,kBAAgB,mBAAmB,KAAK,mBAAmB;AAC3D,kBAAgB,kBAAkB,KAAK,kBAAkB;AACzD,kBAAgB,UAAU,KAAK,UAAU;;CAG3C,OAAO,QAAc;AACnB,kBAAgB,aAAa;AAC7B,kBAAgB,UAAU;AAC1B,kBAAgB,aAAa;AAC7B,kBAAgB,WAAW;AAC3B,kBAAgB,UAAU;AAC1B,kBAAgB,OAAO;AACvB,kBAAgB,WAAW;AAC3B,kBAAgB,iBAAiB;AACjC,kBAAgB,iBAAiB;AACjC,kBAAgB,gBAAgB;AAChC,kBAAgB,WAAW;AAC3B,kBAAgB,iBAAiB;AACjC,kBAAgB,mBAAmB;AACnC,kBAAgB,kBAAkB;AAClC,kBAAgB,UAAU;;CAG5B,OAAO,eAAwB;AAC7B,SACE,gBAAgB,cAAc,QAAQ,gBAAgB,WAAW;;;AAiDvE,SAAS,sBAAsB,MAAoC;CACjE,MAAM,MAAM,kBAAkB;CAE9B,SAAS,QACP,cACA,YACA,SAAkB,MACT;AACT,MAAI,iBAAiB,UAAa,iBAAiB,KACjD,QAAO;AACT,MAAI,WAAW,UAAa,WAAW,KAAM,QAAO;AACpD,MAAI,eAAe,UAAa,eAAe,KAAM,QAAO;AAC5D,SAAO;;AAGT,QAAO;EACL,WAAW,KAAK,aAAa,gBAAgB,cAAc;EAC3D,QAAQ,QAAQ,KAAK,QAAQ,gBAAgB,SAAS,KAAK,OAAO;EAGlE,WAAW,QACT,KAAK,WACL,gBAAgB,YAChB,KAAK,UACN;EACD,SAAS,QAAQ,KAAK,SAAS,gBAAgB,UAAU,KAAK,QAAQ;EAGtE,QAAQ,KAAK,UAAU,gBAAgB;EACvC,KAAK,QAAQ,KAAK,KAAK,gBAAgB,MAAM,KAAK,IAAI;EACtD,SAAS,QAAQ,KAAK,SAAS,gBAAgB,UAAU,KAAK,QAAQ;EAGtE,eAAe,QACb,KAAK,eACL,gBAAgB,gBAChB,KAAK,cACN;EACD,eAAe,QACb,KAAK,eACL,gBAAgB,gBAChB,KAAK,cACN;EACD,cAAc,QACZ,KAAK,cACL,gBAAgB,eAChB,KAAK,aACN;EACD,SAAS,QACP,KAAK,SACL,gBAAgB,UAChB,KAAK,QACN;EACD,eAAe,QACb,KAAK,eACL,gBAAgB,eACjB;EACD,iBAAiB,QACf,KAAK,iBACL,gBAAgB,iBACjB;EACD,gBAAgB,QACd,KAAK,gBACL,gBAAgB,gBACjB;EACD,QAAQ,QACN,KAAK,QACL,gBAAgB,SAChB,KAAK,OACN;EACF;;AAyCH,SAAgB,KACd,UACA,WAC+D;AAC/D,KAAI,OAAO,aAAa,WACtB,QAAO,UAAU,UAAe,aAAa,EAAE,CAAC;AAElD,SAAyB,OACvB,UAAU,IAAI,SAAS;;AAG3B,SAAS,UAA2B,IAAO,MAAmC;CAC5E,MAAM,WAAW,KAAK,QAAQ,GAAG,QAAQ;CACzC,MAAM,aAAa,KAAK,cAAc,KAAK,UAAU,KAAK,YAAY,GAAG;CAEzE,MAAM,UAAU,eAEd,GAAG,MACe;EAClB,MAAM,IAAI,sBAAsB,KAAK;AAErC,MAAI,EAAE,aAAa,QAAQ,EAAE,UAAU,MAAM;AAC3C,cAAW,CAAC,KACV,SAAS,SAAS,+GACnB;AACD,UAAO,GAAG,MAAM,MAAM,KAAK;;EAG7B,MAAM,YACJ,KAAK,WAAW,KAAK,OAAO,KAAK,OAAO,YAAY,KAAK,MAAM,OAC3D,KAAK,KACL,KAAK,SAAS,IACZ,EAAE,MAAM,GACR;EAER,MAAM,YAAY,YAAY,KAAK;EACnC,IAAI,UAAU;EACd,IAAIA,WAA0B;EAC9B,IAAIC,SAAkB;AAEtB,MAAI;AACF,OAAI,KAAK,aAAa,KACpB,UAAS,MAAM,gBACb,GAAG,KAAK,MAAM,GAAG,KAAK,EACtB,KAAK,WACL,SACD;OAED,UAAS,MAAM,GAAG,MAAM,MAAM,KAAK;AAErC,UAAO;WACA,GAAG;AACV,aAAU;AACV,cAAW,aAAa,QAAQ,EAAE,UAAU,OAAO,EAAE;AACrD,OAAI,KAAK,WAAW,KAClB,KAAI;AACF,SAAK,QAAQ,aAAa,QAAQ,IAAI,IAAI,MAAM,OAAO,EAAE,CAAC,EAAE,SAAS;WAC/D;AAIV,cAAW,CAAC,MAAM,SAAS,SAAS,YAAY,WAAW;AAC3D,SAAM;YACE;GACR,MAAM,YAAY,YAAY,KAAK,GAAG;GAEtC,MAAMC,aAAsC,EAAE;AAC9C,OAAI,cAAc,KAChB,YAAW,+BAA+B;GAG5C,MAAM,iBAAiB;IACrB,GAAI,EAAE,mBAAmB,EAAE;IAC3B,GAAG;IACJ;AAED,OAAI;AACF,kBAAc;KACZ,WAAW,EAAE;KACb,QAAQ,EAAE;KACV;KACA;KACA;KACA,WAAW,EAAE;KACb,SAAS,EAAE;KACX,QAAQ,EAAE,UAAU;KACpB;KACA,YAAY,UAAU,SAAS;KAC/B,cAAc;KACd,KAAK,EAAE;KACP,SAAS,EAAE;KACX,eAAe,EAAE;KACjB,eAAe,EAAE;KACjB,cAAc,EAAE;KAChB,SAAS,EAAE;KACX,iBACE,OAAO,KAAK,WAAW,CAAC,SAAS,IAC7B,iBACA,EAAE;KACR,gBAAgB,EAAE;KAClB,QAAQ,EAAE;KACV,eAAe,EAAE;KAClB,CAAC;YACK,YAAY;AACnB,eAAW,CAAC,MACV,8BAA8B,SAAS,KAAK,aAC7C;;;;AAKP,QAAO,eAAe,SAAS,QAAQ;EACrC,OAAO,GAAG;EACV,cAAc;EACf,CAAC;AACF,QAAO;;AAGT,eAAe,gBACb,IACA,WACA,UACkB;CAClB,IAAIC;CACJ,MAAM,iBAAiB,IAAI,SAAgB,UAAU,WAAW;AAC9D,UAAQ,iBAEJ,uBAAO,IAAI,MAAM,SAAS,SAAS,aAAa,UAAU,YAAY,CAAC,EACzE,UACD;GACD;AAEF,KAAI;AACF,SAAO,MAAM,QAAQ,KAAK,CAAC,IAAI,EAAE,eAAe,CAAC;WACzC;AACR,eAAa,MAAM;;;AA+CvB,SAAgB,QACd,UACA,WACqE;AACrE,KAAI,OAAO,aAAa,WACtB,QAAO,aAAa,UAAe,aAAa,EAAE,CAAC;AAErD,SAAyB,OACvB,aAAa,IAAI,SAAS;;AAY9B,SAAS,sBAAsB,MAA6C;CAC1E,MAAM,MAAM,kBAAkB;AAC9B,QAAO;EACL,WAAW,KAAK,aAAa,gBAAgB,cAAc;EAC3D,QAAQ,KAAK,UAAU,KAAK,UAAU,gBAAgB,WAAW;EACjE,SAAS,KAAK,WAAW,KAAK,WAAW,gBAAgB,YAAY;EACrE,KAAK,KAAK,OAAO,KAAK,OAAO,gBAAgB,QAAQ;EACrD,eAAe,KAAK,iBAAiB,gBAAgB,kBAAkB;EACvE,SAAS;EACV;;AAGH,SAAS,gBACP,OACA,IACgC;AAChC,KAAI,SAAS,KAAM,QAAO;AAC1B,KAAI,MAAM,MAAM;EACd,MAAM,OAAO,GAAG;AAChB,MAAI,SAAS,mBAAoB,QAAQ,QAAQ,GAAG,YAClD,QAAO;;AAEX,KAAI,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM,CACpD,QAAO;AACT,QAAO,EAAE,OAAO,OAAO,MAAM,EAAE;;AAGjC,SAAS,aAA8B,IAAO,MAAyB;CACrE,MAAM,WAAW,KAAK,QAAQ,GAAG,QAAQ;CAEzC,MAAM,UAAU,eAEd,GAAG,MACe;EAClB,MAAM,SAAS,sBAAsB,KAAK;EAC1C,MAAM,MAAM,kBAAkB;EAC9B,MAAM,cAAc,OAAO;EAC3B,MAAM,YAAY,KAAK,aAAa,YAAY;EAEhD,MAAM,QAAQ,YAA8B;GAC1C,MAAM,aAAa,gBACjB,KAAK,WAAW,IAAI,KAAK,KAAK,KAAK,SAAS,IAAI,EAAE,MAAM,GAAG,MAC3D,OAAO,cACR;GAED,MAAM,QAAQ,YAAY,KAAK;GAC/B,IAAI,UAAU;GACd,IAAIC,eAA8B;GAClC,IAAIH,SAAkB;AAEtB,OAAI;AACF,aAAS,MAAM,GAAG,MAAM,MAAM,KAAK;AACnC,WAAO;YACA,KAAK;AACZ,cAAU;AACV,mBAAe,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI;AAC/D,UAAM;aACE;IACR,MAAM,YAAY,YAAY,KAAK,GAAG;IACtC,MAAM,cAAc,gBAAgB,QAAQ,OAAO,cAAc;IACjE,MAAM,YAAY,kBAAkB;AAEpC,QAAI,OAAO,aAAa,MAAM;AAC5B,SAAI;AACF,gBAAU;OACR,WAAW,OAAO;OAClB,QAAQ,OAAO;OACf;OACA,SAAS,WAAW,WAAW,YAAY;OAC3C;OACA;OACA;OACA;OACA;OACA;OACA,SAAS,OAAO,WAAW;OAC3B,KAAK,OAAO;OACZ,eAAe,OAAO;OACvB,CAAC;cACK,GAAG;AACV,iBAAW,CAAC,MACV,kCAAkC,SAAS,KAAK,IACjD;;AAGH,SAAI,YACF,KAAI;AACF,sBAAgB;OACd,WAAW,OAAO;OAClB,QAAQ,OAAO;OACf;OACA,KAAK,OAAO;OACZ,SAAS,OAAO,WAAW;OAC3B,eAAe,OAAO;OACvB,CAAC;cACK,GAAG;AACV,iBAAW,CAAC,MACV,mCAAmC,SAAS,KAAK,IAClD;;;;;AAOX,MAAI,aAAa;GACf,MAAM,SAAS,IAAI,eAAe;IAChC;IACA,QAAQ,OAAO;IACf,SAAS,OAAO,WAAW;IAC3B,KAAK,OAAO;IACb,CAAC;AACF,UAAO,gBAAgB,IAAI,QAAQ,MAAM;;AAE3C,SAAO,OAAO;;AAGhB,QAAO,eAAe,SAAS,QAAQ;EACrC,OAAO,GAAG;EACV,cAAc;EACf,CAAC;AACF,QAAO"}

package/dist/exceptions.d.ts

@@ -0,0 +1,19 @@
+//#region src/exceptions.d.ts
+declare class AmplitudeAIError extends Error {
+  constructor(message: string);
+}
+declare class ConfigurationError extends AmplitudeAIError {
+  constructor(message: string);
+}
+declare class TrackingError extends AmplitudeAIError {
+  constructor(message: string);
+}
+declare class ProviderError extends AmplitudeAIError {
+  constructor(message: string);
+}
+declare class ValidationError extends AmplitudeAIError {
+  constructor(message: string);
+}
+//#endregion
+export { AmplitudeAIError, ConfigurationError, ProviderError, TrackingError, ValidationError };
+//# sourceMappingURL=exceptions.d.ts.map

package/dist/exceptions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"exceptions.d.ts","names":[],"sources":["../src/exceptions.ts"],"sourcesContent":[],"mappings":";cAAa,gBAAA,SAAyB,KAAA;EAAzB,WAAA,CAAA,OAAiB,EAAA,MAAA;AAO9B;AAOa,cAPA,kBAAA,SAA2B,gBAAA,CAOW;EAOtC,WAAA,CAAA,OAAc,EAAA,MAAA;AAO3B;cAda,aAAA,SAAsB,gBAAA;;;cAOtB,aAAA,SAAsB,gBAAA;;;cAOtB,eAAA,SAAwB,gBAAA"}

package/dist/exceptions.js

@@ -0,0 +1,35 @@
+//#region src/exceptions.ts
+var AmplitudeAIError = class extends Error {
+	constructor(message) {
+		super(message);
+		this.name = "AmplitudeAIError";
+	}
+};
+var ConfigurationError = class extends AmplitudeAIError {
+	constructor(message) {
+		super(message);
+		this.name = "ConfigurationError";
+	}
+};
+var TrackingError = class extends AmplitudeAIError {
+	constructor(message) {
+		super(message);
+		this.name = "TrackingError";
+	}
+};
+var ProviderError = class extends AmplitudeAIError {
+	constructor(message) {
+		super(message);
+		this.name = "ProviderError";
+	}
+};
+var ValidationError = class extends AmplitudeAIError {
+	constructor(message) {
+		super(message);
+		this.name = "ValidationError";
+	}
+};
+
+//#endregion
+export { AmplitudeAIError, ConfigurationError, ProviderError, TrackingError, ValidationError };
+//# sourceMappingURL=exceptions.js.map

package/dist/exceptions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"exceptions.js","names":[],"sources":["../src/exceptions.ts"],"sourcesContent":["export class AmplitudeAIError extends Error {\n  constructor(message: string) {\n    super(message);\n    this.name = 'AmplitudeAIError';\n  }\n}\n\nexport class ConfigurationError extends AmplitudeAIError {\n  constructor(message: string) {\n    super(message);\n    this.name = 'ConfigurationError';\n  }\n}\n\nexport class TrackingError extends AmplitudeAIError {\n  constructor(message: string) {\n    super(message);\n    this.name = 'TrackingError';\n  }\n}\n\nexport class ProviderError extends AmplitudeAIError {\n  constructor(message: string) {\n    super(message);\n    this.name = 'ProviderError';\n  }\n}\n\nexport class ValidationError extends AmplitudeAIError {\n  constructor(message: string) {\n    super(message);\n    this.name = 'ValidationError';\n  }\n}\n"],"mappings":";AAAA,IAAa,mBAAb,cAAsC,MAAM;CAC1C,YAAY,SAAiB;AAC3B,QAAM,QAAQ;AACd,OAAK,OAAO;;;AAIhB,IAAa,qBAAb,cAAwC,iBAAiB;CACvD,YAAY,SAAiB;AAC3B,QAAM,QAAQ;AACd,OAAK,OAAO;;;AAIhB,IAAa,gBAAb,cAAmC,iBAAiB;CAClD,YAAY,SAAiB;AAC3B,QAAM,QAAQ;AACd,OAAK,OAAO;;;AAIhB,IAAa,gBAAb,cAAmC,iBAAiB;CAClD,YAAY,SAAiB;AAC3B,QAAM,QAAQ;AACd,OAAK,OAAO;;;AAIhB,IAAa,kBAAb,cAAqC,iBAAiB;CACpD,YAAY,SAAiB;AAC3B,QAAM,QAAQ;AACd,OAAK,OAAO"}

package/dist/index.d.ts

@@ -0,0 +1,37 @@
+import { PrivacyConfig } from "./core/privacy.js";
+import { AIConfig, AIConfigOptions, ContentMode } from "./config.js";
+import { EvidenceQuote, MessageLabel, RubricScore, SessionEnrichments, TopicClassification } from "./core/enrichments.js";
+import { TenantHandle } from "./tenant.js";
+import { AmplitudeAILike, AmplitudeClientLike, AmplitudeEvent, AmplitudeLike, AmplitudeOrAI, AnthropicParams, AnthropicResponse, Attachment, BedrockConverseParams, BedrockConverseResponse, ChatCompletionParams, ChatCompletionResponse, ChatMessage, ContentBlock, MistralChatParams, MistralChatResponse, ToolCallShape, TrackCallOptions, TrackFn, resolveAmplitude } from "./types.js";
+import { AmplitudeAI } from "./client.js";
+import { Session } from "./session.js";
+import { AiMessageOpts, BoundAgent, EmbeddingOpts, ScoreOpts, SessionEndOpts, SessionEnrichmentOpts, SpanOpts, ToolCallOpts, UserMessageOpts } from "./bound-agent.js";
+import { SessionContext, getActiveContext, runWithContext, runWithContextAsync } from "./context.js";
+import { EVENT_AI_RESPONSE, EVENT_EMBEDDING, EVENT_SCORE, EVENT_SESSION_END, EVENT_SESSION_ENRICHMENT, EVENT_SPAN, EVENT_TOOL_CALL, EVENT_USER_MESSAGE, MAX_SERIALIZED_LENGTH, PROP_ABANDONMENT_TURN, PROP_AGENT_CHAIN, PROP_AGENT_ID, PROP_AGENT_VERSION, PROP_ATTACHMENTS, PROP_ATTACHMENT_COUNT, PROP_ATTACHMENT_TYPES, PROP_CACHE_CREATION_TOKENS, PROP_CACHE_READ_TOKENS, PROP_COMMENT, PROP_COMPONENT_TYPE, PROP_CONTEXT, PROP_COST_USD, PROP_CUSTOMER_ORG_ID, PROP_EDITED_MESSAGE_ID, PROP_EMBEDDING_DIMENSIONS, PROP_ENRICHMENTS, PROP_ENV, PROP_ERROR_MESSAGE, PROP_EVALUATION_SOURCE, PROP_FINISH_REASON, PROP_HAS_ATTACHMENTS, PROP_HAS_REASONING, PROP_IDLE_TIMEOUT_MINUTES, PROP_INPUT_STATE, PROP_INPUT_TOKENS, PROP_INVOCATION_ID, PROP_IS_EDIT, PROP_IS_ERROR, PROP_IS_REGENERATION, PROP_IS_STREAMING, PROP_LATENCY_MS, PROP_LOCALE, PROP_MAX_OUTPUT_TOKENS, PROP_MESSAGE_ID, PROP_MESSAGE_LABELS, PROP_MESSAGE_LABEL_MAP, PROP_MODEL_NAME, PROP_MODEL_TIER, PROP_OUTPUT_STATE, PROP_OUTPUT_TOKENS, PROP_PARENT_AGENT_ID, PROP_PARENT_MESSAGE_ID, PROP_PARENT_SPAN_ID, PROP_PROMPT_ID, PROP_PROVIDER, PROP_QUALITY_SCORE, PROP_REASONING_CONTENT, PROP_REASONING_TOKENS, PROP_REQUEST_COMPLEXITY, PROP_ROOT_AGENT_NAME, PROP_RUNTIME, PROP_SCORE_NAME, PROP_SCORE_VALUE, PROP_SDK_VERSION, PROP_SENTIMENT_SCORE, PROP_SESSION_ID, PROP_SESSION_REPLAY_ID, PROP_SPAN_ID, PROP_SPAN_KIND, PROP_SPAN_NAME, PROP_SYSTEM_PROMPT, PROP_SYSTEM_PROMPT_LENGTH, PROP_TARGET_ID, PROP_TARGET_TYPE, PROP_TASK_FAILURE_REASON, PROP_TASK_FAILURE_TYPE, PROP_TEMPERATURE, PROP_TOOL_CALLS, PROP_TOOL_INPUT, PROP_TOOL_NAME, PROP_TOOL_OUTPUT, PROP_TOOL_SUCCESS, PROP_TOP_P, PROP_TOTAL_ATTACHMENT_SIZE, PROP_TOTAL_TOKENS, PROP_TRACE_ID, PROP_TTFB_MS, PROP_TURN_ID, PROP_WAS_CACHED, PROP_WAS_COPIED, SDK_RUNTIME, SDK_VERSION } from "./core/constants.js";
+import { serializeToJsonString, trackAiMessage, trackConversation, trackEmbedding, trackScore, trackSessionEnd, trackSessionEnrichment, trackSpan, trackToolCall, trackUserMessage } from "./core/tracking.js";
+import { ObserveWrapped, ToolCallTracker, ToolWrapped, observe, tool } from "./decorators.js";
+import { AmplitudeAIError, ConfigurationError, ProviderError, TrackingError, ValidationError } from "./exceptions.js";
+import { MockAmplitudeAI } from "./testing.js";
+import { GENERATED_FILES, MCP_PROMPTS, MCP_RESOURCES, MCP_SERVER_NAME, MCP_TOOLS } from "./mcp/contract.js";
+import { extractContext, injectContext } from "./propagation.js";
+import { MiddlewareOptions, createAmplitudeAIMiddleware } from "./middleware.js";
+import { StreamingAccumulator } from "./utils/streaming.js";
+import { ANTHROPIC_AVAILABLE, Anthropic, extractAnthropicContent, extractAnthropicSystemPrompt } from "./providers/anthropic.js";
+import { OPENAI_AVAILABLE, OpenAI, WrappedResponses, extractSystemPrompt } from "./providers/openai.js";
+import { AzureOpenAI } from "./providers/azure-openai.js";
+import { AmplitudeAIWrapError, wrap } from "./wrappers.js";
+import { patch, patchAnthropic, patchAzureOpenAI, patchBedrock, patchGemini, patchMistral, patchOpenAI, patchedProviders, unpatch, unpatchAnthropic, unpatchAzureOpenAI, unpatchBedrock, unpatchGemini, unpatchMistral, unpatchOpenAI } from "./patching.js";
+import { GEMINI_AVAILABLE, Gemini, extractGeminiResponse } from "./providers/gemini.js";
+import { BEDROCK_AVAILABLE, Bedrock, extractBedrockResponse } from "./providers/bedrock.js";
+import { MISTRAL_AVAILABLE, Mistral } from "./providers/mistral.js";
+import { AmplitudeCallbackHandler, createAmplitudeCallback } from "./integrations/langchain.js";
+import { AmplitudeAgentExporter, AmplitudeGenAIExporter, AmplitudeSpanExporter } from "./integrations/opentelemetry.js";
+import { AmplitudeLlamaIndexHandler, createAmplitudeLlamaIndexHandler } from "./integrations/llamaindex.js";
+import { AmplitudeTracingProcessor } from "./integrations/openai-agents.js";
+import { AmplitudeToolLoop } from "./integrations/anthropic-tools.js";
+import { AmplitudeCrewAIHooks } from "./integrations/crewai.js";
+import { inferProviderFromModel } from "./utils/providers.js";
+import { calculateCost, getGenaiPriceLookupCandidates, inferProvider, stripProviderPrefix } from "./utils/costs.js";
+import { countMessageTokens, countTokens, estimateTokens } from "./utils/tokens.js";
+import { TIER_FAST, TIER_REASONING, TIER_STANDARD, inferModelTier } from "./utils/model-tiers.js";
+export { AIConfig, type AIConfigOptions, ANTHROPIC_AVAILABLE, OPENAI_AVAILABLE as AZURE_OPENAI_AVAILABLE, type AiMessageOpts, AmplitudeAI, AmplitudeAIError, type AmplitudeAILike, AmplitudeAIWrapError, AmplitudeAgentExporter, AmplitudeCallbackHandler, type AmplitudeClientLike, AmplitudeCrewAIHooks, type AmplitudeEvent, AmplitudeGenAIExporter, type AmplitudeLike, AmplitudeLlamaIndexHandler, type AmplitudeOrAI, AmplitudeSpanExporter, AmplitudeToolLoop, AmplitudeTracingProcessor, Anthropic, type AnthropicParams, type AnthropicResponse, type Attachment, AzureOpenAI, BEDROCK_AVAILABLE, Bedrock, type BedrockConverseParams, type BedrockConverseResponse, BoundAgent, type ChatCompletionParams, type ChatCompletionResponse, type ChatMessage, ConfigurationError, type ContentBlock, ContentMode, EVENT_AI_RESPONSE, EVENT_EMBEDDING, EVENT_SCORE, EVENT_SESSION_END, EVENT_SESSION_ENRICHMENT, EVENT_SPAN, EVENT_TOOL_CALL, EVENT_USER_MESSAGE, type EmbeddingOpts, EvidenceQuote, GEMINI_AVAILABLE, GENERATED_FILES, Gemini, MAX_SERIALIZED_LENGTH, MCP_PROMPTS, MCP_RESOURCES, MCP_SERVER_NAME, MCP_TOOLS, MISTRAL_AVAILABLE, MessageLabel, type MiddlewareOptions, Mistral, type MistralChatParams, type MistralChatResponse, MockAmplitudeAI, OPENAI_AVAILABLE, type ObserveWrapped, OpenAI, PROP_ABANDONMENT_TURN, PROP_AGENT_CHAIN, PROP_AGENT_ID, PROP_AGENT_VERSION, PROP_ATTACHMENTS, PROP_ATTACHMENT_COUNT, PROP_ATTACHMENT_TYPES, PROP_CACHE_CREATION_TOKENS, PROP_CACHE_READ_TOKENS, PROP_COMMENT, PROP_COMPONENT_TYPE, PROP_CONTEXT, PROP_COST_USD, PROP_CUSTOMER_ORG_ID, PROP_EDITED_MESSAGE_ID, PROP_EMBEDDING_DIMENSIONS, PROP_ENRICHMENTS, PROP_ENV, PROP_ERROR_MESSAGE, PROP_EVALUATION_SOURCE, PROP_FINISH_REASON, PROP_HAS_ATTACHMENTS, PROP_HAS_REASONING, PROP_IDLE_TIMEOUT_MINUTES, PROP_INPUT_STATE, PROP_INPUT_TOKENS, PROP_INVOCATION_ID, PROP_IS_EDIT, PROP_IS_ERROR, PROP_IS_REGENERATION, PROP_IS_STREAMING, PROP_LATENCY_MS, PROP_LOCALE, PROP_MAX_OUTPUT_TOKENS, PROP_MESSAGE_ID, PROP_MESSAGE_LABELS, PROP_MESSAGE_LABEL_MAP, PROP_MODEL_NAME, PROP_MODEL_TIER, PROP_OUTPUT_STATE, PROP_OUTPUT_TOKENS, PROP_PARENT_AGENT_ID, PROP_PARENT_MESSAGE_ID, PROP_PARENT_SPAN_ID, PROP_PROMPT_ID, PROP_PROVIDER, PROP_QUALITY_SCORE, PROP_REASONING_CONTENT, PROP_REASONING_TOKENS, PROP_REQUEST_COMPLEXITY, PROP_ROOT_AGENT_NAME, PROP_RUNTIME, PROP_SCORE_NAME, PROP_SCORE_VALUE, PROP_SDK_VERSION, PROP_SENTIMENT_SCORE, PROP_SESSION_ID, PROP_SESSION_REPLAY_ID, PROP_SPAN_ID, PROP_SPAN_KIND, PROP_SPAN_NAME, PROP_SYSTEM_PROMPT, PROP_SYSTEM_PROMPT_LENGTH, PROP_TARGET_ID, PROP_TARGET_TYPE, PROP_TASK_FAILURE_REASON, PROP_TASK_FAILURE_TYPE, PROP_TEMPERATURE, PROP_TOOL_CALLS, PROP_TOOL_INPUT, PROP_TOOL_NAME, PROP_TOOL_OUTPUT, PROP_TOOL_SUCCESS, PROP_TOP_P, PROP_TOTAL_ATTACHMENT_SIZE, PROP_TOTAL_TOKENS, PROP_TRACE_ID, PROP_TTFB_MS, PROP_TURN_ID, PROP_WAS_CACHED, PROP_WAS_COPIED, PrivacyConfig, ProviderError, RubricScore, SDK_RUNTIME, SDK_VERSION, type ScoreOpts, Session, SessionContext, type SessionEndOpts, type SessionEnrichmentOpts, SessionEnrichments, type SpanOpts, StreamingAccumulator, TIER_FAST, TIER_REASONING, TIER_STANDARD, TenantHandle, type ToolCallOpts, type ToolCallShape, ToolCallTracker, type ToolWrapped, TopicClassification, type TrackCallOptions, type TrackFn, TrackingError, type UserMessageOpts, ValidationError, WrappedResponses, calculateCost, countMessageTokens, countTokens, createAmplitudeAIMiddleware, createAmplitudeCallback, createAmplitudeLlamaIndexHandler, estimateTokens, extractAnthropicContent, extractAnthropicSystemPrompt, extractBedrockResponse, extractContext, extractGeminiResponse, extractSystemPrompt, getActiveContext, getGenaiPriceLookupCandidates, inferModelTier, inferProvider, inferProviderFromModel, injectContext, observe, patch, patchAnthropic, patchAzureOpenAI, patchBedrock, patchGemini, patchMistral, patchOpenAI, patchedProviders, resolveAmplitude, runWithContext, runWithContextAsync, serializeToJsonString, stripProviderPrefix, tool, trackAiMessage, trackConversation, trackEmbedding, trackScore, trackSessionEnd, trackSessionEnrichment, trackSpan, trackToolCall, trackUserMessage, unpatch, unpatchAnthropic, unpatchAzureOpenAI, unpatchBedrock, unpatchGemini, unpatchMistral, unpatchOpenAI, wrap };

package/dist/index.js

@@ -0,0 +1,38 @@
+import { AmplitudeAIError, ConfigurationError, ProviderError, TrackingError, ValidationError } from "./exceptions.js";
+import { TIER_FAST, TIER_REASONING, TIER_STANDARD, inferModelTier } from "./utils/model-tiers.js";
+import { EVENT_AI_RESPONSE, EVENT_EMBEDDING, EVENT_SCORE, EVENT_SESSION_END, EVENT_SESSION_ENRICHMENT, EVENT_SPAN, EVENT_TOOL_CALL, EVENT_USER_MESSAGE, MAX_SERIALIZED_LENGTH, PROP_ABANDONMENT_TURN, PROP_AGENT_CHAIN, PROP_AGENT_ID, PROP_AGENT_VERSION, PROP_ATTACHMENTS, PROP_ATTACHMENT_COUNT, PROP_ATTACHMENT_TYPES, PROP_CACHE_CREATION_TOKENS, PROP_CACHE_READ_TOKENS, PROP_COMMENT, PROP_COMPONENT_TYPE, PROP_CONTEXT, PROP_COST_USD, PROP_CUSTOMER_ORG_ID, PROP_EDITED_MESSAGE_ID, PROP_EMBEDDING_DIMENSIONS, PROP_ENRICHMENTS, PROP_ENV, PROP_ERROR_MESSAGE, PROP_EVALUATION_SOURCE, PROP_FINISH_REASON, PROP_HAS_ATTACHMENTS, PROP_HAS_REASONING, PROP_IDLE_TIMEOUT_MINUTES, PROP_INPUT_STATE, PROP_INPUT_TOKENS, PROP_INVOCATION_ID, PROP_IS_EDIT, PROP_IS_ERROR, PROP_IS_REGENERATION, PROP_IS_STREAMING, PROP_LATENCY_MS, PROP_LOCALE, PROP_MAX_OUTPUT_TOKENS, PROP_MESSAGE_ID, PROP_MESSAGE_LABELS, PROP_MESSAGE_LABEL_MAP, PROP_MODEL_NAME, PROP_MODEL_TIER, PROP_OUTPUT_STATE, PROP_OUTPUT_TOKENS, PROP_PARENT_AGENT_ID, PROP_PARENT_MESSAGE_ID, PROP_PARENT_SPAN_ID, PROP_PROMPT_ID, PROP_PROVIDER, PROP_QUALITY_SCORE, PROP_REASONING_CONTENT, PROP_REASONING_TOKENS, PROP_REQUEST_COMPLEXITY, PROP_ROOT_AGENT_NAME, PROP_RUNTIME, PROP_SCORE_NAME, PROP_SCORE_VALUE, PROP_SDK_VERSION, PROP_SENTIMENT_SCORE, PROP_SESSION_ID, PROP_SESSION_REPLAY_ID, PROP_SPAN_ID, PROP_SPAN_KIND, PROP_SPAN_NAME, PROP_SYSTEM_PROMPT, PROP_SYSTEM_PROMPT_LENGTH, PROP_TARGET_ID, PROP_TARGET_TYPE, PROP_TASK_FAILURE_REASON, PROP_TASK_FAILURE_TYPE, PROP_TEMPERATURE, PROP_TOOL_CALLS, PROP_TOOL_INPUT, PROP_TOOL_NAME, PROP_TOOL_OUTPUT, PROP_TOOL_SUCCESS, PROP_TOP_P, PROP_TOTAL_ATTACHMENT_SIZE, PROP_TOTAL_TOKENS, PROP_TRACE_ID, PROP_TTFB_MS, PROP_TURN_ID, PROP_WAS_CACHED, PROP_WAS_COPIED, SDK_RUNTIME, SDK_VERSION } from "./core/constants.js";
+import { PrivacyConfig } from "./core/privacy.js";
+import { serializeToJsonString, trackAiMessage, trackConversation, trackEmbedding, trackScore, trackSessionEnd, trackSessionEnrichment, trackSpan, trackToolCall, trackUserMessage } from "./core/tracking.js";
+import { SessionContext, getActiveContext, runWithContext, runWithContextAsync } from "./context.js";
+import { Session } from "./session.js";
+import { BoundAgent } from "./bound-agent.js";
+import { AIConfig, ContentMode } from "./config.js";
+import { extractContext, injectContext } from "./propagation.js";
+import { inferProviderFromModel } from "./utils/providers.js";
+import { calculateCost, getGenaiPriceLookupCandidates, inferProvider, stripProviderPrefix } from "./utils/costs.js";
+import { StreamingAccumulator } from "./utils/streaming.js";
+import { resolveAmplitude } from "./types.js";
+import { ANTHROPIC_AVAILABLE, Anthropic, extractAnthropicContent, extractAnthropicSystemPrompt } from "./providers/anthropic.js";
+import { BEDROCK_AVAILABLE, Bedrock, extractBedrockResponse } from "./providers/bedrock.js";
+import { GEMINI_AVAILABLE, Gemini, extractGeminiResponse } from "./providers/gemini.js";
+import { MISTRAL_AVAILABLE, Mistral } from "./providers/mistral.js";
+import { OPENAI_AVAILABLE, OpenAI, WrappedResponses, extractSystemPrompt } from "./providers/openai.js";
+import { patch, patchAnthropic, patchAzureOpenAI, patchBedrock, patchGemini, patchMistral, patchOpenAI, patchedProviders, unpatch, unpatchAnthropic, unpatchAzureOpenAI, unpatchBedrock, unpatchGemini, unpatchMistral, unpatchOpenAI } from "./patching.js";
+import { TenantHandle } from "./tenant.js";
+import { AmplitudeAI } from "./client.js";
+import { ToolCallTracker, observe, tool } from "./decorators.js";
+import { EvidenceQuote, MessageLabel, RubricScore, SessionEnrichments, TopicClassification } from "./core/enrichments.js";
+import { MockAmplitudeAI } from "./testing.js";
+import { GENERATED_FILES, MCP_PROMPTS, MCP_RESOURCES, MCP_SERVER_NAME, MCP_TOOLS } from "./mcp/contract.js";
+import { createAmplitudeAIMiddleware } from "./middleware.js";
+import { AzureOpenAI } from "./providers/azure-openai.js";
+import { AmplitudeAIWrapError, wrap } from "./wrappers.js";
+import { AmplitudeCallbackHandler, createAmplitudeCallback } from "./integrations/langchain.js";
+import { AmplitudeAgentExporter, AmplitudeGenAIExporter, AmplitudeSpanExporter } from "./integrations/opentelemetry.js";
+import { AmplitudeLlamaIndexHandler, createAmplitudeLlamaIndexHandler } from "./integrations/llamaindex.js";
+import { AmplitudeTracingProcessor } from "./integrations/openai-agents.js";
+import { AmplitudeToolLoop } from "./integrations/anthropic-tools.js";
+import { AmplitudeCrewAIHooks } from "./integrations/crewai.js";
+import { countMessageTokens, countTokens, estimateTokens } from "./utils/tokens.js";
+
+export { AIConfig, ANTHROPIC_AVAILABLE, OPENAI_AVAILABLE as AZURE_OPENAI_AVAILABLE, AmplitudeAI, AmplitudeAIError, AmplitudeAIWrapError, AmplitudeAgentExporter, AmplitudeCallbackHandler, AmplitudeCrewAIHooks, AmplitudeGenAIExporter, AmplitudeLlamaIndexHandler, AmplitudeSpanExporter, AmplitudeToolLoop, AmplitudeTracingProcessor, Anthropic, AzureOpenAI, BEDROCK_AVAILABLE, Bedrock, BoundAgent, ConfigurationError, ContentMode, EVENT_AI_RESPONSE, EVENT_EMBEDDING, EVENT_SCORE, EVENT_SESSION_END, EVENT_SESSION_ENRICHMENT, EVENT_SPAN, EVENT_TOOL_CALL, EVENT_USER_MESSAGE, EvidenceQuote, GEMINI_AVAILABLE, GENERATED_FILES, Gemini, MAX_SERIALIZED_LENGTH, MCP_PROMPTS, MCP_RESOURCES, MCP_SERVER_NAME, MCP_TOOLS, MISTRAL_AVAILABLE, MessageLabel, Mistral, MockAmplitudeAI, OPENAI_AVAILABLE, OpenAI, PROP_ABANDONMENT_TURN, PROP_AGENT_CHAIN, PROP_AGENT_ID, PROP_AGENT_VERSION, PROP_ATTACHMENTS, PROP_ATTACHMENT_COUNT, PROP_ATTACHMENT_TYPES, PROP_CACHE_CREATION_TOKENS, PROP_CACHE_READ_TOKENS, PROP_COMMENT, PROP_COMPONENT_TYPE, PROP_CONTEXT, PROP_COST_USD, PROP_CUSTOMER_ORG_ID, PROP_EDITED_MESSAGE_ID, PROP_EMBEDDING_DIMENSIONS, PROP_ENRICHMENTS, PROP_ENV, PROP_ERROR_MESSAGE, PROP_EVALUATION_SOURCE, PROP_FINISH_REASON, PROP_HAS_ATTACHMENTS, PROP_HAS_REASONING, PROP_IDLE_TIMEOUT_MINUTES, PROP_INPUT_STATE, PROP_INPUT_TOKENS, PROP_INVOCATION_ID, PROP_IS_EDIT, PROP_IS_ERROR, PROP_IS_REGENERATION, PROP_IS_STREAMING, PROP_LATENCY_MS, PROP_LOCALE, PROP_MAX_OUTPUT_TOKENS, PROP_MESSAGE_ID, PROP_MESSAGE_LABELS, PROP_MESSAGE_LABEL_MAP, PROP_MODEL_NAME, PROP_MODEL_TIER, PROP_OUTPUT_STATE, PROP_OUTPUT_TOKENS, PROP_PARENT_AGENT_ID, PROP_PARENT_MESSAGE_ID, PROP_PARENT_SPAN_ID, PROP_PROMPT_ID, PROP_PROVIDER, PROP_QUALITY_SCORE, PROP_REASONING_CONTENT, PROP_REASONING_TOKENS, PROP_REQUEST_COMPLEXITY, PROP_ROOT_AGENT_NAME, PROP_RUNTIME, PROP_SCORE_NAME, PROP_SCORE_VALUE, PROP_SDK_VERSION, PROP_SENTIMENT_SCORE, PROP_SESSION_ID, PROP_SESSION_REPLAY_ID, PROP_SPAN_ID, PROP_SPAN_KIND, PROP_SPAN_NAME, PROP_SYSTEM_PROMPT, PROP_SYSTEM_PROMPT_LENGTH, PROP_TARGET_ID, PROP_TARGET_TYPE, PROP_TASK_FAILURE_REASON, PROP_TASK_FAILURE_TYPE, PROP_TEMPERATURE, PROP_TOOL_CALLS, PROP_TOOL_INPUT, PROP_TOOL_NAME, PROP_TOOL_OUTPUT, PROP_TOOL_SUCCESS, PROP_TOP_P, PROP_TOTAL_ATTACHMENT_SIZE, PROP_TOTAL_TOKENS, PROP_TRACE_ID, PROP_TTFB_MS, PROP_TURN_ID, PROP_WAS_CACHED, PROP_WAS_COPIED, PrivacyConfig, ProviderError, RubricScore, SDK_RUNTIME, SDK_VERSION, Session, SessionContext, SessionEnrichments, StreamingAccumulator, TIER_FAST, TIER_REASONING, TIER_STANDARD, TenantHandle, ToolCallTracker, TopicClassification, TrackingError, ValidationError, WrappedResponses, calculateCost, countMessageTokens, countTokens, createAmplitudeAIMiddleware, createAmplitudeCallback, createAmplitudeLlamaIndexHandler, estimateTokens, extractAnthropicContent, extractAnthropicSystemPrompt, extractBedrockResponse, extractContext, extractGeminiResponse, extractSystemPrompt, getActiveContext, getGenaiPriceLookupCandidates, inferModelTier, inferProvider, inferProviderFromModel, injectContext, observe, patch, patchAnthropic, patchAzureOpenAI, patchBedrock, patchGemini, patchMistral, patchOpenAI, patchedProviders, resolveAmplitude, runWithContext, runWithContextAsync, serializeToJsonString, stripProviderPrefix, tool, trackAiMessage, trackConversation, trackEmbedding, trackScore, trackSessionEnd, trackSessionEnrichment, trackSpan, trackToolCall, trackUserMessage, unpatch, unpatchAnthropic, unpatchAzureOpenAI, unpatchBedrock, unpatchGemini, unpatchMistral, unpatchOpenAI, wrap };