@posthog/ai 8.6.3 → 8.6.5

package/dist/openai/index.cjs

@@ -45,14 +45,16 @@ const FILE_FAMILY_TYPES = new Set(['file', 'input_file', 'document', 'media', 'f
 const KNOWN_AUDIO_FORMATS = new Set(['wav', 'mp3', 'ogg', 'flac', 'm4a', 'aac', 'webm']);
 class MediaTypeContext {
   static EMPTY = new MediaTypeContext(undefined, undefined);
-  constructor(parent, key) {
+  constructor(parent, key, explicitMediaType) {
     this.parent = parent;
     this.key = key;
+    this.explicitMediaType = explicitMediaType;
   }
   inferMediaType() {
     return this.inferFromSiblingMime() ?? this.inferFromSiblingFormat() ?? this.inferFromParentType() ?? this.inferFromKey();
   }
   inferFromSiblingMime() {
+    if (this.explicitMediaType) return this.explicitMediaType;
     if (!this.parent) return undefined;
     for (const hint of MIME_HINT_KEYS) {
       const v = this.parent[hint];
@@ -87,7 +89,13 @@ class MediaTypeContext {
     if (key.includes('file') || key.includes('document')) return 'application/octet-stream';
     return undefined;
   }
+  hasExplicitBinaryMediaType() {
+    if (!this.explicitMediaType && (!this.parent || !this.key || !STRONG_CONTEXT_KEYS.has(this.key))) return false;
+    const mediaType = this.inferFromSiblingMime();
+    return mediaType !== undefined && !mediaType.toLowerCase().startsWith('text/');
+  }
   signalsBinary() {
+    if (this.explicitMediaType) return true;
     if (this.parent) {
       for (const hint of MIME_HINT_KEYS) {
         if (typeof this.parent[hint] === 'string') return true;
@@ -109,10 +117,10 @@ class BinaryContentRedactor {
   constructor(recognizer = new Base64Recognizer()) {
     this.recognizer = recognizer;
   }
-  redact(value) {
+  redact(value, mediaType) {
     if (this.isMultimodalEnabled()) return value;
     this.visited = new WeakSet();
-    return this.walk(value, MediaTypeContext.EMPTY);
+    return this.walk(value, mediaType ? new MediaTypeContext(undefined, undefined, mediaType) : MediaTypeContext.EMPTY);
   }
   walk(value, ctx) {
     if (value === null || value === undefined) return value;
@@ -136,8 +144,10 @@ class BinaryContentRedactor {
     return out;
   }
   redactString(value, ctx) {
-    const minLength = ctx.signalsBinary() ? STRONG_CONTEXT_MIN_LENGTH : WEAK_CONTEXT_MIN_LENGTH;
-    const recognition = this.recognizer.recognize(value, minLength);
+    const hasExplicitBinaryMediaType = ctx.hasExplicitBinaryMediaType();
+    const recognitionValue = hasExplicitBinaryMediaType ? value.replace(/[\r\n]/g, '') : value;
+    const minLength = hasExplicitBinaryMediaType ? Math.min(recognitionValue.length, STRONG_CONTEXT_MIN_LENGTH) : ctx.signalsBinary() ? STRONG_CONTEXT_MIN_LENGTH : WEAK_CONTEXT_MIN_LENGTH;
+    const recognition = this.recognizer.recognize(recognitionValue, minLength);
     switch (recognition.kind) {
       case 'data-url':
         return this.placeholderFor(recognition.mediaType);
@@ -538,7 +548,7 @@ function formatOpenAIResponsesInput(input, instructions) {
   return messages;
 }
 
-var version = "8.6.3";
+var version = "8.6.5";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -671,6 +681,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;
         }
@@ -836,7 +848,7 @@ function extractRequestId(result) {
  * Assembles the `$ai_provider_metadata` blob for OpenAI / Azure OpenAI events.
  * Provider-specific fields (system fingerprint, request id) live here rather
  * than in the shared, provider-agnostic `$ai_*` namespace. Only keys with a
- * truthy value are included, and `undefined` is returned when there is nothing
+ * meaningful value are included, and `undefined` is returned when there is nothing
  * to report so the property can be omitted from the event entirely.
  */
 function buildProviderMetadata(fields) {
@@ -847,8 +859,125 @@ function buildProviderMetadata(fields) {
   if (fields.requestId) {
     metadata.request_id = fields.requestId;
   }
+  if (fields.incompleteDetails != null) {
+    metadata.incomplete_details = fields.incompleteDetails;
+  }
   return Object.keys(metadata).length > 0 ? metadata : undefined;
 }
+const TERMINAL_RESPONSE_STATUSES = new Set(['completed', 'failed', 'cancelled', 'incomplete']);
+
+/**
+ * Checks whether a Responses API response has reached a status that should
+ * produce a final `$ai_generation` event.
+ */
+function isTerminalResponse(response) {
+  return !!response?.status && TERMINAL_RESPONSE_STATUSES.has(response.status);
+}
+
+/**
+ * Returns an isolated copy of a failed Responses API error for `$ai_error`, or
+ * creates a fallback error when the provider omitted failure details.
+ */
+function getResponseFailure(response) {
+  if (response?.status !== 'failed') {
+    return undefined;
+  }
+  return response.error ? {
+    ...response.error
+  } : new Error(`OpenAI response ${response.id} failed without error details`);
+}
+
+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
@@ -1105,7 +1234,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends openai.AzureOpenAI.C
     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 streaming.Stream(iterator, controller));
           (async () => {
@@ -1236,7 +1365,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends openai.AzureOpenAI.C
                 model: openAIParams.model ?? modelFromResponse,
                 provider: 'azure',
                 input: sanitizeOpenAI(openAIParams.messages),
-                output: formattedOutput,
+                output: sanitizeOpenAIResponse(formattedOutput),
                 latency,
                 timeToFirstToken,
                 baseURL: this.baseURL,
@@ -1283,6 +1412,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends openai.AzureOpenAI.C
         }
         return value;
       });
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     } else {
       const wrappedPromise = parentPromise.then(async result => {
         if ('choices' in result) {
@@ -1291,8 +1421,8 @@ let WrappedCompletions$1 = class WrappedCompletions extends openai.AzureOpenAI.C
             ...posthogParams,
             model: openAIParams.model ?? result.model,
             provider: 'azure',
-            input: openAIParams.messages,
-            output: formatResponseOpenAI(result),
+            input: sanitizeOpenAI(openAIParams.messages),
+            output: sanitizeOpenAIResponse(formatResponseOpenAI(result)),
             latency,
             baseURL: this.baseURL,
             modelParameters: getModelParams(body, result.service_tier),
@@ -1317,7 +1447,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends openai.AzureOpenAI.C
           ...posthogParams,
           model: openAIParams.model,
           provider: 'azure',
-          input: openAIParams.messages,
+          input: sanitizeOpenAI(openAIParams.messages),
           output: [],
           latency: 0,
           baseURL: this.baseURL,
@@ -1331,7 +1461,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends openai.AzureOpenAI.C
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     }
   }
 };
@@ -1357,7 +1487,7 @@ let WrappedResponses$1 = class WrappedResponses extends openai.AzureOpenAI.Respo
     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 streaming.Stream(iterator, controller));
           (async () => {
@@ -1373,6 +1503,7 @@ let WrappedResponses$1 = class WrappedResponses extends openai.AzureOpenAI.Respo
                 inputTokens: 0,
                 outputTokens: 0
               };
+              let terminalResponse;
               for await (const chunk of stream1) {
                 // Track first token time on content delta events
                 if (firstTokenTime === undefined && isResponseTokenChunk(chunk)) {
@@ -1389,9 +1520,10 @@ let WrappedResponses$1 = class WrappedResponses extends openai.AzureOpenAI.Respo
                   if (chunk.response.service_tier != null) {
                     serviceTierFromResponse = chunk.response.service_tier;
                   }
-                }
-                if (chunk.type === 'response.completed' && 'response' in chunk && chunk.response?.output && chunk.response.output.length > 0) {
-                  finalContent = chunk.response.output;
+                  if (isTerminalResponse(chunk.response)) {
+                    terminalResponse = chunk.response;
+                    finalContent = chunk.response.output ?? [];
+                  }
                 }
                 if ('response' in chunk && chunk.response?.usage) {
                   usage = {
@@ -1408,22 +1540,27 @@ let WrappedResponses$1 = class WrappedResponses extends openai.AzureOpenAI.Respo
                 ...posthogParams,
                 model: openAIParams.model ?? modelFromResponse,
                 provider: 'azure',
-                input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
-                output: finalContent,
+                input: formatOpenAIResponsesInput(sanitizeOpenAIResponse(openAIParams.input), openAIParams.instructions),
+                output: sanitizeOpenAIResponse(finalContent),
                 latency,
                 timeToFirstToken,
                 baseURL: this.baseURL,
                 modelParameters: getModelParams(body, serviceTierFromResponse),
                 httpStatus: 200,
                 usage,
-                completionId: completionIdFromResponse
+                stopReason: terminalResponse?.status ?? undefined,
+                completionId: completionIdFromResponse,
+                providerMetadata: buildProviderMetadata({
+                  incompleteDetails: terminalResponse?.incomplete_details
+                }),
+                error: getResponseFailure(terminalResponse)
               });
             } catch (error) {
               await captureAiGeneration(this.phClient, {
                 ...posthogParams,
                 model: openAIParams.model,
                 provider: 'azure',
-                input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
+                input: formatOpenAIResponsesInput(sanitizeOpenAIResponse(openAIParams.input), openAIParams.instructions),
                 output: [],
                 latency: 0,
                 baseURL: this.baseURL,
@@ -1447,6 +1584,7 @@ let WrappedResponses$1 = class WrappedResponses extends openai.AzureOpenAI.Respo
         }
         return value;
       });
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     } else {
       const wrappedPromise = parentPromise.then(async result => {
         if ('output' in result) {
@@ -1455,8 +1593,8 @@ let WrappedResponses$1 = class WrappedResponses extends openai.AzureOpenAI.Respo
             ...posthogParams,
             model: openAIParams.model ?? result.model,
             provider: 'azure',
-            input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
-            output: result.output,
+            input: formatOpenAIResponsesInput(sanitizeOpenAIResponse(openAIParams.input), openAIParams.instructions),
+            output: sanitizeOpenAIResponse(result.output),
             latency,
             baseURL: this.baseURL,
             modelParameters: getModelParams(body, result.service_tier),
@@ -1467,10 +1605,13 @@ let WrappedResponses$1 = class WrappedResponses extends openai.AzureOpenAI.Respo
               reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
               cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0
             },
+            stopReason: result.status ?? undefined,
             completionId: result.id,
             providerMetadata: buildProviderMetadata({
-              requestId: extractRequestId(result)
-            })
+              requestId: extractRequestId(result),
+              incompleteDetails: result.incomplete_details
+            }),
+            error: getResponseFailure(result)
           });
         }
         return result;
@@ -1480,7 +1621,7 @@ let WrappedResponses$1 = class WrappedResponses extends openai.AzureOpenAI.Respo
           ...posthogParams,
           model: openAIParams.model,
           provider: 'azure',
-          input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
+          input: formatOpenAIResponsesInput(sanitizeOpenAIResponse(openAIParams.input), openAIParams.instructions),
           output: [],
           latency: 0,
           baseURL: this.baseURL,
@@ -1494,7 +1635,7 @@ let WrappedResponses$1 = class WrappedResponses extends openai.AzureOpenAI.Respo
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     }
   }
   parse(body, options) {
@@ -1503,15 +1644,15 @@ let WrappedResponses$1 = class WrappedResponses extends openai.AzureOpenAI.Respo
       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, {
         ...posthogParams,
         model: openAIParams.model ?? result.model,
         provider: 'azure',
-        input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
-        output: result.output,
+        input: formatOpenAIResponsesInput(sanitizeOpenAIResponse(openAIParams.input), openAIParams.instructions),
+        output: sanitizeOpenAIResponse(result.output),
         latency,
         baseURL: this.baseURL,
         modelParameters: getModelParams(body, result.service_tier),
@@ -1522,10 +1663,13 @@ let WrappedResponses$1 = class WrappedResponses extends openai.AzureOpenAI.Respo
           reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
           cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0
         },
+        stopReason: result.status ?? undefined,
         completionId: result.id,
         providerMetadata: buildProviderMetadata({
-          requestId: extractRequestId(result)
-        })
+          requestId: extractRequestId(result),
+          incompleteDetails: result.incomplete_details
+        }),
+        error: getResponseFailure(result)
       });
       return result;
     }, async error => {
@@ -1533,7 +1677,7 @@ let WrappedResponses$1 = class WrappedResponses extends openai.AzureOpenAI.Respo
         ...posthogParams,
         model: openAIParams.model,
         provider: 'azure',
-        input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
+        input: formatOpenAIResponsesInput(sanitizeOpenAIResponse(openAIParams.input), openAIParams.instructions),
         output: [],
         latency: 0,
         baseURL: this.baseURL,
@@ -1547,7 +1691,7 @@ let WrappedResponses$1 = class WrappedResponses extends openai.AzureOpenAI.Respo
       });
       throw error;
     });
-    return wrappedPromise;
+    return preserveProviderPromise(parentPromise, wrappedPromise);
   }
 };
 let WrappedEmbeddings$1 = class WrappedEmbeddings extends openai.AzureOpenAI.Embeddings {
@@ -1602,7 +1746,7 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends openai.AzureOpenAI.Emb
       });
       throw error;
     });
