Upload app/agent/prompts.py with huggingface_hub

app/agent/prompts.py

@@ -1,57 +1,82 @@
-SYSTEM_PROMPT = """Ты - AI ассистент, который помогает заполнить чеклист созвона с клиентом.
-Твоя задача - задавать вопросы и анализировать ответы, чтобы собрать всю необходимую информацию о проекте клиента.
-
-Основные темы для выяснения:
-1. Общая информация о проекте (название, описание, контакты)
-2. Цели и задачи (что хотят достичь, ключевые метрики)
-3. Сроки и бюджет (дедлайны, финансовые ограничения)
-4. Технические требования (интеграции, платформы, технологии)
-5. Дополнительная информация (риски, особенности, пожелания)
-
-Правила:
-- Задавай открытые вопросы
-- Адаптируй следующие вопросы на основе полученных ответов
-- Будь вежливым и профессиональным
-- Все общение ведется на русском языке
-"""
-
-INITIAL_QUESTIONS_PROMPT = """Сгенерируй 3 начальных вопроса для клиента.
-
-Вопросы должны быть направлены на выяснение:
-1. Общей информации о проекте
-2. Целей и ожидаемых результатов
-3. Текущей ситуации и контекста
-
-Формат ответа - JSON:
-{
-    "questions": [
-        {"id": "q1", "text": "..."},
-        {"id": "q2", "text": "..."},
-        {"id": "q3", "text": "..."}
-    ]
-}
-"""
-
-ANALYZE_ROUND_PROMPT = """Проанализируй ответы клиента и:
-1. Создай краткое саммари раунда (2-3 предложения)
-2. Определи, какая информация уже получена
-3. Определи, что еще нужно уточнить
-4. Сгенерируй 3 уточняющих вопроса для следующего раунда
-
-Фокусируйся на недостающей информации и углубляй понимание проекта.
-"""
-
-GENERATE_CHECKLIST_PROMPT = """На основе всех полученных ответов создай структурированный чеклист.
-
-Используй категории:
-- Общая информация
-- Цели и задачи
-- Сроки и бюджет
-- Технические требования
-- Дополнительные заметки
-
-Для каждого пункта укажи статус:
-- confirmed - информация подтверждена
-- needs_clarification - требует уточнения
-- not_discussed - не обсуждалось
-"""
+SYSTEM_PROMPT = """Ты — AI-ассистент для заполнения чеклиста созвона с клиентом.
+Твоя задача — задавать вопросы, анализировать ответы и формировать структурированный чеклист.
+Всегда отвечай на русском языке.
+Формат ответа — строго JSON, без лишнего текста."""
+
+INITIAL_QUESTIONS_PROMPT = """Сгенерируй 3 начальных вопроса для чеклиста созвона с клиентом.
+Вопросы должны быть открытыми и покрывать основные темы:
+1. Информация о проекте и клиенте
+2. Цели и задачи проекта
+3. Сроки и бюджет
+
+Верни JSON массив из 3 объектов:
+[{"id": "q1", "text": "текст вопроса"}, ...]"""
+
+
+def analyze_round_prompt(round_number: int, questions_and_answers: str, previous_summaries: str) -> str:
+    return f"""Проанализируй ответы раунда {round_number}.
+
+Вопросы и ответы:
+{questions_and_answers}
+
+{"Предыдущие резюме раундов:" + chr(10) + previous_summaries if previous_summaries else ""}
+
+Создай краткое резюме раунда, выделив:
+- Что было выяснено
+- Что требует уточнения
+- Какие темы ещё не затронуты
+
+Верни JSON объект:
+{{"summary": "текст резюме"}}"""
+
+
+def next_questions_prompt(round_number: int, all_summaries: str) -> str:
+    return f"""На основе анализа предыдущих раундов, сгенерируй 3 уточняющих вопроса для раунда {round_number}.
+
+Резюме предыдущих раундов:
+{all_summaries}
+
+Вопросы должны:
+- Уточнять неясные моменты из предыдущих ответов
+- Раскрывать незатронутые темы
+- Быть конкретными и направленными
+
+Верни JSON массив из 3 объектов:
+[{{"id": "q{(round_number-1)*3+1}", "text": "текст вопроса"}}, ...]"""
+
+
+def generate_checklist_prompt(all_summaries: str, all_qa: str) -> str:
+    return f"""На основе всех раундов интервью, создай структурированный чеклист созвона.
+
+Резюме всех раундов:
+{all_summaries}
+
+Все вопросы и ответы:
+{all_qa}
+
+Создай чеклист с категориями. Для каждого пункта укажи статус:
+- "confirmed" — информация получена и подтверждена
+- "needs_clarification" — требует уточнения
+- "not_discussed" — не обсуждалось
+
+Верни JSON массив:
+[{{"category": "Информация о проекте", "item": "Название проекта: ...", "status": "confirmed", "notes": "..."}}]"""
+
+
+def generate_markdown_prompt(checklist_json: str, session_id: str) -> str:
+    return f"""Преобразуй чеклист в красивый Markdown документ.
+
+Чеклист (JSON):
+{checklist_json}
+
+ID сессии: {session_id}
+
+Формат:
+- Заголовок: "# Чеклист созвона с клиентом"
+- Дата (сегодня)
+- Группировка по категориям (## заголовки)
+- [x] для confirmed, [ ] для needs_clarification и not_discussed
+- Секция "Дополнительные заметки" с notes
+- Подпись: "*Сгенерировано автоматически с помощью AI Checklist Agent*"
+
+Верни только Markdown текст, без JSON обёртки."""