Update agents/tools/ai_tools.py

agents/tools/ai_tools.py

@@ -3,7 +3,7 @@
 # https://docs.crewai.com/introduction
 # https://ai.google.dev/gemini-api/docs
 
-import base64, chess, os, time
+import base64, chess, os, re, time
 from agents.models.llms import (
     LLM_WEB_SEARCH,
     LLM_WEB_BROWSER,
@@ -535,6 +535,22 @@ class AITools():
                         result = part.text
                         break
 
+                fen_pattern = r'\b([rnbqkpRNBQKP1-8\/]+\s+[wb]\s+(?:-|[KQkq]+)\s+(?:-|[a-h][36])\s+\d+\s+\d+)\b'
+
+                match = re.search(fen_pattern, result)
+                
+                if match:
+                    result = match.group(1)
+                else:
+                    lines = result.strip().split("\n")
+
+                    for line in lines:
+                        line = line.strip()
+                        
+                        if "/" in line and (" w " in line or " b " in line):
+                            result = line
+                            break
+                
                 board = chess.Board(result) # FEN validation
 
                 print(f"🛠️ AITools: img_to_fen_tool: model={model}")