@envive-ai/react-hooks 0.3.44 → 0.3.46

package/src/services/hardcopyService/__tests__/hardcopyService.test.ts

@@ -1,7 +1,7 @@
 import { WidgetTypeV3 } from 'src/contexts/typesV3';
 import type { WidgetText } from 'src/application/models/api/widgetText';
 import type { AppDetails } from 'src/atoms/app';
-import { EnviveMetricsEventName } from 'src/services/amplitudeService';
+import { EnviveMetricsEventName, TrackEventParams } from 'src/services/amplitudeService';
 import { getAtomStore } from 'src/atoms/atomStore/atomStore';
 import { resolvedBaseConfigVersionAtom } from 'src/atoms/envive/resolvedBaseConfigVersion';
 import { HardcopyService, MOCK_HARDCOPY_RESPONSE, MockHardcopyService } from '../hardcopyService';
@@ -269,6 +269,52 @@ describe('HardcopyService', () => {
       );
     });
 
+    it('forwards ai_suggestions_attribution_by_slot on WidgetTextResponse when present', async () => {
+      const trackEvent = vi.fn();
+      const attribution = {
+        prompt_button_texts: {
+          served_method_id: 'variant_b',
+          served_method_type: 'bandit_pbm',
+        },
+      };
+      mockGetHardcopy.mockResolvedValueOnce(
+        makeWidgetTextResponse({ aiSuggestionsAttributionBySlot: attribution }),
+      );
+
+      await makeService(trackEvent).getHardcopy({
+        widgetType: WidgetTypeV3.ChatPreviewV3,
+        userEvent: {} as any,
+      });
+
+      expect(trackEvent).toHaveBeenCalledWith(
+        expect.objectContaining({
+          eventName: EnviveMetricsEventName.WidgetTextResponse,
+          eventProps: expect.objectContaining({
+            ai_suggestions_attribution_by_slot: attribution,
+          }),
+        }),
+      );
+    });
+
+    it('omits ai_suggestions_attribution_by_slot when the response carries none', async () => {
+      const trackEvent = vi.fn();
+      mockGetHardcopy.mockResolvedValueOnce(makeWidgetTextResponse());
+
+      await makeService(trackEvent).getHardcopy({
+        widgetType: WidgetTypeV3.ChatPreviewV3,
+        userEvent: {} as any,
+      });
+
+      const responseEvent = trackEvent.mock.calls
+        .map(([params]: [TrackEventParams]) => params)
+        .find(
+          (params: TrackEventParams) =>
+            params.eventName === EnviveMetricsEventName.WidgetTextResponse,
+        );
+      expect(responseEvent).toBeDefined();
+      expect(responseEvent!.eventProps).not.toHaveProperty('ai_suggestions_attribution_by_slot');
+    });
+
     it('does not crash when trackEvent is not provided', async () => {
       const service = new HardcopyService({
         getAppDetails: () => mockAppDetails,

package/src/services/hardcopyService/hardcopyService.ts

@@ -237,6 +237,11 @@ export class HardcopyService implements IHardcopyService {
             response_id: response.responseId,
             widget_text: response.values,
             request_id: widgetTextRequest.requestId,
+            // Which method served each overridden slot — training/attribution
+            // reads this from the analytics event; passed through verbatim.
+            ...(response.aiSuggestionsAttributionBySlot
+              ? { ai_suggestions_attribution_by_slot: response.aiSuggestionsAttributionBySlot }
+              : {}),
           },
         });
         return convertedResponse;