medium thinking level

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",
     ),
 )