@amplitude/ai 0.1.0

package/dist/integrations/anthropic-tools.d.ts

@@ -0,0 +1,38 @@
+import { AmplitudeAI } from "../client.js";
+
+//#region src/integrations/anthropic-tools.d.ts
+
+interface ToolLoopOptions {
+  amplitudeAI: AmplitudeAI;
+  userId?: string;
+  sessionId?: string;
+  agentId?: string;
+  env?: string;
+  maxTurns?: number;
+}
+declare class AmplitudeToolLoop {
+  private _ai;
+  private _userId;
+  private _sessionId;
+  private _agentId;
+  private _env;
+  private _maxTurns;
+  constructor(options: ToolLoopOptions);
+  /**
+   * Run a tool loop with the Anthropic API.
+   *
+   * This method orchestrates multi-turn tool_use conversations,
+   * tracking each AI response and tool call along the way.
+   */
+  run(options: {
+    client: unknown;
+    model: string;
+    messages: Array<Record<string, unknown>>;
+    tools: Array<Record<string, unknown>>;
+    system?: string;
+    toolExecutor: (name: string, input: Record<string, unknown>) => Promise<unknown>;
+  }): Promise<Array<Record<string, unknown>>>;
+}
+//#endregion
+export { AmplitudeToolLoop, ToolLoopOptions };
+//# sourceMappingURL=anthropic-tools.d.ts.map

package/dist/integrations/anthropic-tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic-tools.d.ts","names":[],"sources":["../../src/integrations/anthropic-tools.ts"],"sourcesContent":[],"mappings":";;;;AA6CoB,UAnCH,eAAA,CAmCG;EAAN,WAAA,EAlCC,WAkCD;EACG,MAAA,CAAA,EAAA,MAAA;EAAN,SAAA,CAAA,EAAA,MAAA;EAIE,OAAA,CAAA,EAAA,MAAA;EACJ,GAAA,CAAA,EAAA,MAAA;EACW,QAAA,CAAA,EAAA,MAAA;;AAAd,cAjCO,iBAAA,CAiCP;EAAO,QAAA,GAAA;;;;;;uBAzBU;;;;;;;;;;cAkBT,MAAM;WACT,MAAM;;wCAIJ,4BACJ;MACH,QAAQ,MAAM"}

package/dist/integrations/anthropic-tools.js

@@ -0,0 +1,175 @@
+import { getActiveContext } from "../context.js";
+import { calculateCost } from "../utils/costs.js";
+
+//#region src/integrations/anthropic-tools.ts
+var AmplitudeToolLoop = class {
+	_ai;
+	_userId;
+	_sessionId;
+	_agentId;
+	_env;
+	_maxTurns;
+	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;
+		this._maxTurns = options.maxTurns ?? 10;
+	}
+	/**
+	* Run a tool loop with the Anthropic API.
+	*
+	* This method orchestrates multi-turn tool_use conversations,
+	* tracking each AI response and tool call along the way.
+	*/
+	async run(options) {
+		const ctx = getActiveContext();
+		const userId = this._userId ?? ctx?.userId ?? "unknown";
+		const sessionId = this._sessionId ?? ctx?.sessionId ?? "tool-loop-session";
+		const agentId = this._agentId ?? ctx?.agentId ?? void 0;
+		const env = this._env ?? ctx?.env ?? void 0;
+		const messages = [...options.messages];
+		const allResponses = [];
+		let currentTurnId = 1;
+		for (const msg of messages) {
+			const userText = _extractUserText(msg);
+			if (msg.role === "user" && userText.length > 0) {
+				this._ai.trackUserMessage({
+					userId,
+					content: userText,
+					sessionId,
+					agentId,
+					env,
+					turnId: currentTurnId
+				});
+				currentTurnId += 1;
+			}
+		}
+		for (let turn = 0; turn < this._maxTurns; turn++) {
+			const messagesApi = options.client.messages;
+			const createFn = messagesApi.create;
+			const startTime = performance.now();
+			const response = await createFn.call(messagesApi, {
+				model: options.model,
+				messages,
+				tools: options.tools,
+				system: options.system
+			});
+			const latencyMs = performance.now() - startTime;
+			allResponses.push(response);
+			const content = _normalizeContentBlocks(response.content);
+			const responseText = _extractAssistantText(content);
+			const toolUseBlocks = content.filter((b) => b.type === "tool_use");
+			const usage = response.usage;
+			const cacheRead = usage?.cache_read_input_tokens ?? 0;
+			const cacheCreation = usage?.cache_creation_input_tokens ?? 0;
+			const rawInput = usage?.input_tokens ?? 0;
+			const normalizedInput = cacheRead || cacheCreation ? rawInput + cacheRead + cacheCreation : rawInput;
+			let costUsd;
+			if (usage?.input_tokens != null && usage?.output_tokens != null) try {
+				const cost = calculateCost({
+					modelName: options.model,
+					inputTokens: normalizedInput,
+					outputTokens: usage.output_tokens,
+					cacheReadInputTokens: cacheRead,
+					cacheCreationInputTokens: cacheCreation
+				});
+				if (cost > 0) costUsd = cost;
+			} catch {}
+			this._ai.trackAiMessage({
+				userId,
+				content: responseText,
+				sessionId,
+				model: options.model,
+				provider: "anthropic",
+				latencyMs,
+				turnId: currentTurnId,
+				agentId,
+				env,
+				inputTokens: normalizedInput || void 0,
+				outputTokens: usage?.output_tokens,
+				cacheReadTokens: cacheRead || void 0,
+				cacheCreationTokens: cacheCreation || void 0,
+				totalCostUsd: costUsd
+			});
+			currentTurnId += 1;
+			if (response.stop_reason !== "tool_use" || toolUseBlocks.length === 0) break;
+			messages.push({
+				role: "assistant",
+				content
+			});
+			const toolResults = [];
+			for (const toolUse of toolUseBlocks) {
+				const toolName = String(toolUse.name);
+				const toolInput = toolUse.input;
+				const toolStartTime = performance.now();
+				try {
+					const output = await options.toolExecutor(toolName, toolInput);
+					const toolLatencyMs = performance.now() - toolStartTime;
+					this._ai.trackToolCall({
+						userId,
+						toolName,
+						latencyMs: toolLatencyMs,
+						success: true,
+						sessionId,
+						agentId,
+						env,
+						input: toolInput,
+						output: output == null ? void 0 : String(output)
+					});
+					toolResults.push({
+						type: "tool_result",
+						tool_use_id: toolUse.id,
+						content: output == null ? "" : String(output)
+					});
+				} catch (error) {
+					const toolLatencyMs = performance.now() - toolStartTime;
+					this._ai.trackToolCall({
+						userId,
+						toolName,
+						latencyMs: toolLatencyMs,
+						success: false,
+						sessionId,
+						agentId,
+						env,
+						input: toolInput,
+						errorMessage: error instanceof Error ? error.message : String(error)
+					});
+					toolResults.push({
+						type: "tool_result",
+						tool_use_id: toolUse.id,
+						content: `Error: ${error instanceof Error ? error.message : String(error)}`,
+						is_error: true
+					});
+				}
+			}
+			messages.push({
+				role: "user",
+				content: toolResults
+			});
+		}
+		return allResponses;
+	}
+};
+function _extractUserText(msg) {
+	if (typeof msg.content === "string") return msg.content;
+	if (!Array.isArray(msg.content)) return "";
+	return msg.content.map((block) => {
+		if (typeof block === "string") return block;
+		const item = block;
+		return typeof item.text === "string" ? item.text : "";
+	}).join("");
+}
+function _extractAssistantText(content) {
+	if (!Array.isArray(content)) return "";
+	return content.filter((b) => b.type === "text").map((b) => String(b.text ?? "")).join("");
+}
+function _normalizeContentBlocks(value) {
+	if (!Array.isArray(value)) return [];
+	return value.filter((v) => v != null && typeof v === "object");
+}
+
+//#endregion
+export { AmplitudeToolLoop };
+//# sourceMappingURL=anthropic-tools.js.map

