diff --git a/packages/vertexai/src/requests/request.test.ts b/packages/vertexai/src/requests/request.test.ts index 0cd2d2686ce..6b90cac87d9 100644 --- a/packages/vertexai/src/requests/request.test.ts +++ b/packages/vertexai/src/requests/request.test.ts @@ -112,7 +112,9 @@ describe('request methods', () => { ); it('adds client headers', async () => { const headers = await getHeaders(fakeUrl); - expect(headers.get('x-goog-api-client')).to.include('gl-js/ fire/'); + expect(headers.get('x-goog-api-client')).to.match( + /gl-js\/[0-9\.]+ fire\/[0-9\.]+/ + ); }); it('adds api key', async () => { const headers = await getHeaders(fakeUrl); diff --git a/packages/vertexai/src/requests/request.ts b/packages/vertexai/src/requests/request.ts index 58a22b4ed33..d6df5ce0b5b 100644 --- a/packages/vertexai/src/requests/request.ts +++ b/packages/vertexai/src/requests/request.ts @@ -71,7 +71,7 @@ export class RequestUrl { */ function getClientHeaders(): string { const loggingTags = []; - loggingTags.push(`${LANGUAGE_TAG}/`); + loggingTags.push(`${LANGUAGE_TAG}/${PACKAGE_VERSION}`); loggingTags.push(`fire/${PACKAGE_VERSION}`); return loggingTags.join(' '); }