Update agents/tools/ai_tools.py

agents/tools/ai_tools.py

@@ -1,7 +1,6 @@
 # References:
 
-# # https://docs.crewai.com/introduction
-# https://docs.crewai.com/en/tools/web-scraping/stagehandtool
+# https://docs.crewai.com/introduction
 # https://ai.google.dev/gemini-api/docs
 
 import base64, chess, os
@@ -108,11 +107,14 @@ class AITools():
                 wait_for_captcha_solves=True,
                 verbose=3
             ) as stagehand_tool:
-                return stagehand_tool.run(
+                result = stagehand_tool.run(
                     instruction=question, 
                     url=url_str,
                     command_type="act" # TODO: act, extract, observe
                 )
+
+                print(f"🤖 Result: {result}")
+                return result
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
 
@@ -146,8 +148,10 @@ class AITools():
                     )
                 )
             )
-          
-            return response.text.strip()
+
+            result = response.text.strip()
+            print(f"🤖 Result: {result}")
+            return result
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
     
@@ -181,8 +185,10 @@ class AITools():
                     )
                 )
             )
-          
-            return response.text.strip()
+
+            result = response.text.strip()
+            print(f"🤖 Result: {result}")
+            return result
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
     
@@ -216,8 +222,10 @@ class AITools():
                     )
                 )
             )
-    
-            return response.text.strip()
+
+            result = response.text.strip()
+            print(f"🤖 Result: {result}")
+            return result
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
             
@@ -240,7 +248,7 @@ class AITools():
         try:
             client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
     
-            return client.models.generate_content(
+            result = client.models.generate_content(
                 model=LLM_YOUTUBE_ANALYSIS,
                 contents=types.Content(
                     parts=[types.Part(file_data=types.FileData(file_uri=url)),
@@ -252,6 +260,9 @@ class AITools():
                     )
                 )
             )
+
+            print(f"🤖 Result: {result}")
+            return result
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
     
@@ -298,7 +309,9 @@ class AITools():
                 )
             )
           
-            return response.text.strip()
+            result = response.text.strip()
+            print(f"🤖 Result: {result}")
+            return result
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
     
@@ -333,7 +346,11 @@ class AITools():
             
             for part in response.candidates[0].content.parts:
                 if part.code_execution_result is not None:
-                    return part.code_execution_result.output
+                    result part.code_execution_result.output
+                    break
+
+            print(f"🤖 Result: {result}")
+            return result
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
 
@@ -371,7 +388,11 @@ class AITools():
             
             for part in response.candidates[0].content.parts:
                 if part.code_execution_result is not None:
-                    return part.code_execution_result.output
+                    result = part.code_execution_result.output
+                    break
+
+            print(f"🤖 Result: {result}")
+            return result
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
 
@@ -421,19 +442,18 @@ class AITools():
                     )
                 )
             )
-            
-            fen = None
-            
+
+            result = None
+
             for part in response.parts:
                 if part.text is not None:
-                    fen = part.text.strip()
+                    result = part.text.strip()
                     break
 
-            board = chess.Board(fen) # FEN validation
+            board = chess.Board(result) # FEN validation
             
-            print(f"🤖 FEN: {fen}")
-
-            return fen;
+            print(f"🤖 Result: {result}")
+            return result
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
 
@@ -490,9 +510,8 @@ class AITools():
                     result = part.text.strip()
                     break
 
-            print(f"🤖 Algebraic notation: {result}")
-
-            return result;
+            print(f"🤖 Result: {result}")
+            return result
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
 
@@ -512,5 +531,7 @@ class AITools():
                 )
             )
         )
-        
-        return response.text
+
+        result = response.text.strip()
+        print(f"🤖 Result: {result}")
+        return result