Hanlding avgLogprobs infinity

google-gemini-php · Jan 10, 2025 · 17f2ef5 · 17f2ef5
1 parent fd77c6b
commit 17f2ef5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/Data/Candidate.php b/src/Data/Candidate.php
@@ -61,6 +61,10 @@ public static function from(array $attributes): self
             default => null,
         };
 
+        if (($attributes['avgLogprobs'] ?? null) === 'Infinity') { // @phpstan-ignore-line
+            $attributes['avgLogprobs'] = INF;
+        }
+
         return new self(
             content: $content,
             finishReason: $finishReason,