From 2ab8e711d37217fcd711f31415b85c16abd9239a Mon Sep 17 00:00:00 2001
From: Debanjum <debanjum@gmail.com>
Date: Tue, 18 Mar 2025 16:53:28 +0530
Subject: [PATCH] Fix Gemini models to output valid json when configured

---
 src/khoj/processor/conversation/google/utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/khoj/processor/conversation/google/utils.py b/src/khoj/processor/conversation/google/utils.py
index 2a5f994f..ebe91527 100644
--- a/src/khoj/processor/conversation/google/utils.py
+++ b/src/khoj/processor/conversation/google/utils.py
@@ -65,6 +65,7 @@ def gemini_completion_with_backoff(
         temperature=temperature,
         max_output_tokens=MAX_OUTPUT_TOKENS_GEMINI,
         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]