@posthog/ai 7.2.2 → 7.3.1

package/dist/index.d.ts

@@ -3,10 +3,10 @@ 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';
+import { GoogleGenAI, GenerateContentParameters, GenerateContentResponse, GoogleGenAIOptions } from '@google/genai';
 import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
 import { Serialized } from '@langchain/core/load/serializable';
 import { ChainValues } from '@langchain/core/utils/types';
@@ -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;
@@ -186,12 +191,7 @@ declare class WrappedMessages extends AnthropicOriginal.Messages {
     create(body: MessageCreateParamsBase & MonitoringParams, options?: RequestOptions): APIPromise$1<Stream$1<RawMessageStreamEvent> | Message>;
 }
 
-interface MonitoringGeminiConfig {
-    apiKey?: string;
-    vertexai?: boolean;
-    project?: string;
-    location?: string;
-    apiVersion?: string;
+interface MonitoringGeminiConfig extends GoogleGenAIOptions {
     posthog: PostHog;
 }
 declare class PostHogGoogleGenAI {