Update agents/tools/ai_tools.py

agents/tools/ai_tools.py

@@ -48,10 +48,13 @@ class AITools():
                 model=LLM_WEB_SEARCH,
                 contents=question,
                 config=types.GenerateContentConfig(
-                    tools=[types.Tool(google_search=types.GoogleSearch())]
+                    tools=[types.Tool(google_search=types.GoogleSearch())],
+                    thinking_config=types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL
+                    )
                 )
             )
-    
+
             return response.text.strip()
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
@@ -133,7 +136,7 @@ class AITools():
                 config=types.GenerateContentConfig(
                     thinking_config=types.ThinkingConfig(
                         thinking_level=THINKING_LEVEL
-                    ),
+                    )
                 )
             )
             
@@ -173,7 +176,12 @@ class AITools():
     
             response = client.models.generate_content(
                 model=LLM_IMAGE_ANALYSIS,
-                contents=[file, question]
+                contents=[file, question],
+                config=types.GenerateContentConfig(
+                    thinking_config=types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL
+                    )
+                )
             )
           
             return response.text.strip()
@@ -201,7 +209,12 @@ class AITools():
             
             response = client.models.generate_content(
                 model=LLM_AUDIO_ANALYSIS, 
-                contents=[file, question]
+                contents=[file, question],
+                config=types.GenerateContentConfig(
+                    thinking_config=types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL
+                    )
+                )
             )
           
             return response.text.strip()
@@ -229,7 +242,12 @@ class AITools():
             
             response = client.models.generate_content(
                 model=LLM_VIDEO_ANALYSIS, 
-                contents=[file, question]
+                contents=[file, question],
+                config=types.GenerateContentConfig(
+                    thinking_config=types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL
+                    )
+                )
             )
     
             return response.text.strip()
@@ -258,6 +276,11 @@ class AITools():
                 contents=types.Content(
                     parts=[types.Part(file_data=types.FileData(file_uri=url)),
                            types.Part(text=question)]
+                ),
+                config=types.GenerateContentConfig(
+                    thinking_config=types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL
+                    )
                 )
             )
         except Exception as e:
@@ -296,7 +319,12 @@ class AITools():
             
             response = client.models.generate_content(
                 model=LLM_DOCUMENT_ANALYSIS,
-                contents=contents
+                contents=contents,
+                config=types.GenerateContentConfig(
+                    thinking_config=types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL
+                    )
+                )
             )
           
             return response.text.strip()
@@ -323,7 +351,10 @@ class AITools():
                 model=LLM_CODE_GENERATION,
                 contents=[f"{question}\n{json_data}"],
                 config=types.GenerateContentConfig(
-                    tools=[types.Tool(code_execution=types.ToolCodeExecution)]
+                    tools=[types.Tool(code_execution=types.ToolCodeExecution)],
+                    thinking_config=types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL
+                    )
                 ),
             )
             
@@ -356,7 +387,10 @@ class AITools():
                 model=LLM_CODE_EXECUTION,
                 contents=[file, question],
                 config=types.GenerateContentConfig(
-                    tools=[types.Tool(code_execution=types.ToolCodeExecution)]
+                    tools=[types.Tool(code_execution=types.ToolCodeExecution)],
+                    thinking_config=types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL
+                    )
                 ),
             )