105 lines
7.6 KiB
Python
105 lines
7.6 KiB
Python
"""Prompts CRUD."""
|
||
from .connection import query, execute, execute_returning
|
||
|
||
|
||
def _serialize(row):
|
||
"""Convert datetime fields to strings for JSON serialization."""
|
||
if row and row.get("created_at"):
|
||
row["created_at"] = str(row["created_at"])
|
||
return row
|
||
|
||
|
||
def get_active(role):
|
||
rows = query(
|
||
"SELECT * FROM prompts WHERE role = %s AND is_active = true ORDER BY created_at DESC LIMIT 1",
|
||
(role,),
|
||
)
|
||
return _serialize(rows[0]) if rows else None
|
||
|
||
|
||
def get(prompt_id):
|
||
rows = query("SELECT * FROM prompts WHERE id = %s", (prompt_id,))
|
||
return rows[0] if rows else None
|
||
|
||
|
||
def seed_defaults():
|
||
"""Auto-seed default prompts if table is empty."""
|
||
count = query("SELECT COUNT(*) as cnt FROM prompts")
|
||
if count[0]["cnt"] > 0:
|
||
return
|
||
|
||
execute("""
|
||
INSERT INTO prompts (role, name, body, is_active, notes) VALUES (
|
||
'extract', 'default-v1',
|
||
'Ты — анализатор договоров облачного провайдера.\n\nНиже текст спецификации услуг из ПЕРВОГО документа (базовый договор).\nИзвлеки ВСЕ строки спецификации в JSON-массив.\n\nВерни СТРОГО JSON без пояснений. Не используй markdown-блоки.\n\nФОРМАТ:\n{\n "mode": "partial",\n "ops": [\n {"action": "ADD", "new_row": {"name": "полное наименование", "price": число, "qty": число, "sum": число, "date_start": "YYYY-MM-DD"}, "comment": ""}\n ]\n}\n\nПРАВИЛА:\n1. Извлеки КАЖДУЮ строку таблицы спецификации как отдельную ADD-операцию.\n2. Пропускай итоговые строки ("Итого...") и строки с подписями.\n3. Если ячейка пустая — ставь null (не пиши 0).\n4. price, qty, sum — ЧИСЛА, не строки.\n\nТЕКСТ ДОКУМЕНТА:\n---\n{doc_text}\n---',
|
||
true, 'Авто-создан из llm_prompt.py _build_initial'
|
||
)
|
||
""")
|
||
execute("""
|
||
INSERT INTO prompts (role, name, body, is_active, notes) VALUES (
|
||
'diff', 'default-v1',
|
||
'Ты — анализатор допсоглашений к договорам облачного провайдера.\n\nУ тебя есть текущая спецификация услуг и текст нового допсоглашения (ДС).\nТвоя задача — определить, какие изменения вносит ДС в текущую спецификацию.\n\nВерни СТРОГО JSON без пояснений. Не используй markdown-блоки.\n\nФОРМАТ:\n{\n "mode": "partial" | "full_replace",\n "ops": [\n {"action": "ADD", "new_row": {"name": "...", "price": число, "qty": число, "sum": число, "date_start": "YYYY-MM-DD"}, "comment": "..."},\n {"action": "UPDATE", "target_id": "rN", "new_values": {"price": число}, "comment": "..."},\n {"action": "DELETE", "target_id": "rN", "comment": "..."},\n {"action": "UNRESOLVED", "new_values": {"name": "...", "price": число, ...}, "reason": "почему не смог сопоставить"}\n ]\n}\n\nПРАВИЛА:\n1. mode = "full_replace" — если в тексте есть фразы: «в следующей редакции», «заменить приложение», «излагается в следующей редакции». При full_replace — опиши ВСЕ новые строки как ADD.\n2. mode = "partial" — если ДС меняет только отдельные строки.\n3. Для UPDATE/DELETE — укажи target_id (r1, r2...) из списка текущей спецификации. НЕ придумывай новые id.\n4. new_values в UPDATE — только ИЗМЕНЁННЫЕ поля (не все).\n5. Если не можешь однозначно сопоставить строку — action: UNRESOLVED с reason.\n6. Пропускай итоговые строки и подписи.\n7. price, qty, sum — ЧИСЛА (не строки).\n\nТЕКУЩАЯ СПЕЦИФИКАЦИЯ:\n{spec_current}\n\nТЕКСТ ДОПСОГЛАШЕНИЯ:\n---\n{doc_text}\n---',
|
||
true, 'Авто-создан из llm_prompt.py _build_diff'
|
||
)
|
||
""")
|
||
_ensure_classify_prompt()
|
||
|
||
|
||
def list_by_role(role):
|
||
"""List all versions for a role, newest first."""
|
||
rows = query(
|
||
"SELECT id, role, name, is_active, created_by, notes, created_at FROM prompts WHERE role=%s ORDER BY created_at DESC",
|
||
(role,),
|
||
)
|
||
for r in rows:
|
||
if r.get("created_at"):
|
||
r["created_at"] = str(r["created_at"])
|
||
return rows
|
||
|
||
|
||
def save_new_version(role, name, body, notes="", is_active=True):
|
||
"""Save new prompt version. Deactivates all others for this role, inserts new one."""
|
||
if is_active:
|
||
execute("UPDATE prompts SET is_active=false WHERE role=%s", (role,))
|
||
return execute_returning(
|
||
"""INSERT INTO prompts (role, name, body, is_active, notes)
|
||
VALUES (%s, %s, %s, %s, %s) RETURNING *""",
|
||
(role, name, body, is_active, notes),
|
||
)
|
||
|
||
|
||
def activate(prompt_id):
|
||
"""Activate a prompt version (deactivates others for same role)."""
|
||
row = query("SELECT role FROM prompts WHERE id=%s", (prompt_id,))
|
||
if not row:
|
||
return False
|
||
role = row[0]["role"]
|
||
execute("UPDATE prompts SET is_active=false WHERE role=%s", (role,))
|
||
execute("UPDATE prompts SET is_active=true WHERE id=%s", (prompt_id,))
|
||
return True
|
||
|
||
|
||
def delete_prompt(prompt_id):
|
||
"""Delete a prompt version (cannot delete active one)."""
|
||
row = query("SELECT is_active FROM prompts WHERE id=%s", (prompt_id,))
|
||
if not row:
|
||
return False
|
||
if row[0]["is_active"]:
|
||
return False # cannot delete active
|
||
execute("DELETE FROM prompts WHERE id=%s", (prompt_id,))
|
||
return True
|
||
|
||
|
||
def _ensure_classify_prompt():
|
||
"""Ensure classify prompt exists (idempotent)."""
|
||
existing = query("SELECT id FROM prompts WHERE role = 'classify' AND is_active = true LIMIT 1")
|
||
if existing:
|
||
return
|
||
execute("""
|
||
INSERT INTO prompts (role, name, body, is_active, notes) VALUES (
|
||
'classify', 'default-v1',
|
||
'Ты — система классификации договорных документов. Проанализируй текст и верни СТРОГО ВАЛИДНЫЙ JSON ОДНОЙ СТРОКОЙ (без переносов строк, без markdown, без лишних пробелов в начале/конце).\n\nПоля:\n- doc_type: "contract" (договор) / "supplement" (допсоглашение) / "specification" (спецификация/приложение) / "other"\n- own_number: номер ЭТОГО документа, строка без лишних пробелов (или null)\n- parent_number: номер родительского договора из фраз «к Договору №...» (или null)\n- doc_date: дата в YYYY-MM-DD. «27 февраля 2026» → 2026-02-27 (или null)\n- counterparty: полное название контрагента (Заказчик/Арендатор), без сокращений (или null)\n\nПример вывода:\n{"doc_type":"supplement","own_number":"1","parent_number":"01300_2","doc_date":"2026-02-27","counterparty":"АО XXX003"}\n\nДОКУМЕНТ:\n---\n{header_text}\n---',
|
||
true, 'Авто-создан: classify prompt v2 — LLM сам предложил'
|
||
)
|
||
""")
|