@posthog/ai 6.1.1 → 6.1.2

package/dist/index.d.ts

@@ -6,7 +6,7 @@ import { LanguageModelV2 } from '@ai-sdk/provider';
 import AnthropicOriginal from '@anthropic-ai/sdk';
 import { RequestOptions as RequestOptions$2, APIPromise as APIPromise$1 } from '@anthropic-ai/sdk/core';
 import { Stream as Stream$1 } from '@anthropic-ai/sdk/streaming';
-import { GoogleGenAI } from '@google/genai';
+import { GoogleGenAI, GenerateContentParameters, GenerateContentResponse } from '@google/genai';
 import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
 import { Serialized } from '@langchain/core/load/serializable';
 import { ChainValues } from '@langchain/core/utils/types';
@@ -139,24 +139,6 @@ declare class WrappedMessages extends AnthropicOriginal.Messages {
     create(body: MessageCreateParamsBase & MonitoringParams, options?: RequestOptions$2): APIPromise$1<Stream$1<RawMessageStreamEvent> | Message>;
 }
 
-type GenerateContentRequest = {
-    model: string;
-    contents: any;
-    config?: any;
-    [key: string]: any;
-};
-type GenerateContentResponse = {
-    text?: string;
-    candidates?: any[];
-    usageMetadata?: {
-        promptTokenCount?: number;
-        candidatesTokenCount?: number;
-        totalTokenCount?: number;
-        thoughtsTokenCount?: number;
-        cachedContentTokenCount?: number;
-    };
-    [key: string]: any;
-};
 interface MonitoringGeminiConfig {
     apiKey?: string;
     vertexai?: boolean;
@@ -175,8 +157,8 @@ declare class WrappedModels {
     private readonly phClient;
     private readonly client;
     constructor(client: GoogleGenAI, phClient: PostHog);
-    generateContent(params: GenerateContentRequest & MonitoringParams): Promise<GenerateContentResponse>;
-    generateContentStream(params: GenerateContentRequest & MonitoringParams): AsyncGenerator<any, void, unknown>;
+    generateContent(params: GenerateContentParameters & MonitoringParams): Promise<GenerateContentResponse>;
+    generateContentStream(params: GenerateContentParameters & MonitoringParams): AsyncGenerator<GenerateContentResponse, void, unknown>;
     private formatInput;
     private formatInputForPostHog;
 }