From b18b7b2e333ab49e23fdfca0b0964ae42a869ca0 Mon Sep 17 00:00:00 2001 From: Debanjum Date: Thu, 19 Jun 2025 11:36:45 -0700 Subject: [PATCH] Handle unset response thoughts. Useful when retry on failed request Previously resulted in unbound local variable response_thoughts error --- src/khoj/processor/conversation/google/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/khoj/processor/conversation/google/utils.py b/src/khoj/processor/conversation/google/utils.py index acda998d..18ab33cc 100644 --- a/src/khoj/processor/conversation/google/utils.py +++ b/src/khoj/processor/conversation/google/utils.py @@ -105,6 +105,7 @@ def gemini_completion_with_backoff( gemini_clients[api_key] = client formatted_messages, system_instruction = format_messages_for_gemini(messages, system_prompt) + response_thoughts: str | None = None # format model response schema response_schema = None @@ -328,7 +329,7 @@ def generate_safety_response(safety_ratings: list[gtypes.SafetyRating]): def format_messages_for_gemini( original_messages: list[ChatMessage], system_prompt: str = None -) -> tuple[list[str], str]: +) -> tuple[list[gtypes.Content], str]: # Extract system message system_prompt = system_prompt or "" messages = deepcopy(original_messages)