feat(db/llm/ui): настроена работа кнопок подтверждающих изменения внеcенные инструментом и финальные для выхода из инструмента и очистки сессии.
This commit is contained in:
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -1,25 +1,28 @@
|
|||||||
"""
|
"""
|
||||||
===============================================================================
|
===============================================================================
|
||||||
FILE: modules/web_api/llm/agent.py
|
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)
|
MODULE: web_api / llm (Core Agent & Function Calling Dispatcher)
|
||||||
ROLE: Главный оркестратор взаимодействия с Ollama LLM (Qwen 2.5), разбор вызовов
|
ROLE: Главный оркестратор взаимодействия с Ollama LLM (Qwen 2.5), разбор вызовов
|
||||||
инструментов (Function Calling), генерация превью промпта и логирование.
|
инструментов (Function Calling), интеграция с декларативным реестром
|
||||||
|
действий SQLite (tool_action_registry) и управление сессионными кнопками.
|
||||||
|
|
||||||
AI-CONTEXT-ANCHORS & INVARIANTS:
|
AI-CONTEXT-ANCHORS & INVARIANTS:
|
||||||
- ANCHOR[LOGGING_CONFIG]: Явный вывод логов в stdout для мгновенной видимости
|
- ANCHOR[LOGGING_CONFIG]: Настройка StreamHandler в sys.stdout с propagate=False
|
||||||
вызовов тулов в systemd journalctl.
|
для предотвращения дублирования строк в systemd journalctl.
|
||||||
- ANCHOR[DYNAMIC_CONTEXT]: Сборка системного контекста (календарь, сессия, промпт).
|
- ANCHOR[DYNAMIC_CONTEXT]: Сборка системного контекста (календарь, сессия, промпт).
|
||||||
- ANCHOR[INFERENCE_OPTIONS]: Параметры инференса (repeat_penalty, ctx_size) для
|
- ANCHOR[INFERENCE_OPTIONS]: Параметры инференса (repeat_penalty, ctx_size) для
|
||||||
предотвращения урезания длинных списков моделью Qwen 2.5.
|
предотвращения урезания длинных списков моделью Qwen 2.5.
|
||||||
- ANCHOR[TOOL_ROUTER]: Диспетчеризация функций SQLite (CRUD задач, снапшотов, KB).
|
- ANCHOR[TOOL_ROUTER]: Диспетчеризация функций SQLite (CRUD задач, снапшотов, KB).
|
||||||
- ANCHOR[PROMPT_MERGE_LOGIC]: Универсальный парсер точечного добавления и
|
- ANCHOR[PROMPT_MERGE_LOGIC]: Универсальный парсер точечного добавления и
|
||||||
удаления пунктов системного промпта в режиме предпросмотра (PROMPT_PREVIEW).
|
удаления пунктов системного промпта в режиме предпросмотра (PROMPT_PREVIEW).
|
||||||
|
- ANCHOR[DATA_DRIVEN_ACTIONS]: Универсальная обработка инструментов на базе
|
||||||
|
таблицы tool_action_registry (мгновенный возврат ответов и динамических кнопок).
|
||||||
- ANCHOR[SECONDARY_PASS]: Вторичный вызов LLM для формирования текстового ответа
|
- ANCHOR[SECONDARY_PASS]: Вторичный вызов LLM для формирования текстового ответа
|
||||||
на основе полученного tool_result.
|
на основе полученного tool_result.
|
||||||
|
|
||||||
DEPENDENCIES:
|
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/schemas.py (TOOLS_SCHEMA)
|
||||||
- modules/web_api/llm/core/calendar_utils.py (get_dynamic_calendar_context)
|
- 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)
|
- 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 (
|
from .db_tools import (
|
||||||
db_get_active_system_prompt,
|
db_get_active_system_prompt,
|
||||||
db_add_system_prompt,
|
db_add_system_prompt,
|
||||||
|
db_get_tool_action,
|
||||||
db_get_tasks,
|
db_get_tasks,
|
||||||
db_update_task_status,
|
db_update_task_status,
|
||||||
db_delete_task,
|
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]
|
# --- [SECTION 2: LOGGING CONFIGURATION] --- # ANCHOR[LOGGING_CONFIG]
|
||||||
logger = logging.getLogger("SCUD_AGENT")
|
logger = logging.getLogger("SCUD_AGENT")
|
||||||
logger.setLevel(logging.INFO)
|
logger.setLevel(logging.INFO)
|
||||||
logger.propagate = False
|
logger.propagate = False # Исключаем дублирование сообщений в родительском логгере Uvicorn
|
||||||
|
|
||||||
if not logger.handlers:
|
if not logger.handlers:
|
||||||
handler = logging.StreamHandler(sys.stdout)
|
handler = logging.StreamHandler(sys.stdout)
|
||||||
@@ -87,13 +91,14 @@ def process_chat_message(
|
|||||||
image_b64: Optional[str] = None,
|
image_b64: Optional[str] = None,
|
||||||
chat_history: List[Dict[str, Any]] = None,
|
chat_history: List[Dict[str, Any]] = None,
|
||||||
session_id: str = "web_session_main"
|
session_id: str = "web_session_main"
|
||||||
) -> Tuple[str, List[Dict[str, Any]]]:
|
) -> Tuple[str, List[Dict[str, Any]], Optional[Dict[str, Any]]]:
|
||||||
"""
|
"""
|
||||||
Главный конвейер обработки входящего сообщения:
|
Главный конвейер обработки входящего сообщения:
|
||||||
1. Сохранение сообщения пользователя.
|
1. Сохранение сообщения пользователя.
|
||||||
2. Формирование системного контекста и вызов Ollama.
|
2. Формирование системного контекста и вызов Ollama.
|
||||||
3. Выполнение вызванного Tool (если сгенерирован).
|
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}")
|
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}")
|
logger.info(f"🚀 Вызов функции (Tool): {fn_name} с аргументами: {fn_args}")
|
||||||
tool_result_content = ""
|
tool_result_content = ""
|
||||||
|
|
||||||
# Роутинг инструментов
|
# --- 5.1. УНИВЕРСАЛЬНАЯ ПРОВЕРКА DATA-DRIVEN РЕЕСТРА (tool_action_registry) --- # ANCHOR[DATA_DRIVEN_ACTIONS]
|
||||||
if fn_name == "db_get_snapshots":
|
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)
|
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)
|
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)
|
tool_result_content = json.dumps(db_get_session_states(), ensure_ascii=False)
|
||||||
|
|
||||||
elif fn_name == "db_delete_snapshots":
|
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":
|
elif fn_name == "db_get_reference":
|
||||||
tool_result_content = json.dumps(db_get_reference(category=fn_args.get("category")), ensure_ascii=False)
|
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"Для применения изменений подтвердите действие («подтверждаю») или отмените («отмена»)."
|
f"Для применения изменений подтвердите действие («подтверждаю») или отмените («отмена»)."
|
||||||
)
|
)
|
||||||
db_save_chat_message(session_id, "assistant", preview_reply)
|
db_save_chat_message(session_id, "assistant", preview_reply)
|
||||||
# Возвращаем "PROMPT_PREVIEW" как третий параметр
|
return clean_raw_tool_tags(preview_reply), db_get_chat_history(session_id), {
|
||||||
return clean_raw_tool_tags(preview_reply), db_get_chat_history(session_id), "PROMPT_PREVIEW"
|
"type": "PROMPT_PREVIEW",
|
||||||
|
"buttons": [
|
||||||
elif fn_name == "db_confirm_prompt_preview":
|
{"label": "Подтвердить", "value": "подтверждаю", "style": "primary"},
|
||||||
if session_state and session_state.get("state_type") == "PROMPT_PREVIEW":
|
{"label": "Отменить", "value": "отмена", "style": "danger"}
|
||||||
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)
|
|
||||||
|
|
||||||
elif fn_name == "db_get_rules":
|
elif fn_name == "db_get_rules":
|
||||||
tool_result_content = json.dumps(db_get_rules(), ensure_ascii=False)
|
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":
|
elif fn_name == "db_delete_task":
|
||||||
res = db_delete_task(user_id=user_id, task_id=str(fn_args.get("task_id", "")).upper())
|
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)
|
tool_result_content = json.dumps(res, ensure_ascii=False)
|
||||||
|
|
||||||
messages.append({"role": "tool", "content": tool_result_content})
|
messages.append({"role": "tool", "content": tool_result_content})
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
"""
|
"""
|
||||||
FILE: modules/web_api/llm/db/db_prompts.py
|
FILE: modules/web_api/llm/db/db_prompts.py
|
||||||
"""
|
"""
|
||||||
|
import json
|
||||||
import logging
|
import logging
|
||||||
from typing import List, Dict, Any, Optional
|
from typing import List, Dict, Any, Optional
|
||||||
from .connection import get_db_connection
|
from .connection import get_db_connection
|
||||||
@@ -15,6 +16,25 @@ def db_get_active_system_prompt() -> str:
|
|||||||
conn.close()
|
conn.close()
|
||||||
return row["prompt_text"] if row else "Ты — ИИ-ассистент SCUD Orion AI."
|
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]:
|
def db_add_system_prompt(name: str, prompt_text: str) -> Dict[str, Any]:
|
||||||
try:
|
try:
|
||||||
with get_db_connection() as conn:
|
with get_db_connection() as conn:
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ from .db.db_snapshots import db_get_snapshots, db_delete_snapshots
|
|||||||
from .db.db_prompts import (
|
from .db.db_prompts import (
|
||||||
db_get_active_system_prompt,
|
db_get_active_system_prompt,
|
||||||
db_add_system_prompt,
|
db_add_system_prompt,
|
||||||
|
db_get_tool_action,
|
||||||
db_get_rules,
|
db_get_rules,
|
||||||
db_set_session_state,
|
db_set_session_state,
|
||||||
db_get_session_state,
|
db_get_session_state,
|
||||||
@@ -16,7 +17,8 @@ from .db.db_prompts import (
|
|||||||
db_get_session_states,
|
db_get_session_states,
|
||||||
db_get_stats,
|
db_get_stats,
|
||||||
db_get_anomalies,
|
db_get_anomalies,
|
||||||
db_get_reference
|
db_get_reference,
|
||||||
|
db_get_tool_action
|
||||||
)
|
)
|
||||||
|
|
||||||
def db_get_current_server_time():
|
def db_get_current_server_time():
|
||||||
|
|||||||
@@ -133,19 +133,37 @@ async function sendMessage(e) {
|
|||||||
|
|
||||||
// Генерация блока кнопок подтверждения при необходимости
|
// Генерация блока кнопок подтверждения при необходимости
|
||||||
let actionButtonsHtml = "";
|
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 = '<i class="fa-solid fa-arrow-right text-[10px] opacity-60"></i>';
|
||||||
|
|
||||||
|
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 = '<i class="fa-solid fa-check"></i>';
|
||||||
|
} else if (isDanger) {
|
||||||
|
btnClasses = "bg-rose-50 hover:bg-rose-100 active:bg-rose-200 text-rose-700 border border-rose-300";
|
||||||
|
iconMarkup = '<i class="fa-solid fa-xmark"></i>';
|
||||||
|
}
|
||||||
|
|
||||||
|
return `
|
||||||
|
<button type="button" onclick="handleActionButtonClick('${escapeHtml(btn.value)}')"
|
||||||
|
class="${btnClasses} font-semibold px-3 py-1.5 rounded-xl text-xs flex items-center gap-1.5 transition cursor-pointer">
|
||||||
|
${iconMarkup}
|
||||||
|
<span>${escapeHtml(btn.label)}</span>
|
||||||
|
</button>
|
||||||
|
`;
|
||||||
|
}).join("");
|
||||||
|
|
||||||
actionButtonsHtml = `
|
actionButtonsHtml = `
|
||||||
<div class="action-buttons-container flex items-center gap-2 mt-3 pt-2.5 border-t border-slate-100">
|
<div class="action-buttons-container flex flex-wrap items-center gap-2 mt-3 pt-2.5 border-t border-slate-100">
|
||||||
<button type="button" onclick="handleActionButtonClick('подтверждаю')"
|
${buttonsMarkup}
|
||||||
class="bg-emerald-600 hover:bg-emerald-700 active:bg-emerald-800 text-white font-semibold px-3.5 py-1.5 rounded-xl text-xs flex items-center gap-1.5 shadow-sm transition">
|
|
||||||
<i class="fa-solid fa-check"></i>
|
|
||||||
<span>Подтвердить</span>
|
|
||||||
</button>
|
|
||||||
<button type="button" onclick="handleActionButtonClick('отмена')"
|
|
||||||
class="bg-slate-100 hover:bg-slate-200 active:bg-slate-300 text-slate-700 font-semibold px-3.5 py-1.5 rounded-xl text-xs flex items-center gap-1.5 border border-slate-300 transition">
|
|
||||||
<i class="fa-solid fa-xmark"></i>
|
|
||||||
<span>Отменить</span>
|
|
||||||
</button>
|
|
||||||
</div>
|
</div>
|
||||||
`;
|
`;
|
||||||
}
|
}
|
||||||
|
|||||||
+31
-1
@@ -18,6 +18,31 @@ from core.database import (
|
|||||||
|
|
||||||
DB_PATH = os.path.join(DATA_DIR, "scud_orion_ai.db")
|
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():
|
def print_stats():
|
||||||
"""Выводит общую статистику по записям в таблицах БД."""
|
"""Выводит общую статистику по записям в таблицах БД."""
|
||||||
@@ -255,6 +280,7 @@ def print_session_states():
|
|||||||
print("=" * 80 + "\n")
|
print("=" * 80 + "\n")
|
||||||
|
|
||||||
|
|
||||||
|
Python
|
||||||
HELP_TEXT = """
|
HELP_TEXT = """
|
||||||
CLI-утилита инспекции и управления SQLite базой данных СКУД (scud_orion_ai.db)
|
CLI-утилита инспекции и управления SQLite базой данных СКУД (scud_orion_ai.db)
|
||||||
|
|
||||||
@@ -266,6 +292,7 @@ CLI-утилита инспекции и управления SQLite базой
|
|||||||
anomalies -- Посмотреть историю найденных аномалий СКУД ⟷ 1С
|
anomalies -- Посмотреть историю найденных аномалий СКУД ⟷ 1С
|
||||||
rules -- Посмотреть правила Базы Знаний ИИ из SQLite
|
rules -- Посмотреть правила Базы Знаний ИИ из SQLite
|
||||||
prompts -- Посмотреть системные промпты (system_prompts)
|
prompts -- Посмотреть системные промпты (system_prompts)
|
||||||
|
tools -- Посмотреть реестр действий инструментов и кнопок (tool_action_registry)
|
||||||
sessions -- Посмотреть активные сессии и превью (session_states)
|
sessions -- Посмотреть активные сессии и превью (session_states)
|
||||||
dump [output.xlsx] -- Полный дамп всех таблиц БД в многостраничный Excel
|
dump [output.xlsx] -- Полный дамп всех таблиц БД в многостраничный Excel
|
||||||
snapshot del [ID] или [--day ДД.ММ.ГГГГ] -- Удаление снапшота по ID или всех за выбранный день
|
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 scud --snapshot Y20260805-007
|
||||||
python scripts/db_cli.py absences 07.08.2026
|
python scripts/db_cli.py absences 07.08.2026
|
||||||
python scripts/db_cli.py prompts
|
python scripts/db_cli.py prompts
|
||||||
|
python scripts/db_cli.py tools
|
||||||
python scripts/db_cli.py sessions
|
python scripts/db_cli.py sessions
|
||||||
python scripts/db_cli.py snapshot del Y20260805-007
|
python scripts/db_cli.py snapshot del Y20260805-007
|
||||||
python scripts/db_cli.py snapshot del --day 04.08.2026
|
python scripts/db_cli.py snapshot del --day 04.08.2026
|
||||||
@@ -294,7 +322,7 @@ def main():
|
|||||||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||||||
add_help=False
|
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('action', nargs='?', default=None, help="Дополнительное действие (например, 'del')")
|
||||||
parser.add_argument('param', nargs='?', default=None, help="Параметр (дата, ID снапшота или имя файла)")
|
parser.add_argument('param', nargs='?', default=None, help="Параметр (дата, ID снапшота или имя файла)")
|
||||||
parser.add_argument('--snapshot', type=str, default=None, help="ID конкретного снапшота для инспекции")
|
parser.add_argument('--snapshot', type=str, default=None, help="ID конкретного снапшота для инспекции")
|
||||||
@@ -325,6 +353,8 @@ def main():
|
|||||||
print_system_prompts()
|
print_system_prompts()
|
||||||
elif args.command == 'sessions':
|
elif args.command == 'sessions':
|
||||||
print_session_states()
|
print_session_states()
|
||||||
|
elif args.command == 'tools':
|
||||||
|
print_tool_actions()
|
||||||
elif args.command == 'dump':
|
elif args.command == 'dump':
|
||||||
filename = args.param if args.param else "db_dump_full.xlsx"
|
filename = args.param if args.param else "db_dump_full.xlsx"
|
||||||
dump_all_to_excel(filename)
|
dump_all_to_excel(filename)
|
||||||
|
|||||||
@@ -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()
|
||||||
Reference in New Issue
Block a user