Compare commits
2
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c0a5145604 | ||
|
|
7ee7520c98 |
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -20,7 +20,8 @@ import logging
|
||||
import re
|
||||
from typing import List, Dict, Any, Tuple, Optional
|
||||
|
||||
# Импорт фасада базы данных
|
||||
# Импорт фасада базы данных и прямого подключения
|
||||
from .db.connection import get_db_connection
|
||||
from .db_tools import (
|
||||
db_get_active_system_prompt,
|
||||
db_add_system_prompt,
|
||||
@@ -81,7 +82,7 @@ def process_chat_message(
|
||||
2. Перехват завершения работы ('нет, закончить настройку') с автоочисткой эфемерных сообщений.
|
||||
3. Формирование системного контекста с учетом активного действия (черновика).
|
||||
4. Выполнение вызванного Tool и опрос Data-Driven реестра действий.
|
||||
5. Проверка счетчика отвлечений (idle_turns) и Context Guard на 3-м шаге.
|
||||
5. Проверка счетчика отвлечений (idle_turns), Context Guard на 3-м шаге и автоочистка при N > 3.
|
||||
6. Возврат кортежа: (reply_text, chat_history, action_metadata).
|
||||
"""
|
||||
logger.info(f"Получено сообщение от user_id={user_id}, session_id={session_id}: {user_message}")
|
||||
@@ -114,7 +115,9 @@ def process_chat_message(
|
||||
|
||||
if user_msg_clean in ["подтверждаю", "подтвердить", "да", "сохранить", "применить", "ок", "хорошо"]:
|
||||
db_add_system_prompt("main_agent", draft_text)
|
||||
db_clear_session_state(session_id)
|
||||
|
||||
# Переводим сессию в режим follow-up диалога
|
||||
db_set_session_state(session_id, "PROMPT_FOLLOWUP", {"idle_turns": 0})
|
||||
|
||||
action_cfg = db_get_tool_action("db_confirm_prompt_preview")
|
||||
reply_text = action_cfg["success_template"] if action_cfg else "Системный промпт успешно сохранен и применен в базе данных."
|
||||
@@ -123,15 +126,16 @@ def process_chat_message(
|
||||
|
||||
reply_text = reply_text.replace("✅", "").replace("❌", "").replace("**", "").replace("*", "").strip()
|
||||
|
||||
is_eph = action_cfg.get("is_ephemeral", 1) if action_cfg else 1
|
||||
db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=1)
|
||||
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=is_eph)
|
||||
return reply_text, db_get_chat_history(session_id), {
|
||||
"type": action_cfg.get("action_type") if action_cfg else "FOLLOW_UP_ACTION",
|
||||
"buttons": action_cfg.get("buttons", []) if action_cfg else []
|
||||
}
|
||||
|
||||
elif user_msg_clean in ["отмена", "отменить", "отклонить", "назад", "стоп"]:
|
||||
db_clear_session_state(session_id)
|
||||
db_set_session_state(session_id, "PROMPT_FOLLOWUP", {"idle_turns": 0})
|
||||
|
||||
action_cfg = db_get_tool_action("db_cancel_prompt_preview")
|
||||
reply_text = action_cfg["success_template"] if action_cfg else "Изменения системного промпта отменены."
|
||||
@@ -140,8 +144,9 @@ def process_chat_message(
|
||||
|
||||
reply_text = reply_text.replace("✅", "").replace("❌", "").replace("**", "").replace("*", "").strip()
|
||||
|
||||
is_eph = action_cfg.get("is_ephemeral", 1) if action_cfg else 1
|
||||
db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=1)
|
||||
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=is_eph)
|
||||
return reply_text, db_get_chat_history(session_id), {
|
||||
"type": action_cfg.get("action_type") if action_cfg else "FOLLOW_UP_ACTION",
|
||||
"buttons": action_cfg.get("buttons", []) if action_cfg else []
|
||||
@@ -160,11 +165,11 @@ def process_chat_message(
|
||||
|
||||
# Информируем модель о наличии незавершенного действия в сессии
|
||||
active_state_context = ""
|
||||
if session_state and session_state.get("state_type") == "PROMPT_PREVIEW":
|
||||
if session_state and session_state.get("state_type") in ["PROMPT_PREVIEW", "PROMPT_FOLLOWUP"]:
|
||||
active_state_context = (
|
||||
"\n[АКТИВНОЕ ДЕЙСТВИЕ В СЕССИИ]\n"
|
||||
"В данный момент оператор рассматривает подготовленный черновик системного промпта.\n"
|
||||
"- Если оператор просит продолжить правки или уточняет детали по черновику — продолжай работу с ним.\n"
|
||||
"В данный момент оператор находится в процессе настройки системного промпта.\n"
|
||||
"- Если оператор просит продолжить правки или уточняет детали — продолжай работу с ним.\n"
|
||||
"- Если оператор переключился на другую тему или вызвал другой инструмент — выполни его команду штатно.\n"
|
||||
)
|
||||
|
||||
@@ -250,7 +255,7 @@ def process_chat_message(
|
||||
draft_text = session_state.get("pending_data", "")
|
||||
|
||||
db_add_system_prompt("main_agent", draft_text)
|
||||
db_clear_session_state(session_id)
|
||||
db_set_session_state(session_id, "PROMPT_FOLLOWUP", {"idle_turns": 0})
|
||||
|
||||
if action_cfg and action_cfg.get("bypass_llm"):
|
||||
reply_text = action_cfg["success_template"]
|
||||
@@ -271,7 +276,7 @@ def process_chat_message(
|
||||
return err_reply, db_get_chat_history(session_id), None
|
||||
|
||||
elif fn_name == "db_cancel_prompt_preview":
|
||||
db_clear_session_state(session_id)
|
||||
db_set_session_state(session_id, "PROMPT_FOLLOWUP", {"idle_turns": 0})
|
||||
if action_cfg and action_cfg.get("bypass_llm"):
|
||||
reply_text = action_cfg["success_template"]
|
||||
if action_cfg.get("follow_up_question"):
|
||||
@@ -374,6 +379,16 @@ def process_chat_message(
|
||||
"idle_turns": 0
|
||||
}
|
||||
db_set_session_state(session_id, "PROMPT_PREVIEW", state_payload)
|
||||
|
||||
# Помечаем последнее сообщение пользователя как эфемерное
|
||||
with get_db_connection() as conn_fix:
|
||||
cursor_fix = conn_fix.cursor()
|
||||
cursor_fix.execute("""
|
||||
UPDATE chat_messages
|
||||
SET is_ephemeral = 1
|
||||
WHERE id = (SELECT MAX(id) FROM chat_messages WHERE session_id = ? AND role = 'user')
|
||||
""", (session_id,))
|
||||
conn_fix.commit()
|
||||
|
||||
preview_reply = (
|
||||
f"Ваше изменение успешно предпросмотрено. Полный обновленный системный промпт теперь выглядит так:\n\n"
|
||||
@@ -425,24 +440,32 @@ def process_chat_message(
|
||||
raw_content = sec_res_data.get("message", {}).get("content", "").strip().replace("**", "").replace("*", "")
|
||||
final_content = clean_raw_tool_tags(clean_output(raw_content))
|
||||
|
||||
# Если была активна фоновая транзакция, инкрементируем счетчик Topic Shift
|
||||
if session_state and session_state.get("state_type") == "PROMPT_PREVIEW":
|
||||
# Определяем, был ли вызов служебного инструмента просмотра промпта
|
||||
tool_names_called = [t["function"]["name"] for t in tool_calls]
|
||||
is_output_ephemeral = 1 if any(name in ["db_get_system_prompt", "db_get_system_prompts", "db_preview_prompt_merge"] for name in tool_names_called) else 0
|
||||
|
||||
# Отслеживание отвлечений при активном процессе настройки
|
||||
if session_state and session_state.get("state_type") in ["PROMPT_PREVIEW", "PROMPT_FOLLOWUP"]:
|
||||
idle_count = db_increment_session_idle(session_id)
|
||||
logger.info(f"Выполнен инструмент вне черновика. Текущий idle_turns: {idle_count}")
|
||||
logger.info(f"Смена темы ({session_state.get('state_type')}). Текущий idle_turns: {idle_count}")
|
||||
|
||||
if idle_count >= 3:
|
||||
guard_note = "\n\nНапоминание: У вас остался непримененный черновик системного промпта. Сохранить его или сбросить?"
|
||||
if idle_count == 3:
|
||||
guard_note = "\n\nНапоминание: Завершить настройку системного промпта и очистить сессию?"
|
||||
final_content += guard_note
|
||||
db_save_chat_message(session_id, "assistant", final_content, is_ephemeral=1)
|
||||
db_save_chat_message(session_id, "assistant", final_content, is_ephemeral=is_output_ephemeral)
|
||||
return final_content, db_get_chat_history(session_id), {
|
||||
"type": "PROMPT_PREVIEW_GUARD",
|
||||
"type": "FOLLOW_UP_ACTION",
|
||||
"buttons": [
|
||||
{"label": "Применить черновик", "value": "подтверждаю", "style": "primary"},
|
||||
{"label": "Сбросить черновик", "value": "отмена", "style": "danger"}
|
||||
{"label": "Да, закончить", "value": "нет, закончить настройку", "style": "primary"},
|
||||
{"label": "Показать промпт", "value": "покажи системный промпт", "style": "secondary"}
|
||||
]
|
||||
}
|
||||
elif idle_count > 3:
|
||||
logger.info(f"Автоочистка сессии: оператор переключился на другую тему (idle_turns={idle_count}).")
|
||||
db_clear_session_state(session_id)
|
||||
db_purge_ephemeral_messages(session_id)
|
||||
|
||||
db_save_chat_message(session_id, "assistant", final_content, is_ephemeral=0)
|
||||
db_save_chat_message(session_id, "assistant", final_content, is_ephemeral=is_output_ephemeral)
|
||||
return final_content, db_get_chat_history(session_id), None
|
||||
|
||||
# Если вызовов инструментов не было (обычный текстовый диалог)
|
||||
@@ -450,20 +473,25 @@ def process_chat_message(
|
||||
content_str = clean_raw_tool_tags(clean_output(raw_str))
|
||||
final_reply = content_str or "Запрос обработан."
|
||||
|
||||
if session_state and session_state.get("state_type") == "PROMPT_PREVIEW":
|
||||
if session_state and session_state.get("state_type") in ["PROMPT_PREVIEW", "PROMPT_FOLLOWUP"]:
|
||||
idle_count = db_increment_session_idle(session_id)
|
||||
logger.info(f"Текстовый диалог вне черновика. Текущий idle_turns: {idle_count}")
|
||||
if idle_count >= 3:
|
||||
guard_note = "\n\nНапоминание: У вас остался непримененный черновик системного промпта. Сохранить его или сбросить?"
|
||||
logger.info(f"Текстовый диалог вне настройки ({session_state.get('state_type')}). Текущий idle_turns: {idle_count}")
|
||||
|
||||
if idle_count == 3:
|
||||
guard_note = "\n\nНапоминание: Завершить настройку системного промпта и очистить сессию?"
|
||||
final_reply += guard_note
|
||||
db_save_chat_message(session_id, "assistant", final_reply, is_ephemeral=1)
|
||||
db_save_chat_message(session_id, "assistant", final_reply, is_ephemeral=0)
|
||||
return final_reply, db_get_chat_history(session_id), {
|
||||
"type": "PROMPT_PREVIEW_GUARD",
|
||||
"type": "FOLLOW_UP_ACTION",
|
||||
"buttons": [
|
||||
{"label": "Применить черновик", "value": "подтверждаю", "style": "primary"},
|
||||
{"label": "Сбросить черновик", "value": "отмена", "style": "danger"}
|
||||
{"label": "Да, закончить", "value": "нет, закончить настройку", "style": "primary"},
|
||||
{"label": "Показать промпт", "value": "покажи системный промпт", "style": "secondary"}
|
||||
]
|
||||
}
|
||||
elif idle_count > 3:
|
||||
logger.info(f"Автоочистка сессии: оператор переключился на другую тему (idle_turns={idle_count}).")
|
||||
db_clear_session_state(session_id)
|
||||
db_purge_ephemeral_messages(session_id)
|
||||
|
||||
db_save_chat_message(session_id, "assistant", final_reply, is_ephemeral=0)
|
||||
return final_reply, db_get_chat_history(session_id), None
|
||||
|
||||
@@ -0,0 +1,58 @@
|
||||
import os
|
||||
import sqlite3
|
||||
|
||||
DB_PATH = os.path.join(os.path.dirname(__file__), "../../data/scud_orion_ai.db")
|
||||
if not os.path.exists(DB_PATH):
|
||||
DB_PATH = "data/scud_orion_ai.db"
|
||||
|
||||
conn = sqlite3.connect(DB_PATH)
|
||||
cursor = conn.cursor()
|
||||
|
||||
print("=" * 90)
|
||||
print(f"🔍 ИНСПЕКЦИЯ СООБЩЕНИЙ В БАЗЕ: {DB_PATH}")
|
||||
print("=" * 90)
|
||||
|
||||
# 1. Проверяем наличие колонки is_ephemeral
|
||||
cursor.execute("PRAGMA table_info(chat_messages);")
|
||||
cols = [c[1] for c in cursor.fetchall()]
|
||||
|
||||
if "is_ephemeral" not in cols:
|
||||
print("❌ Колонка is_ephemeral отсутствует в таблице chat_messages!")
|
||||
conn.close()
|
||||
exit(1)
|
||||
|
||||
# 2. Считаем общее количество и распределение
|
||||
cursor.execute("SELECT COUNT(*) FROM chat_messages")
|
||||
total_count = cursor.fetchone()[0]
|
||||
|
||||
cursor.execute("SELECT COUNT(*) FROM chat_messages WHERE is_ephemeral = 1")
|
||||
ephemeral_count = cursor.fetchone()[0]
|
||||
|
||||
cursor.execute("SELECT COUNT(*) FROM chat_messages WHERE is_ephemeral = 0")
|
||||
permanent_count = cursor.fetchone()[0]
|
||||
|
||||
print(f"Всего записей в истории: {total_count}")
|
||||
print(f" • Постоянные (is_ephemeral = 0): {permanent_count}")
|
||||
print(f" • Фантомные / Эфемерные (is_ephemeral = 1): {ephemeral_count}")
|
||||
print("-" * 90)
|
||||
|
||||
# 3. Выводим все фантомные записи
|
||||
if ephemeral_count == 0:
|
||||
print("В базе нет фантомных записей (is_ephemeral = 1 пуст).")
|
||||
else:
|
||||
print(f"📋 СПИСОК ВСЕХ ФАНТОМНЫХ СООБЩЕНИЙ ({ephemeral_count} шт.):")
|
||||
cursor.execute("""
|
||||
SELECT id, session_id, role, created_at, content
|
||||
FROM chat_messages
|
||||
WHERE is_ephemeral = 1
|
||||
ORDER BY id ASC
|
||||
""")
|
||||
for row in cursor.fetchall():
|
||||
msg_id, session_id, role, created_at, content = row
|
||||
print(f"\n[ID: {msg_id}] | Сессия: {session_id} | Роль: {role} | Время: {created_at}")
|
||||
print("Содержимое:")
|
||||
print(content.strip())
|
||||
print("-" * 70)
|
||||
|
||||
conn.close()
|
||||
print("=" * 90)
|
||||
Reference in New Issue
Block a user