Commit
·
b1f7ea4
1
Parent(s):
743b676
enable thinking
Browse files- app/api_helpers.py +1 -0
app/api_helpers.py
CHANGED
|
@@ -244,6 +244,7 @@ def create_generation_config(request: OpenAIRequest) -> Dict[str, Any]:
|
|
| 244 |
types.SafetySetting(category="HARM_CATEGORY_HARASSMENT", threshold="OFF"),
|
| 245 |
types.SafetySetting(category="HARM_CATEGORY_CIVIC_INTEGRITY", threshold="OFF")
|
| 246 |
]
|
|
|
|
| 247 |
return config
|
| 248 |
|
| 249 |
def is_gemini_response_valid(response: Any) -> bool:
|
|
|
|
| 244 |
types.SafetySetting(category="HARM_CATEGORY_HARASSMENT", threshold="OFF"),
|
| 245 |
types.SafetySetting(category="HARM_CATEGORY_CIVIC_INTEGRITY", threshold="OFF")
|
| 246 |
]
|
| 247 |
+
config["thinking_config"] = types.ThinkingConfig(include_thoughts=True)
|
| 248 |
return config
|
| 249 |
|
| 250 |
def is_gemini_response_valid(response: Any) -> bool:
|