@posthog/ai 8.9.0 → 8.9.1

package/dist/anthropic/index.mjs

@@ -330,7 +330,7 @@ function addDefaults(params) {
   };
 }
 
-var version = "8.9.0";
+var version = "8.9.1";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -589,14 +589,23 @@ const captureAiGeneration = async (client, options) => {
   }
 };
 
-function addRequestId(result, response, requestIdHeader) {
+function addResponseIds(result, response, requestIdHeader, workspaceIdHeader) {
   if (!result || typeof result !== 'object' || Array.isArray(result)) {
     return result;
   }
-  return Object.defineProperty(result, '_request_id', {
-    value: response.headers.get(requestIdHeader),
-    enumerable: false
-  });
+  const properties = {
+    _request_id: {
+      value: response.headers.get(requestIdHeader),
+      enumerable: false
+    }
+  };
+  if (workspaceIdHeader) {
+    properties._workspace_id = {
+      value: response.headers.get(workspaceIdHeader),
+      enumerable: false
+    };
+  }
+  return Object.defineProperties(result, properties);
 }
 function getResponsePropsPromise(parentPromise) {
   const responsePromise = parentPromise.responsePromise;
@@ -605,7 +614,7 @@ function getResponsePropsPromise(parentPromise) {
   }
   return responsePromise;
 }
-function decorateProviderPromise(wrappedPromise, responsePropsPromise, requestIdHeader, preserveThenUnwrap) {
+function decorateProviderPromise(wrappedPromise, responsePropsPromise, requestIdHeader, workspaceIdHeader, preserveThenUnwrap) {
   const providerPromise = wrappedPromise;
   if (responsePropsPromise) {
     providerPromise.asResponse = async () => (await responsePropsPromise).response;
@@ -614,7 +623,10 @@ function decorateProviderPromise(wrappedPromise, responsePropsPromise, requestId
       return {
         response: props.response,
         data,
-        request_id: props.response.headers.get(requestIdHeader)
+        request_id: props.response.headers.get(requestIdHeader),
+        ...(workspaceIdHeader ? {
+          workspace_id: props.response.headers.get(workspaceIdHeader)
+        } : {})
       };
     };
   }
@@ -623,8 +635,8 @@ function decorateProviderPromise(wrappedPromise, responsePropsPromise, requestId
       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);
+      const transformedPromise = Promise.all([wrappedPromise, responsePropsPromise]).then(([data, props]) => addResponseIds(transform(data, props), props.response, requestIdHeader, workspaceIdHeader));
+      return decorateProviderPromise(transformedPromise, responsePropsPromise, requestIdHeader, workspaceIdHeader, true);
     };
   }
   return providerPromise;
@@ -640,7 +652,7 @@ function decorateProviderPromise(wrappedPromise, responsePropsPromise, requestId
 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);
+  const providerPromise = decorateProviderPromise(wrappedPromise, responsePropsPromise, options.requestIdHeader ?? 'x-request-id', options.workspaceIdHeader, preserveThenUnwrap);
   if (!responsePropsPromise) {
     const asResponse = parentPromise.asResponse?.bind(parentPromise);
     if (asResponse) {
@@ -1082,7 +1094,8 @@ class WrappedMessages extends AnthropicOriginal.Messages {
         return value;
       });
       return preserveProviderPromise(parentPromise, wrappedPromise, {
-        requestIdHeader: 'request-id'
+        requestIdHeader: 'request-id',
+        workspaceIdHeader: 'anthropic-workspace-id'
       });
     } else {
       const parentPromise = super.create(anthropicParams, options);
@@ -1133,7 +1146,8 @@ class WrappedMessages extends AnthropicOriginal.Messages {
         throw error;
       });
       return preserveProviderPromise(parentPromise, wrappedPromise, {
-        requestIdHeader: 'request-id'
+        requestIdHeader: 'request-id',
+        workspaceIdHeader: 'anthropic-workspace-id'
       });
     }
   }