forcing web tool

src/ui/callbacks.py

@@ -110,18 +110,18 @@ def bot_response(
         ) + "🤔 Агент анализирует необходимость поиска..."
         yield history
 
-        # Промпт Роутера на английском (модели Qwen/Claude понимают его лучше)
         agent_messages = [
             {
                 "role": "system",
                 "content": (
-                    "You are an internal routing assistant. Decide if a web search is needed to answer the user's latest message.\n"
-                    "Respond ONLY with a valid JSON. No explanations.\n"
-                    'Format if search needed: {"search": true, "query": "exact search terms"}\n'
-                    'Format if NOT needed: {"search": false}'
-                ),
-            }
-        ]
+                    "Ты строгий системный классификатор. Твоя единственная цель - определить, нужен ли интернет-поиск.\n"
+                    "ПРАВИЛО 1: Если вопрос касается недавних событий, атак, инцидентов, новостей, дат или меняющихся фактов — поиск НУЖЕН ОБЯЗАТЕЛЬНО, даже если ты думаешь, что знаешь ответ.\n"
+                    "ПРАВИЛО 2: Отвечай СТРОГО и ТОЛЬКО в формате JSON. Запрещено использовать теги <think> и писать текст.\n\n"
+                    "Формат (если нужен поиск):\n"
+                    '{"search": true, "query": "оптимальный и короткий поисковый запрос"}\n\n'
+                    "Формат (если это просто болтовня или математика):\n"
+                    '{"search": false, "query": null}'
+                ]
 
         for msg in history[-3:]:
             if msg["role"] == "user":
@@ -131,11 +131,15 @@ def bot_response(
             eval_response = engine.generate(
                 messages=agent_messages,
                 max_tokens=150,
-                temperature=0.0,  # Обязательно 0 для точности!
+                temperature=0.0, # 0.0 для максимальной предсказуемости
                 stream=False,
             )
 
             raw_response = eval_response["choices"][0]["message"]["content"]
+
+            # ВАЖНО: Выводим в терминал (консоль), что ответил роутер, чтобы отлаживать
+            print(f"\n--- [DEBUG] ОТВЕТ РОУТЕРА ---\n{raw_response}\n---------------------------\n")
+
             clean_query = extract_search_query(raw_response)
 
             if clean_query:

src/utils/helpers.py

@@ -68,42 +68,41 @@ def web_search(query: str, max_results: int = 3) -> list:
 
 
 def extract_search_query(text: str):
-    """Извлекает поисковый запрос из JSON, Python-массивов или обычного текста."""
+    """Надежно извлекает запрос из ответа роутера."""
     if not text:
         return None
 
-    # 1. Ищем строгий JSON {"search": true, "query": "..."}
+    # Вырезаем теги мыслей, если упрямая модель их всё же сгенерировала
+    text_clean = re.sub(r"<think>.*?</think>", "", text, flags=re.DOTALL).strip()
+    if not text_clean:
+        text_clean = text  # Если ответ состоял только из тегов, ищем внутри них
+
     try:
-        json_match = re.search(r"```(?:json)?\s*(\{.*?\})\s*```", text, re.DOTALL)
+        # 1. Ищем блок ```json ... ```
+        json_match = re.search(r"```(?:json)?\s*(\{.*?\})\s*```", text_clean, re.DOTALL)
         if json_match:
             data = json.loads(json_match.group(1))
-            return data.get("query") if data.get("search") else None
+            # Проверяем флаг search (строго True)
+            if data.get("search") is True or str(data.get("search")).lower() == "true":
+                return data.get("query")
+            return None
 
-        bare_json = re.search(r"\{.*?\}", text, re.DOTALL)
+        # 2. Ищем просто фигурные скобки (голый JSON)
+        bare_json = re.search(r"\{.*?\}", text_clean, re.DOTALL)
         if bare_json:
             data = json.loads(bare_json.group(0))
-            if "query" in data and ("search" in data or "search_needed" in data):
+            if data.get("search") is True or str(data.get("search")).lower() == "true":
                 return data.get("query")
-    except Exception:
-        pass
+            return None
+    except Exception as e:
+        print(f"Ошибка парсинга JSON роутера: {e}")
 
-    # 2. Ищем специфичный формат массива из логов модели (как на скриншоте)
-    # Пример: [{'text': '🔍 Поиск по запросу: "attacks on..."', 'type': 'search'}]
+    # 3. Резервный вариант на случай старых галлюцинаций
     if "'type': 'search'" in text or '"type": "search"' in text:
         match = re.search(r"['\"]text['\"]\s*:\s*['\"](.*?)['\"]", text)
         if match:
             raw_q = match.group(1)
-            # Вытаскиваем то, что внутри двойных кавычек (сам запрос)
             sub_match = re.search(r'"([^"]+)"', raw_q)
-            clean_q = sub_match.group(1) if sub_match else raw_q
-            # Очищаем от мусорных слов, если они остались
-            return re.sub(r"^(?:🔍\s*)?Поиск по запросу:\s*", "", clean_q).strip()
-
-    # 3. Фолбэк: просто ищем текст в кавычках после слова query/запрос
-    match = re.search(
-        r'(?:запрос|query|ищем)\s*:?\s*["\']([^"\']+)["\']', text, re.IGNORECASE
-    )
-    if match:
-        return match.group(1)
+            return (sub_match.group(1) if sub_match else raw_q).strip()
 
     return None