@posthog/ai 8.6.3 → 8.6.4

package/dist/openai/index.d.ts

@@ -3,6 +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 { AzureClientOptions } from 'openai/azure';
 
 interface MonitoringEventPropertiesWithDefaults {
     distinctId?: string;
@@ -34,10 +35,9 @@ type ResponsesCreateParamsNonStreaming$1 = OpenAIOrignal.Responses.ResponseCreat
 type ResponsesCreateParamsStreaming$1 = OpenAIOrignal.Responses.ResponseCreateParamsStreaming;
 type CreateEmbeddingResponse$1 = OpenAIOrignal.CreateEmbeddingResponse;
 type EmbeddingCreateParams$1 = OpenAIOrignal.EmbeddingCreateParams;
-interface MonitoringOpenAIConfig$1 {
+interface MonitoringOpenAIConfig$1 extends AzureClientOptions {
     apiKey: string;
     posthog: PostHog;
-    baseURL?: string;
 }
 type RequestOptions$1 = Record<string, any>;
 declare class PostHogAzureOpenAI extends AzureOpenAI {

package/dist/openai/index.mjs

@@ -534,7 +534,7 @@ function formatOpenAIResponsesInput(input, instructions) {
   return messages;
 }
 
-var version = "8.6.3";
+var version = "8.6.4";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -667,6 +667,8 @@ const captureAiGeneration$1 = async (client, options) => {
       if (httpStatus === undefined) {
         if (typeof options.error === 'object' && 'status' in options.error && typeof options.error.status === 'number') {
           httpStatus = options.error.status;
+        } else if (typeof options.error === 'object' && 'statusCode' in options.error && typeof options.error.statusCode === 'number') {
+          httpStatus = options.error.statusCode;
         } else {
           httpStatus = 500;
         }
@@ -846,6 +848,98 @@ function buildProviderMetadata(fields) {
   return Object.keys(metadata).length > 0 ? metadata : undefined;
 }
 
+function addRequestId(result, response, requestIdHeader) {
+  if (!result || typeof result !== 'object' || Array.isArray(result)) {
+    return result;
+  }
+  return Object.defineProperty(result, '_request_id', {
+    value: response.headers.get(requestIdHeader),
+    enumerable: false
+  });
+}
+function getResponsePropsPromise(parentPromise) {
+  const responsePromise = parentPromise.responsePromise;
+  if (!responsePromise || typeof responsePromise.then !== 'function') {
+    return undefined;
+  }
+  return responsePromise;
+}
+function decorateProviderPromise(wrappedPromise, responsePropsPromise, requestIdHeader, preserveThenUnwrap) {
+  const providerPromise = wrappedPromise;
+  if (responsePropsPromise) {
+    providerPromise.asResponse = async () => (await responsePropsPromise).response;
+    providerPromise.withResponse = async () => {
+      const [props, data] = await Promise.all([responsePropsPromise, wrappedPromise]);
+      return {
+        response: props.response,
+        data,
+        request_id: props.response.headers.get(requestIdHeader)
+      };
+    };
+  }
+  if (preserveThenUnwrap) {
+    providerPromise._thenUnwrap = transform => {
+      if (!responsePropsPromise) {
+        throw new Error('The provider promise response metadata is unavailable');
+      }
+      const transformedPromise = Promise.all([wrappedPromise, responsePropsPromise]).then(([data, props]) => addRequestId(transform(data, props), props.response, requestIdHeader));
+      return decorateProviderPromise(transformedPromise, responsePropsPromise, requestIdHeader, true);
+    };
+  }
+  return providerPromise;
+}
+
+/**
+ * Keep the provider SDK helpers on a promise whose resolved value is instrumented.
+ * OpenAI's parse helpers compose create calls through `_thenUnwrap`, while both
+ * OpenAI and Anthropic expose the raw response through `asResponse` and
+ * `withResponse`.
+ */
+
+function preserveProviderPromise(parentPromise, wrappedPromise, options = {}) {
+  const responsePropsPromise = getResponsePropsPromise(parentPromise);
+  const preserveThenUnwrap = typeof parentPromise._thenUnwrap === 'function';
+  const providerPromise = decorateProviderPromise(wrappedPromise, responsePropsPromise, options.requestIdHeader ?? 'x-request-id', preserveThenUnwrap);
+  if (!responsePropsPromise) {
+    const asResponse = parentPromise.asResponse?.bind(parentPromise);
+    if (asResponse) {
+      providerPromise.asResponse = asResponse;
+    }
+    const withResponse = parentPromise.withResponse?.bind(parentPromise);
+    if (withResponse) {
+      providerPromise.withResponse = async () => {
+        const [response, data] = await Promise.all([withResponse(), wrappedPromise]);
+        return {
+          ...response,
+          data
+        };
+      };
+    }
+  }
+  return providerPromise;
+}
+
+/**
+ * OpenAI's `Responses.parse` dispatches through `this._client.responses.create`.
+ * Temporarily use the provider's original `create` implementation so parsing a
+ * wrapped response does not capture the same request twice.
+ */
+function callWithOriginalCreate(resource, originalCreate, callback) {
+  const resourceRecord = resource;
+  const hadOwnCreate = Object.prototype.hasOwnProperty.call(resource, 'create');
+  const wrappedCreate = resourceRecord['create'];
+  resourceRecord['create'] = originalCreate;
+  try {
+    return callback();
+  } finally {
+    if (hadOwnCreate) {
+      resourceRecord['create'] = wrappedCreate;
+    } else {
+      delete resourceRecord['create'];
+    }
+  }
+}
+
 /**
  * Splits an SDK stream into a monitoring branch and a caller branch without
  * allowing either branch to read ahead of the other. Unlike the SDKs' `tee()`
@@ -1101,7 +1195,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends AzureOpenAI.Chat.Com
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     if (openAIParams.stream) {
-      return parentPromise.then(value => {
+      const wrappedPromise = parentPromise.then(value => {
         if (Symbol.asyncIterator in value) {
           const [stream1, stream2] = monitoredStreamTee(value, (iterator, controller) => new Stream(iterator, controller));
           (async () => {
@@ -1279,6 +1373,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends AzureOpenAI.Chat.Com
         }
         return value;
       });
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     } else {
       const wrappedPromise = parentPromise.then(async result => {
         if ('choices' in result) {
@@ -1327,7 +1422,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends AzureOpenAI.Chat.Com
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     }
   }
 };
@@ -1353,7 +1448,7 @@ let WrappedResponses$1 = class WrappedResponses extends AzureOpenAI.Responses {
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     if (openAIParams.stream) {
-      return parentPromise.then(value => {
+      const wrappedPromise = parentPromise.then(value => {
         if (Symbol.asyncIterator in value) {
           const [stream1, stream2] = monitoredStreamTee(value, (iterator, controller) => new Stream(iterator, controller));
           (async () => {
@@ -1443,6 +1538,7 @@ let WrappedResponses$1 = class WrappedResponses extends AzureOpenAI.Responses {
         }
         return value;
       });
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     } else {
       const wrappedPromise = parentPromise.then(async result => {
         if ('output' in result) {
@@ -1490,7 +1586,7 @@ let WrappedResponses$1 = class WrappedResponses extends AzureOpenAI.Responses {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     }
   }
   parse(body, options) {
@@ -1499,7 +1595,7 @@ let WrappedResponses$1 = class WrappedResponses extends AzureOpenAI.Responses {
       posthogParams
     } = extractPosthogParams(body);
     const startTime = Date.now();
-    const parentPromise = super.parse(openAIParams, options);
+    const parentPromise = callWithOriginalCreate(this, super.create.bind(this), () => super.parse(openAIParams, options));
     const wrappedPromise = parentPromise.then(async result => {
       const latency = (Date.now() - startTime) / 1000;
       await captureAiGeneration(this.phClient, {
@@ -1543,7 +1639,7 @@ let WrappedResponses$1 = class WrappedResponses extends AzureOpenAI.Responses {
       });
       throw error;
     });
-    return wrappedPromise;
+    return preserveProviderPromise(parentPromise, wrappedPromise);
   }
 };
 let WrappedEmbeddings$1 = class WrappedEmbeddings extends AzureOpenAI.Embeddings {
@@ -1598,7 +1694,7 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends AzureOpenAI.Embeddings
       });
       throw error;
     });
-    return wrappedPromise;
+    return preserveProviderPromise(parentPromise, wrappedPromise);
   }
 };
 
@@ -1619,25 +1715,6 @@ async function captureAiGenerationAfterSuccess(...args) {
     captureAiGenerationInBackground(...args);
   }
 }
-function preserveAPIPromiseHelpers(parentPromise, wrappedPromise) {
-  const apiPromise = wrappedPromise;
-  if (typeof parentPromise.asResponse === 'function') {
-    apiPromise.asResponse = () => parentPromise.asResponse();
-  }
-  if (typeof parentPromise.withResponse === 'function') {
-    apiPromise.withResponse = async () => {
-      const [response, data] = await Promise.all([parentPromise.withResponse(), wrappedPromise]);
-      // swaps the `data` payload inside the SDK's generic withResponse shape; the compiler
-      // cannot relate the Input- and Output-instantiated conditional types
-      // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
-      return {
-        ...response,
-        data
-      };
-    };
-  }
-  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);
@@ -1884,7 +1961,7 @@ class WrappedCompletions extends Completions {
         }
         return value;
       });
-      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     } else {
       const wrappedPromise = parentPromise.then(async result => {
         if ('choices' in result) {
@@ -1939,7 +2016,7 @@ class WrappedCompletions extends Completions {
         });
         throw error;
       });
-      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     }
   }
 }
@@ -2077,7 +2154,7 @@ class WrappedResponses extends Responses {
         }
         return value;
       });
-      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     } else {
       const wrappedPromise = parentPromise.then(async result => {
         if ('output' in result) {
@@ -2136,7 +2213,7 @@ class WrappedResponses extends Responses {
         });
         throw error;
       });
-      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     }
   }
   parse(body, options) {
@@ -2145,64 +2222,55 @@ class WrappedResponses extends Responses {
       posthogParams
     } = extractPosthogParams(body);
     const startTime = Date.now();
-    const originalCreate = super.create.bind(this);
-    const originalSelfRecord = this;
-    const tempCreate = originalSelfRecord['create'];
-    originalSelfRecord['create'] = originalCreate;
-    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,
-          model: openAIParams.model ?? result.model,
-          provider: 'openai',
-          input: formatOpenAIResponsesInput(sanitizeOpenAIResponse(openAIParams.input), openAIParams.instructions),
-          output: result.output,
-          latency,
-          baseURL: this.baseURL,
-          modelParameters: getModelParams(body, result.service_tier),
-          httpStatus: 200,
-          usage: {
-            inputTokens: result.usage?.input_tokens ?? 0,
-            outputTokens: result.usage?.output_tokens ?? 0,
-            reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
-            cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0,
-            rawUsage: result.usage
-          },
-          stopReason: result.status ?? undefined,
-          completionId: result.id,
-          providerMetadata: buildProviderMetadata({
-            requestId: extractRequestId(result)
-          })
-        });
+    const parentPromise = callWithOriginalCreate(this, super.create.bind(this), () => super.parse(openAIParams, options));
+    const wrappedPromise = parentPromise.then(async result => {
+      if (isPendingBackgroundResponse(openAIParams, result)) {
         return result;
-      }, async error => {
-        await captureAiGeneration(this.phClient, {
-          ...posthogParams,
-          model: openAIParams.model,
-          provider: 'openai',
-          input: formatOpenAIResponsesInput(sanitizeOpenAIResponse(openAIParams.input), openAIParams.instructions),
-          output: [],
-          latency: 0,
-          baseURL: this.baseURL,
-          modelParameters: getModelParams(body),
-          usage: {
-            inputTokens: 0,
-            outputTokens: 0
-          },
-          error
-        });
-        throw error;
+      }
+      const latency = (Date.now() - startTime) / 1000;
+      await captureAiGeneration(this.phClient, {
+        ...posthogParams,
+        model: openAIParams.model ?? result.model,
+        provider: 'openai',
+        input: formatOpenAIResponsesInput(sanitizeOpenAIResponse(openAIParams.input), openAIParams.instructions),
+        output: result.output,
+        latency,
+        baseURL: this.baseURL,
+        modelParameters: getModelParams(body, result.service_tier),
+        httpStatus: 200,
+        usage: {
+          inputTokens: result.usage?.input_tokens ?? 0,
+          outputTokens: result.usage?.output_tokens ?? 0,
+          reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
+          cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0,
+          rawUsage: result.usage
+        },
+        stopReason: result.status ?? undefined,
+        completionId: result.id,
+        providerMetadata: buildProviderMetadata({
+          requestId: extractRequestId(result)
+        })
       });
-      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
-    } finally {
-      // Restore our wrapped create method
-      originalSelfRecord['create'] = tempCreate;
-    }
+      return result;
+    }, async error => {
+      await captureAiGeneration(this.phClient, {
+        ...posthogParams,
+        model: openAIParams.model,
+        provider: 'openai',
+        input: formatOpenAIResponsesInput(sanitizeOpenAIResponse(openAIParams.input), openAIParams.instructions),
+        output: [],
+        latency: 0,
+        baseURL: this.baseURL,
+        modelParameters: getModelParams(body),
+        usage: {
+          inputTokens: 0,
+          outputTokens: 0
+        },
+        error
+      });
+      throw error;
+    });
+    return preserveProviderPromise(parentPromise, wrappedPromise);
   }
 }
 class WrappedEmbeddings extends Embeddings {
@@ -2259,7 +2327,7 @@ class WrappedEmbeddings extends Embeddings {
       });
       throw error;
     });
-    return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
+    return preserveProviderPromise(parentPromise, wrappedPromise);
   }
 }
 class WrappedAudio extends Audio {
@@ -2369,7 +2437,7 @@ class WrappedTranscriptions extends Transcriptions {
         }
         return value;
       });
-      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     } else {
       const wrappedPromise = parentPromise.then(async result => {
         if (result && typeof result === 'object' && 'text' in result) {
@@ -2410,7 +2478,7 @@ class WrappedTranscriptions extends Transcriptions {
         });
         throw error;
       });
-      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     }
   }
 }