@posthog/ai 7.19.7 → 7.20.0

package/dist/index.cjs

@@ -733,7 +733,7 @@ function formatOpenAIResponsesInput(input, instructions) {
   return messages;
 }
 
-var version = "7.19.7";
+var version = "7.20.0";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -894,6 +894,12 @@ const captureAiGeneration = async (client, options) => {
     ...(options.tools ? {
       $ai_tools: options.tools
     } : {}),
+    ...(options.completionId ? {
+      $ai_completion_id: options.completionId
+    } : {}),
+    ...(options.providerMetadata && Object.keys(options.providerMetadata).length > 0 ? {
+      $ai_provider_metadata: options.providerMetadata
+    } : {}),
     ...errorData,
     ...costOverrideData
   };
@@ -917,6 +923,32 @@ const captureAiGeneration = async (client, options) => {
 function isResponseTokenChunk(chunk) {
   return chunk.type === 'response.output_item.added' || chunk.type === 'response.content_part.added' || chunk.type === 'response.output_text.delta' || chunk.type === 'response.reasoning_text.delta' || chunk.type === 'response.reasoning_summary_text.delta' || chunk.type === 'response.audio.delta' || chunk.type === 'response.audio.transcript.delta' || chunk.type === 'response.refusal.delta';
 }
+/**
+ * Reads the OpenAI SDK's `_request_id` field from a response object. The SDK
+ * attaches the `x-request-id` response header here, but it is not part of the
+ * public response types, so it has to be read through a cast. Used to populate
+ * `$ai_provider_metadata.request_id`.
+ */
+function extractRequestId(result) {
+  return result?._request_id ?? undefined;
+}
+/**
+ * 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
+ * to report so the property can be omitted from the event entirely.
+ */
+function buildProviderMetadata(fields) {
+  const metadata = {};
+  if (fields.systemFingerprint) {
+    metadata.system_fingerprint = fields.systemFingerprint;
+  }
+  if (fields.requestId) {
+    metadata.request_id = fields.requestId;
+  }
+  return Object.keys(metadata).length > 0 ? metadata : undefined;
+}
 
 const Chat = openai.OpenAI.Chat;
 const Completions = Chat.Completions;
@@ -963,6 +995,10 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
         if ('tee' in value) {
           const [stream1, stream2] = value.tee();
           (async () => {
+            // Hoisted so the catch block can surface whatever was accumulated
+            // from the streamed chunks before the failure.
+            let completionIdFromResponse;
+            let systemFingerprintFromResponse;
             try {
               const contentBlocks = [];
               let accumulatedContent = '';
@@ -978,10 +1014,16 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
               const toolCallsInProgress = new Map();
               let rawUsageData;
               for await (const chunk of stream1) {
-                // Extract model from chunk (Chat Completions chunks have model field)
+                // Extract model and completion metadata from chunk (Chat Completions chunks carry these fields)
                 if (!modelFromResponse && chunk.model) {
                   modelFromResponse = chunk.model;
                 }
+                if (!completionIdFromResponse && chunk.id) {
+                  completionIdFromResponse = chunk.id;
+                }
+                if (!systemFingerprintFromResponse && chunk.system_fingerprint) {
+                  systemFingerprintFromResponse = chunk.system_fingerprint;
+                }
                 const choice = chunk?.choices?.[0];
                 if (choice?.finish_reason) {
                   stopReason = choice.finish_reason;
@@ -1097,7 +1139,11 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                   rawUsage: rawUsageData
                 },
                 stopReason,
-                tools: availableTools
+                tools: availableTools,
+                completionId: completionIdFromResponse,
+                providerMetadata: buildProviderMetadata({
+                  systemFingerprint: systemFingerprintFromResponse
+                })
               });
             } catch (error) {
               await captureAiGeneration(this.phClient, {
@@ -1113,6 +1159,13 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
                   inputTokens: 0,
                   outputTokens: 0
                 },
+                // If the stream fails mid-flight, surface whatever completion
+                // metadata the consumed chunks already provided so the error
+                // event can still be correlated to OpenAI's Logs dashboard.
+                completionId: completionIdFromResponse,
+                providerMetadata: buildProviderMetadata({
+                  systemFingerprint: systemFingerprintFromResponse
+                }),
                 error
               });
               throw error;
@@ -1148,7 +1201,12 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
               rawUsage: result.usage
             },
             stopReason: result.choices[0]?.finish_reason ?? undefined,
-            tools: availableTools
+            tools: availableTools,
+            completionId: result.id,
+            providerMetadata: buildProviderMetadata({
+              systemFingerprint: result.system_fingerprint,
+              requestId: extractRequestId(result)
+            })
           });
         }
         return result;
