put back parameters in test case

google-gemini · Sep 26, 2024 · c0a2cff · c0a2cff
1 parent 21be624
commit c0a2cff
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/src/models/generative-model.test.ts b/src/models/generative-model.test.ts
@@ -123,7 +123,11 @@ describe("GenerativeModel", () => {
           value.includes("be friendly") &&
           value.includes("temperature") &&
           value.includes("testField") &&
-          value.includes(HarmBlockThreshold.BLOCK_LOW_AND_ABOVE)
+          value.includes(HarmBlockThreshold.BLOCK_LOW_AND_ABOVE) &&
+          value.includes("presencePenalty") &&
+          value.includes("frequencyPenalty") &&
+          value.includes("responseLogprobs") &&
+          value.includes("logprobs")
         );
       }),
       match((value) => {
@@ -206,6 +210,10 @@ describe("GenerativeModel", () => {
             },
           },
         },
+        presencePenalty: 0.6,
+        frequencyPenalty: 0.5,
+        responseLogprobs: true,
+        logprobs: 2,
       },
       safetySettings: [
         {
@@ -235,7 +243,11 @@ describe("GenerativeModel", () => {
           value.includes("topK") &&
           value.includes("newTestField") &&
           !value.includes("testField") &&
-          value.includes(HarmCategory.HARM_CATEGORY_HARASSMENT)
+          value.includes(HarmCategory.HARM_CATEGORY_HARASSMENT) &&
+          value.includes("presencePenalty") &&
+          value.includes("frequencyPenalty") &&
+          value.includes("responseLogprobs") &&
+          value.includes("logprobs")
         );
       }),
       {},