File tree 2 files changed +3
-2
lines changed
src/khoj/processor/conversation/google
2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -143,7 +143,8 @@ def gemini_send_message_to_model(
143
143
# This caused unwanted behavior and terminates response early for gemini 1.5 series. Monitor for flakiness with 2.0 series.
144
144
if response_type == "json_object" and model in ["gemini-2.0-flash" ]:
145
145
model_kwargs ["response_mime_type" ] = "application/json"
146
- model_kwargs ["response_schema" ] = response_schema
146
+ if response_schema :
147
+ model_kwargs ["response_schema" ] = response_schema
147
148
148
149
# Get Response from Gemini
149
150
return gemini_completion_with_backoff (
Original file line number Diff line number Diff line change @@ -89,7 +89,7 @@ def gemini_completion_with_backoff(
89
89
90
90
# format model response schema
91
91
response_schema = None
92
- if model_kwargs and "response_schema" in model_kwargs :
92
+ if model_kwargs and not is_none_or_empty ( model_kwargs . get ( "response_schema" )) :
93
93
response_schema = clean_response_schema (model_kwargs ["response_schema" ])
94
94
95
95
seed = int (os .getenv ("KHOJ_LLM_SEED" )) if os .getenv ("KHOJ_LLM_SEED" ) else None
You can’t perform that action at this time.
0 commit comments