Update agents/tools/ai_tools.py

agents/tools/ai_tools.py

@@ -64,12 +64,14 @@ class AITools():
             print(f"🤖 AITools: {tool_name}: result={result}")
             return result
         except Exception as e:
+            print(f"🤖 AITools: {tool_name}: result={str(e)}")
             raise RuntimeError(f"Processing failed: {str(e)}")
 
     def _extract_execution_result(response):
         for part in response.candidates[0].content.parts:
             if part.code_execution_result is not None:
                 return part.code_execution_result.output.strip()
+
         return None
 
     @tool("Web Search Tool")
@@ -105,6 +107,7 @@ class AITools():
             print(f"🤖 AITools: web_search_tool: result={result}")
             return result
         except Exception as e:
+            print(f"🤖 AITools: web_search_tool: result={str(e)}")
             raise RuntimeError(f"Processing failed: {str(e)}")
 
     @tool("Web Browser Tool")
@@ -149,6 +152,7 @@ class AITools():
                 print(f"🤖 AITools: web_browser_tool: result={result}")
                 return result
         except Exception as e:
+            print(f"🤖 AITools: web_browser_tool: result={str(e)}")
             raise RuntimeError(f"Processing failed: {str(e)}")
 
     @tool("Image Analysis Tool")
@@ -234,6 +238,7 @@ class AITools():
             print(f"🤖 AITools: youtube_analysis_tool: result={result}")
             return result
         except Exception as e:
+            print(f"🤖 AITools: youtube_analysis_tool: result={str(e)}")
             raise RuntimeError(f"Processing failed: {str(e)}")
     
     @tool("Document Analysis Tool")
@@ -283,6 +288,7 @@ class AITools():
             print(f"🤖 AITools: document_analysis_tool: result={result}")
             return result
         except Exception as e:
+            print(f"🤖 AITools: document_analysis_tool: result={str(e)}")
             raise RuntimeError(f"Processing failed: {str(e)}")
     
     @tool("Code Generation Tool")
@@ -319,6 +325,7 @@ class AITools():
             print(f"🤖 AITools: code_generation_tool: result={result}")
             return result
         except Exception as e:
+            print(f"🤖 AITools: code_generation_tool: result={str(e)}")
             raise RuntimeError(f"Processing failed: {str(e)}")
 
     @tool("Code Execution Tool")
@@ -358,6 +365,7 @@ class AITools():
             print(f"🤖 AITools: code_execution_tool: result={result}")
             return result
         except Exception as e:
+            print(f"🤖 AITools: code_execution_tool: result={str(e)}")
             raise RuntimeError(f"Processing failed: {str(e)}")
 
     @tool("Image to FEN Tool")
@@ -421,6 +429,7 @@ class AITools():
             print(f"🤖 AITools: img_to_fen_tool: result={result}")
             return result
         except Exception as e:
+            print(f"🤖 AITools: img_to_fen_tool: result={str(e)}")
             raise RuntimeError(f"Processing failed: {str(e)}")
 
     @tool("Algebraic Notation Tool")
@@ -440,7 +449,7 @@ class AITools():
         """
         print(f"🤖 AITools: algebraic_notation_tool: question={question}, file_path={file_path}, best_move={best_move}")
 
-        try:         
+        try:
             client = AITools._get_client()
 
             with open(file_path, "rb") as f:
@@ -483,27 +492,32 @@ class AITools():
             print(f"🤖 AITools: algebraic_notation_tool: result={result}")
             return result
         except Exception as e:
+            print(f"🤖 AITools: algebraic_notation_tool: result={str(e)}")
             raise RuntimeError(f"Processing failed: {str(e)}")
 
     def get_final_answer(question, answer):
         print(f"🤖 AITools: get_final_answer: question={question}, answer={answer}")
 
-        client = AITools._get_client()
-    
-        prompt = PROMPT_FINAL_ANSWER.format(question=question, answer=answer)
-
-        print(f"🤖 AITools: get_final_answer: prompt={prompt}")
+        try:
+            client = AITools._get_client()
+        
+            prompt = PROMPT_FINAL_ANSWER.format(question=question, answer=answer)
     
-        response = client.models.generate_content(
-            model=LLM_FINAL_ANSWER, 
-            contents=[prompt],
-            config=types.GenerateContentConfig(
-                thinking_config=types.ThinkingConfig(
-                    thinking_level=THINKING_LEVEL_FINAL_ANSWER
+            print(f"🤖 AITools: get_final_answer: prompt={prompt}")
+        
+            response = client.models.generate_content(
+                model=LLM_FINAL_ANSWER, 
+                contents=[prompt],
+                config=types.GenerateContentConfig(
+                    thinking_config=types.ThinkingConfig(
+                        thinking_level=THINKING_LEVEL_FINAL_ANSWER
+                    )
                 )
             )
-        )
-
-        result = response.text.strip()
-        print(f"🤖 AITools: get_final_answer: result={result}")
-        return result
+    
+            result = response.text.strip()
+            print(f"🤖 AITools: get_final_answer: result={result}")
+            return result
+        except Exception as e:
+            print(f"🤖 AITools: get_final_answer: result={str(e)}")
+            raise RuntimeError(f"Processing failed: {str(e)}")