Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
hsubox76 committed May 29, 2024
1 parent d28a012 commit 72c0e5c
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions packages/main/src/models/generative-model.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -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",
);
Expand Down Expand Up @@ -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,
Expand Down

0 comments on commit 72c0e5c

Please sign in to comment.