update vertexai names in tests

Author: hsubox76

packages/vertexai/src/api.test.ts

@@ -21,7 +21,7 @@ import { VertexAI } from './public-types';
 import { GenerativeModel } from './models/generative-model';
 import { VertexError } from './errors';
 
-const fakeVertex: VertexAI = {
+const fakeVertexAI: VertexAI = {
   app: {
     name: 'DEFAULT',
     automaticDataCollectionEnabled: true,
@@ -35,13 +35,13 @@ const fakeVertex: VertexAI = {
 
 describe('Top level API', () => {
   it('getGenerativeModel throws if no model is provided', () => {
-    expect(() => getGenerativeModel(fakeVertex, {} as ModelParams)).to.throw(
+    expect(() => getGenerativeModel(fakeVertexAI, {} as ModelParams)).to.throw(
       VertexError.NO_MODEL
     );
   });
   it('getGenerativeModel throws if no apiKey is provided', () => {
     const fakeVertexNoApiKey = {
-      ...fakeVertex,
+      ...fakeVertexAI,
       app: { options: { projectId: 'my-project' } }
     } as VertexAI;
     expect(() =>
@@ -50,15 +50,15 @@ describe('Top level API', () => {
   });
   it('getGenerativeModel throws if no projectId is provided', () => {
     const fakeVertexNoProject = {
-      ...fakeVertex,
+      ...fakeVertexAI,
       app: { options: { apiKey: 'my-key' } }
     } as VertexAI;
     expect(() =>
       getGenerativeModel(fakeVertexNoProject, { model: 'my-model' })
     ).to.throw(VertexError.NO_PROJECT_ID);
   });
   it('getGenerativeModel gets a GenerativeModel', () => {
-    const genModel = getGenerativeModel(fakeVertex, { model: 'my-model' });
+    const genModel = getGenerativeModel(fakeVertexAI, { model: 'my-model' });
     expect(genModel).to.be.an.instanceOf(GenerativeModel);
     expect(genModel.model).to.equal('publishers/google/models/my-model');
   });

packages/vertexai/src/models/generative-model.test.ts

@@ -18,7 +18,7 @@ import { expect } from 'chai';
 import { GenerativeModel } from './generative-model';
 import { VertexAI } from '../public-types';
 
-const fakeVertex: VertexAI = {
+const fakeVertexAI: VertexAI = {
   app: {
     name: 'DEFAULT',
     automaticDataCollectionEnabled: true,
@@ -32,23 +32,23 @@ const fakeVertex: VertexAI = {
 
 describe('GenerativeModel', () => {
   it('handles plain model name', () => {
-    const genModel = new GenerativeModel(fakeVertex, { model: 'my-model' });
+    const genModel = new GenerativeModel(fakeVertexAI, { model: 'my-model' });
     expect(genModel.model).to.equal('publishers/google/models/my-model');
   });
   it('handles models/ prefixed model name', () => {
-    const genModel = new GenerativeModel(fakeVertex, {
+    const genModel = new GenerativeModel(fakeVertexAI, {
       model: 'models/my-model'
     });
     expect(genModel.model).to.equal('publishers/google/models/my-model');
   });
   it('handles full model name', () => {
-    const genModel = new GenerativeModel(fakeVertex, {
+    const genModel = new GenerativeModel(fakeVertexAI, {
       model: 'publishers/google/models/my-model'
     });
     expect(genModel.model).to.equal('publishers/google/models/my-model');
   });
   it('handles prefixed tuned model name', () => {
-    const genModel = new GenerativeModel(fakeVertex, {
+    const genModel = new GenerativeModel(fakeVertexAI, {
       model: 'tunedModels/my-model'
     });
     expect(genModel.model).to.equal('tunedModels/my-model');