-    return wrappedPromise;
+    return preserveProviderPromise(parentPromise, wrappedPromise);
   }
 };
 
@@ -1623,28 +1767,8 @@ 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);
+  return params.background === true && !response.usage && !!response.status && !isTerminalResponse(response);
 }
 class PostHogOpenAI extends openai.OpenAI {
   constructor(config) {
@@ -1832,7 +1956,7 @@ class WrappedCompletions extends Completions {
                 model: openAIParams.model ?? modelFromResponse,
                 provider: 'openai',
                 input: sanitizeOpenAI(openAIParams.messages),
-                output: formattedOutput,
+                output: sanitizeOpenAIResponse(formattedOutput),
                 latency,
                 timeToFirstToken,
                 baseURL: this.baseURL,
@@ -1888,7 +2012,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) {
@@ -1900,7 +2024,7 @@ class WrappedCompletions extends Completions {
             model: openAIParams.model ?? result.model,
             provider: 'openai',
             input: sanitizeOpenAI(openAIParams.messages),
-            output: formattedOutput,
+            output: sanitizeOpenAIResponse(formattedOutput),
             latency,
             baseURL: this.baseURL,
             modelParameters: getModelParams(body, result.service_tier),
@@ -1943,7 +2067,7 @@ class WrappedCompletions extends Completions {
         });
         throw error;
       });
-      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     }
   }
 }
