Update agents/tools/ai_tools.py

agents/tools/ai_tools.py

@@ -71,14 +71,16 @@ class AITools():
                         raise RuntimeError("Media file processing failed")
                     time.sleep(1)            
 
+                config_params = {}
+                if current_model != LLM_FALLBACK:
+                    config_params["thinking_config"] = types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL_MEDIA_ANALYSIS
+                    )
+
                 response = client.models.generate_content(
                     model=current_model,
                     contents=[file, question],
-                    config=types.GenerateContentConfig(
-                        thinking_config=types.ThinkingConfig(
-                            thinking_level=THINKING_LEVEL_MEDIA_ANALYSIS
-                        )
-                    )
+                    config=types.GenerateContentConfig(**config_params)
                 )
 
                 result = response.text
@@ -124,15 +126,16 @@ class AITools():
         
         for attempt in range(2):
             try:
+                config_params = {"tools": [types.Tool(google_search=types.GoogleSearch())]}
+                if model != LLM_FALLBACK:
+                    config_params["thinking_config"] = types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL_WEB_SEARCH
+                    )
+
                 response = client.models.generate_content(
                     model=model,
                     contents=question,
-                    config=types.GenerateContentConfig(
-                        tools=[types.Tool(google_search=types.GoogleSearch())],
-                        thinking_config=types.ThinkingConfig(
-                            thinking_level=THINKING_LEVEL_WEB_SEARCH
-                        )
-                    )
+                    config=types.GenerateContentConfig(**config_params)
                 )
 
                 result = response.text
@@ -269,17 +272,19 @@ class AITools():
         
         for attempt in range(2):
             try:
+                config_params = {}
+                if model != LLM_FALLBACK:
+                    config_params["thinking_config"] = types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL_YOUTUBE_ANALYSIS
+                    )
+
                 result = client.models.generate_content(
                     model=model,
                     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_YOUTUBE_ANALYSIS
-                        )
-                    )
+                    config=types.GenerateContentConfig(**config_params)
                 )
 
                 print(f"🛠️ AITools: youtube_analysis_tool: model={model}")
@@ -331,14 +336,16 @@ class AITools():
                     file = client.files.upload(file=file_path)
                     contents = [file, question]
                 
+                config_params = {}
+                if model != LLM_FALLBACK:
+                    config_params["thinking_config"] = types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL_DOCUMENT_ANALYSIS
+                    )
+
                 response = client.models.generate_content(
                     model=model,
                     contents=contents,
-                    config=types.GenerateContentConfig(
-                        thinking_config=types.ThinkingConfig(
-                            thinking_level=THINKING_LEVEL_DOCUMENT_ANALYSIS
-                        )
-                    )
+                    config=types.GenerateContentConfig(**config_params)
                 )
               
                 result = response.text
@@ -377,15 +384,16 @@ class AITools():
         
         for attempt in range(2):
             try:
+                config_params = {"tools": [types.Tool(code_execution=types.ToolCodeExecution)]}
+                if model != LLM_FALLBACK:
+                    config_params["thinking_config"] = types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL_CODE_GENERATION
+                    )
+
                 response = client.models.generate_content(
                     model=model,
                     contents=[f"{question}\n{json_data}"],
-                    config=types.GenerateContentConfig(
-                        tools=[types.Tool(code_execution=types.ToolCodeExecution)],
-                        thinking_config=types.ThinkingConfig(
-                            thinking_level=THINKING_LEVEL_CODE_GENERATION
-                        )
-                    ),
+                    config=types.GenerateContentConfig(**config_params),
                 )
 
                 result = AITools._extract_execution_result(response)
@@ -427,15 +435,16 @@ class AITools():
             try:
                 file = client.files.upload(file=file_path)
 
+                config_params = {"tools": [types.Tool(code_execution=types.ToolCodeExecution)]}
+                if model != LLM_FALLBACK:
+                    config_params["thinking_config"] = types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL_CODE_EXECUTION
+                    )
+
                 response = client.models.generate_content(
                     model=model,
                     contents=[file, question],
-                    config=types.GenerateContentConfig(
-                        tools=[types.Tool(code_execution=types.ToolCodeExecution)],
-                        thinking_config=types.ThinkingConfig(
-                            thinking_level=THINKING_LEVEL_CODE_EXECUTION
-                        )
-                    ),
+                    config=types.GenerateContentConfig(**config_params),
                 )
 
                 result = AITools._extract_execution_result(response)
@@ -494,14 +503,16 @@ class AITools():
                     ]
                 )
 
+                config_params = {}
+                if model != LLM_FALLBACK:
+                    config_params["thinking_config"] = types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL_IMAGE_TO_FEN
+                    )
+
                 response = client.models.generate_content(
                     model=model,
                     contents=[content],
-                    config=types.GenerateContentConfig(
-                        thinking_config=types.ThinkingConfig(
-                            thinking_level=THINKING_LEVEL_IMAGE_TO_FEN
-                        )
-                    )
+                    config=types.GenerateContentConfig(**config_params)
                 )
 
                 result = None
@@ -567,14 +578,16 @@ class AITools():
                     ]
                 )
                 
+                config_params = {}
+                if model != LLM_FALLBACK:
+                    config_params["thinking_config"] = types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL_ALGEBRAIC_NOTATION
+                    )
+
                 response = client.models.generate_content(
                     model=model,
                     contents=[content],
-                    config=types.GenerateContentConfig(
-                        thinking_config=types.ThinkingConfig(
-                            thinking_level=THINKING_LEVEL_ALGEBRAIC_NOTATION
-                        )
-                    )
+                    config=types.GenerateContentConfig(**config_params)
                 )
 
                 result = None
@@ -620,14 +633,16 @@ class AITools():
             try:
                 prompt = PROMPT_FINAL_ANSWER.format(question=question, answer=answer)
 
+                config_params = {}
+                if model != LLM_FALLBACK:
+                    config_params["thinking_config"] = types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL_FINAL_ANSWER
+                    )
+
                 response = client.models.generate_content(
                     model=model, 
                     contents=[prompt],
-                    config=types.GenerateContentConfig(
-                        thinking_config=types.ThinkingConfig(
-                            thinking_level=THINKING_LEVEL_FINAL_ANSWER
-                        )
-                    )
+                    config=types.GenerateContentConfig(**config_params)
                 )
         
                 result = response.text.strip()