From 111e47cb77550d8226166093daec9ac7fe2375e4 Mon Sep 17 00:00:00 2001 From: Matteo Rosati Date: Thu, 29 Jan 2026 10:23:23 +0100 Subject: [PATCH] medium thinking level --- llm_config.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/llm_config.py b/llm_config.py index e11a33b..e52843d 100644 --- a/llm_config.py +++ b/llm_config.py @@ -9,7 +9,8 @@ tools = [ types.Tool( retrieval=types.Retrieval( vertex_rag_store=types.VertexRagStore( - rag_resources=[types.VertexRagStoreRagResource(rag_corpus=CORPUS)], + rag_resources=[ + types.VertexRagStoreRagResource(rag_corpus=CORPUS)], ) ) ) @@ -20,17 +21,19 @@ generate_content_config = types.GenerateContentConfig( top_p=0.95, max_output_tokens=65535, safety_settings=[ - types.SafetySetting(category="HARM_CATEGORY_HATE_SPEECH", threshold="OFF"), + types.SafetySetting( + category="HARM_CATEGORY_HATE_SPEECH", threshold="OFF"), types.SafetySetting( category="HARM_CATEGORY_DANGEROUS_CONTENT", threshold="OFF" ), types.SafetySetting( category="HARM_CATEGORY_SEXUALLY_EXPLICIT", threshold="OFF" ), - types.SafetySetting(category="HARM_CATEGORY_HARASSMENT", threshold="OFF"), + types.SafetySetting( + category="HARM_CATEGORY_HARASSMENT", threshold="OFF"), ], tools=tools, thinking_config=types.ThinkingConfig( - thinking_level="HIGH", + thinking_level="MEDIUM", ), )