@@ -1988,6 +2112,7 @@ class WrappedResponses extends Responses {
                 webSearchCount: 0
               };
               let rawUsageData;
+              let terminalResponse;
               for await (const chunk of stream1) {
                 // Track first token time on content delta events
                 if (firstTokenTime === undefined && isResponseTokenChunk(chunk)) {
@@ -2008,10 +2133,9 @@ class WrappedResponses extends Responses {
                   if (chunkWebSearchCount > 0 && chunkWebSearchCount > (usage.webSearchCount ?? 0)) {
                     usage.webSearchCount = chunkWebSearchCount;
                   }
-                }
-                if (chunk.type === 'response.completed' && 'response' in chunk && chunk.response?.output && chunk.response.output.length > 0) {
-                  finalContent = chunk.response.output;
-                  if (chunk.response.status) {
+                  if (isTerminalResponse(chunk.response)) {
+                    terminalResponse = chunk.response;
+                    finalContent = chunk.response.output ?? [];
                     stopReason = chunk.response.status;
                   }
                 }
@@ -2034,7 +2158,7 @@ class WrappedResponses extends Responses {
                 model: openAIParams.model ?? modelFromResponse,
                 provider: 'openai',
                 input: formatOpenAIResponsesInput(sanitizeOpenAIResponse(openAIParams.input), openAIParams.instructions),
-                output: finalContent,
+                output: sanitizeOpenAIResponse(finalContent),
                 latency,
                 timeToFirstToken,
                 baseURL: this.baseURL,
@@ -2050,7 +2174,11 @@ class WrappedResponses extends Responses {
                 },
                 stopReason,
                 tools: availableTools,
-                completionId: completionIdFromResponse
+                completionId: completionIdFromResponse,
+                providerMetadata: buildProviderMetadata({
+                  incompleteDetails: terminalResponse?.incomplete_details
+                }),
+                error: getResponseFailure(terminalResponse)
               });
             } catch (error) {
               await captureAiGeneration(this.phClient, {
@@ -2081,7 +2209,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) {
@@ -2098,7 +2226,7 @@ class WrappedResponses extends Responses {
             model: openAIParams.model ?? result.model,
             provider: 'openai',
             input: formatOpenAIResponsesInput(sanitizeOpenAIResponse(openAIParams.input), openAIParams.instructions),
-            output: formattedOutput,
+            output: sanitizeOpenAIResponse(formattedOutput),
             latency,
             baseURL: this.baseURL,
             modelParameters: getModelParams(body, result.service_tier),
@@ -2115,8 +2243,10 @@ class WrappedResponses extends Responses {
             tools: availableTools,
             completionId: result.id,
             providerMetadata: buildProviderMetadata({
-              requestId: extractRequestId(result)
-            })
+              requestId: extractRequestId(result),
+              incompleteDetails: result.incomplete_details
+            }),
+            error: getResponseFailure(result)
           });
         }
         return result;
@@ -2140,7 +2270,7 @@ class WrappedResponses extends Responses {
         });
         throw error;
       });
