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

Fix Gemini models to output valid json when configured

Browse Source
This commit is contained in:
Debanjum
2025-03-18 16:53:28 +05:30
parent ce60cb9779
commit 2ab8e711d3
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -65,6 +65,7 @@ def gemini_completion_with_backoff(
temperature=temperature, temperature=temperature,
max_output_tokens=MAX_OUTPUT_TOKENS_GEMINI, max_output_tokens=MAX_OUTPUT_TOKENS_GEMINI,
safety_settings=SAFETY_SETTINGS, safety_settings=SAFETY_SETTINGS,
response_mime_type=model_kwargs.get("response_mime_type", "text/plain") if model_kwargs else "text/plain",
) )
formatted_messages = [gtypes.Content(role=message.role, parts=message.content) for message in messages] formatted_messages = [gtypes.Content(role=message.role, parts=message.content) for message in messages]