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;
  parentAgentId?: string;
  env?: string;
  privacyConfig?: PrivacyConfig | null;
}
declare class AmplitudeCallbackHandler {
  private _ai;
  private _identities;
  private _privacyConfig;
  private _runStartTimes;
  private _runModelNames;
  private _toolInputs;
  private _toolNames;
  constructor(options: CallbackHandlerOptions);
  private _startRun;
  private _forgetRun;
  private _rememberModelName;
  private _trackUserContents;
  handleLLMStart(serialized: Record<string, unknown>, prompts: string[], runId: string, parentRunId?: string): void;
  /**
   * Chat-model path (ChatOpenAI, ChatAnthropic, …). Without this, modern
   * LangChain stacks never emit `[Agent] User Message` because they call
   * `handleChatModelStart` instead of `handleLLMStart`.
   */
  handleChatModelStart(serialized: Record<string, unknown>, messages: unknown[][], runId: string, parentRunId?: string): void;
  handleLLMEnd(output: Record<string, unknown>, runId: string, parentRunId?: string): void;
  handleToolStart(serialized: Record<string, unknown>, input: string, runId: string, parentRunId?: string): void;
  handleToolEnd(output: string, runId: string, parentRunId?: string): void;
  handleToolError(error: unknown, runId: string, parentRunId?: string): void;
  handleChainStart(_serialized: Record<string, unknown>, _inputs: unknown, runId: string, parentRunId?: string): void;
  handleChainEnd(_outputs: unknown, runId: string): void;
  handleChainError(_error: unknown, runId: string): void;
  handleLLMError(error: unknown, runId: string, parentRunId?: string): void;
}
declare function createAmplitudeCallback(options: CallbackHandlerOptions): AmplitudeCallbackHandler;
//#endregion
export { AmplitudeCallbackHandler, CallbackHandlerOptions, createAmplitudeCallback };
//# sourceMappingURL=langchain.d.ts.map