Add logging headers (#293)

Author: hsubox76

packages/app/src/constants.ts

@@ -38,6 +38,7 @@ import { name as remoteConfigCompatName } from '../../../packages/remote-config-
 import { name as storageName } from '../../../packages/storage/package.json';
 import { name as storageCompatName } from '../../../packages/storage-compat/package.json';
 import { name as firestoreName } from '../../../packages/firestore/package.json';
+import { name as vertexName } from '../../../packages/vertexai/package.json';
 import { name as firestoreCompatName } from '../../../packages/firestore-compat/package.json';
 import { name as packageName } from '../../../packages/firebase/package.json';
 
@@ -73,6 +74,7 @@ export const PLATFORM_LOG_STRING = {
   [storageCompatName]: 'fire-gcs-compat',
   [firestoreName]: 'fire-fst',
   [firestoreCompatName]: 'fire-fst-compat',
+  [vertexName]: 'fire-vertex',
   'fire-js': 'fire-js', // Platform identifier for JS SDK.
   [packageName]: 'fire-js-all'
 } as const;

packages/vertexai/src/constants.ts

@@ -15,6 +15,16 @@
  * limitations under the License.
  */
 
+import { version } from '../package.json';
+
 export const VERTEX_TYPE = 'vertex';
 
 export const DEFAULT_LOCATION = 'us-central1';
+
+export const DEFAULT_BASE_URL = 'https://firebaseml.googleapis.com';
+
+export const DEFAULT_API_VERSION = 'v2beta';
+
+export const PACKAGE_VERSION = version;
+
+export const LANGUAGE_TAG = 'gl-js';

packages/vertexai/src/requests/request.test.ts

@@ -19,8 +19,9 @@ import { expect, use } from 'chai';
 import { restore, stub } from 'sinon';
 import sinonChai from 'sinon-chai';
 import chaiAsPromised from 'chai-as-promised';
-import { DEFAULT_API_VERSION, RequestUrl, Task, makeRequest } from './request';
+import { RequestUrl, Task, makeRequest } from './request';
 import { ApiSettings } from '../types/internal';
+import { DEFAULT_API_VERSION } from '../constants';
 
 use(sinonChai);
 use(chaiAsPromised);

packages/vertexai/src/requests/request.ts

@@ -18,14 +18,12 @@
 import { RequestOptions } from '../types';
 import { ERROR_FACTORY, VertexError } from '../errors';
 import { ApiSettings } from '../types/internal';
-import { version } from '../../package.json';
-
-const DEFAULT_BASE_URL = 'https://firebaseml.googleapis.com';
-
-export const DEFAULT_API_VERSION = 'v2beta';
-
-const PACKAGE_VERSION = version;
-const PACKAGE_LOG_HEADER = 'firebase-vertexai-js';
+import {
+  DEFAULT_API_VERSION,
+  DEFAULT_BASE_URL,
+  LANGUAGE_TAG,
+  PACKAGE_VERSION
+} from '../constants';
 
 export enum Task {
   GENERATE_CONTENT = 'generateContent',
@@ -68,10 +66,13 @@ export class RequestUrl {
 }
 
 /**
- * Simple, but may become more complex if we add more versions to log.
+ * Log language and "fire/version" to x-goog-api-client
  */
 function getClientHeaders(): string {
-  return `${PACKAGE_LOG_HEADER}/${PACKAGE_VERSION}`;
+  const loggingTags = [];
+  loggingTags.push(`${LANGUAGE_TAG}/`);
+  loggingTags.push(`fire/${PACKAGE_VERSION}`);
+  return loggingTags.join(' ');
 }
 
 export async function makeRequest(