klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 21:19:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle unset response thoughts. Useful when retry on failed request

Browse Source 
Previously resulted in unbound local variable response_thoughts error
This commit is contained in:
Debanjum
2025-06-19 11:36:45 -07:00
parent 906ff46e6c
commit b18b7b2e33
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -105,6 +105,7 @@ def gemini_completion_with_backoff(
gemini_clients[api_key] = client gemini_clients[api_key] = client
formatted_messages, system_instruction = format_messages_for_gemini(messages, system_prompt) formatted_messages, system_instruction = format_messages_for_gemini(messages, system_prompt)
response_thoughts: str | None = None
# format model response schema # format model response schema
response_schema = None response_schema = None
@@ -328,7 +329,7 @@ def generate_safety_response(safety_ratings: list[gtypes.SafetyRating]):
def format_messages_for_gemini( def format_messages_for_gemini(
original_messages: list[ChatMessage], system_prompt: str = None original_messages: list[ChatMessage], system_prompt: str = None
) -> tuple[list[str], str]: ) -> tuple[list[gtypes.Content], str]:
# Extract system message # Extract system message
system_prompt = system_prompt or "" system_prompt = system_prompt or ""
messages = deepcopy(original_messages) messages = deepcopy(original_messages)