@@ -1195,6 +1253,9 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         if ('tee' in value && typeof value.tee === 'function') {
           const [stream1, stream2] = value.tee();
           (async () => {
+            // Hoisted so the catch block can surface the completion ID that
+            // was accumulated from the streamed chunks before the failure.
+            let completionIdFromResponse;
             try {
               let finalContent = [];
               let modelFromResponse;
@@ -1212,10 +1273,13 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                   firstTokenTime = Date.now();
                 }
                 if ('response' in chunk && chunk.response) {
-                  // Extract model from response object in chunk (for stored prompts)
+                  // Extract model and completion ID from the response object in the chunk (for stored prompts)
                   if (!modelFromResponse && chunk.response.model) {
                     modelFromResponse = chunk.response.model;
                   }
+                  if (!completionIdFromResponse && chunk.response.id) {
+                    completionIdFromResponse = chunk.response.id;
+                  }
                   const chunkWebSearchCount = calculateWebSearchCount(chunk.response);
                   if (chunkWebSearchCount > 0 && chunkWebSearchCount > (usage.webSearchCount ?? 0)) {
                     usage.webSearchCount = chunkWebSearchCount;
@@ -1261,7 +1325,8 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                   rawUsage: rawUsageData
                 },
                 stopReason,
-                tools: availableTools
+                tools: availableTools,
+                completionId: completionIdFromResponse
               });
             } catch (error) {
               await captureAiGeneration(this.phClient, {
@@ -1277,6 +1342,9 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
                   inputTokens: 0,
                   outputTokens: 0
                 },
+                // Surface the completion ID from any chunks consumed before
+                // the stream failed so the error event remains correlatable.
+                completionId: completionIdFromResponse,
                 error
               });
               throw error;
@@ -1313,7 +1381,11 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
               rawUsage: result.usage
             },
             stopReason: result.status ?? undefined,
-            tools: availableTools
+            tools: availableTools,
+            completionId: result.id,
+            providerMetadata: buildProviderMetadata({
+              requestId: extractRequestId(result)
+            })
           });
         }
         return result;
@@ -1371,7 +1443,11 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
             cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0,
             rawUsage: result.usage
           },
