Add FinishReason.LANGUAGE enum value. (#198)

google-gemini · Jul 8, 2024 · 05b8631 · 05b8631
1 parent e23e99a
commit 05b8631
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 1 deletion.
diff --git a/.changeset/gentle-kangaroos-relax.md b/.changeset/gentle-kangaroos-relax.md
@@ -0,0 +1,5 @@
+---
+"@google/generative-ai": minor
+---
+
+Add FinishReason.LANGUAGE enum value.
diff --git a/common/api-review/generative-ai.api.md b/common/api-review/generative-ai.api.md
@@ -258,6 +258,8 @@ export enum FinishReason {
     // (undocumented)
     FINISH_REASON_UNSPECIFIED = "FINISH_REASON_UNSPECIFIED",
     // (undocumented)
+    LANGUAGE = "LANGUAGE",
+    // (undocumented)
     MAX_TOKENS = "MAX_TOKENS",
     // (undocumented)
     OTHER = "OTHER",

diff --git a/docs/reference/main/generative-ai.finishreason.md b/docs/reference/main/generative-ai.finishreason.md
@@ -17,6 +17,7 @@ export declare enum FinishReason
 |  Member | Value | Description |
 |  --- | --- | --- |
 |  FINISH\_REASON\_UNSPECIFIED | <code>&quot;FINISH_REASON_UNSPECIFIED&quot;</code> |  |
+|  LANGUAGE | <code>&quot;LANGUAGE&quot;</code> |  |
 |  MAX\_TOKENS | <code>&quot;MAX_TOKENS&quot;</code> |  |
 |  OTHER | <code>&quot;OTHER&quot;</code> |  |
 |  RECITATION | <code>&quot;RECITATION&quot;</code> |  |

diff --git a/packages/main/src/requests/response-helpers.ts b/packages/main/src/requests/response-helpers.ts
@@ -163,7 +163,11 @@ export function getFunctionCalls(
   }
 }
 
-const badFinishReasons = [FinishReason.RECITATION, FinishReason.SAFETY];
+const badFinishReasons = [
+  FinishReason.RECITATION,
+  FinishReason.SAFETY,
+  FinishReason.LANGUAGE,
+];
 
 function hadBadFinishReason(candidate: GenerateContentCandidate): boolean {
   return (

diff --git a/packages/main/types/enums.ts b/packages/main/types/enums.ts
@@ -95,6 +95,8 @@ export enum FinishReason {
   SAFETY = "SAFETY",
   // The candidate content was flagged for recitation reasons.
   RECITATION = "RECITATION",
+  // The candidate content was flagged for using an unsupported language.
+  LANGUAGE = "LANGUAGE",
   // Unknown reason.
   OTHER = "OTHER",
 }