Update helpers.py

src/utils/helpers.py

@@ -68,41 +68,47 @@ def web_search(query: str, max_results: int = 3) -> list:
 
 
 def extract_search_query(text: str):
-    """Надежно извлекает запрос из ответа роутера."""
+    """Супер-надежный парсер запроса (с поддержкой XML, JSON и логов)."""
     if not text:
         return None
 
-    # Вырезаем теги мыслей, если упрямая модель их всё же сгенерировала
-    text_clean = re.sub(r"<think>.*?</think>", "", text, flags=re.DOTALL).strip()
-    if not text_clean:
-        text_clean = text  # Если ответ состоял только из тегов, ищем внутри них
+    # ВЫВОДИМ В КОНСОЛЬ ТО, ЧТО РЕАЛЬНО ПОДУМАЛ РОУТЕР
+    print(f"\n--- [DEBUG] RAW ROUTER OUTPUT ---\n{text}\n---------------------------\n")
 
-    try:
-        # 1. Ищем блок ```json ... ```
-        json_match = re.search(r"```(?:json)?\s*(\{.*?\})\s*```", text_clean, re.DOTALL)
-        if json_match:
-            data = json.loads(json_match.group(1))
-            # Проверяем флаг search (строго True)
-            if data.get("search") is True or str(data.get("search")).lower() == "true":
-                return data.get("query")
-            return None
+    # 1. Очищаем от размышлений
+    clean = re.sub(r"<think>.*?</think>", "", text, flags=re.DOTALL).strip()
+    if not clean:
+        clean = text
 
-        # 2. Ищем просто фигурные скобки (голый JSON)
-        bare_json = re.search(r"\{.*?\}", text_clean, re.DOTALL)
-        if bare_json:
-            data = json.loads(bare_json.group(0))
+    # 2. Ищем XML-тег (наш новый основной метод)
+    xml_match = re.search(r"<search>(.*?)</search>", clean, re.IGNORECASE)
+    if xml_match:
+        return xml_match.group(1).strip()
+
+    # 3. Резерв на случай JSON
+    json_match = re.search(r"\{.*?\}", clean, re.DOTALL)
+    if json_match:
+        try:
+            data = json.loads(json_match.group(0))
             if data.get("search") is True or str(data.get("search")).lower() == "true":
                 return data.get("query")
-            return None
-    except Exception as e:
-        print(f"Ошибка парсинга JSON роутера: {e}")
+        except:
+            pass
 
-    # 3. Резервный вариант на случай старых галлюцинаций
-    if "'type': 'search'" in text or '"type": "search"' in text:
-        match = re.search(r"['\"]text['\"]\s*:\s*['\"](.*?)['\"]", text)
+    # 4. Резерв на случай "старых галлюцинаций" с массивами
+    if "search" in text.lower():
+        match = re.search(
+            r"(?:запрос|query|text)['\"]\s*:\s*['\"]([^'\"]+)['\"]", text, re.IGNORECASE
+        )
         if match:
-            raw_q = match.group(1)
-            sub_match = re.search(r'"([^"]+)"', raw_q)
-            return (sub_match.group(1) if sub_match else raw_q).strip()
+            val = match.group(1)
+            val = re.sub(
+                r"^(?:🔍\s*)?(?:Поиск(?:\s*по\s*запросу)?|Search):\s*",
+                "",
+                val,
+                flags=re.IGNORECASE,
+            ).strip()
+            if val and val.lower() not in ["no", "false", "none", "no_search"]:
+                return val
 
     return None