-      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     }
   }
   parse(body, options) {
@@ -2149,64 +2279,57 @@ 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: sanitizeOpenAIResponse(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),
+          incompleteDetails: result.incomplete_details
+        }),
+        error: getResponseFailure(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 {
@@ -2263,7 +2386,7 @@ class WrappedEmbeddings extends Embeddings {
       });
       throw error;
     });
-    return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
+    return preserveProviderPromise(parentPromise, wrappedPromise);
   }
 }
 class WrappedAudio extends Audio {
@@ -2338,7 +2461,7 @@ class WrappedTranscriptions extends Transcriptions {
                 model: openAIParams.model,
                 provider: 'openai',
                 input: openAIParams.prompt,
-                output: finalContent,
+                output: sanitizeOpenAIResponse(finalContent),
                 latency,
                 timeToFirstToken,
                 baseURL: this.baseURL,
@@ -2373,7 +2496,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) {
@@ -2383,7 +2506,7 @@ class WrappedTranscriptions extends Transcriptions {
             model: openAIParams.model,
             provider: 'openai',
             input: openAIParams.prompt,
-            output: result.text,
+            output: sanitizeOpenAIResponse(result.text),
             latency,
             baseURL: this.baseURL,
             modelParameters: getModelParams(body),
@@ -2414,7 +2537,7 @@ class WrappedTranscriptions extends Transcriptions {
         });
         throw error;
       });
-      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
+      return preserveProviderPromise(parentPromise, wrappedPromise);
     }
   }
 }