@posthog/ai 7.11.2 → 7.12.0

package/dist/index.d.ts

@@ -4,8 +4,7 @@ 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 { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-http';
 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';
@@ -58,10 +57,6 @@ 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;
@@ -193,63 +188,35 @@ 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;
+interface PostHogTraceExporterOptions {
+    /**
+     * Your PostHog project API key.
+     */
+    apiKey: string;
+    /**
+     * PostHog host URL. Defaults to `https://us.i.posthog.com`.
+     */
+    host?: string;
 }
-
-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>;
+/**
+ * An OpenTelemetry SpanExporter that sends traces to PostHog's OTLP
+ * ingestion endpoint. PostHog converts `gen_ai.*` spans into
+ * `$ai_generation` events server-side.
+ *
+ * @example
+ * ```ts
+ * import { PostHogTraceExporter } from '@posthog/ai/otel'
+ * import { NodeSDK } from '@opentelemetry/sdk-node'
+ *
+ * const sdk = new NodeSDK({
+ *   traceExporter: new PostHogTraceExporter({ apiKey: 'phc_...' }),
+ * })
+ * sdk.start()
+ * ```
+ */
+declare class PostHogTraceExporter extends OTLPTraceExporter {
+    constructor(options: PostHogTraceExporterOptions);
 }
-declare function createPostHogSpanProcessor(phClient: PostHog, options?: PostHogTelemetryOptions): SpanProcessor;
 
 type MessageCreateParamsNonStreaming = AnthropicOriginal.Messages.MessageCreateParamsNonStreaming;
 type MessageCreateParamsStreaming = AnthropicOriginal.Messages.MessageCreateParamsStreaming;
@@ -436,4 +403,4 @@ declare class Prompts {
     private fetchPromptFromApi;
 }
 
-export { PostHogAnthropic as Anthropic, PostHogAzureOpenAI as AzureOpenAI, PostHogGoogleGenAI as GoogleGenAI, LangChainCallbackHandler, PostHogOpenAI as OpenAI, PostHogSpanProcessor, Prompts, captureSpan, createPostHogSpanProcessor, wrapVercelLanguageModel as withTracing };
+export { PostHogAnthropic as Anthropic, PostHogAzureOpenAI as AzureOpenAI, PostHogGoogleGenAI as GoogleGenAI, LangChainCallbackHandler, PostHogOpenAI as OpenAI, PostHogTraceExporter, Prompts, wrapVercelLanguageModel as withTracing };