qwen new promt 1

app.py

@@ -18,40 +18,43 @@ async def health_check():
         "model_loaded": llm is not None
     }
 
+# ... (остальное без изменений)
+
 @app.post("/api/v1/generate_plantuml", response_model=DiagramResponse)
 async def generate_plantuml(req: DiagramRequest):
     global llm
     if llm is None:
-        raise HTTPException(503, "Модель ещё не загружена или ошибка при загрузке")
+        raise HTTPException(503, "Модель ещё не загружена")
 
     desc = preprocess_description(req.description)
     if not desc:
         raise HTTPException(400, "Description cannot be empty")
 
-    prompt = get_prompt(desc)  # ← твой промпт с примерами
-
-    output = llm(
-        prompt=prompt,
-        max_tokens=1200,
-        temperature=0.65,
-        top_p=0.92,
-        top_k=40,
-        repeat_penalty=1.1,
-        stop=["</s>", "@enduml", "\n\n"],  # чтобы не генерил лишнего
-        echo=False
+    messages = get_prompt(desc)  # теперь list[dict]
+
+    output = llm.create_chat_completion(
+        messages,
+        max_tokens=1500,           # чуть больше — примеры длинные
+        temperature=0.6,           # ниже — меньше креатива, строже синтаксис
+        top_p=0.9,
+        top_k=35,
+        repeat_penalty=1.15,       # меньше повторений
+        stop=["</s>", "<|im_end|>", "@enduml\n\n", "```"],  # стоп на конце
     )
 
-    generated_text = output["choices"][0]["text"].strip()
+    generated_text = output["choices"][0]["message"]["content"].strip()
 
-    # Пытаемся аккуратно извлечь PlantUML
+    # Извлечение только кода (на всякий случай)
     start_marker = "@startuml"
     end_marker = "@enduml"
     if start_marker in generated_text and end_marker in generated_text:
         start_idx = generated_text.find(start_marker)
-        end_idx = generated_text.find(end_marker, start_idx) + len(end_marker)
+        end_idx = generated_text.rfind(end_marker) + len(end_marker)
         plantuml_code = generated_text[start_idx:end_idx]
     else:
-        # fallback — весь текст, если маркеры потерялись
-        plantuml_code = generated_text
+        plantuml_code = generated_text  # fallback
+
+    # Финальная чистка: убираем лишние пустые строки внутри
+    plantuml_code = re.sub(r'\n\s*\n', '\n', plantuml_code).strip()
 
-    return {"plantuml_code": plantuml_code.strip()}
+    return {"plantuml_code": plantuml_code}

utils.py

@@ -1,308 +1,250 @@
 from pydantic import BaseModel
+import re
+
 
 class DiagramRequest(BaseModel):
     description: str
 
+
 class DiagramResponse(BaseModel):
     plantuml_code: str
 
-# Простой препроцесс: чистим текст, добавляем маркеры если нужно
-def preprocess_description(desc: str) -> str:
-    return desc.strip().replace("\n", " ").replace("\r", "")
-
-# Шаблон промпта (few-shot с твоими примерами + инструкция)
-def get_prompt(description: str) -> str:
-    return f"""You are an expert in generating PlantUML code from diagram descriptions. Detect the diagram type (BPMN, UML, C4, flowchart) based on the text.
-
-- For BPMN-like processes with actors/pools and branches: Use @startuml with activity diagram, partitions for pools (e.g., partition Client { ... }), start, end, if/else for gateways.
-- For UML: Use class/object/sequence diagrams as appropriate.
-- For C4 models (systems/containers/relations): Include !include <C4/C4_Container> and use macros like System, Container, Rel.
-- For flowcharts/tables of steps: Use activity diagram with sequential actions.
-
-Output ONLY the PlantUML code inside @startuml ... @enduml. No explanations.
-
-Examples:
-
-Description: Клиент: (start Потребность оформить заказ)
-
-Клиент: Адрес и способ доставки
-
-Система Додо: Проверка адреса и доступности доставки
-
-Клиент: Выбор блюд и акций
-
-Система Додо: Расчёт стоимости и ETA
-
-Клиент: Выбор способа оплаты (онлайн)
-
-Система Додо: Инициация онлайн-оплаты
-
-Платёжный провайдер: Авторизация платежа
-
-Система Додо: Обработка результата авторизации
-
-Система Додо: Оплата успешна
-
-Система Додо: Создать заказ и присвоить ETA
-
-Система Додо: Уведомление клиенту: заказ принят
-
-Ресторан: Печать чек-листа / стикера
-
-Ресторан: Приготовление заказа
-
-Ресторан: Фото-верификация и упаковка
-
-Ресторан: Готовность заказа
-
-Система Додо: Автоназначение курьера
-
-Система Додо: Построение маршрута
-
-Ресторан: Передача заказа курьеру
-
-Курьер: Доставка к адресу клиента
-
-Курьер: Клиент доступен
-
-Курьер: Вручение заказа
-
-Система Додо: Закрыть заказ (доставлен)
-
-(finish Заказ получен)
-
-Клиент: (start Потребность оформить заказ)
-
-Клиент: Адрес и способ доставки
-
-Система Додо: Проверка адреса и доступности доставки
-
-Клиент: Выбор блюд и акций
-
-Система Додо: Расчёт стоимости и ETA
-
-Клиент: Выбор способа оплаты (онлайн)
-
-Система Додо: Инициация онлайн-оплаты
-
-Платёжный провайдер: Авторизация платежа
-
-Система Додо: Обработка результата авторизации
-
-Система Додо: Оплата успешна
-
-Система Додо: Создать заказ и присвоить ETA
-
-Система Додо: Уведомление клиенту: заказ принят
-
-Ресторан: Печать чек-листа / стикера
-
-Ресторан: Приготовление заказа
-
-Ресторан: Фото-верификация и упаковка
-
-Ресторан: Готовность заказа
-
-Система Додо: Автоназначение курьера
-
-Система Додо: Построение маршрута
-
-Ресторан: Передача заказа курьеру
-
-Курьер: Доставка к адресу клиента
-
-Курьер: Клиент недоступен
-
-Курьер: Ожидание по политике (15 минут)
-
-Курьер: Повторная попытка / процедура по политике
-
-Система Додо: Закрыть заказ (клиент недоступен, заказ отменён)
-
-(finish Клиент недоступен, заказ отменён)
-
-Клиент: (start Потребность оформить заказ)
-
-Клиент: Адрес и способ доставки
-
-Система Додо: Проверка адреса и доступности доставки
-
-Клиент: Выбор блюд и акций
-
-Система Додо: Расчёт стоимости и ETA
-
-Клиент: Выбор способа оплаты (онлайн)
-
-Система Додо: Инициация онлайн-оплаты
-
-Платёжный провайдер: Авторизация платежа
-
-Система Додо: Обработка результата авторизации
-
-Система Додо: Оплата неуспешна
-
-Система Додо: Ретраи / каскад по платежу
 
-Система Додо: Предложить альтернативный способ оплаты
-
-Клиент: Выбор альтернативного способа оплаты
-
-Система Додо: Инициация онлайн-оплаты
-
-Платёжный провайдер: Авторизация платежа
-
-Система Додо: Обработка результата авторизации
-
-Система Додо: Оплата успешна
-
-Система Додо: Создать заказ и присвоить ETA
-
-Система Додо: Уведомление клиенту: заказ принят
-
-Ресторан: Печать чек-листа / стикера
-
-Ресторан: Приготовление заказа
-
-Ресторан: Фото-верификация и упаковка
-
-Ресторан: Готовность заказа
-
-Система Додо: Автоназначение курьера
-
-Система Додо: Построение маршрута
-
-Ресторан: Передача заказа курьеру
-
-Курьер: Доставка к адресу клиента
-
-Курьер: Клиент доступен
-
-Курьер: Вручение заказа
-
-Система Додо: Закрыть заказ (доставлен)
-
-(finish Заказ получен)
-
-Клиент: (start Потребность оформить заказ)
-
-Клиент: Адрес и способ доставки
-
-Система Додо: Проверка адреса и доступности доставки
-
-Клиент: Выбор блюд и акций
-
-Система Додо: Расчёт стоимости и ETA
-
-Клиент: Выбор способа оплаты (онлайн)
-
-Система Додо: Инициация онлайн-оплаты
-
-Платёжный провайдер: Авторизация платежа
-
-Система Додо: Обработка результата авторизации
-
-Система Додо: Оплата неуспешна
-
-Система Додо: Ретраи / каскад по платежу
-
-Система Додо: Предложить альтернативный способ оплаты
-
-Клиент: Отказаться от альтернативного спосо
-
-ба оплаты
-
-Система Додо: Отменить заказ (не оплачен)
-
-(finish Заказ отменён, не оплачен)  
-PlantUML:
-@startuml
-partition Client {{
-start
-:Потребность оформить заказ;
-:Адрес и способ доставки;
-:Выбор блюд и акций;
-:Выбор способа оплаты (онлайн);
-}}
-partition "Система Додо" {{
-:Проверка адреса и доступности доставки;
-:Расчёт стоимости и ETA;
-:Инициация онлайн-оплаты;
-if (Оплата успешна?) then (yes)
-:Создать заказ и присвоить ETA;
-:Уведомление клиенту: заказ принят;
-else (no)
-:Ретраи / каскад по платежу;
-:Предложить альтернативный способ оплаты;
-endif
-}}
-partition "Платёжный провайдер" {{
-:Авторизация платежа;
-}}
-' ... continue with branches and ends
-stop
-@enduml
-
-Description: Шаг | Роль
+def preprocess_description(desc: str) -> str:
+    """
+    Улучшенный препроцессинг:
+    - Убираем лишние пустые строки в начале/конце
+    - Нормализуем множественные пробелы и табы
+    - Сохраняем \n внутри — это важно для таблиц "Шаг | Роль" и списков
+    - Убираем только trailing/leading whitespace
+    """
+    # Удаляем BOM, если есть
+    desc = desc.strip().lstrip('\ufeff')
+
+    # Заменяем множественные пробелы/табы на один, но сохраняем строки
+    desc = re.sub(r'[ \t]+', ' ', desc)
+
+    # Убираем пустые строки в начале и конце, но оставляем внутри
+    lines = [line.rstrip() for line in desc.splitlines() if line.strip()]
+    desc = '\n'.join(lines).strip()
+
+    return desc
+
+
+def get_prompt(description: str) -> list[dict]:
+    """
+    Возвращает список сообщений в формате ChatML для Qwen2.5-Instruct.
+    Это сильно улучшает качество по сравнению с plain text.
+    """
+    system_prompt = """You are an expert PlantUML diagram generator. Your ONLY task is to convert the given process description into a valid, complete PlantUML activity diagram.
+
+Rules you MUST follow strictly:
+- Always output ONLY the PlantUML code: from @startuml to @enduml inclusive.
+- NEVER add explanations, comments, markdown, extra text, apologies or anything else.
+- Use the new activity diagram syntax (beta).
+- Use swimlanes/partitions (|Actor|) for different roles or actors when mentioned (Шаг | Роль, Actor: action, etc.).
+- Use start / stop / (*) / (*) for begin/end.
+- Use :action; for steps.
+- Use if (Condition?) then (yes) ... else (no) ... endif for decisions.
+- Use -> [label]; for arrows with conditions if needed.
+- Make diagrams clean, logical and complete — close all branches, ifs, etc.
+- Detect type: simple sequence → linear; table Шаг/Роль → swimlanes; branches (Найдена?, Автоматически?) → if/else.
+
+Examples follow. Follow their style and completeness exactly."""
+
+    messages = [
+        {"role": "system", "content": system_prompt},
+    ]
+
+    # Few-shot примеры (полные, без '...', валидные)
+    examples = [
+        # Пример 1: Таблица с ролями (тех. стек)
+        {
+            "user": """Шаг | Роль
 
 1. Создание запроса | Инициатор
-
 2. Внесение технологии в тех. стек в статусе "Consideration" | Координатор
-
 3. Оценка по техническим критериям | Инициатор
-
 4. Уточнение требований к технологии | Совет по технологиям, Координатор, Владелец продукта
-
 5. Оценка ограничений и рисков принятия новой мажорной версии | Координатор
-
 6. Презентация новой мажорной версии | Инициатор
-
 7. Принятие решения | Главный корпоративный архитектор, Совет по технологиям
-
 8. Внесение записи в тех. стек, оповещение | Координатор
-
 9. Перевод прошлой мажорной версии в статус Deprecated | Координатор
+10. Оповещение | Координатор
+11. Планирование обновления продукта | Координатор
+12. Планирование автоматизации ресурсов | Координатор
+13. Изменение атрибутов технологии в стеке | Координатор""",
+            "assistant": """@startuml
+skinparam shadowing false
+skinparam monochrome true
+
+|Инициатор|
+start
+:Создание запроса;
 
-10.Оповещение | Координатор
+|Координатор|
+:Внесение технологии в тех. стек (Consideration);
 
-11.Планирование обновления продукта, в рамках которого поставляется технология | Координатор
+|Инициатор|
+:Оценка по техническим критериям;
 
-12.Планирование автоматизации ресурсов на стороне ДИТ | Координатор
+|Совет по технологиям, Координатор, Владелец продукта|
+:Уточнение требований к технологии;
+
+|Координатор|
+:Оценка ограничений и рисков;
+
+|Инициатор|
+:Презентация новой мажорной версии;
+
+|Главный корпоративный архитектор, Совет по технологиям|
+:Принятие решения;
+
+|Координатор|
+:Внесение записи в тех. стек;
+:Оповещение;
+:Перевод прошлой версии в Deprecated;
+:Оповещение;
+:Планирование обновления продукта;
+:Планирование автоматизации ресурсов;
+
+:Изменение атрибутов технологии;
 
-13.Изменение атрибутов технологии в технологическом стеке | Координатор
-PlantUML:
-@startuml
-start
-:1. Создание запроса;  <<Инициатор>>
-:2. Внесение технологии в тех. стек в статусе "Consideration";  <<Координатор>>
-' ... sequential
 stop
-@enduml
+@enduml"""
+        },
 
-Description: Шаг
+        # Пример 2: Ещё одна таблица с ролями
+        {
+            "user": """Шаг | Роль
 
-1. Выполнить сценарий выбора СФЛ при назначени роли
+1. Создание запроса | Инициатор
+2. Подтверждение потребности в изменении условий использования | Координатор
+3. Определение пользователей технологии | Совет по технологиям
+4. Оценка трудозатрат | Совет по технологиям
+5. Принятие решения | Главный корпоративный архитектор, Совет по технологиям
+6. Изменение статуса технологии | Координатор
+7. Оповещение клиентов | Команда разработки
+8. Выполнение работ | ДЭиС""",
+            "assistant": """@startuml
+skinparam shadowing false
+
+|Инициатор|
+start
+:Создание запроса;
 
-2. Выдать сообщение о невозможности совмещения ролей
+|Координатор|
+:Подтверждение потребности в изменении;
 
-3. Сохранить скан документа,удостоверяющего полномочия
+|Совет по технологиям|
+:Определение пользователей технологии;
+:Оценка трудозатрат;
 
-4. Выбрать все нужные полномочия из списка
+|Главный корпоративный архитектор, Совет по технологиям|
+:Принятие решения;
 
-5. Запросить информацию о праве передоверие
+|Координатор|
+:Изменение статуса технологии;
 
-6. Запросить срок действия документа, удостоверяющего полномочия
+|Команда разработки|
+:Оповещение клиентов;
 
-7. Запросить прочие атрибуты роли "Доверенное лицо"
+|ДЭиС|
+:Выполнение работ;
 
-8. Запросить все обязательные атрибуты роли  # Третий
-PlantUML:
-@startuml
+stop
+@enduml"""
+        },
+
+        # Пример 3: Простая последовательность без ролей
+        {
+            "user": """Шаг
+
+1. Создание запроса
+2. Рассмотрение запроса
+3. Анализ запроса
+4. Принятие решения о реализации изменения
+5. Реализация улучшений и исправление несоответствий
+6. Оценка результатов""",
+            "assistant": """@startuml
 start
-:1. Выполнить сценарий выбора СФЛ при назначени роли;
-:2. Выдать сообщение о невозможности совмещения ролей;
-' ...
+:Создание запроса;
+:Рассмотрение запроса;
+:Анализ запроса;
+:Принятие решения о реализации;
+:Реализация улучшений и исправлений;
+:Оценка результатов;
 stop
-@enduml
+@enduml"""
+        },
+
+        # Пример 4: Чайник с событием и ветвлением
+        {
+            "user": """Шаг
+
+1. Начало
+2. Наполнить чайник
+3. Включить чайник
+4. Ждать закипания
+5. Событие: Чайник закипел
+6. Таймер: 30 секунд
+7. Автовыключение
+8. Ручное выключение
+9. Чай готов""",
+            "assistant": """@startuml
+start
+:Наполнить чайник;
+:Включить чайник;
+:Ждать закипания;
+note right: Событие: Чайник закипел
+:Таймер: 30 секунд;
+if (Выключен автоматически?) then (да)
+  :Автовыключение;
+else (нет)
+  :Ручное выключение;
+endif
+:Чай готов;
+stop
+@enduml"""
+        },
+
+        # Пример 5: Решение с ветвлением (работа/учёба)
+        {
+            "user": """Шаг
+
+1. Зачисление
+2. Обновить резюме
+3. Искать гибкую работу
+4. Решение: Найдена?
+5. Договориться о графике
+6. Совмещение работы и учебы
+7. Фокус на учебе
+8. Только учеба""",
+            "assistant": """@startuml
+start
+:Зачисление;
+:Обновить резюме;
+:Искать гибкую работу;
+if (Работа найдена?) then (да)
+  :Договориться о графике;
+  :Совмещение работы и учебы;
+else (нет)
+  :Фокус на учебе;
+  :Только учеба;
+endif
+stop
+@enduml"""
+        },
+    ]
+
+    # Добавляем все few-shot
+    for ex in examples:
+        messages.append({"role": "user", "content": ex["user"]})
+        messages.append({"role": "assistant", "content": ex["assistant"]})
+
+    # Финальный запрос
+    messages.append(
+        {"role": "user", "content": f"Generate PlantUML activity diagram for this description:\n\n{description}"})
 
-Now, generate for this description: {description}
-"""
+    return messages