@amplitude/ai 0.2.0 → 0.3.0

package/dist/bound-agent.d.ts

@@ -12,26 +12,29 @@ type SpanOpts = Partial<Omit<Parameters<AmplitudeAI['trackSpan']>[0], 'spanName'
 type SessionEndOpts = Partial<Parameters<AmplitudeAI['trackSessionEnd']>[0]>;
 type SessionEnrichmentOpts = Partial<Omit<Parameters<AmplitudeAI['trackSessionEnrichment']>[0], 'enrichments'>>;
 type ScoreOpts = Partial<Omit<Parameters<AmplitudeAI['score']>[0], 'name' | 'value' | 'targetId'>>;
+interface AgentOptions {
+  userId?: string | null;
+  parentAgentId?: string | null;
+  customerOrgId?: string | null;
+  agentVersion?: string | null;
+  description?: string | null;
+  context?: Record<string, unknown> | null;
+  env?: string | null;
+  sessionId?: string | null;
+  traceId?: string | null;
+  groups?: Record<string, unknown> | null;
+  deviceId?: string | null;
+  browserSessionId?: string | null;
+}
 declare class BoundAgent {
   readonly _ai: AmplitudeAI;
   readonly _defaults: Record<string, unknown>;
-  constructor(ai: AmplitudeAI, opts: {
+  constructor(ai: AmplitudeAI, opts: AgentOptions & {
     agentId: string;
-    userId?: string | null;
-    parentAgentId?: string | null;
-    customerOrgId?: string | null;
-    agentVersion?: string | null;
-    context?: Record<string, unknown> | null;
-    env?: string | null;
-    sessionId?: string | null;
-    traceId?: string | null;
-    groups?: Record<string, unknown> | null;
-    deviceId?: string | null;
-    browserSessionId?: string | null;
   });
   get agentId(): string;
   get ai(): AmplitudeAI;
-  child(agentId: string, overrides?: Record<string, unknown>): BoundAgent;
+  child(agentId: string, overrides?: AgentOptions): BoundAgent;
   private _merge;
   trackUserMessage(content: string, opts?: UserMessageOpts): string;
   trackAiMessage(content: string, model: string, provider: string, latencyMs: number, opts?: AiMessageOpts): string;
@@ -47,10 +50,11 @@ declare class BoundAgent {
     userId?: string | null;
     deviceId?: string | null;
     browserSessionId?: string | null;
+    autoFlush?: boolean;
   }): Session;
   flush(): unknown;
   shutdown(): void;
 }
 //#endregion
-export { AiMessageOpts, BoundAgent, EmbeddingOpts, ScoreOpts, SessionEndOpts, SessionEnrichmentOpts, SpanOpts, ToolCallOpts, UserMessageOpts };
+export { AgentOptions, AiMessageOpts, BoundAgent, EmbeddingOpts, ScoreOpts, SessionEndOpts, SessionEnrichmentOpts, SpanOpts, ToolCallOpts, UserMessageOpts };
 //# sourceMappingURL=bound-agent.d.ts.map

package/dist/bound-agent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"bound-agent.d.ts","names":[],"sources":["../src/bound-agent.ts"],"sourcesContent":[],"mappings":";;;;;;AAY8B,KAAlB,eAAA,GAAkB,OAAA,CAC5B,IAD4B,CACvB,UADuB,CACZ,WADY,CAAA,kBAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,SAAA,CAAA,CAAA;AAAO,KAIzB,aAAA,GAAgB,OAJS,CAKnC,IALmC,CAMjC,UANiC,CAMtB,WANsB,CAAA,gBAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,SAAA,GAAA,OAAA,GAAA,UAAA,GAAA,WAAA,CAAA,CAAA;AAIzB,KAOA,YAAA,GAAe,OAPF,CAQvB,IARuB,CASrB,UATqB,CASV,WATU,CAAA,eAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,UAAA,GAAA,WAAA,GAAA,SAAA,CAAA,CAAA;AAEV,KAYH,aAAA,GAAgB,OAZb,CAab,IAba,CAcX,UAdW,CAcA,WAdA,CAAA,gBAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,OAAA,GAAA,UAAA,GAAA,WAAA,CAAA,CAAA;AAAX,KAmBQ,QAAA,GAAW,OAnBnB,CAoBF,IApBE,CAoBG,UApBH,CAoBc,WApBd,CAAA,WAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,UAAA,GAAA,WAAA,CAAA,CAAA;AADF,KAwBU,cAAA,GAAiB,OAxB3B,CAyBA,UAzBA,CAyBW,WAzBX,CAAA,iBAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA;AAD0B,KA6BhB,qBAAA,GAAwB,OA7BR,CA8B1B,IA9B0B,CA8BrB,UA9BqB,CA8BV,WA9BU,CAAA,wBAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,aAAA,CAAA,CAAA;AAAO,KAiCvB,SAAA,GAAY,OAjCW,CAkCjC,IAlCiC,CAkC5B,UAlC4B,CAkCjB,WAlCiB,CAAA,OAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,MAAA,GAAA,OAAA,GAAA,UAAA,CAAA,CAAA;AAOvB,cA2CC,UAAA,CA3CW;EAET,SAAA,GAAA,EA0CC,WA1CD;EAAX,SAAA,SAAA,EA2CkB,MA3ClB,CAAA,MAAA,EAAA,OAAA,CAAA;EADF,WAAA,CAAA,EAAA,EA+CM,WA/CN,EAAA,IAAA,EAAA;IADyB,OAAA,EAAA,MAAA;IAAO,MAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAOtB,aAAa,CAAA,EAAA,MAAA,GAAA,IAAA;IAEV,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAAX,YAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IADF,OAAA,CAAA,EA+Cc,MA/Cd,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;IAD0B,GAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAAO,SAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAOvB,OAAQ,CAAA,EAAA,MAAA,GAAA,IAAA;IACF,MAAA,CAAA,EA4CH,MA5CG,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;IAAX,QAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAAL,gBAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EADqB,CAAA;EAAO,IAAA,OAAA,CAAA,CAAA,EAAA,MAAA;EAIlB,IAAA,EAAA,CAAA,CAAA,EAmEA,WAnEc;EACb,KAAA,CAAA,OAAA,EAAA,MAAA,EAAA,SAAA,CAAA,EAsEuB,MAtEvB,CAAA,MAAA,EAAA,OAAA,CAAA,CAAA,EAsEsD,UAtEtD;EAAX,QAAA,MAAA;EAD2B,gBAAA,CAAA,OAAA,EAAA,MAAA,EAAA,IAAA,CAAA,EA4Ia,eA5Ib,CAAA,EAAA,MAAA;EAAO,cAAA,CAAA,OAAA,EAAA,MAAA,EAAA,KAAA,EAAA,MAAA,EAAA,QAAA,EAAA,MAAA,EAAA,SAAA,EAAA,MAAA,EAAA,IAAA,CAAA,EAwJ1B,aAxJ0B,CAAA,EAAA,MAAA;EAIxB,aAAA,CAAA,QAAA,EAAqB,MAAA,EAAA,SAAA,EAAA,MAAA,EAAA,OAAA,EAAA,OAAA,EAAA,IAAA,CAAA,EAmKvB,YAnKuB,CAAA,EAAA,MAAA;EACf,cAAA,CAAA,KAAA,EAAA,MAAA,EAAA,QAAA,EAAA,MAAA,EAAA,SAAA,EAAA,MAAA,EAAA,IAAA,CAAA,EAgLR,aAhLQ,CAAA,EAAA,MAAA;EAAX,SAAA,CAAA,QAAA,EAAA,MAAA,EAAA,SAAA,EAAA,MAAA,EAAA,IAAA,CAAA,EA0LgD,QA1LhD,CAAA,EAAA,MAAA;EAAL,eAAA,CAAA,IAAA,CAAA,EAkMsB,cAlMtB,CAAA,EAAA,IAAA;EADkC,sBAAA,CAAA,WAAA,EA0MnB,kBA1MmB,EAAA,IAAA,CAAA,EA2M1B,qBA3M0B,CAAA,EAAA,IAAA;EAAO,KAAA,CAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA,MAAA,EAAA,QAAA,EAAA,MAAA,EAAA,IAAA,CAAA,EAuNjC,SAvNiC,CAAA,EAAA,IAAA;EAI/B,OAAA,CAAA,IAiBC,CAjBQ,EAAA;IACH,SAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAAX,kBAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAAL,MAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IADsB,QAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAAO,gBAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAiBlB,CAAA,CAAA,EAoNR,OApNQ;EACG,KAAA,CAAA,CAAA,EAAA,OAAA;EACM,QAAA,CAAA,CAAA,EAAA,IAAA"}
+{"version":3,"file":"bound-agent.d.ts","names":[],"sources":["../src/bound-agent.ts"],"sourcesContent":[],"mappings":";;;;;;AAY8B,KAAlB,eAAA,GAAkB,OAAA,CAC5B,IAD4B,CACvB,UADuB,CACZ,WADY,CAAA,kBAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,SAAA,CAAA,CAAA;AAAO,KAIzB,aAAA,GAAgB,OAJS,CAKnC,IALmC,CAMjC,UANiC,CAMtB,WANsB,CAAA,gBAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,SAAA,GAAA,OAAA,GAAA,UAAA,GAAA,WAAA,CAAA,CAAA;AAIzB,KAOA,YAAA,GAAe,OAPF,CAQvB,IARuB,CASrB,UATqB,CASV,WATU,CAAA,eAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,UAAA,GAAA,WAAA,GAAA,SAAA,CAAA,CAAA;AAEV,KAYH,aAAA,GAAgB,OAZb,CAab,IAba,CAcX,UAdW,CAcA,WAdA,CAAA,gBAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,OAAA,GAAA,UAAA,GAAA,WAAA,CAAA,CAAA;AAAX,KAmBQ,QAAA,GAAW,OAnBnB,CAoBF,IApBE,CAoBG,UApBH,CAoBc,WApBd,CAAA,WAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,UAAA,GAAA,WAAA,CAAA,CAAA;AADF,KAwBU,cAAA,GAAiB,OAxB3B,CAyBA,UAzBA,CAyBW,WAzBX,CAAA,iBAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA;AAD0B,KA6BhB,qBAAA,GAAwB,OA7BR,CA8B1B,IA9B0B,CA8BrB,UA9BqB,CA8BV,WA9BU,CAAA,wBAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,aAAA,CAAA,CAAA;AAAO,KAiCvB,SAAA,GAAY,OAjCW,CAkCjC,IAlCiC,CAkC5B,UAlC4B,CAkCjB,WAlCiB,CAAA,OAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAA,MAAA,GAAA,OAAA,GAAA,UAAA,CAAA,CAAA;AAOvB,UA4CK,YAAA,CA5CO;EAET,MAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAAX,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EADF,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EADyB,YAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAAO,WAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAOtB,OAAA,CAAA,EA2CA,MA3Ca,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EAEV,GAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAAX,SAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EADF,OAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAD0B,MAAA,CAAA,EA+CjB,MA/CiB,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EAAO,QAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAOvB,gBAAQ,CAAA,EAAA,MAAA,GAAA,IAAA;;AACb,cA4CM,UAAA,CA5CN;EAAL,SAAA,GAAA,EA6Cc,WA7Cd;EADqB,SAAA,SAAA,EA+CD,MA/CC,CAAA,MAAA,EAAA,OAAA,CAAA;EAAO,WAAA,CAAA,EAAA,EAkDtB,WAlDsB,EAAA,IAAA,EAmDpB,YAnDoB,GAAA;IAIlB,OAAA,EAAA,MAAc;EACb,CAAA;EAAX,IAAA,OAAA,CAAA,CAAA,EAAA,MAAA;EAD2B,IAAA,EAAA,CAAA,CAAA,EAuEjB,WAvEiB;EAAO,KAAA,CAAA,OAAA,EAAA,MAAA,EAAA,SAAA,CAAA,EA2EA,YA3EA,CAAA,EA2EoB,UA3EpB;EAIxB,QAAA,MAAA;EACM,gBAAA,CAAA,OAAA,EAAA,MAAA,EAAA,IAAA,CAAA,EAyIwB,eAzIxB,CAAA,EAAA,MAAA;EAAX,cAAA,CAAA,OAAA,EAAA,MAAA,EAAA,KAAA,EAAA,MAAA,EAAA,QAAA,EAAA,MAAA,EAAA,SAAA,EAAA,MAAA,EAAA,IAAA,CAAA,EAqJG,aArJH,CAAA,EAAA,MAAA;EAAL,aAAA,CAAA,QAAA,EAAA,MAAA,EAAA,SAAA,EAAA,MAAA,EAAA,OAAA,EAAA,OAAA,EAAA,IAAA,CAAA,EAoKQ,YApKR,CAAA,EAAA,MAAA;EADkC,cAAA,CAAA,KAAA,EAAA,MAAA,EAAA,QAAA,EAAA,MAAA,EAAA,SAAA,EAAA,MAAA,EAAA,IAAA,CAAA,EAmL1B,aAnL0B,CAAA,EAAA,MAAA;EAAO,SAAA,CAAA,QAAA,EAAA,MAAA,EAAA,SAAA,EAAA,MAAA,EAAA,IAAA,CAAA,EA6LY,QA7LZ,CAAA,EAAA,MAAA;EAI/B,eAAS,CAAA,IAAA,CAAA,EAiMG,cAjMH,CAAA,EAAA,IAAA;EACH,sBAAA,CAAA,WAAA,EAuMD,kBAvMC,EAAA,IAAA,CAAA,EAwMR,qBAxMQ,CAAA,EAAA,IAAA;EAAX,KAAA,CAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA,MAAA,EAAA,QAAA,EAAA,MAAA,EAAA,IAAA,CAAA,EAoNG,SApNH,CAAA,EAAA,IAAA;EAAL,OAAA,CAAA,IAqCM,CArCN,EAAA;IADsB,SAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAAO,kBAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAkBd,MAAA,CAAA,EAAA,MAAY,GAAA,IAAA;IAehB,QAAA,CAAU,EAAA,MAAA,GAAA,IAAA;IACP,gBAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IACM,SAAA,CAAA,EAAA,OAAA;EAGd,CAAA,CAAA,EAkMH,OAlMG;EACE,KAAA,CAAA,CAAA,EAAA,OAAA;EAwBE,QAAA,CAAA,CAAA,EAAA,IAAA"}

