Spaces:
Paused
Paused
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,
|