refactor(gemini): set empty response when no candidates

evalplus · May 19, 2024 · ac3192f · ac3192f
1 parent 77c1782
commit ac3192f
Showing 1 changed file with 3 additions and 4 deletions.
diff --git 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))