klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 13:25:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle gemini returning simple string in response candidates

Browse Source
This commit is contained in:
Debanjum
2025-05-19 18:35:09 -07:00
parent 626ced8b8b
commit 33dd4c8c33
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/khoj/processor/conversation/google/utils.py
View File

@@ -259,6 +259,10 @@ def handle_gemini_response(
if len(candidates) == 0 and prompt_feedback: 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." message = f"\nI'd prefer to not respond to that due to **{prompt_feedback.block_reason.name}** issues with your query."
stopped = True 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 # Check if the response was blocked due to safety concerns with the generated content
elif candidates[0].finish_reason == gtypes.FinishReason.SAFETY: elif candidates[0].finish_reason == gtypes.FinishReason.SAFETY:
message = generate_safety_response(candidates[0].safety_ratings) message = generate_safety_response(candidates[0].safety_ratings)