From ac3192f33a929138abb7ddf43eadfbdec6448b2a Mon Sep 17 00:00:00 2001
From: ganler <jaway.liu@gmail.com>
Date: Sun, 19 May 2024 13:40:55 -0500
Subject: [PATCH] refactor(gemini): set empty response when no candidates

---
 repoqa/provider/google.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/repoqa/provider/google.py b/repoqa/provider/google.py
index 073158f..63ec7b4 100644
--- a/repoqa/provider/google.py
+++ b/repoqa/provider/google.py
@@ -31,9 +31,8 @@ def generate_reply(
             system_msg=system_msg,
         )
 
-        assert (
-            len(replies.candidates) == n
-        ), f"# replies = {len(replies.candidates)} != {n = }"
+        if len(replies.candidates) != n:
+            print(f"[WARNING] # replies = {len(replies.candidates)} != {n = }")
 
         ret_texts = []
         for candidate in replies.candidates:
@@ -45,4 +44,4 @@ def generate_reply(
                 ret_texts.append("")
                 print(f"{candidate.safety_ratings = }")
 
-        return ret_texts
+        return ret_texts + [""] * (n - len(ret_texts))