From 72c0e5ca72f502f4316cb2aac097849b1bf85cb9 Mon Sep 17 00:00:00 2001 From: Christina Holland Date: Wed, 29 May 2024 13:45:50 -0700 Subject: [PATCH] format --- packages/main/src/models/generative-model.test.ts | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/packages/main/src/models/generative-model.test.ts b/packages/main/src/models/generative-model.test.ts index e6940446..b1881aac 100644 --- a/packages/main/src/models/generative-model.test.ts +++ b/packages/main/src/models/generative-model.test.ts @@ -317,7 +317,8 @@ describe("GenerativeModel", () => { systemInstruction: { role: "system", parts: [{ text: "be friendly" }] }, }); expect(genModel.systemInstruction?.parts[0].text).to.equal("be friendly"); - expect(genModel.generationConfig.responseSchema.properties.testField).to.exist; + expect(genModel.generationConfig.responseSchema.properties.testField).to + .exist; const mockResponse = getMockResponse( "unary-success-basic-reply-short.json", ); @@ -357,7 +358,8 @@ describe("GenerativeModel", () => { toolConfig: { functionCallingConfig: { mode: FunctionCallingMode.NONE } }, systemInstruction: { role: "system", parts: [{ text: "be friendly" }] }, }); - expect(genModel.generationConfig.responseSchema.properties.testField).to.exist; + expect(genModel.generationConfig.responseSchema.properties.testField).to + .exist; expect(genModel.tools?.length).to.equal(1); expect(genModel.toolConfig?.functionCallingConfig.mode).to.equal( FunctionCallingMode.NONE,