Spaces:
Running
Running
Commit
·
e02039d
1
Parent(s):
4fe6125
added thinking support for gemini mode
Browse files- 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 |
-
|
303 |
-
|
304 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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,
|