package/dist/integrations/anthropic-tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic-tools.js","names":["allResponses: Array<Record<string, unknown>>","costUsd: number | undefined","toolResults: Array<Record<string, unknown>>"],"sources":["../../src/integrations/anthropic-tools.ts"],"sourcesContent":["/**\n * Anthropic tool loop integration.\n *\n * Runs Anthropic's multi-turn tool_use loop and tracks each turn.\n */\n\nimport type { AmplitudeAI } from '../client.js';\nimport { getActiveContext } from '../context.js';\nimport { calculateCost } from '../utils/costs.js';\n\nexport interface ToolLoopOptions {\n  amplitudeAI: AmplitudeAI;\n  userId?: string;\n  sessionId?: string;\n  agentId?: string;\n  env?: string;\n  maxTurns?: number;\n}\n\nexport class AmplitudeToolLoop {\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 _maxTurns: number;\n\n  constructor(options: ToolLoopOptions) {\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    this._maxTurns = options.maxTurns ?? 10;\n  }\n\n  /**\n   * Run a tool loop with the Anthropic API.\n   *\n   * This method orchestrates multi-turn tool_use conversations,\n   * tracking each AI response and tool call along the way.\n   */\n  async run(options: {\n    client: unknown;\n    model: string;\n    messages: Array<Record<string, unknown>>;\n    tools: Array<Record<string, unknown>>;\n    system?: string;\n    toolExecutor: (\n      name: string,\n      input: Record<string, unknown>,\n    ) => Promise<unknown>;\n  }): Promise<Array<Record<string, unknown>>> {\n    const ctx = getActiveContext();\n    const userId = this._userId ?? ctx?.userId ?? 'unknown';\n    const sessionId = this._sessionId ?? ctx?.sessionId ?? 'tool-loop-session';\n    const agentId = this._agentId ?? ctx?.agentId ?? undefined;\n    const env = this._env ?? ctx?.env ?? undefined;\n\n    const messages = [...options.messages];\n    const allResponses: Array<Record<string, unknown>> = [];\n    let currentTurnId = 1;\n\n    for (const msg of messages) {\n      const userText = _extractUserText(msg);\n      if (msg.role === 'user' && userText.length > 0) {\n        this._ai.trackUserMessage({\n          userId,\n          content: userText,\n          sessionId,\n          agentId,\n          env,\n          turnId: currentTurnId,\n        });\n        currentTurnId += 1;\n      }\n    }\n\n    for (let turn = 0; turn < this._maxTurns; turn++) {\n      const clientObj = options.client as Record<string, unknown>;\n      const messagesApi = clientObj.messages as Record<string, unknown>;\n      const createFn = messagesApi.create as (\n        ...args: unknown[]\n      ) => Promise<unknown>;\n\n      const startTime = performance.now();\n      const response = (await createFn.call(messagesApi, {\n        model: options.model,\n        messages,\n        tools: options.tools,\n        system: options.system,\n      })) as Record<string, unknown>;\n\n      const latencyMs = performance.now() - startTime;\n      allResponses.push(response);\n\n      const content = _normalizeContentBlocks(response.content);\n      const responseText = _extractAssistantText(content);\n      const toolUseBlocks = content.filter((b) => b.type === 'tool_use');\n      const usage = response.usage as Record<string, number> | undefined;\n\n      const cacheRead = usage?.cache_read_input_tokens ?? 0;\n      const cacheCreation = usage?.cache_creation_input_tokens ?? 0;\n      const rawInput = usage?.input_tokens ?? 0;\n      const normalizedInput =\n        cacheRead || cacheCreation\n          ? rawInput + cacheRead + cacheCreation\n          : rawInput;\n\n      let costUsd: number | undefined;\n      if (usage?.input_tokens != null && usage?.output_tokens != null) {\n        try {\n          const cost = calculateCost({\n            modelName: options.model,\n            inputTokens: normalizedInput,\n            outputTokens: usage.output_tokens,\n            cacheReadInputTokens: cacheRead,\n            cacheCreationInputTokens: cacheCreation,\n          });\n          if (cost > 0) costUsd = cost;\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      this._ai.trackAiMessage({\n        userId,\n        content: responseText,\n        sessionId,\n        model: options.model,\n        provider: 'anthropic',\n        latencyMs,\n        turnId: currentTurnId,\n        agentId,\n        env,\n        inputTokens: normalizedInput || undefined,\n        outputTokens: usage?.output_tokens,\n        cacheReadTokens: cacheRead || undefined,\n        cacheCreationTokens: cacheCreation || undefined,\n        totalCostUsd: costUsd,\n      });\n      currentTurnId += 1;\n\n      if (response.stop_reason !== 'tool_use' || toolUseBlocks.length === 0) {\n        break;\n      }\n\n      messages.push({ role: 'assistant', content });\n\n      const toolResults: Array<Record<string, unknown>> = [];\n      for (const toolUse of toolUseBlocks) {\n        const toolName = String(toolUse.name);\n        const toolInput = toolUse.input as Record<string, unknown>;\n        const toolStartTime = performance.now();\n\n        try {\n          const output = await options.toolExecutor(toolName, toolInput);\n          const toolLatencyMs = performance.now() - toolStartTime;\n\n          this._ai.trackToolCall({\n            userId,\n            toolName,\n            latencyMs: toolLatencyMs,\n            success: true,\n            sessionId,\n            agentId,\n            env,\n            input: toolInput,\n            output: output == null ? undefined : String(output),\n          });\n\n          toolResults.push({\n            type: 'tool_result',\n            tool_use_id: toolUse.id,\n            content: output == null ? '' : String(output),\n          });\n        } catch (error) {\n          const toolLatencyMs = performance.now() - toolStartTime;\n\n          this._ai.trackToolCall({\n            userId,\n            toolName,\n            latencyMs: toolLatencyMs,\n            success: false,\n            sessionId,\n            agentId,\n            env,\n            input: toolInput,\n            errorMessage:\n              error instanceof Error ? error.message : String(error),\n          });\n\n          toolResults.push({\n            type: 'tool_result',\n            tool_use_id: toolUse.id,\n            content: `Error: ${error instanceof Error ? error.message : String(error)}`,\n            is_error: true,\n          });\n        }\n      }\n\n      messages.push({ role: 'user', content: toolResults });\n    }\n\n    return allResponses;\n  }\n}\n\nfunction _extractUserText(msg: Record<string, unknown>): string {\n  if (typeof msg.content === 'string') return msg.content;\n  if (!Array.isArray(msg.content)) return '';\n  return msg.content\n    .map((block) => {\n      if (typeof block === 'string') return block;\n      const item = block as Record<string, unknown>;\n      return typeof item.text === 'string' ? item.text : '';\n    })\n    .join('');\n}\n\nfunction _extractAssistantText(\n  content: Array<Record<string, unknown>> | undefined,\n): string {\n  if (!Array.isArray(content)) return '';\n  return content\n    .filter((b) => b.type === 'text')\n    .map((b) => String(b.text ?? ''))\n    .join('');\n}\n\nfunction _normalizeContentBlocks(\n  value: unknown,\n): Array<Record<string, unknown>> {\n  if (!Array.isArray(value)) return [];\n  return value.filter((v) => v != null && typeof v === 'object') as Array<\n    Record<string, unknown>\n  >;\n}\n"],"mappings":";;;;AAmBA,IAAa,oBAAb,MAA+B;CAC7B,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CAER,YAAY,SAA0B;AACpC,OAAK,MAAM,QAAQ;AACnB,OAAK,UAAU,QAAQ,UAAU;AACjC,OAAK,aAAa,QAAQ,aAAa;AACvC,OAAK,WAAW,QAAQ,WAAW;AACnC,OAAK,OAAO,QAAQ,OAAO;AAC3B,OAAK,YAAY,QAAQ,YAAY;;;;;;;;CASvC,MAAM,IAAI,SAUkC;EAC1C,MAAM,MAAM,kBAAkB;EAC9B,MAAM,SAAS,KAAK,WAAW,KAAK,UAAU;EAC9C,MAAM,YAAY,KAAK,cAAc,KAAK,aAAa;EACvD,MAAM,UAAU,KAAK,YAAY,KAAK,WAAW;EACjD,MAAM,MAAM,KAAK,QAAQ,KAAK,OAAO;EAErC,MAAM,WAAW,CAAC,GAAG,QAAQ,SAAS;EACtC,MAAMA,eAA+C,EAAE;EACvD,IAAI,gBAAgB;AAEpB,OAAK,MAAM,OAAO,UAAU;GAC1B,MAAM,WAAW,iBAAiB,IAAI;AACtC,OAAI,IAAI,SAAS,UAAU,SAAS,SAAS,GAAG;AAC9C,SAAK,IAAI,iBAAiB;KACxB;KACA,SAAS;KACT;KACA;KACA;KACA,QAAQ;KACT,CAAC;AACF,qBAAiB;;;AAIrB,OAAK,IAAI,OAAO,GAAG,OAAO,KAAK,WAAW,QAAQ;GAEhD,MAAM,cADY,QAAQ,OACI;GAC9B,MAAM,WAAW,YAAY;GAI7B,MAAM,YAAY,YAAY,KAAK;GACnC,MAAM,WAAY,MAAM,SAAS,KAAK,aAAa;IACjD,OAAO,QAAQ;IACf;IACA,OAAO,QAAQ;IACf,QAAQ,QAAQ;IACjB,CAAC;GAEF,MAAM,YAAY,YAAY,KAAK,GAAG;AACtC,gBAAa,KAAK,SAAS;GAE3B,MAAM,UAAU,wBAAwB,SAAS,QAAQ;GACzD,MAAM,eAAe,sBAAsB,QAAQ;GACnD,MAAM,gBAAgB,QAAQ,QAAQ,MAAM,EAAE,SAAS,WAAW;GAClE,MAAM,QAAQ,SAAS;GAEvB,MAAM,YAAY,OAAO,2BAA2B;GACpD,MAAM,gBAAgB,OAAO,+BAA+B;GAC5D,MAAM,WAAW,OAAO,gBAAgB;GACxC,MAAM,kBACJ,aAAa,gBACT,WAAW,YAAY,gBACvB;GAEN,IAAIC;AACJ,OAAI,OAAO,gBAAgB,QAAQ,OAAO,iBAAiB,KACzD,KAAI;IACF,MAAM,OAAO,cAAc;KACzB,WAAW,QAAQ;KACnB,aAAa;KACb,cAAc,MAAM;KACpB,sBAAsB;KACtB,0BAA0B;KAC3B,CAAC;AACF,QAAI,OAAO,EAAG,WAAU;WAClB;AAKV,QAAK,IAAI,eAAe;IACtB;IACA,SAAS;IACT;IACA,OAAO,QAAQ;IACf,UAAU;IACV;IACA,QAAQ;IACR;IACA;IACA,aAAa,mBAAmB;IAChC,cAAc,OAAO;IACrB,iBAAiB,aAAa;IAC9B,qBAAqB,iBAAiB;IACtC,cAAc;IACf,CAAC;AACF,oBAAiB;AAEjB,OAAI,SAAS,gBAAgB,cAAc,cAAc,WAAW,EAClE;AAGF,YAAS,KAAK;IAAE,MAAM;IAAa;IAAS,CAAC;GAE7C,MAAMC,cAA8C,EAAE;AACtD,QAAK,MAAM,WAAW,eAAe;IACnC,MAAM,WAAW,OAAO,QAAQ,KAAK;IACrC,MAAM,YAAY,QAAQ;IAC1B,MAAM,gBAAgB,YAAY,KAAK;AAEvC,QAAI;KACF,MAAM,SAAS,MAAM,QAAQ,aAAa,UAAU,UAAU;KAC9D,MAAM,gBAAgB,YAAY,KAAK,GAAG;AAE1C,UAAK,IAAI,cAAc;MACrB;MACA;MACA,WAAW;MACX,SAAS;MACT;MACA;MACA;MACA,OAAO;MACP,QAAQ,UAAU,OAAO,SAAY,OAAO,OAAO;MACpD,CAAC;AAEF,iBAAY,KAAK;MACf,MAAM;MACN,aAAa,QAAQ;MACrB,SAAS,UAAU,OAAO,KAAK,OAAO,OAAO;MAC9C,CAAC;aACK,OAAO;KACd,MAAM,gBAAgB,YAAY,KAAK,GAAG;AAE1C,UAAK,IAAI,cAAc;MACrB;MACA;MACA,WAAW;MACX,SAAS;MACT;MACA;MACA;MACA,OAAO;MACP,cACE,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;MACzD,CAAC;AAEF,iBAAY,KAAK;MACf,MAAM;MACN,aAAa,QAAQ;MACrB,SAAS,UAAU,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;MACzE,UAAU;MACX,CAAC;;;AAIN,YAAS,KAAK;IAAE,MAAM;IAAQ,SAAS;IAAa,CAAC;;AAGvD,SAAO;;;AAIX,SAAS,iBAAiB,KAAsC;AAC9D,KAAI,OAAO,IAAI,YAAY,SAAU,QAAO,IAAI;AAChD,KAAI,CAAC,MAAM,QAAQ,IAAI,QAAQ,CAAE,QAAO;AACxC,QAAO,IAAI,QACR,KAAK,UAAU;AACd,MAAI,OAAO,UAAU,SAAU,QAAO;EACtC,MAAM,OAAO;AACb,SAAO,OAAO,KAAK,SAAS,WAAW,KAAK,OAAO;GACnD,CACD,KAAK,GAAG;;AAGb,SAAS,sBACP,SACQ;AACR,KAAI,CAAC,MAAM,QAAQ,QAAQ,CAAE,QAAO;AACpC,QAAO,QACJ,QAAQ,MAAM,EAAE,SAAS,OAAO,CAChC,KAAK,MAAM,OAAO,EAAE,QAAQ,GAAG,CAAC,CAChC,KAAK,GAAG;;AAGb,SAAS,wBACP,OACgC;AAChC,KAAI,CAAC,MAAM,QAAQ,MAAM,CAAE,QAAO,EAAE;AACpC,QAAO,MAAM,QAAQ,MAAM,KAAK,QAAQ,OAAO,MAAM,SAAS"}

package/dist/integrations/crewai.d.ts

@@ -0,0 +1,14 @@
+//#region src/integrations/crewai.d.ts
+/**
+ * CrewAI integration — stub for TypeScript.
+ *
+ * CrewAI is a Python-only framework. This module is a placeholder
+ * to maintain API parity with the Python SDK. It will throw an
+ * informative error if someone tries to use it.
+ */
+declare class AmplitudeCrewAIHooks {
+  constructor();
+}
+//#endregion
+export { AmplitudeCrewAIHooks };
+//# sourceMappingURL=crewai.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"crewai.d.ts","names":[],"sources":["../../src/integrations/crewai.ts"],"sourcesContent":[],"mappings":";;AAUA;;;;;;cAAa,oBAAA"}

package/dist/integrations/crewai.js

@@ -0,0 +1,19 @@
+import { ProviderError } from "../exceptions.js";
+
+//#region src/integrations/crewai.ts
+/**
+* CrewAI integration — stub for TypeScript.
+*
+* CrewAI is a Python-only framework. This module is a placeholder
+* to maintain API parity with the Python SDK. It will throw an
+* informative error if someone tries to use it.
+*/
+var AmplitudeCrewAIHooks = class {
+	constructor() {
+		throw new ProviderError("CrewAI is a Python-only framework. The @amplitude/ai CrewAI integration is not available in Node.js. Use the LangChain or OpenTelemetry integration instead.");
+	}
+};
+
+//#endregion
+export { AmplitudeCrewAIHooks };
+//# sourceMappingURL=crewai.js.map

package/dist/integrations/crewai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"crewai.js","names":[],"sources":["../../src/integrations/crewai.ts"],"sourcesContent":["/**\n * CrewAI integration — stub for TypeScript.\n *\n * CrewAI is a Python-only framework. This module is a placeholder\n * to maintain API parity with the Python SDK. It will throw an\n * informative error if someone tries to use it.\n */\n\nimport { ProviderError } from '../exceptions.js';\n\nexport class AmplitudeCrewAIHooks {\n  constructor() {\n    throw new ProviderError(\n      'CrewAI is a Python-only framework. The @amplitude/ai CrewAI integration ' +\n        'is not available in Node.js. Use the LangChain or OpenTelemetry integration instead.',\n    );\n  }\n}\n"],"mappings":";;;;;;;;;;AAUA,IAAa,uBAAb,MAAkC;CAChC,cAAc;AACZ,QAAM,IAAI,cACR,+JAED"}

package/dist/integrations/langchain.d.ts

@@ -0,0 +1,37 @@
+import { PrivacyConfig } from "../core/privacy.js";
+import { AmplitudeAI } from "../client.js";
+
+//#region src/integrations/langchain.d.ts
+
+interface CallbackHandlerOptions {
+  amplitudeAI: AmplitudeAI;
+  userId?: string;
+  sessionId?: string;
+  agentId?: string;
+  env?: string;
+  privacyConfig?: PrivacyConfig | null;
+}
+declare class AmplitudeCallbackHandler {
+  private _ai;
+  private _userId;
+  private _sessionId;
+  private _agentId;
+  private _env;
+  private _privacyConfig;
+  private _runStartTimes;
+  private _runModelNames;
+  private _toolInputs;
+  private _toolNames;
+  constructor(options: CallbackHandlerOptions);
+  private _getContext;
+  handleLLMStart(serialized: Record<string, unknown>, prompts: string[], runId: string): void;
+  handleLLMEnd(output: Record<string, unknown>, runId: string): void;
+  handleToolStart(serialized: Record<string, unknown>, input: string, runId: string): void;
+  handleToolEnd(output: string, runId: string): void;
+  handleToolError(error: unknown, runId: string): void;
+  handleLLMError(error: unknown, runId: string): void;
+}
+declare function createAmplitudeCallback(options: CallbackHandlerOptions): AmplitudeCallbackHandler;
+//#endregion
+export { AmplitudeCallbackHandler, CallbackHandlerOptions, createAmplitudeCallback };
+//# sourceMappingURL=langchain.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"langchain.d.ts","names":[],"sources":["../../src/integrations/langchain.ts"],"sourcesContent":[],"mappings":";;;;;AA6FuB,UAjFN,sBAAA,CAiFM;EAgEP,WAAA,EAhJD,WAgJC;EAAM,MAAA,CAAA,EAAA,MAAA;EAoFN,SAAA,CAAA,EAAA,MAAA;;;kBA/NE;;cAGL,wBAAA;;;;;;;;;;;uBAYU;;6BAqBP;uBAuCO;8BAgEP;;;;;iBAoFA,uBAAA,UACL,yBACR"}

package/dist/integrations/langchain.js

@@ -0,0 +1,189 @@
+import { getActiveContext } from "../context.js";
+import { calculateCost } from "../utils/costs.js";
+
+//#region src/integrations/langchain.ts
+var AmplitudeCallbackHandler = class {
+	_ai;
+	_userId;
+	_sessionId;
+	_agentId;
+	_env;
+	_privacyConfig;
+	_runStartTimes = /* @__PURE__ */ new Map();
+	_runModelNames = /* @__PURE__ */ new Map();
+	_toolInputs = /* @__PURE__ */ new Map();
+	_toolNames = /* @__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;
+		this._privacyConfig = options.privacyConfig ?? 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
+		};
+	}
+	handleLLMStart(serialized, prompts, runId) {
+		this._runStartTimes.set(runId, performance.now());
+		const kwargs = serialized.kwargs;
+		const idModel = Array.isArray(serialized.id) ? serialized.id.find((v) => typeof v === "string" && (v.includes("gpt") || v.includes("claude") || v.includes("gemini"))) : void 0;
+		const modelName = String(kwargs?.model ?? kwargs?.modelName ?? kwargs?.model_name ?? idModel ?? "");
+		if (modelName) this._runModelNames.set(runId, modelName);
+		const trackUserMessage = this._ai.trackUserMessage;
+		if (typeof trackUserMessage === "function") {
+			const ctx = this._getContext();
+			for (const prompt of prompts) {
+				if (!prompt) continue;
+				trackUserMessage({
+					userId: ctx.userId,
+					content: prompt,
+					sessionId: ctx.sessionId ?? "langchain-session",
+					traceId: ctx.traceId,
+					agentId: ctx.agentId,
+					env: ctx.env
+				});
+			}
+		}
+	}
+	handleLLMEnd(output, runId) {
+		const startTime = this._runStartTimes.get(runId) ?? performance.now();
+		this._runStartTimes.delete(runId);
+		const latencyMs = performance.now() - startTime;
+		const firstGen = output.generations?.[0]?.[0];
+		const content = _extractLangchainText(firstGen);
+		const llmOutput = output.llmOutput;
+		const usage = llmOutput?.tokenUsage ?? llmOutput?.usage ?? llmOutput?.usage_metadata;
+		const modelFromStart = this._runModelNames.get(runId);
+		this._runModelNames.delete(runId);
+		const ctx = this._getContext();
+		const modelName = String(llmOutput?.modelName ?? modelFromStart ?? "unknown");
+		const inputTokens = _safeNumber(usage?.promptTokens ?? usage?.prompt_tokens ?? usage?.input_tokens);
+		const outputTokens = _safeNumber(usage?.completionTokens ?? usage?.completion_tokens ?? usage?.output_tokens);
+		let costUsd;
+		if (modelName !== "unknown" && inputTokens != null && outputTokens != null) try {
+			const cost = calculateCost({
+				modelName,
+				inputTokens,
+				outputTokens
+			});
+			if (cost > 0) costUsd = cost;
+		} catch {}
+		this._ai.trackAiMessage({
+			userId: ctx.userId,
+			content,
+			sessionId: ctx.sessionId ?? "langchain-session",
+			model: modelName,
+			provider: "langchain",
+			latencyMs,
+			traceId: ctx.traceId,
+			agentId: ctx.agentId,
+			env: ctx.env,
+			inputTokens,
+			outputTokens,
+			totalTokens: _safeNumber(usage?.totalTokens ?? usage?.total_tokens),
+			totalCostUsd: costUsd,
+			privacyConfig: this._privacyConfig
+		});
+	}
+	handleToolStart(serialized, input, runId) {
+		this._runStartTimes.set(runId, performance.now());
+		this._toolInputs.set(runId, input);
+		const name = String(serialized.name ?? serialized.id?.slice(-1)[0] ?? "langchain-tool");
+		this._toolNames.set(runId, name);
+	}
+	handleToolEnd(output, runId) {
+		const startTime = this._runStartTimes.get(runId) ?? performance.now();
+		this._runStartTimes.delete(runId);
+		const latencyMs = performance.now() - startTime;
+		const toolInput = this._toolInputs.get(runId);
+		this._toolInputs.delete(runId);
+		const toolName = this._toolNames.get(runId) ?? "langchain-tool";
+		this._toolNames.delete(runId);
+		const ctx = this._getContext();
+		this._ai.trackToolCall({
+			userId: ctx.userId,
+			toolName,
+			latencyMs,
+			success: true,
+			sessionId: ctx.sessionId,
+			traceId: ctx.traceId,
+			agentId: ctx.agentId,
+			env: ctx.env,
+			input: toolInput,
+			output
+		});
+	}
+	handleToolError(error, runId) {
+		const startTime = this._runStartTimes.get(runId) ?? performance.now();
+		this._runStartTimes.delete(runId);
+		const latencyMs = performance.now() - startTime;
+		const toolInput = this._toolInputs.get(runId);
+		this._toolInputs.delete(runId);
+		const toolName = this._toolNames.get(runId) ?? "langchain-tool";
+		this._toolNames.delete(runId);
+		const ctx = this._getContext();
+		this._ai.trackToolCall({
+			userId: ctx.userId,
+			toolName,
+			latencyMs,
+			success: false,
+			sessionId: ctx.sessionId,
+			traceId: ctx.traceId,
+			agentId: ctx.agentId,
+			env: ctx.env,
+			input: toolInput,
+			errorMessage: error instanceof Error ? error.message : String(error)
+		});
+	}
+	handleLLMError(error, runId) {
+		const startTime = this._runStartTimes.get(runId) ?? performance.now();
+		this._runStartTimes.delete(runId);
+		const latencyMs = performance.now() - startTime;
+		const ctx = this._getContext();
+		this._ai.trackAiMessage({
+			userId: ctx.userId,
+			content: "",
+			sessionId: ctx.sessionId ?? "langchain-session",
+			model: "unknown",
+			provider: "langchain",
+			latencyMs,
+			traceId: ctx.traceId,
+			agentId: ctx.agentId,
+			env: ctx.env,
+			isError: true,
+			errorMessage: error instanceof Error ? error.message : String(error)
+		});
+	}
+};
+function createAmplitudeCallback(options) {
+	return new AmplitudeCallbackHandler(options);
+}
+function _safeNumber(value) {
+	return typeof value === "number" ? value : void 0;
+}
+function _extractLangchainText(generation) {
+	if (generation == null || typeof generation !== "object") return "";
+	const gen = generation;
+	if (typeof gen.text === "string") return gen.text;
+	const message = gen.message;
+	if (message == null) return "";
+	if (typeof message.content === "string") return message.content;
+	if (!Array.isArray(message.content)) return "";
+	return message.content.map((part) => {
+		if (typeof part === "string") return part;
+		const item = part;
+		return typeof item.text === "string" ? item.text : "";
+	}).join("");
+}
+
+//#endregion
+export { AmplitudeCallbackHandler, createAmplitudeCallback };
+//# sourceMappingURL=langchain.js.map

package/dist/integrations/langchain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"langchain.js","names":["costUsd: number | undefined"],"sources":["../../src/integrations/langchain.ts"],"sourcesContent":["/**\n * LangChain integration — AmplitudeCallbackHandler.\n *\n * Tracks LLM calls, tool calls, and chain events via LangChain's\n * callback system.\n */\n\nimport type { AmplitudeAI } from '../client.js';\nimport { getActiveContext } from '../context.js';\nimport type { PrivacyConfig } from '../core/privacy.js';\nimport { calculateCost } from '../utils/costs.js';\n\nexport interface CallbackHandlerOptions {\n  amplitudeAI: AmplitudeAI;\n  userId?: string;\n  sessionId?: string;\n  agentId?: string;\n  env?: string;\n  privacyConfig?: PrivacyConfig | null;\n}\n\nexport class AmplitudeCallbackHandler {\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 _privacyConfig: PrivacyConfig | null;\n  private _runStartTimes: Map<string, number> = new Map();\n  private _runModelNames: Map<string, string> = new Map();\n  private _toolInputs: Map<string, unknown> = new Map();\n  private _toolNames: Map<string, string> = new Map();\n\n  constructor(options: CallbackHandlerOptions) {\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    this._privacyConfig = options.privacyConfig ?? 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  handleLLMStart(\n    serialized: Record<string, unknown>,\n    prompts: string[],\n    runId: string,\n  ): void {\n    this._runStartTimes.set(runId, performance.now());\n    const kwargs = serialized.kwargs as Record<string, unknown> | undefined;\n    const idModel = Array.isArray(serialized.id)\n      ? serialized.id.find(\n          (v) =>\n            typeof v === 'string' &&\n            (v.includes('gpt') || v.includes('claude') || v.includes('gemini')),\n        )\n      : undefined;\n    const modelName = String(\n      kwargs?.model ?? kwargs?.modelName ?? kwargs?.model_name ?? idModel ?? '',\n    );\n    if (modelName) this._runModelNames.set(runId, modelName);\n\n    const trackUserMessage = (\n      this._ai as unknown as {\n        trackUserMessage?: (opts: Record<string, unknown>) => void;\n      }\n    ).trackUserMessage;\n    if (typeof trackUserMessage === 'function') {\n      const ctx = this._getContext();\n      for (const prompt of prompts) {\n        if (!prompt) continue;\n        trackUserMessage({\n          userId: ctx.userId,\n          content: prompt,\n          sessionId: ctx.sessionId ?? 'langchain-session',\n          traceId: ctx.traceId,\n          agentId: ctx.agentId,\n          env: ctx.env,\n        });\n      }\n    }\n  }\n\n  handleLLMEnd(output: Record<string, unknown>, runId: string): void {\n    const startTime = this._runStartTimes.get(runId) ?? performance.now();\n    this._runStartTimes.delete(runId);\n    const latencyMs = performance.now() - startTime;\n\n    const generations = output.generations as\n      | Array<Array<Record<string, unknown>>>\n      | undefined;\n    const firstGen = generations?.[0]?.[0];\n    const content = _extractLangchainText(firstGen);\n    const llmOutput = output.llmOutput as Record<string, unknown> | undefined;\n    const usage =\n      (llmOutput?.tokenUsage as Record<string, unknown> | undefined) ??\n      (llmOutput?.usage as Record<string, unknown> | undefined) ??\n      (llmOutput?.usage_metadata as Record<string, unknown> | undefined);\n    const modelFromStart = this._runModelNames.get(runId);\n    this._runModelNames.delete(runId);\n\n    const ctx = this._getContext();\n    const modelName = String(\n      llmOutput?.modelName ?? modelFromStart ?? 'unknown',\n    );\n    const inputTokens = _safeNumber(\n      usage?.promptTokens ?? usage?.prompt_tokens ?? usage?.input_tokens,\n    );\n    const outputTokens = _safeNumber(\n      usage?.completionTokens ??\n        usage?.completion_tokens ??\n        usage?.output_tokens,\n    );\n\n    let costUsd: number | undefined;\n    if (\n      modelName !== 'unknown' &&\n      inputTokens != null &&\n      outputTokens != null\n    ) {\n      try {\n        const cost = calculateCost({ modelName, inputTokens, outputTokens });\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,\n      sessionId: ctx.sessionId ?? 'langchain-session',\n      model: modelName,\n      provider: 'langchain',\n      latencyMs,\n      traceId: ctx.traceId,\n      agentId: ctx.agentId,\n      env: ctx.env,\n      inputTokens,\n      outputTokens,\n      totalTokens: _safeNumber(usage?.totalTokens ?? usage?.total_tokens),\n      totalCostUsd: costUsd,\n      privacyConfig: this._privacyConfig,\n    });\n  }\n\n  handleToolStart(\n    serialized: Record<string, unknown>,\n    input: string,\n    runId: string,\n  ): void {\n    this._runStartTimes.set(runId, performance.now());\n    this._toolInputs.set(runId, input);\n    const name = String(\n      serialized.name ??\n        (serialized.id as string[] | undefined)?.slice(-1)[0] ??\n        'langchain-tool',\n    );\n    this._toolNames.set(runId, name);\n  }\n\n  handleToolEnd(output: string, runId: string): void {\n    const startTime = this._runStartTimes.get(runId) ?? performance.now();\n    this._runStartTimes.delete(runId);\n    const latencyMs = performance.now() - startTime;\n    const toolInput = this._toolInputs.get(runId);\n    this._toolInputs.delete(runId);\n    const toolName = this._toolNames.get(runId) ?? 'langchain-tool';\n    this._toolNames.delete(runId);\n\n    const ctx = this._getContext();\n    this._ai.trackToolCall({\n      userId: ctx.userId,\n      toolName,\n      latencyMs,\n      success: true,\n      sessionId: ctx.sessionId,\n      traceId: ctx.traceId,\n      agentId: ctx.agentId,\n      env: ctx.env,\n      input: toolInput,\n      output,\n    });\n  }\n\n  handleToolError(error: unknown, runId: string): void {\n    const startTime = this._runStartTimes.get(runId) ?? performance.now();\n    this._runStartTimes.delete(runId);\n    const latencyMs = performance.now() - startTime;\n    const toolInput = this._toolInputs.get(runId);\n    this._toolInputs.delete(runId);\n    const toolName = this._toolNames.get(runId) ?? 'langchain-tool';\n    this._toolNames.delete(runId);\n\n    const ctx = this._getContext();\n    this._ai.trackToolCall({\n      userId: ctx.userId,\n      toolName,\n      latencyMs,\n      success: false,\n      sessionId: ctx.sessionId,\n      traceId: ctx.traceId,\n      agentId: ctx.agentId,\n      env: ctx.env,\n      input: toolInput,\n      errorMessage: error instanceof Error ? error.message : String(error),\n    });\n  }\n\n  handleLLMError(error: unknown, runId: string): void {\n    const startTime = this._runStartTimes.get(runId) ?? performance.now();\n    this._runStartTimes.delete(runId);\n    const latencyMs = performance.now() - startTime;\n\n    const ctx = this._getContext();\n    this._ai.trackAiMessage({\n      userId: ctx.userId,\n      content: '',\n      sessionId: ctx.sessionId ?? 'langchain-session',\n      model: 'unknown',\n      provider: 'langchain',\n      latencyMs,\n      traceId: ctx.traceId,\n      agentId: ctx.agentId,\n      env: ctx.env,\n      isError: true,\n      errorMessage: error instanceof Error ? error.message : String(error),\n    });\n  }\n}\n\nexport function createAmplitudeCallback(\n  options: CallbackHandlerOptions,\n): AmplitudeCallbackHandler {\n  return new AmplitudeCallbackHandler(options);\n}\n\nfunction _safeNumber(value: unknown): number | undefined {\n  return typeof value === 'number' ? value : undefined;\n}\n\nfunction _extractLangchainText(generation: unknown): string {\n  if (generation == null || typeof generation !== 'object') return '';\n  const gen = generation as Record<string, unknown>;\n  if (typeof gen.text === 'string') return gen.text;\n  const message = gen.message as Record<string, unknown> | undefined;\n  if (message == null) return '';\n  if (typeof message.content === 'string') return message.content;\n  if (!Array.isArray(message.content)) return '';\n  return message.content\n    .map((part) => {\n      if (typeof part === 'string') return part;\n      const item = part as Record<string, unknown>;\n      return typeof item.text === 'string' ? item.text : '';\n    })\n    .join('');\n}\n"],"mappings":";;;;AAqBA,IAAa,2BAAb,MAAsC;CACpC,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ,iCAAsC,IAAI,KAAK;CACvD,AAAQ,iCAAsC,IAAI,KAAK;CACvD,AAAQ,8BAAoC,IAAI,KAAK;CACrD,AAAQ,6BAAkC,IAAI,KAAK;CAEnD,YAAY,SAAiC;AAC3C,OAAK,MAAM,QAAQ;AACnB,OAAK,UAAU,QAAQ,UAAU;AACjC,OAAK,aAAa,QAAQ,aAAa;AACvC,OAAK,WAAW,QAAQ,WAAW;AACnC,OAAK,OAAO,QAAQ,OAAO;AAC3B,OAAK,iBAAiB,QAAQ,iBAAiB;;CAGjD,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,eACE,YACA,SACA,OACM;AACN,OAAK,eAAe,IAAI,OAAO,YAAY,KAAK,CAAC;EACjD,MAAM,SAAS,WAAW;EAC1B,MAAM,UAAU,MAAM,QAAQ,WAAW,GAAG,GACxC,WAAW,GAAG,MACX,MACC,OAAO,MAAM,aACZ,EAAE,SAAS,MAAM,IAAI,EAAE,SAAS,SAAS,IAAI,EAAE,SAAS,SAAS,EACrE,GACD;EACJ,MAAM,YAAY,OAChB,QAAQ,SAAS,QAAQ,aAAa,QAAQ,cAAc,WAAW,GACxE;AACD,MAAI,UAAW,MAAK,eAAe,IAAI,OAAO,UAAU;EAExD,MAAM,mBACJ,KAAK,IAGL;AACF,MAAI,OAAO,qBAAqB,YAAY;GAC1C,MAAM,MAAM,KAAK,aAAa;AAC9B,QAAK,MAAM,UAAU,SAAS;AAC5B,QAAI,CAAC,OAAQ;AACb,qBAAiB;KACf,QAAQ,IAAI;KACZ,SAAS;KACT,WAAW,IAAI,aAAa;KAC5B,SAAS,IAAI;KACb,SAAS,IAAI;KACb,KAAK,IAAI;KACV,CAAC;;;;CAKR,aAAa,QAAiC,OAAqB;EACjE,MAAM,YAAY,KAAK,eAAe,IAAI,MAAM,IAAI,YAAY,KAAK;AACrE,OAAK,eAAe,OAAO,MAAM;EACjC,MAAM,YAAY,YAAY,KAAK,GAAG;EAKtC,MAAM,WAHc,OAAO,cAGI,KAAK;EACpC,MAAM,UAAU,sBAAsB,SAAS;EAC/C,MAAM,YAAY,OAAO;EACzB,MAAM,QACH,WAAW,cACX,WAAW,SACX,WAAW;EACd,MAAM,iBAAiB,KAAK,eAAe,IAAI,MAAM;AACrD,OAAK,eAAe,OAAO,MAAM;EAEjC,MAAM,MAAM,KAAK,aAAa;EAC9B,MAAM,YAAY,OAChB,WAAW,aAAa,kBAAkB,UAC3C;EACD,MAAM,cAAc,YAClB,OAAO,gBAAgB,OAAO,iBAAiB,OAAO,aACvD;EACD,MAAM,eAAe,YACnB,OAAO,oBACL,OAAO,qBACP,OAAO,cACV;EAED,IAAIA;AACJ,MACE,cAAc,aACd,eAAe,QACf,gBAAgB,KAEhB,KAAI;GACF,MAAM,OAAO,cAAc;IAAE;IAAW;IAAa;IAAc,CAAC;AACpE,OAAI,OAAO,EAAG,WAAU;UAClB;AAKV,OAAK,IAAI,eAAe;GACtB,QAAQ,IAAI;GACZ;GACA,WAAW,IAAI,aAAa;GAC5B,OAAO;GACP,UAAU;GACV;GACA,SAAS,IAAI;GACb,SAAS,IAAI;GACb,KAAK,IAAI;GACT;GACA;GACA,aAAa,YAAY,OAAO,eAAe,OAAO,aAAa;GACnE,cAAc;GACd,eAAe,KAAK;GACrB,CAAC;;CAGJ,gBACE,YACA,OACA,OACM;AACN,OAAK,eAAe,IAAI,OAAO,YAAY,KAAK,CAAC;AACjD,OAAK,YAAY,IAAI,OAAO,MAAM;EAClC,MAAM,OAAO,OACX,WAAW,QACR,WAAW,IAA6B,MAAM,GAAG,CAAC,MACnD,iBACH;AACD,OAAK,WAAW,IAAI,OAAO,KAAK;;CAGlC,cAAc,QAAgB,OAAqB;EACjD,MAAM,YAAY,KAAK,eAAe,IAAI,MAAM,IAAI,YAAY,KAAK;AACrE,OAAK,eAAe,OAAO,MAAM;EACjC,MAAM,YAAY,YAAY,KAAK,GAAG;EACtC,MAAM,YAAY,KAAK,YAAY,IAAI,MAAM;AAC7C,OAAK,YAAY,OAAO,MAAM;EAC9B,MAAM,WAAW,KAAK,WAAW,IAAI,MAAM,IAAI;AAC/C,OAAK,WAAW,OAAO,MAAM;EAE7B,MAAM,MAAM,KAAK,aAAa;AAC9B,OAAK,IAAI,cAAc;GACrB,QAAQ,IAAI;GACZ;GACA;GACA,SAAS;GACT,WAAW,IAAI;GACf,SAAS,IAAI;GACb,SAAS,IAAI;GACb,KAAK,IAAI;GACT,OAAO;GACP;GACD,CAAC;;CAGJ,gBAAgB,OAAgB,OAAqB;EACnD,MAAM,YAAY,KAAK,eAAe,IAAI,MAAM,IAAI,YAAY,KAAK;AACrE,OAAK,eAAe,OAAO,MAAM;EACjC,MAAM,YAAY,YAAY,KAAK,GAAG;EACtC,MAAM,YAAY,KAAK,YAAY,IAAI,MAAM;AAC7C,OAAK,YAAY,OAAO,MAAM;EAC9B,MAAM,WAAW,KAAK,WAAW,IAAI,MAAM,IAAI;AAC/C,OAAK,WAAW,OAAO,MAAM;EAE7B,MAAM,MAAM,KAAK,aAAa;AAC9B,OAAK,IAAI,cAAc;GACrB,QAAQ,IAAI;GACZ;GACA;GACA,SAAS;GACT,WAAW,IAAI;GACf,SAAS,IAAI;GACb,SAAS,IAAI;GACb,KAAK,IAAI;GACT,OAAO;GACP,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;GACrE,CAAC;;CAGJ,eAAe,OAAgB,OAAqB;EAClD,MAAM,YAAY,KAAK,eAAe,IAAI,MAAM,IAAI,YAAY,KAAK;AACrE,OAAK,eAAe,OAAO,MAAM;EACjC,MAAM,YAAY,YAAY,KAAK,GAAG;EAEtC,MAAM,MAAM,KAAK,aAAa;AAC9B,OAAK,IAAI,eAAe;GACtB,QAAQ,IAAI;GACZ,SAAS;GACT,WAAW,IAAI,aAAa;GAC5B,OAAO;GACP,UAAU;GACV;GACA,SAAS,IAAI;GACb,SAAS,IAAI;GACb,KAAK,IAAI;GACT,SAAS;GACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;GACrE,CAAC;;;AAIN,SAAgB,wBACd,SAC0B;AAC1B,QAAO,IAAI,yBAAyB,QAAQ;;AAG9C,SAAS,YAAY,OAAoC;AACvD,QAAO,OAAO,UAAU,WAAW,QAAQ;;AAG7C,SAAS,sBAAsB,YAA6B;AAC1D,KAAI,cAAc,QAAQ,OAAO,eAAe,SAAU,QAAO;CACjE,MAAM,MAAM;AACZ,KAAI,OAAO,IAAI,SAAS,SAAU,QAAO,IAAI;CAC7C,MAAM,UAAU,IAAI;AACpB,KAAI,WAAW,KAAM,QAAO;AAC5B,KAAI,OAAO,QAAQ,YAAY,SAAU,QAAO,QAAQ;AACxD,KAAI,CAAC,MAAM,QAAQ,QAAQ,QAAQ,CAAE,QAAO;AAC5C,QAAO,QAAQ,QACZ,KAAK,SAAS;AACb,MAAI,OAAO,SAAS,SAAU,QAAO;EACrC,MAAM,OAAO;AACb,SAAO,OAAO,KAAK,SAAS,WAAW,KAAK,OAAO;GACnD,CACD,KAAK,GAAG"}

package/dist/integrations/llamaindex.d.ts

@@ -0,0 +1,44 @@
+import { AmplitudeAI } from "../client.js";
+
+//#region src/integrations/llamaindex.d.ts
+
+interface LlamaIndexHandlerOptions {
+  amplitudeAI: AmplitudeAI;
+  userId?: string;
+  sessionId?: string;
+  agentId?: string;
+  env?: string;
+}
+declare class AmplitudeLlamaIndexHandler {
+  private _ai;
+  private _userId;
+  private _sessionId;
+  private _agentId;
+  private _env;
+  private _startTimes;
+  constructor(options: LlamaIndexHandlerOptions);
+  private _getContext;
+  onLLMStart(eventId: string): void;
+  onLLMEnd(eventId: string, response: {
+    content?: string;
+    model?: string;
+    inputTokens?: number;
+    outputTokens?: number;
+  }): void;
+  onEmbeddingStart(eventId: string): void;
+  onEmbeddingEnd(eventId: string, response: {
+    model?: string;
+    inputTokens?: number;
+    dimensions?: number;
+  }): void;
+  onToolStart(eventId: string): void;
+  onToolEnd(eventId: string, response: {
+    toolName?: string;
+    output?: unknown;
+    success?: boolean;
+  }): void;
+}
+declare function createAmplitudeLlamaIndexHandler(options: LlamaIndexHandlerOptions): AmplitudeLlamaIndexHandler;
+//#endregion
+export { AmplitudeLlamaIndexHandler, LlamaIndexHandlerOptions, createAmplitudeLlamaIndexHandler };
+//# sourceMappingURL=llamaindex.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"llamaindex.d.ts","names":[],"sources":["../../src/integrations/llamaindex.ts"],"sourcesContent":[],"mappings":";;;;UAWiB,wBAAA;eACF;;;;;;cAOF,0BAAA;;;;;;;uBAQU;;;;;;;;;;;;;;;;;;;;;;iBAiIP,gCAAA,UACL,2BACR"}