bibibi12345 commited on
Commit
e02039d
·
1 Parent(s): 4fe6125

added thinking support for gemini mode

Browse files
Files changed (1) hide show
  1. src/google_api_client.py +10 -3
src/google_api_client.py CHANGED
@@ -299,9 +299,16 @@ def build_gemini_payload_from_native(native_request: dict, model_from_path: str)
299
  Build a Gemini API payload from a native Gemini request.
300
  This is used for direct Gemini API calls.
301
  """
302
- # Add default safety settings if not provided
303
- if "safetySettings" not in native_request:
304
- native_request["safetySettings"] = DEFAULT_SAFETY_SETTINGS
 
 
 
 
 
 
 
305
 
306
  return {
307
  "model": model_from_path,
 
299
  Build a Gemini API payload from a native Gemini request.
300
  This is used for direct Gemini API calls.
301
  """
302
+ native_request["safetySettings"] = DEFAULT_SAFETY_SETTINGS
303
+
304
+ if "generationConfig" not in native_request:
305
+ native_request["generationConfig"] = {}
306
+
307
+ if "thinkingConfig" not in native_request["generationConfig"]:
308
+ native_request["generationConfig"]["thinkingConfig"] = {}
309
+
310
+ native_request["generationConfig"]["thinkingConfig"]["includeThoughts"] = True
311
+ native_request["generationConfig"]["thinkingConfig"]["thinkingBudget"] = -1
312
 
313
  return {
314
  "model": model_from_path,