@posthog/ai 8.2.0 → 8.2.2

package/dist/openai/index.d.ts

@@ -29,6 +29,9 @@ type ChatCompletionChunk$1 = OpenAIOrignal.ChatCompletionChunk;
 type ChatCompletionCreateParamsBase$1 = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParams;
 type ChatCompletionCreateParamsNonStreaming$1 = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParamsNonStreaming;
 type ChatCompletionCreateParamsStreaming$1 = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParamsStreaming;
+type ResponsesCreateParamsBase$1 = OpenAIOrignal.Responses.ResponseCreateParams;
+type ResponsesCreateParamsNonStreaming$1 = OpenAIOrignal.Responses.ResponseCreateParamsNonStreaming;
+type ResponsesCreateParamsStreaming$1 = OpenAIOrignal.Responses.ResponseCreateParamsStreaming;
 type CreateEmbeddingResponse$1 = OpenAIOrignal.CreateEmbeddingResponse;
 type EmbeddingCreateParams$1 = OpenAIOrignal.EmbeddingCreateParams;
 interface MonitoringOpenAIConfig$1 {
@@ -40,6 +43,7 @@ type RequestOptions$1 = Record<string, any>;
 declare class PostHogAzureOpenAI extends AzureOpenAI {
     private readonly phClient;
     chat: WrappedChat$1;
+    responses: WrappedResponses$1;
     embeddings: WrappedEmbeddings$1;
     constructor(config: MonitoringOpenAIConfig$1);
 }
@@ -55,6 +59,15 @@ declare class WrappedCompletions$1 extends AzureOpenAI.Chat.Completions {
     create(body: ChatCompletionCreateParamsStreaming$1 & MonitoringParams, options?: RequestOptions$1): APIPromise<Stream<ChatCompletionChunk$1>>;
     create(body: ChatCompletionCreateParamsBase$1 & MonitoringParams, options?: RequestOptions$1): APIPromise<ChatCompletion$1 | Stream<ChatCompletionChunk$1>>;
 }
+declare class WrappedResponses$1 extends AzureOpenAI.Responses {
+    private readonly phClient;
+    private readonly baseURL;
+    constructor(client: AzureOpenAI, phClient: PostHog);
+    create(body: ResponsesCreateParamsNonStreaming$1 & MonitoringParams, options?: RequestOptions$1): APIPromise<OpenAIOrignal.Responses.Response>;
+    create(body: ResponsesCreateParamsStreaming$1 & MonitoringParams, options?: RequestOptions$1): APIPromise<Stream<OpenAIOrignal.Responses.ResponseStreamEvent>>;
+    create(body: ResponsesCreateParamsBase$1 & MonitoringParams, options?: RequestOptions$1): APIPromise<OpenAIOrignal.Responses.Response | Stream<OpenAIOrignal.Responses.ResponseStreamEvent>>;
+    parse<Params extends ResponseCreateParamsWithTools, ParsedT = ExtractParsedContentFromParams<Params>>(body: Params & MonitoringParams, options?: RequestOptions$1): APIPromise<ParsedResponse<ParsedT>>;
+}
 declare class WrappedEmbeddings$1 extends AzureOpenAI.Embeddings {
     private readonly phClient;
     private readonly baseURL;

package/dist/openai/index.mjs

@@ -530,7 +530,7 @@ function formatOpenAIResponsesInput(input, instructions) {
   return messages;
 }
 
-var version = "8.2.0";
+var version = "8.2.2";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -803,6 +803,7 @@ class PostHogAzureOpenAI extends AzureOpenAI {
     super(openAIConfig);
     this.phClient = posthog;
     this.chat = new WrappedChat$1(this, this.phClient);
+    this.responses = new WrappedResponses$1(this, this.phClient);
     this.embeddings = new WrappedEmbeddings$1(this, this.phClient);
   }
 }
@@ -1354,6 +1355,10 @@ function preserveAPIPromiseHelpers(parentPromise, wrappedPromise) {
   }
   return apiPromise;
 }
+const TERMINAL_RESPONSE_STATUSES = new Set(['completed', 'failed', 'cancelled', 'incomplete']);
+function isPendingBackgroundResponse(params, response) {
+  return params.background === true && !response.usage && !!response.status && !TERMINAL_RESPONSE_STATUSES.has(response.status);
+}
 class PostHogOpenAI extends OpenAI {
   constructor(config) {
     const {
@@ -1779,6 +1784,9 @@ class WrappedResponses extends Responses {
     } else {
       const wrappedPromise = parentPromise.then(async result => {
         if ('output' in result) {
+          if (isPendingBackgroundResponse(openAIParams, result)) {
+            return result;
+          }
           const latency = (Date.now() - startTime) / 1000;
           const availableTools = extractAvailableToolCalls('openai', openAIParams);
           const formattedOutput = formatResponseOpenAI({
@@ -1847,6 +1855,9 @@ class WrappedResponses extends Responses {
     try {
       const parentPromise = super.parse(openAIParams, options);
       const wrappedPromise = parentPromise.then(async result => {
+        if (isPendingBackgroundResponse(openAIParams, result)) {
+          return result;
+        }
         const latency = (Date.now() - startTime) / 1000;
         await captureAiGeneration(this.phClient, {
           ...posthogParams,