diff --git a/data/scud_orion_ai.db b/data/scud_orion_ai.db index 0867d0b..b24e2af 100644 Binary files a/data/scud_orion_ai.db and b/data/scud_orion_ai.db differ diff --git a/data/scud_orion_ai.db-shm b/data/scud_orion_ai.db-shm index 84061a4..5ca3d3a 100644 Binary files a/data/scud_orion_ai.db-shm and b/data/scud_orion_ai.db-shm differ diff --git a/data/scud_orion_ai.db-wal b/data/scud_orion_ai.db-wal index 1b919fe..ebf0535 100644 Binary files a/data/scud_orion_ai.db-wal and b/data/scud_orion_ai.db-wal differ diff --git a/modules/web_api/llm/agent.py b/modules/web_api/llm/agent.py index e46e3ad..0fd7df5 100644 --- a/modules/web_api/llm/agent.py +++ b/modules/web_api/llm/agent.py @@ -1,25 +1,28 @@ """ =============================================================================== FILE: modules/web_api/llm/agent.py -PROJECT: SCUD Orion AI (Unified Repository) +PROJECT: SCUD Orion AI (Unified Architecture) MODULE: web_api / llm (Core Agent & Function Calling Dispatcher) ROLE: Главный оркестратор взаимодействия с Ollama LLM (Qwen 2.5), разбор вызовов - инструментов (Function Calling), генерация превью промпта и логирование. + инструментов (Function Calling), интеграция с декларативным реестром + действий SQLite (tool_action_registry) и управление сессионными кнопками. AI-CONTEXT-ANCHORS & INVARIANTS: - - ANCHOR[LOGGING_CONFIG]: Явный вывод логов в stdout для мгновенной видимости - вызовов тулов в systemd journalctl. + - ANCHOR[LOGGING_CONFIG]: Настройка StreamHandler в sys.stdout с propagate=False + для предотвращения дублирования строк в systemd journalctl. - ANCHOR[DYNAMIC_CONTEXT]: Сборка системного контекста (календарь, сессия, промпт). - ANCHOR[INFERENCE_OPTIONS]: Параметры инференса (repeat_penalty, ctx_size) для предотвращения урезания длинных списков моделью Qwen 2.5. - ANCHOR[TOOL_ROUTER]: Диспетчеризация функций SQLite (CRUD задач, снапшотов, KB). - ANCHOR[PROMPT_MERGE_LOGIC]: Универсальный парсер точечного добавления и удаления пунктов системного промпта в режиме предпросмотра (PROMPT_PREVIEW). + - ANCHOR[DATA_DRIVEN_ACTIONS]: Универсальная обработка инструментов на базе + таблицы tool_action_registry (мгновенный возврат ответов и динамических кнопок). - ANCHOR[SECONDARY_PASS]: Вторичный вызов LLM для формирования текстового ответа на основе полученного tool_result. DEPENDENCIES: - - modules/web_api/llm/db_tools.py (доступ к SQLite) + - modules/web_api/llm/db_tools.py (доступ к SQLite и реестру действий) - modules/web_api/llm/schemas.py (TOOLS_SCHEMA) - modules/web_api/llm/core/calendar_utils.py (get_dynamic_calendar_context) - modules/web_api/llm/core/tool_injector.py (clean_raw_tool_tags, clean_output) @@ -40,6 +43,7 @@ from typing import List, Dict, Any, Tuple, Optional from .db_tools import ( db_get_active_system_prompt, db_add_system_prompt, + db_get_tool_action, db_get_tasks, db_update_task_status, db_delete_task, @@ -66,7 +70,7 @@ from .core.tool_injector import clean_raw_tool_tags, clean_output, inject_tools_ # --- [SECTION 2: LOGGING CONFIGURATION] --- # ANCHOR[LOGGING_CONFIG] logger = logging.getLogger("SCUD_AGENT") logger.setLevel(logging.INFO) -logger.propagate = False +logger.propagate = False # Исключаем дублирование сообщений в родительском логгере Uvicorn if not logger.handlers: handler = logging.StreamHandler(sys.stdout) @@ -87,13 +91,14 @@ def process_chat_message( image_b64: Optional[str] = None, chat_history: List[Dict[str, Any]] = None, session_id: str = "web_session_main" -) -> Tuple[str, List[Dict[str, Any]]]: +) -> Tuple[str, List[Dict[str, Any]], Optional[Dict[str, Any]]]: """ Главный конвейер обработки входящего сообщения: 1. Сохранение сообщения пользователя. 2. Формирование системного контекста и вызов Ollama. 3. Выполнение вызванного Tool (если сгенерирован). - 4. Вторичный проход генерации и возврат истории. + 4. Опрос Data-Driven реестра действий (tool_action_registry) или вторичный проход LLM. + 5. Возврат кортежа: (reply_text, chat_history, action_metadata). """ logger.info(f"Получено сообщение от user_id={user_id}, session_id={session_id}: {user_message}") @@ -182,8 +187,46 @@ def process_chat_message( logger.info(f"🚀 Вызов функции (Tool): {fn_name} с аргументами: {fn_args}") tool_result_content = "" - # Роутинг инструментов - if fn_name == "db_get_snapshots": + # --- 5.1. УНИВЕРСАЛЬНАЯ ПРОВЕРКА DATA-DRIVEN РЕЕСТРА (tool_action_registry) --- # ANCHOR[DATA_DRIVEN_ACTIONS] + action_cfg = db_get_tool_action(fn_name) + + if fn_name == "db_confirm_prompt_preview": + if session_state and session_state.get("state_type") == "PROMPT_PREVIEW": + db_add_system_prompt("main_agent", session_state.get("pending_data", "")) + db_clear_session_state(session_id) + + if action_cfg and action_cfg.get("bypass_llm"): + reply_text = action_cfg["success_template"] + if action_cfg.get("follow_up_question"): + reply_text += f"\n\n{action_cfg['follow_up_question']}" + + db_save_chat_message(session_id, "assistant", reply_text) + return reply_text, db_get_chat_history(session_id), { + "type": action_cfg.get("action_type"), + "buttons": action_cfg.get("buttons", []) + } + tool_result_content = json.dumps({"status": "success"}, ensure_ascii=False) + else: + err_reply = "⚠️ Нет активного превью для подтверждения." + db_save_chat_message(session_id, "assistant", err_reply) + return err_reply, db_get_chat_history(session_id), None + + elif fn_name == "db_cancel_prompt_preview": + db_clear_session_state(session_id) + if action_cfg and action_cfg.get("bypass_llm"): + reply_text = action_cfg["success_template"] + if action_cfg.get("follow_up_question"): + reply_text += f"\n\n{action_cfg['follow_up_question']}" + + db_save_chat_message(session_id, "assistant", reply_text) + return reply_text, db_get_chat_history(session_id), { + "type": action_cfg.get("action_type"), + "buttons": action_cfg.get("buttons", []) + } + tool_result_content = json.dumps({"status": "cancelled"}, ensure_ascii=False) + + # --- 5.2. СТАНДАРТНЫЙ РОУТИНГ ИНСТРУМЕНТОВ --- + elif fn_name == "db_get_snapshots": snapshots_res = db_get_snapshots(session_id=session_id, date_str=fn_args.get("date_str"), original_user_message=user_message) tool_result_content = json.dumps(snapshots_res, ensure_ascii=False) @@ -206,7 +249,17 @@ def process_chat_message( tool_result_content = json.dumps(db_get_session_states(), ensure_ascii=False) elif fn_name == "db_delete_snapshots": - tool_result_content = json.dumps(db_delete_snapshots(snapshot_id=fn_args.get("snapshot_id"), day_str=fn_args.get("day_str")), ensure_ascii=False) + res = db_delete_snapshots(snapshot_id=fn_args.get("snapshot_id"), day_str=fn_args.get("day_str")) + if action_cfg and action_cfg.get("bypass_llm"): + reply_text = action_cfg["success_template"] + if action_cfg.get("follow_up_question"): + reply_text += f"\n\n{action_cfg['follow_up_question']}" + db_save_chat_message(session_id, "assistant", reply_text) + return reply_text, db_get_chat_history(session_id), { + "type": action_cfg.get("action_type"), + "buttons": action_cfg.get("buttons", []) + } + tool_result_content = json.dumps(res, ensure_ascii=False) elif fn_name == "db_get_reference": tool_result_content = json.dumps(db_get_reference(category=fn_args.get("category")), ensure_ascii=False) @@ -260,20 +313,13 @@ def process_chat_message( f"Для применения изменений подтвердите действие («подтверждаю») или отмените («отмена»)." ) db_save_chat_message(session_id, "assistant", preview_reply) - # Возвращаем "PROMPT_PREVIEW" как третий параметр - return clean_raw_tool_tags(preview_reply), db_get_chat_history(session_id), "PROMPT_PREVIEW" - - elif fn_name == "db_confirm_prompt_preview": - if session_state and session_state.get("state_type") == "PROMPT_PREVIEW": - res = db_add_system_prompt("main_agent", session_state.get("pending_data", "")) - db_clear_session_state(session_id) - tool_result_content = json.dumps(res, ensure_ascii=False) - else: - tool_result_content = json.dumps({"status": "error", "message": "Нет активного превью для подтверждения."}, ensure_ascii=False) - - elif fn_name == "db_cancel_prompt_preview": - db_clear_session_state(session_id) - tool_result_content = json.dumps({"status": "success", "message": "Превью системного промпта отменено."}, ensure_ascii=False) + return clean_raw_tool_tags(preview_reply), db_get_chat_history(session_id), { + "type": "PROMPT_PREVIEW", + "buttons": [ + {"label": "Подтвердить", "value": "подтверждаю", "style": "primary"}, + {"label": "Отменить", "value": "отмена", "style": "danger"} + ] + } elif fn_name == "db_get_rules": tool_result_content = json.dumps(db_get_rules(), ensure_ascii=False) @@ -288,6 +334,15 @@ def process_chat_message( elif fn_name == "db_delete_task": res = db_delete_task(user_id=user_id, task_id=str(fn_args.get("task_id", "")).upper()) + if action_cfg and action_cfg.get("bypass_llm"): + reply_text = action_cfg["success_template"] + if action_cfg.get("follow_up_question"): + reply_text += f"\n\n{action_cfg['follow_up_question']}" + db_save_chat_message(session_id, "assistant", reply_text) + return reply_text, db_get_chat_history(session_id), { + "type": action_cfg.get("action_type"), + "buttons": action_cfg.get("buttons", []) + } tool_result_content = json.dumps(res, ensure_ascii=False) messages.append({"role": "tool", "content": tool_result_content}) diff --git a/modules/web_api/llm/db/db_prompts.py b/modules/web_api/llm/db/db_prompts.py index f7d7334..9978bc4 100644 --- a/modules/web_api/llm/db/db_prompts.py +++ b/modules/web_api/llm/db/db_prompts.py @@ -1,6 +1,7 @@ """ FILE: modules/web_api/llm/db/db_prompts.py """ +import json import logging from typing import List, Dict, Any, Optional from .connection import get_db_connection @@ -15,6 +16,25 @@ def db_get_active_system_prompt() -> str: conn.close() return row["prompt_text"] if row else "Ты — ИИ-ассистент SCUD Orion AI." +def db_get_tool_action(tool_name: str) -> Optional[Dict[str, Any]]: + """Получение шаблона и кнопок действия для инструмента из SQLite.""" + conn = get_db_connection() + cursor = conn.cursor() + cursor.execute(""" + SELECT tool_name, category, bypass_llm, success_template, + follow_up_question, action_type, buttons_json + FROM tool_action_registry + WHERE tool_name = ? AND is_active = 1 + """, (tool_name,)) + row = cursor.fetchone() + conn.close() + + if row: + res = dict(row) + res["buttons"] = json.loads(res["buttons_json"]) if res.get("buttons_json") else [] + return res + return None + def db_add_system_prompt(name: str, prompt_text: str) -> Dict[str, Any]: try: with get_db_connection() as conn: diff --git a/modules/web_api/llm/db_tools.py b/modules/web_api/llm/db_tools.py index 806fb07..df4f588 100644 --- a/modules/web_api/llm/db_tools.py +++ b/modules/web_api/llm/db_tools.py @@ -9,6 +9,7 @@ from .db.db_snapshots import db_get_snapshots, db_delete_snapshots from .db.db_prompts import ( db_get_active_system_prompt, db_add_system_prompt, + db_get_tool_action, db_get_rules, db_set_session_state, db_get_session_state, @@ -16,7 +17,8 @@ from .db.db_prompts import ( db_get_session_states, db_get_stats, db_get_anomalies, - db_get_reference + db_get_reference, + db_get_tool_action ) def db_get_current_server_time(): diff --git a/modules/web_api/static/js/chat.js b/modules/web_api/static/js/chat.js index 4083a01..cedc7bf 100644 --- a/modules/web_api/static/js/chat.js +++ b/modules/web_api/static/js/chat.js @@ -133,19 +133,37 @@ async function sendMessage(e) { // Генерация блока кнопок подтверждения при необходимости let actionButtonsHtml = ""; - if (data.action_type === "PROMPT_PREVIEW") { + const actionData = data.action_type || data.action; + + if (actionData && actionData.buttons && actionData.buttons.length > 0) { + const buttonsMarkup = actionData.buttons.map(btn => { + let btnClasses = "bg-slate-100 hover:bg-slate-200 active:bg-slate-300 text-slate-700 border border-slate-300"; + let iconMarkup = ''; + + const labelLower = (btn.label || "").toLowerCase(); + const isPrimary = btn.style === "primary" || labelLower.includes("подтверд") || labelLower.startsWith("да"); + const isDanger = btn.style === "danger" || labelLower.includes("отмен") || labelLower.startsWith("нет") || labelLower.includes("законч") || labelLower.includes("заверш"); + + if (isPrimary) { + btnClasses = "bg-emerald-600 hover:bg-emerald-700 active:bg-emerald-800 text-white shadow-sm"; + iconMarkup = ''; + } else if (isDanger) { + btnClasses = "bg-rose-50 hover:bg-rose-100 active:bg-rose-200 text-rose-700 border border-rose-300"; + iconMarkup = ''; + } + + return ` + + `; + }).join(""); + actionButtonsHtml = ` -
- - +
+ ${buttonsMarkup}
`; } diff --git a/scripts/db_cli.py b/scripts/db_cli.py index 5fa6101..2049738 100644 --- a/scripts/db_cli.py +++ b/scripts/db_cli.py @@ -18,6 +18,31 @@ from core.database import ( DB_PATH = os.path.join(DATA_DIR, "scud_orion_ai.db") +def print_tool_actions(): + """Выводит реестр декларативных действий инструментов и шаблоны кнопок.""" + print("\n" + "=" * 110) + print("🧰 ДЕКЛАРАТИВНЫЙ РЕЕСТР ДЕЙСТВИЙ ИНСТРУМЕНТОВ (tool_action_registry):") + print("=" * 110) + with get_connection() as conn: + cursor = conn.cursor() + cursor.execute(""" + SELECT id, tool_name, category, bypass_llm, success_template, follow_up_question, buttons_json + FROM tool_action_registry + WHERE is_active = 1 + ORDER BY id ASC + """) + rows = cursor.fetchall() + if not rows: + print("Таблица tool_action_registry пуста.") + else: + for r in rows: + print(f"ID: {r[0]} | Tool: [{r[1]}] | Категория: {r[2]} | Bypass LLM: {'ДА (0.05с)' if r[3] else 'НЕТ'}") + print(f" • Сообщение: {r[4]}") + if r[5]: + print(f" • Вопрос: {r[5]}") + print(f" • Кнопки: {r[6]}") + print("-" * 110) + print("=" * 110 + "\n") def print_stats(): """Выводит общую статистику по записям в таблицах БД.""" @@ -255,6 +280,7 @@ def print_session_states(): print("=" * 80 + "\n") +Python HELP_TEXT = """ CLI-утилита инспекции и управления SQLite базой данных СКУД (scud_orion_ai.db) @@ -266,6 +292,7 @@ CLI-утилита инспекции и управления SQLite базой anomalies -- Посмотреть историю найденных аномалий СКУД ⟷ 1С rules -- Посмотреть правила Базы Знаний ИИ из SQLite prompts -- Посмотреть системные промпты (system_prompts) + tools -- Посмотреть реестр действий инструментов и кнопок (tool_action_registry) sessions -- Посмотреть активные сессии и превью (session_states) dump [output.xlsx] -- Полный дамп всех таблиц БД в многостраничный Excel snapshot del [ID] или [--day ДД.ММ.ГГГГ] -- Удаление снапшота по ID или всех за выбранный день @@ -277,6 +304,7 @@ CLI-утилита инспекции и управления SQLite базой python scripts/db_cli.py scud --snapshot Y20260805-007 python scripts/db_cli.py absences 07.08.2026 python scripts/db_cli.py prompts + python scripts/db_cli.py tools python scripts/db_cli.py sessions python scripts/db_cli.py snapshot del Y20260805-007 python scripts/db_cli.py snapshot del --day 04.08.2026 @@ -294,7 +322,7 @@ def main(): formatter_class=argparse.RawDescriptionHelpFormatter, add_help=False ) - parser.add_argument('command', nargs='?', default=None, choices=['stats', 'snapshots', 'scud', 'absences', 'anomalies', 'rules', 'prompts', 'sessions', 'dump', 'snapshot'], help="Основная команда") + parser.add_argument('command', nargs='?', default=None, choices=['stats', 'snapshots', 'scud', 'absences', 'anomalies', 'rules', 'prompts', 'sessions', 'dump', 'snapshot', 'tools'], help="Основная команда") parser.add_argument('action', nargs='?', default=None, help="Дополнительное действие (например, 'del')") parser.add_argument('param', nargs='?', default=None, help="Параметр (дата, ID снапшота или имя файла)") parser.add_argument('--snapshot', type=str, default=None, help="ID конкретного снапшота для инспекции") @@ -325,6 +353,8 @@ def main(): print_system_prompts() elif args.command == 'sessions': print_session_states() + elif args.command == 'tools': + print_tool_actions() elif args.command == 'dump': filename = args.param if args.param else "db_dump_full.xlsx" dump_all_to_excel(filename) diff --git a/scripts/init_tool_registry.py b/scripts/init_tool_registry.py new file mode 100644 index 0000000..5cc8a1b --- /dev/null +++ b/scripts/init_tool_registry.py @@ -0,0 +1,104 @@ +""" +Скрипт инициализации таблицы tool_action_registry в SQLite БД. +""" +import os +import sys +import json +import sqlite3 + +BASE_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "..")) +DB_PATH = os.path.join(BASE_DIR, "data", "scud_orion_ai.db") + +def init_tool_action_registry(): + conn = sqlite3.connect(DB_PATH) + cursor = conn.cursor() + + # 1. Создаем таблицу + cursor.execute(""" + CREATE TABLE IF NOT EXISTS tool_action_registry ( + id INTEGER PRIMARY KEY AUTOINCREMENT, + tool_name TEXT UNIQUE NOT NULL, + category TEXT DEFAULT 'general', + bypass_llm INTEGER DEFAULT 1, + success_template TEXT NOT NULL, + follow_up_question TEXT, + action_type TEXT DEFAULT 'BUTTONS', + buttons_json TEXT, + is_active INTEGER DEFAULT 1, + updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP + ); + """) + + # 2. Начальное наполнение базовыми сценариями + seed_tools = [ + ( + "db_confirm_prompt_preview", + "prompt", + 1, + "✅ Системный промпт успешно сохранен и применен в базе данных.", + "Желаете продолжить работу с системным промптом?", + "FOLLOW_UP_ACTION", + json.dumps([ + {"label": "Показать промпт", "value": "покажи системный промпт"}, + {"label": "Готово", "value": "нет, спасибо"} + ], ensure_ascii=False) + ), + ( + "db_cancel_prompt_preview", + "prompt", + 1, + "❌ Изменения системного промпта отменены.", + "Желаете продолжить работу с системным промптом?", + "FOLLOW_UP_ACTION", + json.dumps([ + {"label": "Показать промпт", "value": "покажи системный промпт"}, + {"label": "Готово", "value": "нет, спасибо"} + ], ensure_ascii=False) + ), + ( + "db_delete_task", + "tasks", + 1, + "🗑 Задача успешно удалена из вашего реестра.", + "Хотите посмотреть обновленный список задач?", + "FOLLOW_UP_ACTION", + json.dumps([ + {"label": "Показать задачи", "value": "покажи список задач"}, + {"label": "Готово", "value": "нет"} + ], ensure_ascii=False) + ), + ( + "db_delete_snapshots", + "snapshots", + 1, + "🗑 Выбранные снапшоты успешно удалены из базы данных.", + "Вывести актуальный список снапшотов?", + "FOLLOW_UP_ACTION", + json.dumps([ + {"label": "Показать снапшоты", "value": "покажи список снапшотов"}, + {"label": "Готово", "value": "нет"} + ], ensure_ascii=False) + ) + ] + + for item in seed_tools: + cursor.execute(""" + INSERT INTO tool_action_registry ( + tool_name, category, bypass_llm, success_template, follow_up_question, action_type, buttons_json + ) VALUES (?, ?, ?, ?, ?, ?, ?) + ON CONFLICT(tool_name) DO UPDATE SET + category = excluded.category, + bypass_llm = excluded.bypass_llm, + success_template = excluded.success_template, + follow_up_question = excluded.follow_up_question, + action_type = excluded.action_type, + buttons_json = excluded.buttons_json, + updated_at = CURRENT_TIMESTAMP + """, item) + + conn.commit() + conn.close() + print("[✓] Таблица tool_action_registry успешно создана и наполнена!") + +if __name__ == "__main__": + init_tool_action_registry()