@posthog/ai 8.10.0 → 8.10.2

package/dist/adk/index.d.ts

@@ -1,7 +1,7 @@
-import { BasePlugin, Context, InvocationContext, BaseAgent, LlmRequest, LlmResponse, BaseTool } from '@google/adk';
-import { Content } from '@google/genai';
-import { PostHog } from 'posthog-node';
-
+import { BaseAgent, BasePlugin, BaseTool, Context, InvocationContext, LlmRequest, LlmResponse } from "@google/adk";
+import { Content } from "@google/genai";
+import { PostHog } from "posthog-node";
+//#region src/adk/plugin.d.ts
 /**
  * Resolver for the PostHog distinct ID. Either a static string, or a function
  * that derives it from the ADK model callback context (e.g. from
@@ -10,30 +10,30 @@ import { PostHog } from 'posthog-node';
  */
 type DistinctIdResolver = string | ((context: Context) => string | null | undefined);
 interface PostHogADKPluginOptions {
-    /** The PostHog client used to capture events. */
-    client: PostHog;
-    /**
-     * Distinct ID for the captured events. When omitted, the ADK
-     * `callbackContext.userId` is used if present, otherwise events are captured
-     * anonymously (personless) keyed by the trace ID.
-     */
-    distinctId?: DistinctIdResolver;
-    /**
-     * Provider label recorded on `$ai_provider`. Defaults to `'gemini'` since ADK
-     * defaults to Gemini models. Set this when routing ADK to another provider so
-     * PostHog attributes cost to the right model catalog.
-     */
-    provider?: string;
-    /** Redacts captured input/output content when true. Defaults to false. */
-    privacyMode?: boolean;
-    /** Group analytics mapping (group type -> group id) attached to every event. */
-    groups?: Record<string, string | number>;
-    /** Extra properties merged into every captured AI event. */
-    properties?: Record<string, unknown>;
-    /** Awaits event delivery instead of batching. Useful in serverless environments. */
-    captureImmediate?: boolean;
-    /** Invoked when the plugin fails to capture an event. Never throws into the model flow. */
-    onError?: (error: unknown) => void;
+  /** The PostHog client used to capture events. */
+  client: PostHog;
+  /**
+   * Distinct ID for the captured events. When omitted, the ADK
+   * `callbackContext.userId` is used if present, otherwise events are captured
+   * anonymously (personless) keyed by the trace ID.
+   */
+  distinctId?: DistinctIdResolver;
+  /**
+   * Provider label recorded on `$ai_provider`. Defaults to `'gemini'` since ADK
+   * defaults to Gemini models. Set this when routing ADK to another provider so
+   * PostHog attributes cost to the right model catalog.
+   */
+  provider?: string;
+  /** Redacts captured input/output content when true. Defaults to false. */
+  privacyMode?: boolean;
+  /** Group analytics mapping (group type -> group id) attached to every event. */
+  groups?: Record<string, string | number>;
+  /** Extra properties merged into every captured AI event. */
+  properties?: Record<string, unknown>;
+  /** Awaits event delivery instead of batching. Useful in serverless environments. */
+  captureImmediate?: boolean;
+  /** Invoked when the plugin fails to capture an event. Never throws into the model flow. */
+  onError?: (error: unknown) => void;
 }
 /**
  * A Google ADK (`@google/adk`) `BasePlugin` that captures PostHog AI traces,
@@ -65,85 +65,85 @@ interface PostHogADKPluginOptions {
  * ```
  */
 declare class PostHogADKPlugin extends BasePlugin {
-    private readonly _client;
-    private readonly _distinctId?;
-    private readonly _provider;
-    private readonly _privacyMode;
-    private readonly _groups?;
-    private readonly _properties;
-    private readonly _captureImmediate;
-    private readonly _onError?;
-    /** FIFO of in-flight model calls for each invocation branch and agent. */
-    private readonly _pending;
-    private readonly _traces;
-    private readonly _pendingAgents;
-    private readonly _pendingTools;
-    constructor(options: PostHogADKPluginOptions);
-    beforeRunCallback({ invocationContext, }: {
-        invocationContext: InvocationContext;
-    }): Promise<Content | undefined>;
-    afterRunCallback({ invocationContext }: {
-        invocationContext: InvocationContext;
-    }): Promise<void>;
-    beforeAgentCallback({ agent, callbackContext, }: {
-        agent: BaseAgent;
-        callbackContext: Context;
-    }): Promise<Content | undefined>;
-    afterAgentCallback({ callbackContext, }: {
-        agent: BaseAgent;
-        callbackContext: Context;
-    }): Promise<Content | undefined>;
-    beforeModelCallback({ callbackContext, llmRequest, }: {
-        callbackContext: Context;
-        llmRequest: LlmRequest;
-    }): Promise<LlmResponse | undefined>;
-    afterModelCallback({ callbackContext, llmResponse, }: {
-        callbackContext: Context;
-        llmResponse: LlmResponse;
-    }): Promise<LlmResponse | undefined>;
-    onModelErrorCallback({ callbackContext, llmRequest, error, }: {
-        callbackContext: Context;
-        llmRequest: LlmRequest;
-        error: Error;
-    }): Promise<LlmResponse | undefined>;
-    beforeToolCallback({ tool, toolArgs, toolContext, }: {
-        tool: BaseTool;
-        toolArgs: Record<string, unknown>;
-        toolContext: Context;
-    }): Promise<Record<string, unknown> | undefined>;
-    afterToolCallback({ tool, toolContext, result, }: {
-        tool: BaseTool;
-        toolArgs: Record<string, unknown>;
-        toolContext: Context;
-        result: Record<string, unknown>;
-    }): Promise<Record<string, unknown> | undefined>;
-    onToolErrorCallback({ tool, toolContext, error, }: {
-        tool: BaseTool;
-        toolArgs: Record<string, unknown>;
-        toolContext: Context;
-        error: Error;
-    }): Promise<Record<string, unknown> | undefined>;
-    private _capture;
-    private _captureToolSpan;
-    private _captureLifecycleEvent;
-    private _resolveDistinctId;
-    private _resolveInvocationDistinctId;
-    private _rememberContext;
-    private _parentSpanId;
-    private _pendingKey;
-    private _toolKey;
-    private _peekPending;
-    private _takePending;
-    private _takePendingAgent;
-    private _takePendingTool;
-    private _isNonTerminalStreamResponse;
-    private _evictStalePending;
-    private _evictStaleQueueEntries;
-    private _clearPendingInvocation;
-    private _formatInput;
-    private _formatOutput;
-    private _handleError;
+  private readonly _client;
+  private readonly _distinctId?;
+  private readonly _provider;
+  private readonly _privacyMode;
+  private readonly _groups?;
+  private readonly _properties;
+  private readonly _captureImmediate;
+  private readonly _onError?;
+  /** FIFO of in-flight model calls for each invocation branch and agent. */
+  private readonly _pending;
+  private readonly _traces;
+  private readonly _pendingAgents;
+  private readonly _pendingTools;
+  constructor(options: PostHogADKPluginOptions);
+  beforeRunCallback({ invocationContext }: {
+    invocationContext: InvocationContext;
+  }): Promise<Content | undefined>;
+  afterRunCallback({ invocationContext }: {
+    invocationContext: InvocationContext;
+  }): Promise<void>;
+  beforeAgentCallback({ agent, callbackContext }: {
+    agent: BaseAgent;
+    callbackContext: Context;
+  }): Promise<Content | undefined>;
+  afterAgentCallback({ callbackContext }: {
+    agent: BaseAgent;
+    callbackContext: Context;
+  }): Promise<Content | undefined>;
+  beforeModelCallback({ callbackContext, llmRequest }: {
+    callbackContext: Context;
+    llmRequest: LlmRequest;
+  }): Promise<LlmResponse | undefined>;
+  afterModelCallback({ callbackContext, llmResponse }: {
+    callbackContext: Context;
+    llmResponse: LlmResponse;
+  }): Promise<LlmResponse | undefined>;
+  onModelErrorCallback({ callbackContext, llmRequest, error }: {
+    callbackContext: Context;
+    llmRequest: LlmRequest;
+    error: Error;
+  }): Promise<LlmResponse | undefined>;
+  beforeToolCallback({ tool, toolArgs, toolContext }: {
+    tool: BaseTool;
+    toolArgs: Record<string, unknown>;
+    toolContext: Context;
+  }): Promise<Record<string, unknown> | undefined>;
+  afterToolCallback({ tool, toolContext, result }: {
+    tool: BaseTool;
+    toolArgs: Record<string, unknown>;
+    toolContext: Context;
+    result: Record<string, unknown>;
+  }): Promise<Record<string, unknown> | undefined>;
+  onToolErrorCallback({ tool, toolContext, error }: {
+    tool: BaseTool;
+    toolArgs: Record<string, unknown>;
+    toolContext: Context;
+    error: Error;
+  }): Promise<Record<string, unknown> | undefined>;
+  private _capture;
+  private _captureToolSpan;
+  private _captureLifecycleEvent;
+  private _resolveDistinctId;
+  private _resolveInvocationDistinctId;
+  private _rememberContext;
+  private _parentSpanId;
+  private _pendingKey;
+  private _toolKey;
+  private _peekPending;
+  private _takePending;
+  private _takePendingAgent;
+  private _takePendingTool;
+  private _isNonTerminalStreamResponse;
+  private _evictStalePending;
+  private _evictStaleQueueEntries;
+  private _clearPendingInvocation;
+  private _formatInput;
+  private _formatOutput;
+  private _handleError;
 }
-
-export { PostHogADKPlugin };
-export type { DistinctIdResolver, PostHogADKPluginOptions };
+//#endregion
+export { type DistinctIdResolver, PostHogADKPlugin, type PostHogADKPluginOptions };
+//# sourceMappingURL=index.d.ts.map