format and fix generative-model tests

Author: dlarocque

packages/ai/src/methods/chat-session.test.ts

@@ -266,8 +266,8 @@ describe('ChatSession', () => {
         undefined,
         requestOptions
       );
-      await expect(chatSession.sendMessageStream('hello', singleRequestOptions)).to.be
-        .rejected;
+      await expect(chatSession.sendMessageStream('hello', singleRequestOptions))
+        .to.be.rejected;
       expect(generateContentStreamStub).to.be.calledWith(
         fakeApiSettings,
         'a-model',
@@ -297,8 +297,8 @@ describe('ChatSession', () => {
         undefined,
         requestOptions
       );
-      await expect(chatSession.sendMessageStream('hello', singleRequestOptions)).to.be
-        .rejected;
+      await expect(chatSession.sendMessageStream('hello', singleRequestOptions))
+        .to.be.rejected;
       expect(generateContentStreamStub).to.be.calledWith(
         fakeApiSettings,
         'a-model',

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

@@ -293,34 +293,34 @@ describe('GenerativeModel', () => {
         signal: singleRequestOptions.signal
       })
     );
-    it('passes base model params through to ChatSession when there are no startChatParams', async () => {
-      const genModel = new GenerativeModel(fakeAI, {
-        model: 'my-model',
-        generationConfig: {
-          topK: 1
-        }
-      });
-      const chatSession = genModel.startChat();
-      expect(chatSession.params?.generationConfig).to.deep.equal({
+  });
+  it('passes base model params through to ChatSession when there are no startChatParams', async () => {
+    const genModel = new GenerativeModel(fakeAI, {
+      model: 'my-model',
+      generationConfig: {
         topK: 1
-      });
-      restore();
+      }
     });
-    it('overrides base model params with startChatParams', () => {
-      const genModel = new GenerativeModel(fakeAI, {
-        model: 'my-model',
-        generationConfig: {
-          topK: 1
-        }
-      });
-      const chatSession = genModel.startChat({
-        generationConfig: {
-          topK: 2
-        }
-      });
-      expect(chatSession.params?.generationConfig).to.deep.equal({
+    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(fakeAI, {
+      model: 'my-model',
+      generationConfig: {
+        topK: 1
+      }
+    });
+    const chatSession = genModel.startChat({
+      generationConfig: {
         topK: 2
-      });
+      }
+    });
+    expect(chatSession.params?.generationConfig).to.deep.equal({
+      topK: 2
     });
   });
   it('passes params through to chat.sendMessage', async () => {

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

@@ -750,7 +750,10 @@ describe('request methods', () => {
         '{}'
       );
 
-      await expect(requestPromise).to.be.rejectedWith(AIError, /Network failure/);
+      await expect(requestPromise).to.be.rejectedWith(
+        AIError,
+        /Network failure/
+      );
       expect(removeSpy).to.have.been.calledOnce;
     });