v1.0.176: prompts — seed per-role + _serialize non-mutating
This commit is contained in:
+55
-52
@@ -3,8 +3,9 @@ from .connection import query, execute, execute_returning
|
|||||||
|
|
||||||
|
|
||||||
def _serialize(row):
|
def _serialize(row):
|
||||||
"""Convert datetime fields to strings for JSON serialization."""
|
"""Convert datetime fields to strings for JSON serialization (non-mutating)."""
|
||||||
if row and row.get("created_at"):
|
if row and row.get("created_at"):
|
||||||
|
row = dict(row)
|
||||||
row["created_at"] = str(row["created_at"])
|
row["created_at"] = str(row["created_at"])
|
||||||
return row
|
return row
|
||||||
|
|
||||||
@@ -24,57 +25,59 @@ def get(prompt_id):
|
|||||||
|
|
||||||
def seed_defaults():
|
def seed_defaults():
|
||||||
"""Auto-seed default prompts if table is empty."""
|
"""Auto-seed default prompts if table is empty."""
|
||||||
count = query("SELECT COUNT(*) as cnt FROM prompts")
|
# Check each role separately — don't skip if one is missing
|
||||||
if count[0]["cnt"] > 0:
|
existing_roles = set(r["role"] for r in query("SELECT DISTINCT role FROM prompts"))
|
||||||
return
|
|
||||||
|
if "extract" not in existing_roles:
|
||||||
extract_body = (
|
extract_body = (
|
||||||
"Ты — анализатор договоров облачного провайдера.\n\n"
|
"Ты — анализатор договоров облачного провайдера.\n\n"
|
||||||
"Ниже текст спецификации услуг из ПЕРВОГО документа (базовый договор).\n"
|
"Ниже текст спецификации услуг из ПЕРВОГО документа (базовый договор).\n"
|
||||||
"Извлеки ВСЕ строки спецификации в JSON-массив.\n\n"
|
"Извлеки ВСЕ строки спецификации в JSON-массив.\n\n"
|
||||||
"Верни СТРОГО JSON без пояснений. Не используй markdown-блоки.\n\n"
|
"Верни СТРОГО JSON без пояснений. Не используй markdown-блоки.\n\n"
|
||||||
"ФОРМАТ:\n"
|
"ФОРМАТ:\n"
|
||||||
'{\n "mode": "partial",\n "ops": [\n'
|
'{\n "mode": "partial",\n "ops": [\n'
|
||||||
' {"action": "ADD", "new_row": {"name": "полное наименование", "price": число, "qty": число, "sum": число, "date_start": "YYYY-MM-DD"}, "comment": ""}\n'
|
' {"action": "ADD", "new_row": {"name": "полное наименование", "price": число, "qty": число, "sum": число, "date_start": "YYYY-MM-DD"}, "comment": ""}\n'
|
||||||
" ]\n}\n\n"
|
" ]\n}\n\n"
|
||||||
"ПРАВИЛА:\n"
|
"ПРАВИЛА:\n"
|
||||||
"1. Извлеки КАЖДУЮ строку таблицы спецификации как отдельную ADD-операцию.\n"
|
"1. Извлеки КАЖДУЮ строку таблицы спецификации как отдельную ADD-операцию.\n"
|
||||||
'2. Пропускай итоговые строки ("Итого...") и строки с подписями.\n'
|
'2. Пропускай итоговые строки ("Итого...") и строки с подписями.\n'
|
||||||
"3. Если ячейка пустая — ставь null (не пиши 0).\n"
|
"3. Если ячейка пустая — ставь null (не пиши 0).\n"
|
||||||
"4. price, qty, sum — ЧИСЛА, не строки.\n\n"
|
"4. price, qty, sum — ЧИСЛА, не строки.\n\n"
|
||||||
"ТЕКСТ ДОКУМЕНТА:\n---\n{doc_text}\n---"
|
"ТЕКСТ ДОКУМЕНТА:\n---\n{doc_text}\n---"
|
||||||
)
|
)
|
||||||
execute(
|
execute(
|
||||||
"INSERT INTO prompts (role, name, body, is_active, notes) VALUES (%s, %s, %s, %s, %s)",
|
"INSERT INTO prompts (role, name, body, is_active, notes) VALUES (%s, %s, %s, %s, %s)",
|
||||||
("extract", "default-v1", extract_body, True, "Авто-создан из llm_prompt.py _build_initial"),
|
("extract", "default-v1", extract_body, True, "Авто-создан из llm_prompt.py _build_initial"),
|
||||||
)
|
)
|
||||||
diff_body = (
|
|
||||||
"Ты — анализатор допсоглашений к договорам облачного провайдера.\n\n"
|
if "diff" not in existing_roles:
|
||||||
"У тебя есть текущая спецификация услуг и текст нового допсоглашения (ДС).\n"
|
diff_body = (
|
||||||
"Твоя задача — определить, какие изменения вносит ДС в текущую спецификацию.\n\n"
|
"Ты — анализатор допсоглашений к договорам облачного провайдера.\n\n"
|
||||||
"Верни СТРОГО JSON без пояснений. Не используй markdown-блоки.\n\n"
|
"У тебя есть текущая спецификация услуг и текст нового допсоглашения (ДС).\n"
|
||||||
"ФОРМАТ:\n"
|
"Твоя задача — определить, какие изменения вносит ДС в текущую спецификацию.\n\n"
|
||||||
'{\n "mode": "partial" | "full_replace",\n "ops": [\n'
|
"Верни СТРОГО JSON без пояснений. Не используй markdown-блоки.\n\n"
|
||||||
' {"action": "ADD", "new_row": {"name": "...", "price": число, "qty": число, "sum": число, "date_start": "YYYY-MM-DD"}, "comment": "..."},\n'
|
"ФОРМАТ:\n"
|
||||||
' {"action": "UPDATE", "target_id": "rN", "new_values": {"price": число}, "comment": "..."},\n'
|
'{\n "mode": "partial" | "full_replace",\n "ops": [\n'
|
||||||
' {"action": "DELETE", "target_id": "rN", "comment": "..."},\n'
|
' {"action": "ADD", "new_row": {"name": "...", "price": число, "qty": число, "sum": число, "date_start": "YYYY-MM-DD"}, "comment": "..."},\n'
|
||||||
' {"action": "UNRESOLVED", "new_values": {"name": "...", "price": число, ...}, "reason": "почему не смог сопоставить"}\n'
|
' {"action": "UPDATE", "target_id": "rN", "new_values": {"price": число}, "comment": "..."},\n'
|
||||||
" ]\n}\n\n"
|
' {"action": "DELETE", "target_id": "rN", "comment": "..."},\n'
|
||||||
"ПРАВИЛА:\n"
|
' {"action": "UNRESOLVED", "new_values": {"name": "...", "price": число, ...}, "reason": "почему не смог сопоставить"}\n'
|
||||||
'1. mode = "full_replace" — если в тексте есть фразы: «в следующей редакции», «заменить приложение», «излагается в следующей редакции». При full_replace — опиши ВСЕ новые строки как ADD.\n'
|
" ]\n}\n\n"
|
||||||
'2. mode = "partial" — если ДС меняет только отдельные строки.\n'
|
"ПРАВИЛА:\n"
|
||||||
"3. Для UPDATE/DELETE — укажи target_id (r1, r2...) из списка текущей спецификации. НЕ придумывай новые id.\n"
|
'1. mode = "full_replace" — если в тексте есть фразы: «в следующей редакции», «заменить приложение», «излагается в следующей редакции». При full_replace — опиши ВСЕ новые строки как ADD.\n'
|
||||||
"4. new_values в UPDATE — только ИЗМЕНЁННЫЕ поля (не все).\n"
|
'2. mode = "partial" — если ДС меняет только отдельные строки.\n'
|
||||||
"5. Если не можешь однозначно сопоставить строку — action: UNRESOLVED с reason.\n"
|
"3. Для UPDATE/DELETE — укажи target_id (r1, r2...) из списка текущей спецификации. НЕ придумывай новые id.\n"
|
||||||
"6. Пропускай итоговые строки и подписи.\n"
|
"4. new_values в UPDATE — только ИЗМЕНЁННЫЕ поля (не все).\n"
|
||||||
"7. price, qty, sum — ЧИСЛА (не строки).\n\n"
|
"5. Если не можешь однозначно сопоставить строку — action: UNRESOLVED с reason.\n"
|
||||||
"ТЕКУЩАЯ СПЕЦИФИКАЦИЯ:\n{spec_current}\n\n"
|
"6. Пропускай итоговые строки и подписи.\n"
|
||||||
"ТЕКСТ ДОПСОГЛАШЕНИЯ:\n---\n{doc_text}\n---"
|
"7. price, qty, sum — ЧИСЛА (не строки).\n\n"
|
||||||
)
|
"ТЕКУЩАЯ СПЕЦИФИКАЦИЯ:\n{spec_current}\n\n"
|
||||||
execute(
|
"ТЕКСТ ДОПСОГЛАШЕНИЯ:\n---\n{doc_text}\n---"
|
||||||
"INSERT INTO prompts (role, name, body, is_active, notes) VALUES (%s, %s, %s, %s, %s)",
|
)
|
||||||
("diff", "default-v1", diff_body, True, "Авто-создан из llm_prompt.py _build_diff"),
|
execute(
|
||||||
)
|
"INSERT INTO prompts (role, name, body, is_active, notes) VALUES (%s, %s, %s, %s, %s)",
|
||||||
|
("diff", "default-v1", diff_body, True, "Авто-создан из llm_prompt.py _build_diff"),
|
||||||
|
)
|
||||||
_ensure_classify_prompt()
|
_ensure_classify_prompt()
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user