From 33dd4c8c3393694076a63af6d2092a12739aef17 Mon Sep 17 00:00:00 2001 From: Debanjum Date: Mon, 19 May 2025 18:35:09 -0700 Subject: [PATCH] Handle gemini returning simple string in response candidates --- src/khoj/processor/conversation/google/utils.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/khoj/processor/conversation/google/utils.py b/src/khoj/processor/conversation/google/utils.py index ede54ba3..a72b8b84 100644 --- a/src/khoj/processor/conversation/google/utils.py +++ b/src/khoj/processor/conversation/google/utils.py @@ -259,6 +259,10 @@ def handle_gemini_response( if len(candidates) == 0 and prompt_feedback: message = f"\nI'd prefer to not respond to that due to **{prompt_feedback.block_reason.name}** issues with your query." stopped = True + # If response hits rate limit + elif isinstance(candidates[0], str): + message = candidates[0] + stopped = True # Check if the response was blocked due to safety concerns with the generated content elif candidates[0].finish_reason == gtypes.FinishReason.SAFETY: message = generate_safety_response(candidates[0].safety_ratings)