Financial

app.py

@@ -2,11 +2,15 @@ import os
 import re
 import html as ihtml
 import textwrap
+import warnings
 import requests
 import gradio as gr
 
 from smolagents import CodeAgent, LiteLLMModel, tool
 
+# Убираем шумные предупреждения pydantic в логах (не критично для работы)
+warnings.filterwarnings("ignore", category=UserWarning, module="pydantic")
+
 
 # ----------------------------
 # Config (CPU-friendly defaults)
@@ -24,7 +28,7 @@ TIMEOUT = int(os.getenv("LITELLM_TIMEOUT", "3600"))
 
 
 def make_model(model_name: str) -> LiteLLMModel:
-    # flatten_messages_as_text=False помогает избежать проблем формата messages
+    # flatten_messages_as_text=False помогает избежать ошибок формата messages
     return LiteLLMModel(
         model_id=f"ollama_chat/{model_name}",
         api_base=OLLAMA_BASE,
@@ -50,13 +54,13 @@ def _strip_html(raw_html: str) -> str:
 
 @tool
 def web_search(query: str) -> str:
-    """Поиск в интернете (DuckDuckGo HTML), возвращает очищенный текст.
+    """Поиск (DuckDuckGo HTML) -> очищенный текст.
 
     Args:
         query: Текстовый поисковый запрос.
 
     Returns:
-        Очищенный текст (обрезан до ~4500 символов).
+        Очищенный текст (до ~4500 символов).
     """
     resp = requests.get(
         "https://duckduckgo.com/html/",
@@ -76,12 +80,11 @@ def extract_facts(case_description: str) -> str:
         case_description: Текст пользователя.
 
     Returns:
-        Строка с фактами + списком недостающих уточнений.
+        Факты + что уточнить (безопасно).
     """
     text = (case_description or "").strip()
     low = text.lower()
 
-    # Ключевые признаки
     flags = []
     if re.search(r"\b(просроч|просрочка|пропустил платеж)\b", low):
         flags.append("просрочка/пропуск платежа")
@@ -91,33 +94,57 @@ def extract_facts(case_description: str) -> str:
         flags.append("ипотека")
     if re.search(r"\b(кредит|займ|мфо)\b", low):
         flags.append("кредит/займ/МФО")
-    if re.search(r"\b(штраф|пени|неустойк)\b", low):
-        flags.append("штрафы/пени")
 
-    # Числа (в т.ч. суммы)
     nums = re.findall(r"\d[\d\s]{2,}", text)
     nums = [re.sub(r"\s+", "", n) for n in nums][:8]
 
-    missing = []
-    if "просроч" in low:
-        missing.append("Сколько дней/месяцев просрочка?")
-    if "коллект" in low:
-        missing.append("Кто кредитор/банк/МФО? (без номеров договоров)")
-        missing.append("Поступают ли угрозы/давление/звонки родственникам?")
-    if not nums:
-        missing.append("Какая сумма долга/минимального платежа/просрочки? (можно диапазоном)")
-    missing.append("Какой официальный доход/примерные обязательные расходы? (можно диапазоном)")
+    missing = [
+        "Сколько дней/месяцев просрочка?",
+        "Какая сумма просрочки/долга? (можно диапазоном)",
+        "Кто кредитор/банк/МФО? (без номеров договоров)",
+        "Есть ли письма/суды/исполнительное производство?",
+        "Какой официальный доход/обязательные расходы? (диапазоны)",
+    ]
 
     return (
         "ФАКТЫ ИЗ ОПИСАНИЯ (без домыслов):\n"
         f"- Исходный текст: {text}\n"
-        f"- Найденные признаки: {', '.join(flags) if flags else 'не выявлено'}\n"
-        f"- Найденные числа: {', '.join(nums) if nums else 'нет'}\n"
-        f"- Чего не хватает (уточнить безопасно): {('; '.join(missing)) if missing else 'ничего'}\n"
+        f"- Признаки: {', '.join(flags) if flags else 'не выявлено'}\n"
+        f"- Числа: {', '.join(nums) if nums else 'нет'}\n"
+        f"- Что уточнить (безопасно): {'; '.join(missing)}\n"
         "Правило: нельзя добавлять факты, которых нет в исходном тексте."
     )
 
 
+# ВАЖНО: в вашей версии smolagents CodeAgent парсит код из <code>...</code>.
+# Поэтому требуем именно такой формат, и завершаем final_answer(...).
+CODE_FORMAT_RULES = """
+КРИТИЧЕСКИ ВАЖНО: это CodeAgent.
+Ты обязан вернуть ТОЛЬКО Python-код внутри тегов:
+
+<code>
+# python code
+</code>
+
+Никакого текста до или после <code>...</code>.
+В конце обязательно вызови final_answer(...) и больше ничего.
+Если в предыдущей попытке ты не вернул <code>...</code>, ты обязан исправиться и вернуть только код.
+"""
+
+
+def _friendly_agent_error(where: str, err: Exception) -> str:
+    return (
+        f"### Ошибка на шаге: {where}\n\n"
+        "Модель не вернула корректный блок кода для CodeAgent (или код не распарсился).\n\n"
+        "Что помогает:\n"
+        "- Поставить модель поменьше: `qwen2.5-coder:1.5b`\n"
+        "- Уменьшить `MAX_TOKENS` до 768\n"
+        "- Оставить `NUM_CTX` 2048–4096\n\n"
+        "Текст ошибки:\n"
+        f"```text\n{repr(err)}\n```"
+    )
+
+
 def run_fin_aid_multi_agent(case_description: str, region: str, urgency: str, allow_internet: bool):
     case_description = (case_description or "").strip()
     region = (region or "").strip()
@@ -126,105 +153,93 @@ def run_fin_aid_multi_agent(case_description: str, region: str, urgency: str, al
     if not case_description:
         return "Пожалуйста, заполните поле «Описание ситуации»."
 
-    # Всегда даём extract_facts; web_search — только если allow_internet
     tools = [extract_facts] + ([web_search] if allow_internet else [])
 
     # ----------------------------
-    # Agent 1: TRIAGE
+    # TRIAGE
     # ----------------------------
     triage_agent = CodeAgent(
         tools=tools,
         model=make_model(TRIAGE_MODEL),
         add_base_tools=False,
-        max_steps=2,  # достаточно для одной попытки исправить код, но без лишних итераций
+        max_steps=5,  # даём шанс исправиться, если 1 раз не соблюдёт <code>
         additional_authorized_imports=[],
     )
 
     triage_prompt = f"""
-Ты агент TRIAGE (финансовая помощь). Всегда отвечай на РУССКОМ.
-
-КРИТИЧЕСКИ ВАЖНО: Ты работаешь как CodeAgent.
-Ты ДОЛЖЕН вернуть ТОЛЬКО Python-код внутри блока <code>...</code>.
-Ничего вне блока не пиши.
-
-ОБЯЗАТЕЛЬНО закончить через final_answer(triage_text).
+Ты агент TRIAGE (финансовая помощь). Всегда по-русски.
+{CODE_FORMAT_RULES}
 
 Правила:
 - НЕ выдумывай факты. Используй только входные данные и extract_facts().
 - НЕ предполагай взлом/мошенничество/кражу данных, если пользователь этого не писал.
-- НЕ проси чувствительные данные: номера карт, CVV, пароли, SMS-коды, номера документов.
-- Можно просить безопасные данные: суммы диапазонами, сроки, название кредитора без номеров.
+- НЕ проси номера карт/CVV/пароли/SMS-коды/номера документов.
 
-Если web_search доступен (есть в globals), ОБЯЗАТЕЛЬНО сделай 2 вызова:
-1) web_search(f"emergency financial assistance {region}")
-2) web_search("how to avoid financial aid scams")
-Не вставляй большие куски текста — используй только 3–6 коротких тезисов.
-
-Нужно сформировать triage_text в структуре:
-## Сводка ситуации
-## Допущения (только если явно помечаешь как допущение)
-## Приоритеты
-- Сегодня (24–72 часа)
-- На неделе
-- В течение месяца
-## Риски
-## Какие данные подготовить (безопасно)
-## Вопросы для уточнения (до 8)
+Если web_search доступен (проверь: 'web_search' in globals()):
+- сделай РОВНО 2 вызова:
+  s1 = web_search(f"emergency financial assistance {region}")
+  s2 = web_search("how to avoid financial aid scams")
+Используй только 3–6 коротких тезисов из s1/s2 (не вставляй большие куски).
+
+В коде обязательно:
+1) facts = extract_facts(case_description)
+2) has_ws = ('web_search' in globals())
+3) if has_ws: вызвать s1/s2 как выше, иначе s1=s2=""
+4) сформировать triage_text (Markdown) со структурой:
+   ## Сводка ситуации
+   ## Допущения (если есть)
+   ## Приоритеты
+   - Сегодня (24–72 часа)
+   - На неделе
+   - В течение месяца
+   ## Риски
+   ## Какие данные подготовить (безопасно)
+   ## Вопросы (до 8)
+5) final_answer(triage_text)
 
 Входные данные:
 urgency = \"\"\"{urgency}\"\"\"
 region = \"\"\"{region}\"\"\"
 case_description = \"\"\"{case_description}\"\"\"
 """
-
-    triage_result = triage_agent.run(textwrap.dedent(triage_prompt))
+    try:
+        triage_result = triage_agent.run(textwrap.dedent(triage_prompt))
+    except Exception as e:
+        return _friendly_agent_error("TRIAGE", e)
 
     # ----------------------------
-    # Agent 2: ACTIONS
+    # ACTIONS
     # ----------------------------
     actions_agent = CodeAgent(
         tools=tools,
         model=make_model(ACTIONS_MODEL),
         add_base_tools=False,
-        max_steps=2,
+        max_steps=5,
         additional_authorized_imports=[],
     )
 
     actions_prompt = f"""
-Ты агент ACTIONS (финансовая помощь). Всегда отвечай на РУССКОМ.
-
-КРИТИЧЕСКИ ВАЖНО: Ты работаешь как CodeAgent.
-Ты ДОЛЖЕН вернуть ТОЛЬКО Python-код внутри блока <code>...</code>.
-Ничего вне блока не пиши.
-
-ОБЯЗАТЕЛЬНО закончить через final_answer(actions_text).
+Ты агент ACTIONS (финансовая помощь). Всегда по-русски.
+{CODE_FORMAT_RULES}
 
 Правила:
-- НЕ выдумывай факты. Опирайся на triage_result и, при необходимости, extract_facts(case_description).
-- НЕ проси чувствительные данные: номера карт, CVV, пароли, SMS-коды, номера документов.
-- Дай конкретные действия, учитывая срочность, если есть риск визитов коллекторов/эскалации взыскания.
-
-Если web_search доступен, можно добавить 2–4 уточнённых категории помощи/анти-скам (без длинных вставок).
-
-Сформируй actions_text (Markdown) с блоками:
-## Действия
-### Срочно (urgent)
-- 4–8 пунктов
-### На неделе (short_term)
-- 4–8 пунктов
-### В течение месяца (mid_term)
-- 3–6 пунктов
-
-## Варианты помощи в регионе
-- 6–10 пунктов (категории: соцзащита/муниципалитет/НКО/юристы по долгам/банк/МФО/коммуналка/жильё)
-
-## Анти-мошенничество
-- 6–10 коротких пунктов, релевантных долгам и “помощи за предоплату”
-
-## Уточняющие вопросы (до 8)
-- только по делу
-
-Входные данные:
+- НЕ выдумывай факты.
+- НЕ проси номера карт/CVV/пароли/SMS-коды/номера документов.
+- Если кейс про просрочку/коллекторов: приоритет — деэскалация, фиксация, законные шаги, переговоры с кредитором.
+
+В коде обязательно:
+1) facts = extract_facts(case_description)
+2) сформировать actions_text (Markdown) с блоками:
+   ## Действия
+   ### Срочно (urgent) (4–8 пунктов)
+   ### На неделе (short_term) (4–8 пунктов)
+   ### В течение месяца (mid_term) (3–6 пунктов)
+   ## Варианты помощи в регионе (6–10 категорий)
+   ## Анти-мошенничество (6–10 релевантных пунктов для “помощь за предоплату/долги”)
+   ## Уточняющие вопросы (до 8)
+3) final_answer(actions_text)
+
+Вход:
 urgency = \"\"\"{urgency}\"\"\"
 region = \"\"\"{region}\"\"\"
 case_description = \"\"\"{case_description}\"\"\"
@@ -232,56 +247,52 @@ case_description = \"\"\"{case_description}\"\"\"
 triage_result:
 {triage_result}
 """
-
-    actions_result = actions_agent.run(textwrap.dedent(actions_prompt))
+    try:
+        actions_result = actions_agent.run(textwrap.dedent(actions_prompt))
+    except Exception as e:
+        return _friendly_agent_error("ACTIONS", e)
 
     # ----------------------------
-    # Agent 3: WRITER (final)
+    # WRITER (FINAL)
     # ----------------------------
     writer_agent = CodeAgent(
         tools=tools,
         model=make_model(WRITER_MODEL),
         add_base_tools=False,
-        max_steps=2,
+        max_steps=5,
         additional_authorized_imports=[],
     )
 
     writer_prompt = f"""
-Ты агент WRITER (финансовая помощь). Всегда отвечай на РУССКОМ.
-
-КРИТИЧЕСКИ ВАЖНО: Это финальный шаг.
-Ты ДОЛЖЕН вернуть ТОЛЬКО Python-код внутри блока <code>...</code>.
-Ничего вне блока не пиши.
-
-ОБЯЗАТЕЛЬНО закончить через final_answer(report).
+Ты агент WRITER (финансовая помощь). Всегда по-русски.
+{CODE_FORMAT_RULES}
 
 Правила:
 - НЕ выдумывай факты.
 - НЕ проси номера карт/CVV/пароли/SMS-коды/номера документов.
-- Не делай большой таблицы на 30 дней — только краткий “мини-бюджет” и пример.
-- Для кейса “просрочка + коллекторы” приоритеты должны отражать безопасность, деэскалацию, фиксацию коммуникации и легальные шаги.
+- Не делай таблицу на 30 строк — только краткий мини-бюджет и пример.
 
-Сформируй report (Markdown) строго по структуре:
+В коде обязательно:
+1) facts = extract_facts(case_description)
+2) собрать report (Markdown) по структуре:
 
 # План финансовой помощи
 ## Важно
-- не юр/фин совет
-- безопасность: не сообщать коды/пароли/CVV/SMS/номера документов
 ## Сводка ситуации
 ## Приоритеты
 ### Сегодня (24–72 часа)
 ### На неделе
 ### В течение месяца
 ## Пошаговый план
-(используй actions_result)
 ## Варианты помощи в регионе
 ## Мини-бюджет на 30 дней
-- 5–8 правил + 1 короткий пример расчёта (без таблицы на 30 строк)
 ## Анти-мошенничество
 ## Что подготовить
 ## Вопросы для уточнения
 
-Входные данные:
+3) final_answer(report)
+
+Вход:
 urgency = \"\"\"{urgency}\"\"\"
 region = \"\"\"{region}\"\"\"
 case_description = \"\"\"{case_description}\"\"\"
@@ -292,9 +303,11 @@ triage_result:
 actions_result:
 {actions_result}
 """
-
-    final_report = writer_agent.run(textwrap.dedent(writer_prompt))
-    return str(final_report).strip() if final_report is not None else ""
+    try:
+        final_report = writer_agent.run(textwrap.dedent(writer_prompt))
+        return str(final_report).strip() if final_report is not None else ""
+    except Exception as e:
+        return _friendly_agent_error("WRITER", e)
 
 
 with gr.Blocks(title="Financial Aid Navigator (CodeAgent + Ollama)") as demo:
@@ -308,11 +321,7 @@ with gr.Blocks(title="Financial Aid Navigator (CodeAgent + Ollama)") as demo:
         f"- NUM_CTX: `{NUM_CTX}`, MAX_TOKENS: `{MAX_TOKENS}`, TIMEOUT: `{TIMEOUT}`"
     )
 
-    region = gr.Textbox(
-        label="Регион",
-        placeholder="Например: Россия / Нидерланды / Польша",
-        lines=1,
-    )
+    region = gr.Textbox(label="Регион", lines=1, placeholder="Например: Россия")
     urgency = gr.Dropdown(
         ["срочно (24–72 часа)", "в течение недели", "не срочно (в течение месяца)"],
         value="в течение недели",
@@ -321,12 +330,9 @@ with gr.Blocks(title="Financial Aid Navigator (CodeAgent + Ollama)") as demo:
     case_description = gr.Textbox(
         label="Описание ситуации",
         lines=8,
-        placeholder="Например: Просрочил платеж, коллекторы стучатся в дверь. Сумма долга примерно ...",
-    )
-    allow_internet = gr.Checkbox(
-        label="Разрешить агенту поиск в интернете (web_search)",
-        value=False,
+        placeholder="Например: Просрочил платеж по ипотеке, коллекторы стучатся в дверь. Сумма ...",
     )
+    allow_internet = gr.Checkbox(label="Разрешить web_search (интернет)", value=False)
 
     run_btn = gr.Button("Сформировать план помощи (мультиагент)")
     output = gr.Textbox(label="Результат (Markdown)", lines=26)