prompt ban for web

src/ui/callbacks.py

@@ -5,7 +5,7 @@ import gradio as gr
 
 from src.core.engine import engine
 from src.utils.helpers import (
-    extract_json_from_text,
+    extract_search_query,
     extract_text_from_file,
     get_clean_text,
     web_search,
@@ -35,7 +35,13 @@ def set_interactive(is_interactive):
 def bot_response(
     history, system_prompt, temperature, max_tokens, use_search, uploaded_file
 ):
-    messages = [{"role": "system", "content": system_prompt}]
+    # ПРЕДОХРАНИТЕЛЬ: Запрещаем основной модели выводить технические команды текстом.
+    system_guard = (
+        "\n\n[ВАЖНО]: Отвечай пользователю напрямую обычным текстом. "
+        "КАТЕГОРИЧЕСКИ ЗАПРЕЩАЕТСЯ выводить технические команды, массивы или JSON-вызовы (например [{'type': 'search'}]). "
+        "Если тебе нужно рассуждать перед ответом, оборачивай свои мысли СТРОГО в теги <think>твои мысли</think>."
+    )
+    messages = [{"role": "system", "content": system_prompt + system_guard}]
     file_info, file_content = "", ""
 
     # --- 1. ОБРАБОТКА ФАЙЛА ---
@@ -54,12 +60,22 @@ def bot_response(
     for msg in history[-7:]:
         content = str(msg["content"])
         if msg["role"] == "assistant":
-            # Удаляем плашки статусов
+            # Убираем плашки UI
             if "---\n\n" in content:
                 content = content.split("---\n\n", 1)[-1]
-            # Тотально вырезаем теги размышлений и HTML спойлеры из контекста!
+
+            # Тотально вырезаем теги размышлений и HTML спойлеры
             content = re.sub(r"<details.*?>.*?</details>", "", content, flags=re.DOTALL)
             content = re.sub(r"<think>.*?</think>", "", content, flags=re.DOTALL)
+
+            # ИСПРАВЛЕНИЕ: Вырезаем технические массивы поиска, если они проскочили в прошлых ответах
+            content = re.sub(
+                r"\[\s*\{.*?['\"]type['\"]\s*:\s*['\"]search['\"].*?\}\s*\]",
+                "",
+                content,
+                flags=re.DOTALL,
+            )
+
             content = content.strip()
             if not content:
                 content = "*(ответ скрыт)*"
@@ -69,7 +85,7 @@ def bot_response(
     history.append({"role": "assistant", "content": "⏳ Инициализация..."})
     yield history
 
-    # --- 3. АГЕНТ ПОИСКА (ПЕРЕПИСАНО НА JSON) ---
+    # --- 3. АГЕНТ ПОИСКА (РОУТЕР) ---
     search_info = ""
     if use_search:
         history[-1]["content"] = (
@@ -77,22 +93,19 @@ def bot_response(
         ) + "🤔 Агент анализирует необходимость поиска..."
         yield history
 
-        # Просим модель отвечать ТОЛЬКО в JSON, запрещаем теги <think> для этой задачи
+        # Промпт Роутера на английском (модели Qwen/Claude понимают его лучше)
         agent_messages = [
             {
                 "role": "system",
                 "content": (
-                    "Ты системный маршрутизатор. Оцени, нужен ли поиск в интернете для ответа на последний вопрос пользователя. "
-                    "Ответь СТРОГО в формате JSON. Никаких тегов <think>, никаких рассуждений. Только JSON.\n"
-                    "Формат:\n"
-                    '{"search_needed": true, "query": "твой поисковый запрос"}\n'
-                    "или\n"
-                    '{"search_needed": false, "query": null}'
+                    "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}'
                 ),
             }
         ]
 
-        # Добавляем последние сообщения пользователя для контекста
         for msg in history[-3:]:
             if msg["role"] == "user":
                 agent_messages.append({"role": "user", "content": msg["content"]})
@@ -101,17 +114,14 @@ def bot_response(
             eval_response = engine.generate(
                 messages=agent_messages,
                 max_tokens=150,
-                temperature=0.0,  # Обязательно 0 для роутера, чтобы не фантазировал!
+                temperature=0.0,  # Обязательно 0! Нам нужна точность, а не креатив.
                 stream=False,
             )
 
             raw_response = eval_response["choices"][0]["message"]["content"]
-            parsed_json = extract_json_from_text(raw_response)
-
-            search_needed = parsed_json.get("search_needed", False)
-            clean_query = parsed_json.get("query", None)
+            clean_query = extract_search_query(raw_response)
 
-            if search_needed and clean_query:
+            if clean_query:
                 search_info = f'🌐 Ищем: *"{clean_query}"*...'
                 history[-1]["content"] = (
                     f"{file_info + '\n' if file_info else ''}{search_info}"
@@ -122,14 +132,13 @@ def bot_response(
 
                 if search_results:
                     search_info = f'🌐 Найдено {len(search_results)} результатов по запросу *"{clean_query}"*'
-                    search_context = (
-                        "СВЕЖИЕ РЕЗУЛЬТАТЫ ПОИСКА ИЗ ИНТЕРНЕТА ДЛЯ ТВОЕГО ОТВЕТА:\n\n"
-                    )
+                    search_context = "СВЕЖИЕ РЕЗУЛЬТАТЫ ПОИСКА ИЗ ИНТЕРНЕТА:\n\n"
                     for i, r in enumerate(search_results, 1):
                         search_context += f"{i}. {r['title']} ({r['url']})\nСниппет: {r['snippet']}\n\n"
 
-                    # Инжектим результаты в системный промпт
-                    messages[0]["content"] += f"\n\n{search_context}"
+                    messages[0]["content"] += (
+                        f"\n\n{search_context}\n\n[УВЕДОМЛЕНИЕ]: Поиск выполнен. Сформулируй ответ на основе контекста выше."
+                    )
                 else:
                     search_info = (
                         f'🌐 Поиск по запросу *"{clean_query}"* не дал результатов.'
@@ -138,7 +147,7 @@ def bot_response(
                 search_info = "⚡ Агент ответит из своих знаний (поиск не нужен)."
 
         except Exception as e:
-            print(f"Ошибка при логике Агента: {e}")
+            print(f"Ошибка Роутера: {e}")
             search_info = "⚡ Ошибка роутера. Отвечаю из базы знаний."
 
     if file_content:
@@ -170,7 +179,7 @@ def bot_response(
                 partial_text += delta["content"]
                 display_text = partial_text
 
-                # Обработка тегов <think> для UI
+                # Умный UI для тегов <think>
                 if "<think>" in partial_text and "</think>" not in partial_text:
                     display_text = (
                         partial_text.replace(

src/utils/helpers.py

@@ -67,17 +67,43 @@ def web_search(query: str, max_results: int = 3) -> list:
         return []
 
 
-def extract_json_from_text(text: str) -> dict:
-    """Безопасно извлекает JSON из ответа модели (даже если есть маркдаун)."""
+def extract_search_query(text: str):
+    """Извлекает поисковый запрос из JSON, Python-массивов или обычного текста."""
+    if not text:
+        return None
+
+    # 1. Ищем строгий JSON {"search": true, "query": "..."}
     try:
-        # Ищем блок ```json ... ```
-        json_match = re.search(r"```json\s*(.*?)\s*```", text, re.DOTALL)
-        if json_match:
-            return json.loads(json_match.group(1))
-        # Иначе пробуем найти просто фигурные скобки
-        json_match = re.search(r"\{.*\}", text, re.DOTALL)
+        json_match = re.search(r"```(?:json)?\s*(\{.*?\})\s*```", text, re.DOTALL)
         if json_match:
-            return json.loads(json_match.group(0))
-        return {}
+            data = json.loads(json_match.group(1))
+            return data.get("query") if data.get("search") else None
+
+        bare_json = re.search(r"\{.*?\}", text, 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):
+                return data.get("query")
     except Exception:
-        return {}
+        pass
+
+    # 2. Ищем специфичный формат массива из логов модели (как на скриншоте)
+    # Пример: [{'text': '🔍 Поиск по запросу: "attacks on..."', 'type': 'search'}]
+    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 None