@posthog/ai 6.2.0 → 6.3.0

package/dist/index.d.ts

@@ -34,6 +34,7 @@ interface CostOverride {
 declare const Chat: typeof OpenAI.Chat;
 declare const Completions: typeof OpenAI.Chat.Completions;
 declare const Responses: typeof OpenAI.Responses;
+declare const Embeddings: typeof OpenAI.Embeddings;
 type ChatCompletion$1 = OpenAI.ChatCompletion;
 type ChatCompletionChunk$1 = OpenAI.ChatCompletionChunk;
 type ChatCompletionCreateParamsBase$1 = OpenAI.Chat.Completions.ChatCompletionCreateParams;
@@ -42,6 +43,8 @@ type ChatCompletionCreateParamsStreaming$1 = OpenAI.Chat.Completions.ChatComplet
 type ResponsesCreateParamsBase = OpenAI.Responses.ResponseCreateParams;
 type ResponsesCreateParamsNonStreaming = OpenAI.Responses.ResponseCreateParamsNonStreaming;
 type ResponsesCreateParamsStreaming = OpenAI.Responses.ResponseCreateParamsStreaming;
+type CreateEmbeddingResponse$1 = OpenAI.CreateEmbeddingResponse;
+type EmbeddingCreateParams$1 = OpenAI.EmbeddingCreateParams;
 interface MonitoringOpenAIConfig$1 extends ClientOptions$1 {
     apiKey: string;
     posthog: PostHog;
@@ -52,6 +55,7 @@ declare class PostHogOpenAI extends OpenAI {
     private readonly phClient;
     chat: WrappedChat$1;
     responses: WrappedResponses;
+    embeddings: WrappedEmbeddings$1;
     constructor(config: MonitoringOpenAIConfig$1);
 }
 declare class WrappedChat$1 extends Chat {
@@ -60,6 +64,7 @@ declare class WrappedChat$1 extends Chat {
 }
 declare class WrappedCompletions$1 extends Completions {
     private readonly phClient;
+    private readonly baseURL;
     constructor(client: OpenAI, phClient: PostHog);
     create(body: ChatCompletionCreateParamsNonStreaming$1 & MonitoringParams, options?: RequestOptions$1): APIPromise<ChatCompletion$1>;
     create(body: ChatCompletionCreateParamsStreaming$1 & MonitoringParams, options?: RequestOptions$1): APIPromise<Stream<ChatCompletionChunk$1>>;
@@ -67,18 +72,27 @@ declare class WrappedCompletions$1 extends Completions {
 }
 declare class WrappedResponses extends Responses {
     private readonly phClient;
+    private readonly baseURL;
     constructor(client: OpenAI, phClient: PostHog);
     create(body: ResponsesCreateParamsNonStreaming & MonitoringParams, options?: RequestOptions$1): APIPromise<OpenAI.Responses.Response>;
     create(body: ResponsesCreateParamsStreaming & MonitoringParams, options?: RequestOptions$1): APIPromise<Stream<OpenAI.Responses.ResponseStreamEvent>>;
     create(body: ResponsesCreateParamsBase & MonitoringParams, options?: RequestOptions$1): APIPromise<OpenAI.Responses.Response | Stream<OpenAI.Responses.ResponseStreamEvent>>;
     parse<Params extends ResponsesCreateParamsBase, ParsedT = any>(body: Params & MonitoringParams, options?: RequestOptions$1): APIPromise<ParsedResponse<ParsedT>>;
 }
+declare class WrappedEmbeddings$1 extends Embeddings {
+    private readonly phClient;
+    private readonly baseURL;
+    constructor(client: OpenAI, phClient: PostHog);
+    create(body: EmbeddingCreateParams$1 & MonitoringParams, options?: RequestOptions$1): APIPromise<CreateEmbeddingResponse$1>;
+}
 
 type ChatCompletion = OpenAIOrignal.ChatCompletion;
 type ChatCompletionChunk = OpenAIOrignal.ChatCompletionChunk;
 type ChatCompletionCreateParamsBase = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParams;
 type ChatCompletionCreateParamsNonStreaming = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParamsNonStreaming;
 type ChatCompletionCreateParamsStreaming = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParamsStreaming;
+type CreateEmbeddingResponse = OpenAIOrignal.CreateEmbeddingResponse;
+type EmbeddingCreateParams = OpenAIOrignal.EmbeddingCreateParams;
 interface MonitoringOpenAIConfig {
     apiKey: string;
     posthog: PostHog;
@@ -88,6 +102,7 @@ type RequestOptions = Record<string, any>;
 declare class PostHogAzureOpenAI extends AzureOpenAI {
     private readonly phClient;
     chat: WrappedChat;
+    embeddings: WrappedEmbeddings;
     constructor(config: MonitoringOpenAIConfig);
 }
 declare class WrappedChat extends AzureOpenAI.Chat {
@@ -96,11 +111,18 @@ declare class WrappedChat extends AzureOpenAI.Chat {
 }
 declare class WrappedCompletions extends AzureOpenAI.Chat.Completions {
     private readonly phClient;
+    private readonly baseURL;
     constructor(client: AzureOpenAI, phClient: PostHog);
     create(body: ChatCompletionCreateParamsNonStreaming & MonitoringParams, options?: RequestOptions): APIPromise<ChatCompletion>;
     create(body: ChatCompletionCreateParamsStreaming & MonitoringParams, options?: RequestOptions): APIPromise<Stream<ChatCompletionChunk>>;
     create(body: ChatCompletionCreateParamsBase & MonitoringParams, options?: RequestOptions): APIPromise<ChatCompletion | Stream<ChatCompletionChunk>>;
 }
+declare class WrappedEmbeddings extends AzureOpenAI.Embeddings {
+    private readonly phClient;
+    private readonly baseURL;
+    constructor(client: AzureOpenAI, phClient: PostHog);
+    create(body: EmbeddingCreateParams & MonitoringParams, options?: RequestOptions): APIPromise<CreateEmbeddingResponse>;
+}
 
 interface ClientOptions {
     posthogDistinctId?: string;
@@ -133,6 +155,7 @@ declare class PostHogAnthropic extends AnthropicOriginal {
 }
 declare class WrappedMessages extends AnthropicOriginal.Messages {
     private readonly phClient;
+    private readonly baseURL;
     constructor(parentClient: PostHogAnthropic, phClient: PostHog);
     create(body: MessageCreateParamsNonStreaming, options?: RequestOptions$2): APIPromise$1<Message>;
     create(body: MessageCreateParamsStreaming & MonitoringParams, options?: RequestOptions$2): APIPromise$1<Stream$1<RawMessageStreamEvent>>;
@@ -183,17 +206,17 @@ declare class LangChainCallbackHandler extends BaseCallbackHandler {
         groups?: Record<string, any>;
         debug?: boolean;
     });
-    handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, runType?: string, runName?: string): void;
-    handleChainEnd(outputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], kwargs?: {
+    handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, _runType?: string, runName?: string): void;
+    handleChainEnd(outputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], _kwargs?: {
         inputs?: Record<string, unknown>;
     }): void;
-    handleChainError(error: Error, runId: string, parentRunId?: string, tags?: string[], kwargs?: {
+    handleChainError(error: Error, runId: string, parentRunId?: string, tags?: string[], _kwargs?: {
         inputs?: Record<string, unknown>;
     }): void;
     handleChatModelStart(serialized: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): void;
     handleLLMStart(serialized: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): void;
-    handleLLMEnd(output: LLMResult, runId: string, parentRunId?: string, tags?: string[], extraParams?: Record<string, unknown>): void;
-    handleLLMError(err: Error, runId: string, parentRunId?: string, tags?: string[], extraParams?: Record<string, unknown>): void;
+    handleLLMEnd(output: LLMResult, runId: string, parentRunId?: string, tags?: string[], _extraParams?: Record<string, unknown>): void;
+    handleLLMError(err: Error, runId: string, parentRunId?: string, tags?: string[], _extraParams?: Record<string, unknown>): void;
     handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, runName?: string): void;
     handleToolEnd(output: any, runId: string, parentRunId?: string, tags?: string[]): void;
     handleToolError(err: Error, runId: string, parentRunId?: string, tags?: string[]): void;