move test case

Author: dlarocque

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

@@ -165,19 +165,6 @@ describe('GenerativeModel', () => {
     );
     restore();
   });
-  it('passes base model params through to ChatSession when there are no startChatParams', async () => {
-    const genModel = new GenerativeModel(fakeVertexAI, {
-      model: 'my-model',
-      generationConfig: {
-        topK: 1
-      }
-    });
-    const chatSession = genModel.startChat();
-    expect(chatSession.params?.generationConfig).to.deep.equal({
-      topK: 1
-    });
-    restore();
-  });
   it('overrides base model params with startChatParams', () => {
     const genModel = new GenerativeModel(fakeVertexAI, {
       model: 'my-model',
@@ -201,7 +188,10 @@ describe('GenerativeModel', () => {
         { functionDeclarations: [{ name: 'myfunc', description: 'mydesc' }] }
       ],
       toolConfig: { functionCallingConfig: { mode: FunctionCallingMode.NONE } },
-      systemInstruction: { role: 'system', parts: [{ text: 'be friendly' }] }
+      systemInstruction: { role: 'system', parts: [{ text: 'be friendly' }] },
+      generationConfig: {
+        topK: 1
+      }
     });
     expect(genModel.tools?.length).to.equal(1);
     expect(genModel.toolConfig?.functionCallingConfig?.mode).to.equal(
@@ -225,7 +215,8 @@ describe('GenerativeModel', () => {
         return (
           value.includes('myfunc') &&
           value.includes(FunctionCallingMode.NONE) &&
-          value.includes('be friendly')
+          value.includes('be friendly') &&
+          value.includes('topK')
         );
       }),
       {}