Update agent.py

agent.py

@@ -43,9 +43,8 @@ def wiki_search(query: str) -> str:
         docs = WikipediaLoader(query=query, load_max_docs=2).load()
         if not docs:
             return "No Wikipedia results found."
-        return "
-
-".join(d.page_content[:2200] for d in docs)
+        # Fixed: String was broken across lines
+        return "\n\n".join(d.page_content[:2200] for d in docs)
     except Exception as e:
         return f"Wikipedia unavailable: {type(e).__name__}."
 
@@ -67,8 +66,8 @@ def fetch_page(url: str) -> str:
         soup = BeautifulSoup(r.text, "html.parser")
         for tag in soup(["script", "style", "nav", "footer", "header"]):
             tag.decompose()
-        text = soup.get_text(separator="
-", strip=True)
+        # Fixed: String was broken across lines
+        text = soup.get_text(separator="\n", strip=True)
         return text[:9000] if text else "Page appears empty."
     except Exception as e:
         return f"Could not fetch page: {type(e).__name__}."
@@ -104,7 +103,8 @@ TOOLS = [wiki_search, web_search, fetch_page, run_python]
 # PRECHECKS
 # ==========================================================
 
-def maybe_answer_direct(question: str) -> str | None:
+def maybe_answer_direct(question: str) -> str or None:
+    # Fixed: Corrected indentation for this block
     global LAST_PRECHECK
     q = question.strip()
     ql = q.lower()
@@ -183,15 +183,12 @@ Final format:
 FINAL ANSWER: <answer>
 """
 
+# Fixed: String formatting for route prompts
 ROUTE_PROMPTS = {
-    'python': BASE_RULES + "
-This is a structured task. Prefer one precise run_python call when needed.",
-    'wiki': BASE_RULES + "
-This is stable factual knowledge. Prefer wiki_search first.",
-    'web': BASE_RULES + "
-This is a web or URL task. Use web_search and fetch_page sparingly.",
-    'default': BASE_RULES + "
-Choose the shortest reliable path."
+    'python': BASE_RULES + "\nThis is a structured task. Prefer one precise run_python call when needed.",
+    'wiki': BASE_RULES + "\nThis is stable factual knowledge. Prefer wiki_search first.",
+    'web': BASE_RULES + "\nThis is a web or URL task. Use web_search and fetch_page sparingly.",
+    'default': BASE_RULES + "\nChoose the shortest reliable path."
 }
 
 # ==========================================================
@@ -329,20 +326,22 @@ def _validate_by_question(answer: str, question: str) -> str:
         return parts[-1] if len(parts) >= 1 and len(parts) <= 4 else 'N/A'
 
     if 'ioc country code' in q:
-        m = re.search(r'[A-Z]{3}', a)
+        # Fixed: Changed invalid  to \b
+        m = re.search(r'\b[A-Z]{3}\b', a)
         return m.group(0) if m else 'N/A'
 
     if 'how many' in q or 'what is the final numeric output' in q:
-        m = re.search(r'\d+(?:\.\d+)?', a)
+        # Fixed: Changed invalid  to \b
+        m = re.search(r'\b\d+(?:\.\d+)?\b', a)
         return m.group(0) if m else 'N/A'
 
     if 'algebraic notation' in q:
-        m = re.search(r'(?:O-O-O|O-O|[KQRBN]?[a-h]?[1-8]?x?[a-h][1-8](?:=[QRBN])?[+#]?)', a)
+        # Fixed: Changed invalid  to \b
+        m = re.search(r'\b(?:O-O-O|O-O|[KQRBN]?[a-h]?[1-8]?x?[a-h][1-8](?:=[QRBN])?[+#]?)\b', a)
         return m.group(0) if m else 'N/A'
 
     if 'comma-separated' in q or 'use them in the form' in q:
-        if len(a) <= 80 and '
-' not in a:
+        if len(a) <= 80 and '\n' not in a:
             return a
         return 'N/A'
 
@@ -405,4 +404,4 @@ def get_last_trace() -> dict:
         'model_error': LAST_MODEL_ERROR,
         'route': LAST_ROUTE,
         'precheck': LAST_PRECHECK,
-    }
+    }