diff --git a/packages/sdk/server-ai/__tests__/LDAIClientImpl.test.ts b/packages/sdk/server-ai/__tests__/LDAIClientImpl.test.ts index 6c75df56a..e8c9e8cd6 100644 --- a/packages/sdk/server-ai/__tests__/LDAIClientImpl.test.ts +++ b/packages/sdk/server-ai/__tests__/LDAIClientImpl.test.ts @@ -65,6 +65,14 @@ it('returns config with interpolated messagess', async () => { enabled: true, toVercelAISDK: expect.any(Function), }); + + // Verify tracking was called + expect(mockLdClient.track).toHaveBeenCalledWith( + '$ld:ai:config:function:single', + testContext, + key, + 1, + ); }); it('includes context in variables for messages interpolation', async () => { diff --git a/packages/sdk/server-ai/src/LDAIClientImpl.ts b/packages/sdk/server-ai/src/LDAIClientImpl.ts index 331e74800..b90617714 100644 --- a/packages/sdk/server-ai/src/LDAIClientImpl.ts +++ b/packages/sdk/server-ai/src/LDAIClientImpl.ts @@ -144,6 +144,8 @@ export class LDAIClientImpl implements LDAIClient { defaultValue: LDAIDefaults, variables?: Record, ): Promise { + this._ldClient.track('$ld:ai:config:function:single', context, key, 1); + const { tracker, enabled,