package/dist/bound-agent.js

@@ -8,6 +8,7 @@ const CONTEXT_FIELDS = [
 	"parentAgentId",
 	"customerOrgId",
 	"agentVersion",
+	"description",
 	"context",
 	"env",
 	"sessionId",
@@ -25,6 +26,7 @@ var BoundAgent = class BoundAgent {
 			parentAgentId: opts.parentAgentId ?? null,
 			customerOrgId: opts.customerOrgId ?? null,
 			agentVersion: opts.agentVersion ?? null,
+			description: opts.description ?? null,
 			context: opts.context ?? null,
 			env: opts.env ?? null,
 			sessionId: opts.sessionId ?? null,
@@ -46,6 +48,7 @@ var BoundAgent = class BoundAgent {
 			env: this._defaults.env,
 			customerOrgId: this._defaults.customerOrgId,
 			agentVersion: this._defaults.agentVersion,
+			description: this._defaults.description,
 			sessionId: this._defaults.sessionId,
 			traceId: this._defaults.traceId,
 			groups: this._defaults.groups,
@@ -53,7 +56,7 @@ var BoundAgent = class BoundAgent {
 			browserSessionId: this._defaults.browserSessionId
 		};
 		const parentCtx = this._defaults.context;
-		const childCtx = overrides.context;
+		const childCtx = overrides.context ?? null;
 		const overrideEntries = Object.entries(overrides).filter(([k]) => k !== "context");
 		if (parentCtx || childCtx) {
 			const merged = { ...parentCtx ?? {} };

package/dist/bound-agent.js.map

@@ -1 +1 @@
-{"version":3,"file":"bound-agent.js","names":["inherited: Record<string, unknown>"],"sources":["../src/bound-agent.ts"],"sourcesContent":["/**\n * BoundAgent — agent with pre-bound defaults for tracking context fields.\n *\n * Created via `ai.agent(agentId, opts)`. All tracking calls\n * automatically inherit agentId, userId, env, and other defaults.\n */\n\nimport type { AmplitudeAI } from './client.js';\nimport type { SessionEnrichments } from './core/enrichments.js';\nimport { PROP_SESSION_REPLAY_ID } from './core/tracking.js';\nimport { Session } from './session.js';\n\nexport type UserMessageOpts = Partial<\n  Omit<Parameters<AmplitudeAI['trackUserMessage']>[0], 'content'>\n>;\n\nexport type AiMessageOpts = Partial<\n  Omit<\n    Parameters<AmplitudeAI['trackAiMessage']>[0],\n    'content' | 'model' | 'provider' | 'latencyMs'\n  >\n>;\n\nexport type ToolCallOpts = Partial<\n  Omit<\n    Parameters<AmplitudeAI['trackToolCall']>[0],\n    'toolName' | 'latencyMs' | 'success'\n  >\n>;\n\nexport type EmbeddingOpts = Partial<\n  Omit<\n    Parameters<AmplitudeAI['trackEmbedding']>[0],\n    'model' | 'provider' | 'latencyMs'\n  >\n>;\n\nexport type SpanOpts = Partial<\n  Omit<Parameters<AmplitudeAI['trackSpan']>[0], 'spanName' | 'latencyMs'>\n>;\n\nexport type SessionEndOpts = Partial<\n  Parameters<AmplitudeAI['trackSessionEnd']>[0]\n>;\n\nexport type SessionEnrichmentOpts = Partial<\n  Omit<Parameters<AmplitudeAI['trackSessionEnrichment']>[0], 'enrichments'>\n>;\n\nexport type ScoreOpts = Partial<\n  Omit<Parameters<AmplitudeAI['score']>[0], 'name' | 'value' | 'targetId'>\n>;\n\nconst CONTEXT_FIELDS = [\n  'userId',\n  'agentId',\n  'parentAgentId',\n  'customerOrgId',\n  'agentVersion',\n  'context',\n  'env',\n  'sessionId',\n  'traceId',\n  'groups',\n] as const;\n\nexport class BoundAgent {\n  readonly _ai: AmplitudeAI;\n  readonly _defaults: Record<string, unknown>;\n\n  constructor(\n    ai: AmplitudeAI,\n    opts: {\n      agentId: string;\n      userId?: string | null;\n      parentAgentId?: string | null;\n      customerOrgId?: string | null;\n      agentVersion?: string | null;\n      context?: Record<string, unknown> | null;\n      env?: string | null;\n      sessionId?: string | null;\n      traceId?: string | null;\n      groups?: Record<string, unknown> | null;\n      deviceId?: string | null;\n      browserSessionId?: string | null;\n    },\n  ) {\n    this._ai = ai;\n    this._defaults = {\n      userId: opts.userId ?? null,\n      agentId: opts.agentId,\n      parentAgentId: opts.parentAgentId ?? null,\n      customerOrgId: opts.customerOrgId ?? null,\n      agentVersion: opts.agentVersion ?? null,\n      context: opts.context ?? null,\n      env: opts.env ?? null,\n      sessionId: opts.sessionId ?? null,\n      traceId: opts.traceId ?? null,\n      groups: opts.groups ?? null,\n      deviceId: opts.deviceId ?? null,\n      browserSessionId: opts.browserSessionId ?? null,\n    };\n  }\n\n  get agentId(): string {\n    return String(this._defaults.agentId);\n  }\n\n  get ai(): AmplitudeAI {\n    return this._ai;\n  }\n\n  child(agentId: string, overrides: Record<string, unknown> = {}): BoundAgent {\n    const inherited: Record<string, unknown> = {\n      userId: this._defaults.userId,\n      env: this._defaults.env,\n      customerOrgId: this._defaults.customerOrgId,\n      agentVersion: this._defaults.agentVersion,\n      sessionId: this._defaults.sessionId,\n      traceId: this._defaults.traceId,\n      groups: this._defaults.groups,\n      deviceId: this._defaults.deviceId,\n      browserSessionId: this._defaults.browserSessionId,\n    };\n\n    const parentCtx = this._defaults.context as Record<string, unknown> | null;\n    const childCtx = overrides.context as\n      | Record<string, unknown>\n      | null\n      | undefined;\n    const overrideEntries = Object.entries(overrides).filter(\n      ([k]) => k !== 'context',\n    );\n    if (parentCtx || childCtx) {\n      const merged = { ...(parentCtx ?? {}) };\n      if (childCtx) Object.assign(merged, childCtx);\n      inherited.context = merged;\n    }\n\n    for (const [k, v] of overrideEntries) {\n      if (v != null) inherited[k] = v;\n    }\n\n    const { parentAgentId: inheritedParentAgentId, ...inheritedWithoutParent } =\n      inherited;\n    const explicitParent =\n      (inheritedParentAgentId as string | null) ??\n      (this._defaults.agentId as string);\n\n    return new BoundAgent(this._ai, {\n      agentId,\n      parentAgentId: explicitParent,\n      ...(inheritedWithoutParent as Record<string, string | null>),\n    });\n  }\n\n  private _merge<T extends Record<string, unknown>>(\n    kwargs: T,\n    fields: readonly string[] = CONTEXT_FIELDS,\n  ): T {\n    const merged = { ...kwargs } as Record<string, unknown>;\n    for (const field of fields) {\n      if (merged[field] == null && this._defaults[field] != null) {\n        merged[field] = this._defaults[field];\n      }\n    }\n    const deviceId = this._defaults.deviceId as string | null;\n    const browserSessionId = this._defaults.browserSessionId as string | null;\n    if (deviceId && browserSessionId) {\n      const existingEp = merged.eventProperties as\n        | Record<string, unknown>\n        | undefined;\n      const ep = existingEp != null ? { ...existingEp } : {};\n      if (!(PROP_SESSION_REPLAY_ID in ep)) {\n        ep[PROP_SESSION_REPLAY_ID] = `${deviceId}/${browserSessionId}`;\n        merged.eventProperties = ep;\n      }\n    }\n    return merged as T;\n  }\n\n  trackUserMessage(content: string, opts: UserMessageOpts = {}): string {\n    return this._ai.trackUserMessage({\n      ...this._merge(opts),\n      content,\n    } as Parameters<AmplitudeAI['trackUserMessage']>[0]);\n  }\n\n  trackAiMessage(\n    content: string,\n    model: string,\n    provider: string,\n    latencyMs: number,\n    opts: AiMessageOpts = {},\n  ): string {\n    return this._ai.trackAiMessage({\n      ...this._merge(opts),\n      content,\n      model,\n      provider,\n      latencyMs,\n    } as Parameters<AmplitudeAI['trackAiMessage']>[0]);\n  }\n\n  trackToolCall(\n    toolName: string,\n    latencyMs: number,\n    success: boolean,\n    opts: ToolCallOpts = {},\n  ): string {\n    return this._ai.trackToolCall({\n      ...this._merge(opts),\n      toolName,\n      latencyMs,\n      success,\n    } as Parameters<AmplitudeAI['trackToolCall']>[0]);\n  }\n\n  trackEmbedding(\n    model: string,\n    provider: string,\n    latencyMs: number,\n    opts: EmbeddingOpts = {},\n  ): string {\n    return this._ai.trackEmbedding({\n      ...this._merge(opts),\n      model,\n      provider,\n      latencyMs,\n    } as Parameters<AmplitudeAI['trackEmbedding']>[0]);\n  }\n\n  trackSpan(spanName: string, latencyMs: number, opts: SpanOpts = {}): string {\n    return this._ai.trackSpan({\n      ...this._merge(opts),\n      spanName,\n      latencyMs,\n    } as Parameters<AmplitudeAI['trackSpan']>[0]);\n  }\n\n  trackSessionEnd(opts: SessionEndOpts = {}): void {\n    this._ai.trackSessionEnd(\n      this._merge(opts) as Parameters<AmplitudeAI['trackSessionEnd']>[0],\n    );\n  }\n\n  trackSessionEnrichment(\n    enrichments: SessionEnrichments,\n    opts: SessionEnrichmentOpts = {},\n  ): void {\n    this._ai.trackSessionEnrichment({\n      ...this._merge(opts),\n      enrichments,\n    } as Parameters<AmplitudeAI['trackSessionEnrichment']>[0]);\n  }\n\n  score(\n    name: string,\n    value: number,\n    targetId: string,\n    opts: ScoreOpts = {},\n  ): void {\n    this._ai.score({\n      ...this._merge(opts),\n      name,\n      value,\n      targetId,\n    } as Parameters<AmplitudeAI['score']>[0]);\n  }\n\n  session(\n    opts: {\n      sessionId?: string | null;\n      idleTimeoutMinutes?: number | null;\n      userId?: string | null;\n      deviceId?: string | null;\n      browserSessionId?: string | null;\n    } = {},\n  ): Session {\n    return new Session(this, opts);\n  }\n\n  flush(): unknown {\n    return this._ai.flush();\n  }\n\n  shutdown(): void {\n    this._ai.shutdown();\n  }\n}\n"],"mappings":";;;;AAqDA,MAAM,iBAAiB;CACrB;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACD;AAED,IAAa,aAAb,MAAa,WAAW;CACtB,AAAS;CACT,AAAS;CAET,YACE,IACA,MAcA;AACA,OAAK,MAAM;AACX,OAAK,YAAY;GACf,QAAQ,KAAK,UAAU;GACvB,SAAS,KAAK;GACd,eAAe,KAAK,iBAAiB;GACrC,eAAe,KAAK,iBAAiB;GACrC,cAAc,KAAK,gBAAgB;GACnC,SAAS,KAAK,WAAW;GACzB,KAAK,KAAK,OAAO;GACjB,WAAW,KAAK,aAAa;GAC7B,SAAS,KAAK,WAAW;GACzB,QAAQ,KAAK,UAAU;GACvB,UAAU,KAAK,YAAY;GAC3B,kBAAkB,KAAK,oBAAoB;GAC5C;;CAGH,IAAI,UAAkB;AACpB,SAAO,OAAO,KAAK,UAAU,QAAQ;;CAGvC,IAAI,KAAkB;AACpB,SAAO,KAAK;;CAGd,MAAM,SAAiB,YAAqC,EAAE,EAAc;EAC1E,MAAMA,YAAqC;GACzC,QAAQ,KAAK,UAAU;GACvB,KAAK,KAAK,UAAU;GACpB,eAAe,KAAK,UAAU;GAC9B,cAAc,KAAK,UAAU;GAC7B,WAAW,KAAK,UAAU;GAC1B,SAAS,KAAK,UAAU;GACxB,QAAQ,KAAK,UAAU;GACvB,UAAU,KAAK,UAAU;GACzB,kBAAkB,KAAK,UAAU;GAClC;EAED,MAAM,YAAY,KAAK,UAAU;EACjC,MAAM,WAAW,UAAU;EAI3B,MAAM,kBAAkB,OAAO,QAAQ,UAAU,CAAC,QAC/C,CAAC,OAAO,MAAM,UAChB;AACD,MAAI,aAAa,UAAU;GACzB,MAAM,SAAS,EAAE,GAAI,aAAa,EAAE,EAAG;AACvC,OAAI,SAAU,QAAO,OAAO,QAAQ,SAAS;AAC7C,aAAU,UAAU;;AAGtB,OAAK,MAAM,CAAC,GAAG,MAAM,gBACnB,KAAI,KAAK,KAAM,WAAU,KAAK;EAGhC,MAAM,EAAE,eAAe,wBAAwB,GAAG,2BAChD;EACF,MAAM,iBACH,0BACA,KAAK,UAAU;AAElB,SAAO,IAAI,WAAW,KAAK,KAAK;GAC9B;GACA,eAAe;GACf,GAAI;GACL,CAAC;;CAGJ,AAAQ,OACN,QACA,SAA4B,gBACzB;EACH,MAAM,SAAS,EAAE,GAAG,QAAQ;AAC5B,OAAK,MAAM,SAAS,OAClB,KAAI,OAAO,UAAU,QAAQ,KAAK,UAAU,UAAU,KACpD,QAAO,SAAS,KAAK,UAAU;EAGnC,MAAM,WAAW,KAAK,UAAU;EAChC,MAAM,mBAAmB,KAAK,UAAU;AACxC,MAAI,YAAY,kBAAkB;GAChC,MAAM,aAAa,OAAO;GAG1B,MAAM,KAAK,cAAc,OAAO,EAAE,GAAG,YAAY,GAAG,EAAE;AACtD,OAAI,EAAE,0BAA0B,KAAK;AACnC,OAAG,0BAA0B,GAAG,SAAS,GAAG;AAC5C,WAAO,kBAAkB;;;AAG7B,SAAO;;CAGT,iBAAiB,SAAiB,OAAwB,EAAE,EAAU;AACpE,SAAO,KAAK,IAAI,iBAAiB;GAC/B,GAAG,KAAK,OAAO,KAAK;GACpB;GACD,CAAmD;;CAGtD,eACE,SACA,OACA,UACA,WACA,OAAsB,EAAE,EAChB;AACR,SAAO,KAAK,IAAI,eAAe;GAC7B,GAAG,KAAK,OAAO,KAAK;GACpB;GACA;GACA;GACA;GACD,CAAiD;;CAGpD,cACE,UACA,WACA,SACA,OAAqB,EAAE,EACf;AACR,SAAO,KAAK,IAAI,cAAc;GAC5B,GAAG,KAAK,OAAO,KAAK;GACpB;GACA;GACA;GACD,CAAgD;;CAGnD,eACE,OACA,UACA,WACA,OAAsB,EAAE,EAChB;AACR,SAAO,KAAK,IAAI,eAAe;GAC7B,GAAG,KAAK,OAAO,KAAK;GACpB;GACA;GACA;GACD,CAAiD;;CAGpD,UAAU,UAAkB,WAAmB,OAAiB,EAAE,EAAU;AAC1E,SAAO,KAAK,IAAI,UAAU;GACxB,GAAG,KAAK,OAAO,KAAK;GACpB;GACA;GACD,CAA4C;;CAG/C,gBAAgB,OAAuB,EAAE,EAAQ;AAC/C,OAAK,IAAI,gBACP,KAAK,OAAO,KAAK,CAClB;;CAGH,uBACE,aACA,OAA8B,EAAE,EAC1B;AACN,OAAK,IAAI,uBAAuB;GAC9B,GAAG,KAAK,OAAO,KAAK;GACpB;GACD,CAAyD;;CAG5D,MACE,MACA,OACA,UACA,OAAkB,EAAE,EACd;AACN,OAAK,IAAI,MAAM;GACb,GAAG,KAAK,OAAO,KAAK;GACpB;GACA;GACA;GACD,CAAwC;;CAG3C,QACE,OAMI,EAAE,EACG;AACT,SAAO,IAAI,QAAQ,MAAM,KAAK;;CAGhC,QAAiB;AACf,SAAO,KAAK,IAAI,OAAO;;CAGzB,WAAiB;AACf,OAAK,IAAI,UAAU"}
+{"version":3,"file":"bound-agent.js","names":["inherited: Record<string, unknown>"],"sources":["../src/bound-agent.ts"],"sourcesContent":["/**\n * BoundAgent — agent with pre-bound defaults for tracking context fields.\n *\n * Created via `ai.agent(agentId, opts)`. All tracking calls\n * automatically inherit agentId, userId, env, and other defaults.\n */\n\nimport type { AmplitudeAI } from './client.js';\nimport type { SessionEnrichments } from './core/enrichments.js';\nimport { PROP_SESSION_REPLAY_ID } from './core/tracking.js';\nimport { Session } from './session.js';\n\nexport type UserMessageOpts = Partial<\n  Omit<Parameters<AmplitudeAI['trackUserMessage']>[0], 'content'>\n>;\n\nexport type AiMessageOpts = Partial<\n  Omit<\n    Parameters<AmplitudeAI['trackAiMessage']>[0],\n    'content' | 'model' | 'provider' | 'latencyMs'\n  >\n>;\n\nexport type ToolCallOpts = Partial<\n  Omit<\n    Parameters<AmplitudeAI['trackToolCall']>[0],\n    'toolName' | 'latencyMs' | 'success'\n  >\n>;\n\nexport type EmbeddingOpts = Partial<\n  Omit<\n    Parameters<AmplitudeAI['trackEmbedding']>[0],\n    'model' | 'provider' | 'latencyMs'\n  >\n>;\n\nexport type SpanOpts = Partial<\n  Omit<Parameters<AmplitudeAI['trackSpan']>[0], 'spanName' | 'latencyMs'>\n>;\n\nexport type SessionEndOpts = Partial<\n  Parameters<AmplitudeAI['trackSessionEnd']>[0]\n>;\n\nexport type SessionEnrichmentOpts = Partial<\n  Omit<Parameters<AmplitudeAI['trackSessionEnrichment']>[0], 'enrichments'>\n>;\n\nexport type ScoreOpts = Partial<\n  Omit<Parameters<AmplitudeAI['score']>[0], 'name' | 'value' | 'targetId'>\n>;\n\nconst CONTEXT_FIELDS = [\n  'userId',\n  'agentId',\n  'parentAgentId',\n  'customerOrgId',\n  'agentVersion',\n  'description',\n  'context',\n  'env',\n  'sessionId',\n  'traceId',\n  'groups',\n] as const;\n\nexport interface AgentOptions {\n  userId?: string | null;\n  parentAgentId?: string | null;\n  customerOrgId?: string | null;\n  agentVersion?: string | null;\n  description?: string | null;\n  context?: Record<string, unknown> | null;\n  env?: string | null;\n  sessionId?: string | null;\n  traceId?: string | null;\n  groups?: Record<string, unknown> | null;\n  deviceId?: string | null;\n  browserSessionId?: string | null;\n}\n\nexport class BoundAgent {\n  readonly _ai: AmplitudeAI;\n  readonly _defaults: Record<string, unknown>;\n\n  constructor(\n    ai: AmplitudeAI,\n    opts: AgentOptions & { agentId: string },\n  ) {\n    this._ai = ai;\n    this._defaults = {\n      userId: opts.userId ?? null,\n      agentId: opts.agentId,\n      parentAgentId: opts.parentAgentId ?? null,\n      customerOrgId: opts.customerOrgId ?? null,\n      agentVersion: opts.agentVersion ?? null,\n      description: opts.description ?? null,\n      context: opts.context ?? null,\n      env: opts.env ?? null,\n      sessionId: opts.sessionId ?? null,\n      traceId: opts.traceId ?? null,\n      groups: opts.groups ?? null,\n      deviceId: opts.deviceId ?? null,\n      browserSessionId: opts.browserSessionId ?? null,\n    };\n  }\n\n  get agentId(): string {\n    return String(this._defaults.agentId);\n  }\n\n  get ai(): AmplitudeAI {\n    return this._ai;\n  }\n\n  child(agentId: string, overrides: AgentOptions = {}): BoundAgent {\n    const inherited: Record<string, unknown> = {\n      userId: this._defaults.userId,\n      env: this._defaults.env,\n      customerOrgId: this._defaults.customerOrgId,\n      agentVersion: this._defaults.agentVersion,\n      description: this._defaults.description,\n      sessionId: this._defaults.sessionId,\n      traceId: this._defaults.traceId,\n      groups: this._defaults.groups,\n      deviceId: this._defaults.deviceId,\n      browserSessionId: this._defaults.browserSessionId,\n    };\n\n    const parentCtx = this._defaults.context as Record<string, unknown> | null;\n    const childCtx = overrides.context ?? null;\n    const overrideEntries = Object.entries(overrides).filter(\n      ([k]) => k !== 'context',\n    );\n    if (parentCtx || childCtx) {\n      const merged = { ...(parentCtx ?? {}) };\n      if (childCtx) Object.assign(merged, childCtx);\n      inherited.context = merged;\n    }\n\n    for (const [k, v] of overrideEntries) {\n      if (v != null) inherited[k] = v;\n    }\n\n    const { parentAgentId: inheritedParentAgentId, ...inheritedWithoutParent } =\n      inherited;\n    const explicitParent =\n      (inheritedParentAgentId as string | null) ??\n      (this._defaults.agentId as string);\n\n    return new BoundAgent(this._ai, {\n      agentId,\n      parentAgentId: explicitParent,\n      ...(inheritedWithoutParent as Record<string, string | null>),\n    });\n  }\n\n  private _merge<T extends Record<string, unknown>>(\n    kwargs: T,\n    fields: readonly string[] = CONTEXT_FIELDS,\n  ): T {\n    const merged = { ...kwargs } as Record<string, unknown>;\n    for (const field of fields) {\n      if (merged[field] == null && this._defaults[field] != null) {\n        merged[field] = this._defaults[field];\n      }\n    }\n    const deviceId = this._defaults.deviceId as string | null;\n    const browserSessionId = this._defaults.browserSessionId as string | null;\n    if (deviceId && browserSessionId) {\n      const existingEp = merged.eventProperties as\n        | Record<string, unknown>\n        | undefined;\n      const ep = existingEp != null ? { ...existingEp } : {};\n      if (!(PROP_SESSION_REPLAY_ID in ep)) {\n        ep[PROP_SESSION_REPLAY_ID] = `${deviceId}/${browserSessionId}`;\n        merged.eventProperties = ep;\n      }\n    }\n    return merged as T;\n  }\n\n  trackUserMessage(content: string, opts: UserMessageOpts = {}): string {\n    return this._ai.trackUserMessage({\n      ...this._merge(opts),\n      content,\n    } as Parameters<AmplitudeAI['trackUserMessage']>[0]);\n  }\n\n  trackAiMessage(\n    content: string,\n    model: string,\n    provider: string,\n    latencyMs: number,\n    opts: AiMessageOpts = {},\n  ): string {\n    return this._ai.trackAiMessage({\n      ...this._merge(opts),\n      content,\n      model,\n      provider,\n      latencyMs,\n    } as Parameters<AmplitudeAI['trackAiMessage']>[0]);\n  }\n\n  trackToolCall(\n    toolName: string,\n    latencyMs: number,\n    success: boolean,\n    opts: ToolCallOpts = {},\n  ): string {\n    return this._ai.trackToolCall({\n      ...this._merge(opts),\n      toolName,\n      latencyMs,\n      success,\n    } as Parameters<AmplitudeAI['trackToolCall']>[0]);\n  }\n\n  trackEmbedding(\n    model: string,\n    provider: string,\n    latencyMs: number,\n    opts: EmbeddingOpts = {},\n  ): string {\n    return this._ai.trackEmbedding({\n      ...this._merge(opts),\n      model,\n      provider,\n      latencyMs,\n    } as Parameters<AmplitudeAI['trackEmbedding']>[0]);\n  }\n\n  trackSpan(spanName: string, latencyMs: number, opts: SpanOpts = {}): string {\n    return this._ai.trackSpan({\n      ...this._merge(opts),\n      spanName,\n      latencyMs,\n    } as Parameters<AmplitudeAI['trackSpan']>[0]);\n  }\n\n  trackSessionEnd(opts: SessionEndOpts = {}): void {\n    this._ai.trackSessionEnd(\n      this._merge(opts) as Parameters<AmplitudeAI['trackSessionEnd']>[0],\n    );\n  }\n\n  trackSessionEnrichment(\n    enrichments: SessionEnrichments,\n    opts: SessionEnrichmentOpts = {},\n  ): void {\n    this._ai.trackSessionEnrichment({\n      ...this._merge(opts),\n      enrichments,\n    } as Parameters<AmplitudeAI['trackSessionEnrichment']>[0]);\n  }\n\n  score(\n    name: string,\n    value: number,\n    targetId: string,\n    opts: ScoreOpts = {},\n  ): void {\n    this._ai.score({\n      ...this._merge(opts),\n      name,\n      value,\n      targetId,\n    } as Parameters<AmplitudeAI['score']>[0]);\n  }\n\n  session(\n    opts: {\n      sessionId?: string | null;\n      idleTimeoutMinutes?: number | null;\n      userId?: string | null;\n      deviceId?: string | null;\n      browserSessionId?: string | null;\n      autoFlush?: boolean;\n    } = {},\n  ): Session {\n    return new Session(this, opts);\n  }\n\n  flush(): unknown {\n    return this._ai.flush();\n  }\n\n  shutdown(): void {\n    this._ai.shutdown();\n  }\n}\n"],"mappings":";;;;AAqDA,MAAM,iBAAiB;CACrB;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACD;AAiBD,IAAa,aAAb,MAAa,WAAW;CACtB,AAAS;CACT,AAAS;CAET,YACE,IACA,MACA;AACA,OAAK,MAAM;AACX,OAAK,YAAY;GACf,QAAQ,KAAK,UAAU;GACvB,SAAS,KAAK;GACd,eAAe,KAAK,iBAAiB;GACrC,eAAe,KAAK,iBAAiB;GACrC,cAAc,KAAK,gBAAgB;GACnC,aAAa,KAAK,eAAe;GACjC,SAAS,KAAK,WAAW;GACzB,KAAK,KAAK,OAAO;GACjB,WAAW,KAAK,aAAa;GAC7B,SAAS,KAAK,WAAW;GACzB,QAAQ,KAAK,UAAU;GACvB,UAAU,KAAK,YAAY;GAC3B,kBAAkB,KAAK,oBAAoB;GAC5C;;CAGH,IAAI,UAAkB;AACpB,SAAO,OAAO,KAAK,UAAU,QAAQ;;CAGvC,IAAI,KAAkB;AACpB,SAAO,KAAK;;CAGd,MAAM,SAAiB,YAA0B,EAAE,EAAc;EAC/D,MAAMA,YAAqC;GACzC,QAAQ,KAAK,UAAU;GACvB,KAAK,KAAK,UAAU;GACpB,eAAe,KAAK,UAAU;GAC9B,cAAc,KAAK,UAAU;GAC7B,aAAa,KAAK,UAAU;GAC5B,WAAW,KAAK,UAAU;GAC1B,SAAS,KAAK,UAAU;GACxB,QAAQ,KAAK,UAAU;GACvB,UAAU,KAAK,UAAU;GACzB,kBAAkB,KAAK,UAAU;GAClC;EAED,MAAM,YAAY,KAAK,UAAU;EACjC,MAAM,WAAW,UAAU,WAAW;EACtC,MAAM,kBAAkB,OAAO,QAAQ,UAAU,CAAC,QAC/C,CAAC,OAAO,MAAM,UAChB;AACD,MAAI,aAAa,UAAU;GACzB,MAAM,SAAS,EAAE,GAAI,aAAa,EAAE,EAAG;AACvC,OAAI,SAAU,QAAO,OAAO,QAAQ,SAAS;AAC7C,aAAU,UAAU;;AAGtB,OAAK,MAAM,CAAC,GAAG,MAAM,gBACnB,KAAI,KAAK,KAAM,WAAU,KAAK;EAGhC,MAAM,EAAE,eAAe,wBAAwB,GAAG,2BAChD;EACF,MAAM,iBACH,0BACA,KAAK,UAAU;AAElB,SAAO,IAAI,WAAW,KAAK,KAAK;GAC9B;GACA,eAAe;GACf,GAAI;GACL,CAAC;;CAGJ,AAAQ,OACN,QACA,SAA4B,gBACzB;EACH,MAAM,SAAS,EAAE,GAAG,QAAQ;AAC5B,OAAK,MAAM,SAAS,OAClB,KAAI,OAAO,UAAU,QAAQ,KAAK,UAAU,UAAU,KACpD,QAAO,SAAS,KAAK,UAAU;EAGnC,MAAM,WAAW,KAAK,UAAU;EAChC,MAAM,mBAAmB,KAAK,UAAU;AACxC,MAAI,YAAY,kBAAkB;GAChC,MAAM,aAAa,OAAO;GAG1B,MAAM,KAAK,cAAc,OAAO,EAAE,GAAG,YAAY,GAAG,EAAE;AACtD,OAAI,EAAE,0BAA0B,KAAK;AACnC,OAAG,0BAA0B,GAAG,SAAS,GAAG;AAC5C,WAAO,kBAAkB;;;AAG7B,SAAO;;CAGT,iBAAiB,SAAiB,OAAwB,EAAE,EAAU;AACpE,SAAO,KAAK,IAAI,iBAAiB;GAC/B,GAAG,KAAK,OAAO,KAAK;GACpB;GACD,CAAmD;;CAGtD,eACE,SACA,OACA,UACA,WACA,OAAsB,EAAE,EAChB;AACR,SAAO,KAAK,IAAI,eAAe;GAC7B,GAAG,KAAK,OAAO,KAAK;GACpB;GACA;GACA;GACA;GACD,CAAiD;;CAGpD,cACE,UACA,WACA,SACA,OAAqB,EAAE,EACf;AACR,SAAO,KAAK,IAAI,cAAc;GAC5B,GAAG,KAAK,OAAO,KAAK;GACpB;GACA;GACA;GACD,CAAgD;;CAGnD,eACE,OACA,UACA,WACA,OAAsB,EAAE,EAChB;AACR,SAAO,KAAK,IAAI,eAAe;GAC7B,GAAG,KAAK,OAAO,KAAK;GACpB;GACA;GACA;GACD,CAAiD;;CAGpD,UAAU,UAAkB,WAAmB,OAAiB,EAAE,EAAU;AAC1E,SAAO,KAAK,IAAI,UAAU;GACxB,GAAG,KAAK,OAAO,KAAK;GACpB;GACA;GACD,CAA4C;;CAG/C,gBAAgB,OAAuB,EAAE,EAAQ;AAC/C,OAAK,IAAI,gBACP,KAAK,OAAO,KAAK,CAClB;;CAGH,uBACE,aACA,OAA8B,EAAE,EAC1B;AACN,OAAK,IAAI,uBAAuB;GAC9B,GAAG,KAAK,OAAO,KAAK;GACpB;GACD,CAAyD;;CAG5D,MACE,MACA,OACA,UACA,OAAkB,EAAE,EACd;AACN,OAAK,IAAI,MAAM;GACb,GAAG,KAAK,OAAO,KAAK;GACpB;GACA;GACA;GACD,CAAwC;;CAG3C,QACE,OAOI,EAAE,EACG;AACT,SAAO,IAAI,QAAQ,MAAM,KAAK;;CAGhC,QAAiB;AACf,SAAO,KAAK,IAAI,OAAO;;CAGzB,WAAiB;AACf,OAAK,IAAI,UAAU"}

package/dist/cli/status.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"status.d.ts","names":[],"sources":["../../src/cli/status.ts"],"sourcesContent":[],"mappings":";KAIK,YAAA;EAAA,OAAA,EAAA,MAAY;EAYX,SAAA,EAVO,KAkDZ,CAAA;;;;OAjDM;;;cASD,4BAA2B"}
+{"version":3,"file":"status.d.ts","names":[],"sources":["../../src/cli/status.ts"],"sourcesContent":[],"mappings":";KAKK,YAAA;EAAA,OAAA,EAAA,MAAY;EAmCX,SAAA,EAjCO,KAwDZ,CAAA;;;;OAvDM;;;cAgCD,4BAA2B"}

package/dist/cli/status.js

@@ -1,34 +1,46 @@
 import { PROVIDER_ENTRIES } from "./providers.js";
 import { existsSync, readFileSync } from "node:fs";
-import { join } from "node:path";
+import { dirname, join } from "node:path";
+import { fileURLToPath } from "node:url";
 
 //#region src/cli/status.ts
 const isPackageInstalled = (cwd, pkg) => {
 	return existsSync(join(cwd, "node_modules", ...pkg.split("/")));
 };
-const runStatus = (cwd) => {
-	let version = "unknown";
+function resolveVersion(cwd) {
+	try {
+		const nmPkgPath = join(cwd, "node_modules", "@amplitude", "ai", "package.json");
+		if (existsSync(nmPkgPath)) {
+			const pkg = JSON.parse(readFileSync(nmPkgPath, "utf8"));
+			if (pkg.version) return pkg.version;
+		}
+	} catch {}
 	try {
-		const pkgPath = join(cwd, "node_modules", "@amplitude", "ai", "package.json");
-		if (existsSync(pkgPath)) version = JSON.parse(readFileSync(pkgPath, "utf8")).version ?? "unknown";
+		const selfPkgPath = join(dirname(fileURLToPath(import.meta.url)), "..", "..", "package.json");
+		if (existsSync(selfPkgPath)) {
+			const pkg = JSON.parse(readFileSync(selfPkgPath, "utf8"));
+			if (pkg.version) return pkg.version;
+		}
 	} catch {}
+	return "unknown";
+}
+const runStatus = (cwd) => {
+	const version = resolveVersion(cwd);
 	const installedCache = /* @__PURE__ */ new Map();
-	const providers = PROVIDER_ENTRIES.map(({ name, npm }) => {
-		if (!installedCache.has(npm)) installedCache.set(npm, isPackageInstalled(cwd, npm));
-		return {
-			name,
-			installed: installedCache.get(npm) ?? false
-		};
-	});
-	const env = {
-		AMPLITUDE_AI_API_KEY: Boolean(process.env.AMPLITUDE_AI_API_KEY),
-		AMPLITUDE_AI_CONTENT_MODE: Boolean(process.env.AMPLITUDE_AI_CONTENT_MODE),
-		AMPLITUDE_AI_DEBUG: Boolean(process.env.AMPLITUDE_AI_DEBUG)
-	};
 	return {
 		version,
-		providers,
-		env,
+		providers: PROVIDER_ENTRIES.map(({ name, npm }) => {
+			if (!installedCache.has(npm)) installedCache.set(npm, isPackageInstalled(cwd, npm));
+			return {
+				name,
+				installed: installedCache.get(npm) ?? false
+			};
+		}),
+		env: {
+			AMPLITUDE_AI_API_KEY: Boolean(process.env.AMPLITUDE_AI_API_KEY),
+			AMPLITUDE_AI_CONTENT_MODE: Boolean(process.env.AMPLITUDE_AI_CONTENT_MODE),
+			AMPLITUDE_AI_DEBUG: Boolean(process.env.AMPLITUDE_AI_DEBUG)
+		},
 		patchActive: false
 	};
 };

package/dist/cli/status.js.map

@@ -1 +1 @@
-{"version":3,"file":"status.js","names":["env: Record<string, boolean>"],"sources":["../../src/cli/status.ts"],"sourcesContent":["import { existsSync, readFileSync } from 'node:fs';\nimport { join } from 'node:path';\nimport { PROVIDER_ENTRIES } from './providers.js';\n\ntype StatusResult = {\n  version: string;\n  providers: Array<{ name: string; installed: boolean }>;\n  env: Record<string, boolean>;\n  patchActive: boolean;\n};\n\nconst isPackageInstalled = (cwd: string, pkg: string): boolean => {\n  const nmPath = join(cwd, 'node_modules', ...pkg.split('/'));\n  return existsSync(nmPath);\n};\n\nconst runStatus = (cwd: string): StatusResult => {\n  let version = 'unknown';\n  try {\n    const pkgPath = join(\n      cwd,\n      'node_modules',\n      '@amplitude',\n      'ai',\n      'package.json',\n    );\n    if (existsSync(pkgPath)) {\n      const pkg = JSON.parse(readFileSync(pkgPath, 'utf8')) as {\n        version?: string;\n      };\n      version = pkg.version ?? 'unknown';\n    }\n  } catch {\n    // fall through\n  }\n\n  const installedCache = new Map<string, boolean>();\n  const providers = PROVIDER_ENTRIES.map(({ name, npm }) => {\n    if (!installedCache.has(npm)) {\n      installedCache.set(npm, isPackageInstalled(cwd, npm));\n    }\n    return { name, installed: installedCache.get(npm) ?? false };\n  });\n\n  const env: Record<string, boolean> = {\n    AMPLITUDE_AI_API_KEY: Boolean(process.env.AMPLITUDE_AI_API_KEY),\n    AMPLITUDE_AI_CONTENT_MODE: Boolean(process.env.AMPLITUDE_AI_CONTENT_MODE),\n    AMPLITUDE_AI_DEBUG: Boolean(process.env.AMPLITUDE_AI_DEBUG),\n  };\n\n  return {\n    version,\n    providers,\n    env,\n    patchActive: false,\n  };\n};\n\nexport { runStatus, type StatusResult };\n"],"mappings":";;;;;AAWA,MAAM,sBAAsB,KAAa,QAAyB;AAEhE,QAAO,WADQ,KAAK,KAAK,gBAAgB,GAAG,IAAI,MAAM,IAAI,CAAC,CAClC;;AAG3B,MAAM,aAAa,QAA8B;CAC/C,IAAI,UAAU;AACd,KAAI;EACF,MAAM,UAAU,KACd,KACA,gBACA,cACA,MACA,eACD;AACD,MAAI,WAAW,QAAQ,CAIrB,WAHY,KAAK,MAAM,aAAa,SAAS,OAAO,CAAC,CAGvC,WAAW;SAErB;CAIR,MAAM,iCAAiB,IAAI,KAAsB;CACjD,MAAM,YAAY,iBAAiB,KAAK,EAAE,MAAM,UAAU;AACxD,MAAI,CAAC,eAAe,IAAI,IAAI,CAC1B,gBAAe,IAAI,KAAK,mBAAmB,KAAK,IAAI,CAAC;AAEvD,SAAO;GAAE;GAAM,WAAW,eAAe,IAAI,IAAI,IAAI;GAAO;GAC5D;CAEF,MAAMA,MAA+B;EACnC,sBAAsB,QAAQ,QAAQ,IAAI,qBAAqB;EAC/D,2BAA2B,QAAQ,QAAQ,IAAI,0BAA0B;EACzE,oBAAoB,QAAQ,QAAQ,IAAI,mBAAmB;EAC5D;AAED,QAAO;EACL;EACA;EACA;EACA,aAAa;EACd"}
+{"version":3,"file":"status.js","names":[],"sources":["../../src/cli/status.ts"],"sourcesContent":["import { existsSync, readFileSync } from 'node:fs';\nimport { dirname, join } from 'node:path';\nimport { fileURLToPath } from 'node:url';\nimport { PROVIDER_ENTRIES } from './providers.js';\n\ntype StatusResult = {\n  version: string;\n  providers: Array<{ name: string; installed: boolean }>;\n  env: Record<string, boolean>;\n  patchActive: boolean;\n};\n\nconst isPackageInstalled = (cwd: string, pkg: string): boolean => {\n  const nmPath = join(cwd, 'node_modules', ...pkg.split('/'));\n  return existsSync(nmPath);\n};\n\nfunction resolveVersion(cwd: string): string {\n  // First try the user's project node_modules\n  try {\n    const nmPkgPath = join(cwd, 'node_modules', '@amplitude', 'ai', 'package.json');\n    if (existsSync(nmPkgPath)) {\n      const pkg = JSON.parse(readFileSync(nmPkgPath, 'utf8')) as { version?: string };\n      if (pkg.version) return pkg.version;\n    }\n  } catch { /* fall through */ }\n\n  // Fallback: read our own package.json (works when running from the SDK repo)\n  try {\n    const selfDir = dirname(fileURLToPath(import.meta.url));\n    const selfPkgPath = join(selfDir, '..', '..', 'package.json');\n    if (existsSync(selfPkgPath)) {\n      const pkg = JSON.parse(readFileSync(selfPkgPath, 'utf8')) as { version?: string };\n      if (pkg.version) return pkg.version;\n    }\n  } catch { /* fall through */ }\n\n  return 'unknown';\n}\n\nconst runStatus = (cwd: string): StatusResult => {\n  const version = resolveVersion(cwd);\n\n  const installedCache = new Map<string, boolean>();\n  const providers = PROVIDER_ENTRIES.map(({ name, npm }) => {\n    if (!installedCache.has(npm)) {\n      installedCache.set(npm, isPackageInstalled(cwd, npm));\n    }\n    return { name, installed: installedCache.get(npm) ?? false };\n  });\n\n  const env: Record<string, boolean> = {\n    AMPLITUDE_AI_API_KEY: Boolean(process.env.AMPLITUDE_AI_API_KEY),\n    AMPLITUDE_AI_CONTENT_MODE: Boolean(process.env.AMPLITUDE_AI_CONTENT_MODE),\n    AMPLITUDE_AI_DEBUG: Boolean(process.env.AMPLITUDE_AI_DEBUG),\n  };\n\n  return {\n    version,\n    providers,\n    env,\n    patchActive: false,\n  };\n};\n\nexport { runStatus, type StatusResult };\n"],"mappings":";;;;;;AAYA,MAAM,sBAAsB,KAAa,QAAyB;AAEhE,QAAO,WADQ,KAAK,KAAK,gBAAgB,GAAG,IAAI,MAAM,IAAI,CAAC,CAClC;;AAG3B,SAAS,eAAe,KAAqB;AAE3C,KAAI;EACF,MAAM,YAAY,KAAK,KAAK,gBAAgB,cAAc,MAAM,eAAe;AAC/E,MAAI,WAAW,UAAU,EAAE;GACzB,MAAM,MAAM,KAAK,MAAM,aAAa,WAAW,OAAO,CAAC;AACvD,OAAI,IAAI,QAAS,QAAO,IAAI;;SAExB;AAGR,KAAI;EAEF,MAAM,cAAc,KADJ,QAAQ,cAAc,OAAO,KAAK,IAAI,CAAC,EACrB,MAAM,MAAM,eAAe;AAC7D,MAAI,WAAW,YAAY,EAAE;GAC3B,MAAM,MAAM,KAAK,MAAM,aAAa,aAAa,OAAO,CAAC;AACzD,OAAI,IAAI,QAAS,QAAO,IAAI;;SAExB;AAER,QAAO;;AAGT,MAAM,aAAa,QAA8B;CAC/C,MAAM,UAAU,eAAe,IAAI;CAEnC,MAAM,iCAAiB,IAAI,KAAsB;AAcjD,QAAO;EACL;EACA,WAfgB,iBAAiB,KAAK,EAAE,MAAM,UAAU;AACxD,OAAI,CAAC,eAAe,IAAI,IAAI,CAC1B,gBAAe,IAAI,KAAK,mBAAmB,KAAK,IAAI,CAAC;AAEvD,UAAO;IAAE;IAAM,WAAW,eAAe,IAAI,IAAI,IAAI;IAAO;IAC5D;EAWA,KATmC;GACnC,sBAAsB,QAAQ,QAAQ,IAAI,qBAAqB;GAC/D,2BAA2B,QAAQ,QAAQ,IAAI,0BAA0B;GACzE,oBAAoB,QAAQ,QAAQ,IAAI,mBAAmB;GAC5D;EAMC,aAAa;EACd"}

package/dist/client.d.ts

@@ -3,7 +3,7 @@ import { AIConfig } from "./config.js";
 import { MessageLabel, SessionEnrichments } from "./core/enrichments.js";
 import { TenantHandle } from "./tenant.js";
 import { AmplitudeClientLike, Attachment } from "./types.js";
-import { BoundAgent } from "./bound-agent.js";
+import { AgentOptions, BoundAgent } from "./bound-agent.js";
 
 //#region src/client.d.ts
 
@@ -31,11 +31,14 @@ declare class AmplitudeAI {
   protected _config: AIConfig;
   protected _privacyConfig: PrivacyConfig;
   protected _sessionTurnCounters: Map<string, number>;
+  /** @internal Tracks events since last flush() — used by the exit warning. */
+  _trackCountSinceFlush: number;
   constructor(options: {
     amplitude?: AmplitudeClientLike;
     apiKey?: string;
     config?: AIConfig;
   });
+  private _installTrackCounter;
   private _installTrackHook;
   get amplitude(): AmplitudeClientLike;
   get config(): AIConfig;
@@ -47,10 +50,12 @@ declare class AmplitudeAI {
     traceId?: string | null;
     turnId?: number | null;
     messageId?: string | null;
+    messageSource?: string | null;
     agentId?: string | null;
     parentAgentId?: string | null;
     customerOrgId?: string | null;
     agentVersion?: string | null;
+    description?: string | null;
     context?: Record<string, unknown> | null;
     env?: string | null;
     isRegeneration?: boolean;
@@ -96,6 +101,7 @@ declare class AmplitudeAI {
     parentAgentId?: string | null;
     customerOrgId?: string | null;
     agentVersion?: string | null;
+    description?: string | null;
     context?: Record<string, unknown> | null;
     env?: string | null;
     isError?: boolean;
@@ -121,6 +127,7 @@ declare class AmplitudeAI {
     parentAgentId?: string | null;
     customerOrgId?: string | null;
     agentVersion?: string | null;
+    description?: string | null;
     context?: Record<string, unknown> | null;
     env?: string | null;
     errorMessage?: string | null;
@@ -142,6 +149,7 @@ declare class AmplitudeAI {
     parentAgentId?: string | null;
     customerOrgId?: string | null;
     agentVersion?: string | null;
+    description?: string | null;
     context?: Record<string, unknown> | null;
     env?: string | null;
     eventProperties?: Record<string, unknown> | null;
@@ -163,6 +171,7 @@ declare class AmplitudeAI {
     parentAgentId?: string | null;
     customerOrgId?: string | null;
     agentVersion?: string | null;
+    description?: string | null;
     context?: Record<string, unknown> | null;
     env?: string | null;
     eventProperties?: Record<string, unknown> | null;
@@ -178,6 +187,7 @@ declare class AmplitudeAI {
     parentAgentId?: string | null;
     customerOrgId?: string | null;
     agentVersion?: string | null;
+    description?: string | null;
     context?: Record<string, unknown> | null;
     env?: string | null;
     abandonmentTurn?: number | null;
@@ -195,6 +205,7 @@ declare class AmplitudeAI {
     parentAgentId?: string | null;
     customerOrgId?: string | null;
     agentVersion?: string | null;
+    description?: string | null;
     context?: Record<string, unknown> | null;
     env?: string | null;
     eventProperties?: Record<string, unknown> | null;
@@ -215,24 +226,13 @@ declare class AmplitudeAI {
     parentAgentId?: string | null;
     customerOrgId?: string | null;
     agentVersion?: string | null;
+    description?: string | null;
     context?: Record<string, unknown> | null;
     env?: string | null;
     eventProperties?: Record<string, unknown> | null;
     groups?: Record<string, unknown> | null;
   }): void;
-  agent(agentId: string, opts?: {
-    userId?: string | null;
-    parentAgentId?: string | null;
-    customerOrgId?: string | null;
-    agentVersion?: string | null;
-    context?: Record<string, unknown> | null;
-    env?: string | null;
-    sessionId?: string | null;
-    traceId?: string | null;
-    groups?: Record<string, unknown> | null;
-    deviceId?: string | null;
-    browserSessionId?: string | null;
-  }): BoundAgent;
+  agent(agentId: string, opts?: AgentOptions): BoundAgent;
   tenant(customerOrgId: string, opts?: {
     groups?: Record<string, unknown> | null;
     env?: string | null;

package/dist/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","names":[],"sources":["../src/client.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;AA+CA;;;;;;;;;;;;;;;;AAsMkB,cAtML,WAAA,CAsMK;EACL,UAAA,UAAA,EAtMW,mBAsMX;EAQC,UAAA,WAAA,EAAA,OAAA;EAKQ,UAAA,OAAA,EAjND,QAiNC;EACT,UAAA,cAAA,EAjNe,aAiNf;EACO,UAAA,oBAAA,EAjNc,GAiNd,CAAA,MAAA,EAAA,MAAA,CAAA;EA4FN,WAAA,CAAA,OAAA,EAAA;IAGQ,SAAA,CAAA,EA7SN,mBA6SM;IACT,MAAA,CAAA,EAAA,MAAA;IA2CC,MAAA,CAAA,EAvVD,QAuVC;EAEQ,CAAA;EACT,QAAA,iBAAA;EA+BI,IAAA,SAAA,CAAA,CAAA,EAjSE,mBAiSF;EACC,IAAA,MAAA,CAAA,CAAA,EA9RF,QA8RE;EAUJ,WAAA,CAAA,SAAA,EAAA,MAAA,CAAA,EAAA,MAAA;EAEQ,gBAAA,CAAA,IAAA,EAAA;IACT,MAAA,EAAA,MAAA;IAkCK,OAAA,EAAA,MAAA;IAOJ,SAAA,EAAA,MAAA;IAIQ,OAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IACT,MAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IA2BI,SAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAOH,OAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAEQ,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IACT,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAwCC,YAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAEQ,OAAA,CAAA,EAtYR,MAsYQ,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;IACT,GAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAqCG,cAAA,CAAA,EAAA,OAAA;IAID,MAAA,CAAA,EAAA,OAAA;IAIV,eAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAUgB,WAAA,CAAA,EAzbH,UAybG,EAAA,GAAA,IAAA;IAChB,MAAA,CAAA,EAzbQ,YAybR,EAAA,GAAA,IAAA;IAQO,eAAA,CAAA,EAhcU,MAgcV,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;IAAM,MAAA,CAAA,EA/bL,MA+bK,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;;;;;;;;;;;;;;;;;gBApZF,MAAM;;;;;;;;;;;kBAWJ;aACL;;;;;;;;cAQC;;;;;sBAKQ;aACT;oBACO;;;;;;;;;;;;;;;;;;cA4FN;;;sBAGQ;aACT;;;;;;;;;;;;;;;;;cA2CC;;sBAEQ;aACT;;;;;;;iBA+BI;kBACC;;;;;;;;;;cAUJ;;sBAEQ;aACT;;;;;kBAkCK;;;;;;;cAOJ;;;;sBAIQ;aACT;;;;;iBA2BI;;;;;;;cAOH;;sBAEQ;aACT;;;;;;;;;;;;;;;;;cAwCC;;sBAEQ;aACT;;;;;;;cAqCG;;;;aAID;;;MAIV;;aAUgB;;MAChB;YAQO"}
+{"version":3,"file":"client.d.ts","names":[],"sources":["../src/client.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;AAsEA;;;;;;;;;;;;;;;;AAwNkB,cAxNL,WAAA,CAwNK;EACL,UAAA,UAAA,EAxNW,mBAwNX;EASC,UAAA,WAAA,EAAA,OAAA;EAKQ,UAAA,OAAA,EApOD,QAoOC;EACT,UAAA,cAAA,EApOe,aAoOf;EACO,UAAA,oBAAA,EApOc,GAoOd,CAAA,MAAA,EAAA,MAAA,CAAA;EA8FN;EAGQ,qBAAA,EAAA,MAAA;EACT,WAAA,CAAA,OAAA,EAAA;IA6CC,SAAA,CAAA,EA9WE,mBA8WF;IAEQ,MAAA,CAAA,EAAA,MAAA;IACT,MAAA,CAAA,EA/WA,QA+WA;EAgCI,CAAA;EACC,QAAA,oBAAA;EAWJ,QAAA,iBAAA;EAEQ,IAAA,SAAA,CAAA,CAAA,EAzTH,mBAyTG;EACT,IAAA,MAAA,CAAA,CAAA,EAtTG,QAsTH;EAmCK,WAAA,CAAA,SAAA,EAAA,MAAA,CAAA,EAAA,MAAA;EAQJ,gBAAA,CAAA,IAAA,EAAA;IAIQ,MAAA,EAAA,MAAA;IACT,OAAA,EAAA,MAAA;IA4BI,SAAA,EAAA,MAAA;IAQH,OAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAEQ,MAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IACT,SAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IA0CC,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAEQ,OAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IACT,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IA+BkB,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAAoB,YAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAU9B,WAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAChB,OAAA,CAAA,EAhcS,MAgcT,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;IAQO,GAAA,CAAA,EAAA,MAAA,GAAA,IAAA;IAAM,cAAA,CAAA,EAAA,OAAA;;;kBAncA;aACL;sBACS;aACT;;;;;;;;;;;;;;;;;gBA6CG,MAAM;;;;;;;;;;;kBAWJ;aACL;;;;;;;;;cASC;;;;;sBAKQ;aACT;oBACO;;;;;;;;;;;;;;;;;;;cA8FN;;;sBAGQ;aACT;;;;;;;;;;;;;;;;;;cA6CC;;sBAEQ;aACT;;;;;;;iBAgCI;kBACC;;;;;;;;;;;cAWJ;;sBAEQ;aACT;;;;;kBAmCK;;;;;;;;cAQJ;;;;sBAIQ;aACT;;;;;iBA4BI;;;;;;;;cAQH;;sBAEQ;aACT;;;;;;;;;;;;;;;;;;cA0CC;;sBAEQ;aACT;;gCA+BkB,eAAoB;;aAU9B;;MAChB;YAQO"}

package/dist/client.js

@@ -1,5 +1,6 @@
 import { ConfigurationError } from "./exceptions.js";
 import { trackAiMessage, trackEmbedding, trackScore, trackSessionEnd, trackSessionEnrichment, trackSpan, trackToolCall, trackUserMessage } from "./core/tracking.js";
+import { isServerless } from "./serverless.js";
 import { BoundAgent } from "./bound-agent.js";
 import { AIConfig } from "./config.js";
 import { setDefaultPropagateContext } from "./propagation.js";
@@ -11,6 +12,19 @@ import { formatDebugLine, formatDryRunLine } from "./utils/debug.js";
 
 //#region src/client.ts
 const _MAX_SESSION_TURN_COUNTERS = 1e4;
+const _activeInstances = /* @__PURE__ */ new Set();
+let _exitHookRegistered = false;
+function _registerExitHook() {
+	if (_exitHookRegistered) return;
+	_exitHookRegistered = true;
+	process.on("beforeExit", () => {
+		if (!isServerless()) return;
+		for (const instance of _activeInstances) if (instance._trackCountSinceFlush > 0) {
+			console.warn(`⚠️  AmplitudeAI: ${instance._trackCountSinceFlush} event(s) were tracked but never flushed. In serverless environments, call \`await ai.flush()\` before your handler returns, or use session.run() which auto-flushes by default.`);
+			break;
+		}
+	});
+}
 /**
 * Main Amplitude AI client for tracking LLM interactions.
 *
@@ -35,6 +49,8 @@ var AmplitudeAI = class {
 	_config;
 	_privacyConfig;
 	_sessionTurnCounters = /* @__PURE__ */ new Map();
+	/** @internal Tracks events since last flush() — used by the exit warning. */
+	_trackCountSinceFlush = 0;
 	constructor(options) {
 		if (options.amplitude != null) {
 			this._amplitude = options.amplitude;
@@ -50,6 +66,16 @@ var AmplitudeAI = class {
 		this._privacyConfig = this._config.toPrivacyConfig();
 		setDefaultPropagateContext(this._config.propagateContext);
 		if (this._config.debug || this._config.dryRun || this._config.onEventCallback != null) this._installTrackHook();
+		this._installTrackCounter();
+		_activeInstances.add(this);
+		_registerExitHook();
+	}
+	_installTrackCounter() {
+		const originalTrack = this._amplitude.track.bind(this._amplitude);
+		this._amplitude.track = (event) => {
+			this._trackCountSinceFlush++;
+			return originalTrack(event);
+		};
 	}
 	_installTrackHook() {
 		const originalTrack = this._amplitude.track.bind(this._amplitude);
@@ -107,10 +133,12 @@ var AmplitudeAI = class {
 			traceId: opts.traceId,
 			turnId: effectiveTurnId,
 			messageId: opts.messageId,
+			messageSource: opts.messageSource ?? "user",
 			agentId: opts.agentId,
 			parentAgentId: opts.parentAgentId,
 			customerOrgId: opts.customerOrgId,
 			agentVersion: opts.agentVersion,
+			description: opts.description,
 			context: opts.context,
 			env: opts.env,
 			isRegeneration: opts.isRegeneration,
@@ -173,6 +201,7 @@ var AmplitudeAI = class {
 			parentAgentId: opts.parentAgentId,
 			customerOrgId: opts.customerOrgId,
 			agentVersion: opts.agentVersion,
+			description: opts.description,
 			context: opts.context,
 			env: opts.env,
 			isError: opts.isError,
@@ -201,6 +230,7 @@ var AmplitudeAI = class {
 			parentAgentId: opts.parentAgentId,
 			customerOrgId: opts.customerOrgId,
 			agentVersion: opts.agentVersion,
+			description: opts.description,
 			context: opts.context,
 			env: opts.env,
 			errorMessage: opts.errorMessage,
@@ -226,6 +256,7 @@ var AmplitudeAI = class {
 			parentAgentId: opts.parentAgentId,
 			customerOrgId: opts.customerOrgId,
 			agentVersion: opts.agentVersion,
+			description: opts.description,
 			context: opts.context,
 			env: opts.env,
 			eventProperties: opts.eventProperties,
@@ -251,6 +282,7 @@ var AmplitudeAI = class {
 			parentAgentId: opts.parentAgentId,
 			customerOrgId: opts.customerOrgId,
 			agentVersion: opts.agentVersion,
+			description: opts.description,
 			context: opts.context,
 			env: opts.env,
 			eventProperties: opts.eventProperties,
@@ -270,6 +302,7 @@ var AmplitudeAI = class {
 			parentAgentId: opts.parentAgentId,
 			customerOrgId: opts.customerOrgId,
 			agentVersion: opts.agentVersion,
+			description: opts.description,
 			context: opts.context,
 			env: opts.env,
 			abandonmentTurn: opts.abandonmentTurn,
@@ -292,6 +325,7 @@ var AmplitudeAI = class {
 			parentAgentId: opts.parentAgentId,
 			customerOrgId: opts.customerOrgId,
 			agentVersion: opts.agentVersion,
+			description: opts.description,
 			context: opts.context,
 			env: opts.env,
 			eventProperties: opts.eventProperties,
@@ -316,6 +350,7 @@ var AmplitudeAI = class {
 			parentAgentId: opts.parentAgentId,
 			customerOrgId: opts.customerOrgId,
 			agentVersion: opts.agentVersion,
+			description: opts.description,
 			context: opts.context,
 			env: opts.env,
 			eventProperties: opts.eventProperties,
@@ -358,9 +393,12 @@ var AmplitudeAI = class {
 		};
 	}
 	flush() {
+		this._trackCountSinceFlush = 0;
 		return this._amplitude.flush();
 	}
 	shutdown() {
+		_activeInstances.delete(this);
+		this._trackCountSinceFlush = 0;
 		if (this._ownsClient) this._amplitude.shutdown?.();
 	}
 };