@posthog/ai 7.2.1 → 7.3.0

package/dist/index.d.ts

@@ -3,7 +3,7 @@ import { PostHog } from 'posthog-node';
 import { Stream } from 'openai/streaming';
 import { ParsedResponse } from 'openai/resources/responses/responses';
 import { ResponseCreateParamsWithTools, ExtractParsedContentFromParams } from 'openai/lib/ResponsesParser';
-import { LanguageModelV2 } from '@ai-sdk/provider';
+import { LanguageModelV2, LanguageModelV3 } from '@ai-sdk/provider';
 import AnthropicOriginal, { APIPromise as APIPromise$1 } from '@anthropic-ai/sdk';
 import { Stream as Stream$1 } from '@anthropic-ai/sdk/streaming';
 import { GoogleGenAI, GenerateContentParameters, GenerateContentResponse } from '@google/genai';
@@ -147,6 +147,7 @@ declare class WrappedEmbeddings extends AzureOpenAI.Embeddings {
     create(body: EmbeddingCreateParams & MonitoringParams, options?: RequestOptions$1): APIPromise<CreateEmbeddingResponse>;
 }
 
+type LanguageModel = LanguageModelV2 | LanguageModelV3;
 interface ClientOptions {
     posthogDistinctId?: string;
     posthogTraceId?: string;
@@ -158,7 +159,11 @@ interface ClientOptions {
     posthogCostOverride?: CostOverride;
     posthogCaptureImmediate?: boolean;
 }
-declare const wrapVercelLanguageModel: (model: LanguageModelV2, phClient: PostHog, options: ClientOptions) => LanguageModelV2;
+/**
+ * Wraps a Vercel AI SDK language model (V2 or V3) with PostHog tracing.
+ * Automatically detects the model version and applies appropriate instrumentation.
+ */
+declare const wrapVercelLanguageModel: <T extends LanguageModel>(model: T, phClient: PostHog, options: ClientOptions) => T;
 
 type MessageCreateParamsNonStreaming = AnthropicOriginal.Messages.MessageCreateParamsNonStreaming;
 type MessageCreateParamsStreaming = AnthropicOriginal.Messages.MessageCreateParamsStreaming;
@@ -206,6 +211,7 @@ declare class WrappedModels {
     constructor(client: GoogleGenAI, phClient: PostHog);
     generateContent(params: GenerateContentParameters & MonitoringParams): Promise<GenerateContentResponse>;
     generateContentStream(params: GenerateContentParameters & MonitoringParams): AsyncGenerator<GenerateContentResponse, void, unknown>;
+    private formatPartsAsContentBlocks;
     private formatInput;
     private extractSystemInstruction;
     private formatInputForPostHog;