@posthog/ai 7.8.12 → 7.9.0

package/dist/index.d.ts

@@ -4,6 +4,8 @@ import { Stream } from 'openai/streaming';
 import { ParsedResponse } from 'openai/resources/responses/responses';
 import { ResponseCreateParamsWithTools, ExtractParsedContentFromParams } from 'openai/lib/ResponsesParser';
 import { LanguageModelV2, LanguageModelV3 } from '@ai-sdk/provider';
+import { ReadableSpan, SpanProcessor } from '@opentelemetry/sdk-trace-base';
+import { Span, Context } from '@opentelemetry/api';
 import AnthropicOriginal, { APIPromise as APIPromise$1 } from '@anthropic-ai/sdk';
 import { Stream as Stream$1 } from '@anthropic-ai/sdk/streaming';
 import { GoogleGenAI, GenerateContentParameters, GenerateContentResponse, GoogleGenAIOptions } from '@google/genai';
@@ -55,6 +57,10 @@ interface CostOverride {
     inputCost: number;
     outputCost: number;
 }
+declare enum AIEvent {
+    Generation = "$ai_generation",
+    Embedding = "$ai_embedding"
+}
 
 declare const Chat: typeof OpenAI.Chat;
 declare const Completions: typeof OpenAI.Chat.Completions;
@@ -186,6 +192,64 @@ interface ClientOptions {
  */
 declare const wrapVercelLanguageModel: <T extends LanguageModel>(model: T, phClient: PostHog, options: ClientOptions) => T;
 
+type UsageData = Record<string, unknown>;
+interface PostHogSpanMapperContext {
+    options: PostHogTelemetryOptions;
+}
+interface PostHogSpanMapperResult {
+    distinctId?: string;
+    traceId?: string;
+    model?: string;
+    provider: string;
+    input: any;
+    output: any;
+    latency: number;
+    timeToFirstToken?: number;
+    baseURL?: string;
+    httpStatus?: number;
+    eventType?: AIEvent;
+    usage?: UsageData;
+    tools?: any[] | null;
+    modelParams?: Record<string, unknown>;
+    posthogProperties?: Record<string, unknown>;
+    error?: unknown;
+}
+interface PostHogSpanMapper {
+    name: string;
+    canMap: (span: ReadableSpan) => boolean;
+    map: (span: ReadableSpan, context: PostHogSpanMapperContext) => PostHogSpanMapperResult | null;
+}
+type ShouldExportSpan = (params: {
+    otelSpan: ReadableSpan;
+}) => boolean;
+interface PostHogTelemetryOptions {
+    posthogDistinctId?: string;
+    posthogTraceId?: string;
+    posthogProperties?: Record<string, any>;
+    posthogPrivacyMode?: boolean;
+    posthogGroups?: Record<string, any>;
+    posthogModelOverride?: string;
+    posthogProviderOverride?: string;
+    posthogCostOverride?: CostOverride;
+    posthogCaptureImmediate?: boolean;
+    mappers?: PostHogSpanMapper[];
+    shouldExportSpan?: ShouldExportSpan;
+}
+
+declare function captureSpan(span: ReadableSpan, phClient: PostHog, options?: PostHogTelemetryOptions): Promise<void>;
+
+declare class PostHogSpanProcessor implements SpanProcessor {
+    private readonly phClient;
+    private readonly options;
+    private readonly pendingCaptures;
+    constructor(phClient: PostHog, options?: PostHogTelemetryOptions);
+    onStart(_span: Span, _parentContext: Context): void;
+    onEnd(span: ReadableSpan): void;
+    shutdown(): Promise<void>;
+    forceFlush(): Promise<void>;
+}
+declare function createPostHogSpanProcessor(phClient: PostHog, options?: PostHogTelemetryOptions): SpanProcessor;
+
 type MessageCreateParamsNonStreaming = AnthropicOriginal.Messages.MessageCreateParamsNonStreaming;
 type MessageCreateParamsStreaming = AnthropicOriginal.Messages.MessageCreateParamsStreaming;
 type Message = AnthropicOriginal.Messages.Message;
@@ -362,4 +426,4 @@ declare class Prompts {
     private fetchPromptFromApi;
 }
 
-export { PostHogAnthropic as Anthropic, PostHogAzureOpenAI as AzureOpenAI, PostHogGoogleGenAI as GoogleGenAI, LangChainCallbackHandler, PostHogOpenAI as OpenAI, Prompts, wrapVercelLanguageModel as withTracing };
+export { PostHogAnthropic as Anthropic, PostHogAzureOpenAI as AzureOpenAI, PostHogGoogleGenAI as GoogleGenAI, LangChainCallbackHandler, PostHogOpenAI as OpenAI, PostHogSpanProcessor, Prompts, captureSpan, createPostHogSpanProcessor, wrapVercelLanguageModel as withTracing };