@posthog/ai 7.19.7 → 7.20.0

package/dist/openai/index.mjs

@@ -530,7 +530,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;
@@ -697,6 +697,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
   };
@@ -721,6 +727,34 @@ 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.Chat;
 const Completions = Chat.Completions;
 const Responses = OpenAI.Responses;
@@ -773,6 +807,10 @@ 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 = '';
@@ -789,10 +827,16 @@ 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;
@@ -914,7 +958,11 @@ class WrappedCompletions extends Completions {
                   rawUsage: rawUsageData
                 },
                 stopReason,
-                tools: availableTools
+                tools: availableTools,
+                completionId: completionIdFromResponse,
+                providerMetadata: buildProviderMetadata({
+                  systemFingerprint: systemFingerprintFromResponse
+                })
               });
             } catch (error) {
               await captureAiGeneration(this.phClient, {
@@ -930,6 +978,13 @@ 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;
@@ -966,7 +1021,12 @@ 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;
@@ -1020,6 +1080,9 @@ 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;
@@ -1037,10 +1100,13 @@ 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;
@@ -1086,7 +1152,8 @@ class WrappedResponses extends Responses {
                   rawUsage: rawUsageData
                 },
                 stopReason,
-                tools: availableTools
+                tools: availableTools,
+                completionId: completionIdFromResponse
               });
             } catch (error) {
               await captureAiGeneration(this.phClient, {
@@ -1102,6 +1169,9 @@ 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;
@@ -1138,7 +1208,11 @@ 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;
@@ -1196,7 +1270,11 @@ 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 => {