-          stopReason: result.status ?? undefined
+          stopReason: result.status ?? undefined,
+          completionId: result.id,
+          providerMetadata: buildProviderMetadata({
+            requestId: extractRequestId(result)
+          })
         });
         return result;
       }, async error => {
@@ -1627,6 +1703,10 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
         if ('tee' in value) {
           const [stream1, stream2] = value.tee();
           (async () => {
+            // Hoisted so the catch block can surface whatever was accumulated
+            // from the streamed chunks before the failure.
+            let completionIdFromResponse;
+            let systemFingerprintFromResponse;
             try {
               const contentBlocks = [];
               let accumulatedContent = '';
@@ -1639,10 +1719,16 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
               // Map to track in-progress tool calls
               const toolCallsInProgress = new Map();
               for await (const chunk of stream1) {
-                // Extract model from response if not in params
+                // Extract model and completion metadata from chunk (Chat Completions chunks carry these fields)
                 if (!modelFromResponse && chunk.model) {
                   modelFromResponse = chunk.model;
                 }
+                if (!completionIdFromResponse && chunk.id) {
+                  completionIdFromResponse = chunk.id;
+                }
+                if (!systemFingerprintFromResponse && chunk.system_fingerprint) {
+                  systemFingerprintFromResponse = chunk.system_fingerprint;
+                }
                 const choice = chunk?.choices?.[0];
                 // Handle text content
                 const deltaContent = choice?.delta?.content;
@@ -1739,7 +1825,11 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
                 baseURL: this.baseURL,
                 modelParameters: getModelParams(body),
                 httpStatus: 200,
-                usage
+                usage,
+                completionId: completionIdFromResponse,
+                providerMetadata: buildProviderMetadata({
+                  systemFingerprint: systemFingerprintFromResponse
+                })
               });
             } catch (error) {
               await captureAiGeneration(this.phClient, {
@@ -1755,6 +1845,13 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
                   inputTokens: 0,
                   outputTokens: 0
                 },
+                // If the stream fails mid-flight, surface whatever completion
+                // metadata the consumed chunks already provided so the error
+                // event can still be correlated to OpenAI's Logs dashboard.
+                completionId: completionIdFromResponse,
+                providerMetadata: buildProviderMetadata({
+                  systemFingerprint: systemFingerprintFromResponse
+                }),
                 error: error
               });
               throw error;
@@ -1784,7 +1881,12 @@ class WrappedCompletions extends openai.AzureOpenAI.Chat.Completions {
               outputTokens: result.usage?.completion_tokens ?? 0,
               reasoningTokens: result.usage?.completion_tokens_details?.reasoning_tokens ?? 0,
               cacheReadInputTokens: result.usage?.prompt_tokens_details?.cached_tokens ?? 0
-            }
+            },
+            completionId: result.id,
+            providerMetadata: buildProviderMetadata({
+              systemFingerprint: result.system_fingerprint,
+              requestId: extractRequestId(result)
+            })
           });
         }
         return result;
@@ -1831,6 +1933,9 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
         if ('tee' in value && typeof value.tee === 'function') {
           const [stream1, stream2] = value.tee();
           (async () => {
+            // Hoisted so the catch block can surface the completion ID that
+            // was accumulated from the streamed chunks before the failure.
+            let completionIdFromResponse;
             try {
               let finalContent = [];
               let modelFromResponse;
@@ -1845,10 +1950,13 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
                   firstTokenTime = Date.now();
                 }
                 if ('response' in chunk && chunk.response) {
-                  // Extract model from response if not in params (for stored prompts)
+                  // Extract model and completion ID from the response object in the chunk (for stored prompts)
                   if (!modelFromResponse && chunk.response.model) {
                     modelFromResponse = chunk.response.model;
                   }
+                  if (!completionIdFromResponse && chunk.response.id) {
+                    completionIdFromResponse = chunk.response.id;
+                  }
                 }
                 if (chunk.type === 'response.completed' && 'response' in chunk && chunk.response?.output && chunk.response.output.length > 0) {
                   finalContent = chunk.response.output;
@@ -1875,7 +1983,8 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
                 baseURL: this.baseURL,
                 modelParameters: getModelParams(body),
                 httpStatus: 200,
-                usage
+                usage,
+                completionId: completionIdFromResponse
               });
             } catch (error) {
               await captureAiGeneration(this.phClient, {
@@ -1891,6 +2000,9 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
                   inputTokens: 0,
                   outputTokens: 0
                 },
+                // Surface the completion ID from any chunks consumed before
+                // the stream failed so the error event remains correlatable.
+                completionId: completionIdFromResponse,
                 error: error
               });
               throw error;
@@ -1919,7 +2031,11 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
               outputTokens: result.usage?.output_tokens ?? 0,
               reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
               cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0
-            }
+            },
+            completionId: result.id,
+            providerMetadata: buildProviderMetadata({
+              requestId: extractRequestId(result)
+            })
           });
         }
         return result;
@@ -1970,7 +2086,11 @@ class WrappedResponses extends openai.AzureOpenAI.Responses {
           outputTokens: result.usage?.output_tokens ?? 0,
           reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
           cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0
-        }
+        },
+        completionId: result.id,
+        providerMetadata: buildProviderMetadata({
+          requestId: extractRequestId(result)
+        })
       });
       return result;
     }, async error => {