""" =============================================================================== FILE: modules/web_api/llm/agent.py PROJECT: SCUD Orion AI (Unified Architecture) MODULE: web_api / llm (Core Agent Coordinator) ROLE: Нативный оркестратор диалога, диспетчер Function Calling, передача активных срезов в контекст модели и терминальные вызовы. =============================================================================== """ import sys import json import logging 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_apply_prompt_node_action, db_get_tool_action, db_get_tasks, db_update_task_details, db_delete_task, db_add_task, db_tasks_edit, db_export_tasks_markdown, db_get_rules, db_set_session_state, db_get_session_state, db_clear_session_state, db_get_snapshots, db_delete_snapshots, db_get_current_server_time, db_save_chat_message, db_get_chat_history, db_purge_ephemeral_messages, db_get_stats, db_get_anomalies, db_get_reference ) from .schemas import TOOLS_SCHEMA from .core.calendar_utils import get_dynamic_calendar_context from .core.tool_injector import clean_raw_tool_tags, clean_output, inject_tools_if_needed from .core.ollama_client import call_ollama_chat from .core.fast_path import handle_fast_path_intercept from .core.context_manager import ( save_tool_interaction, save_dialog_interaction, mark_last_user_message_ephemeral, close_tool_session_and_cleanup ) logger = logging.getLogger("SCUD_AGENT") logger.setLevel(logging.INFO) logger.propagate = False if not logger.handlers: handler = logging.StreamHandler(sys.stdout) formatter = logging.Formatter("%(asctime)s [%(levelname)s] [%(name)s] %(message)s") handler.setFormatter(formatter) logger.addHandler(handler) def process_chat_message( user_id: int, user_message: str, file_context: str = "", image_b64: Optional[str] = None, chat_history: List[Dict[str, Any]] = None, session_id: str = "web_session_main" ) -> Tuple[str, List[Dict[str, Any]], Optional[Dict[str, Any]]]: """ Главный конвейер обработки входящего сообщения чата на базе нативного Function Calling. """ logger.info(f"Получено сообщение от user_id={user_id}, session_id={session_id}: {user_message}") session_state = db_get_session_state(session_id) if not session_state: db_purge_ephemeral_messages(session_id) full_user_content = f"{user_message}\n\n[СОДЕРЖИМОЕ ПРИКРЕПЛЕННОГО ФАЙЛА]:\n{file_context}" if file_context else user_message # 1. Быстрый перехват системных кнопок UI и терминальных действий без задержек LLM fast_path_res = handle_fast_path_intercept(session_id, user_message, full_user_content, session_state) if fast_path_res: return fast_path_res # 2. Фиксация сообщения пользователя db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=0) db_history = db_get_chat_history(session_id, limit=20) calendar_context = get_dynamic_calendar_context() user_info = f"Пользователь ID={user_id}" if user_id != 0 else "Гость" current_state_type = session_state.get("state_type") if session_state else None state_data = session_state.get("data_json") or {} if session_state else {} if not isinstance(state_data, dict): state_data = {} idle_turns = state_data.get("idle_turns", 0) active_state_context = "" if current_state_type == "PROMPT_PREVIEW": active_state_context = "\n[ТЕКУЩИЙ РЕЖИМ: ПРЕДПРОСМОТР СИСТЕМНОГО ПРОМПТА]\n- Открыт предпросмотр изменений промпта.\n" elif current_state_type == "SNAPSHOTS_VIEW": active_date = state_data.get("query_date", "выбранную дату") active_state_context = f"\n[ТЕКУЩИЙ РЕЖИМ: ПРОСМОТР СНАПШОТОВ СКУД]\n- Отображаются срезы за {active_date}.\n" elif current_state_type == "SNAPSHOT_INSPECT": snap_id = state_data.get("snapshot_id", "") snap_date = state_data.get("log_date", "") records = state_data.get("records", []) lines = [] for r in records: st = "Присутствовал" if r.get("is_present") else "Отсутствовал" lines.append(f"- {r.get('fio')}: Отдел={r.get('department')}, Вход={r.get('time_in')}, Активность={r.get('first_activity')}, Выход={r.get('time_out')}, ВремяВЗдании={r.get('in_building')}, Статус={st}") dump_str = "\n".join(lines) active_state_context = ( f"\n[ТЕКУЩИЙ РЕЖИМ: АКТИВНА ИНСПЕКЦИЯ СРЕЗА СКУД #{snap_id} ЗА {snap_date}]\n" f"Оператор сейчас изучает срез #{snap_id}. При любых вопросах о сотрудниках, их времени входа/выхода, отделах или присутствии — " f"ТЫ ОБЯЗАН брать данные исключительно из этого списка активного среза:\n{dump_str}\n" ) # ⭐️ Промпт с поддержкой Topic Drift и защитой от переспросов по задачам system_prompt_content = ( f"Ты — интеллектуальный ассистент SCUD Orion AI.\n" f"Твоя основная роль — помощь оператору в кадровом аудите СКУД/1С, управлении задачами и настройками системы.\n\n" f"СТРОГИЕ ПРАВИЛА:\n" f"1. К оператору всегда обращайся на Вы.\n" f"2. Для работы с данными системы ВСЕГДА вызывай соответствующие инструменты (tools):\n" f" - Снапшоты и срезы СКУД -> db_get_snapshots\n" f" - Удаление срезов -> db_delete_snapshots\n" f" - Задачи и бэклог -> db_get_tasks, db_tasks_edit\n" f" - Системный промпт -> db_get_system_prompt, db_prompt_node_edit\n" f" - База знаний и регламенты -> db_get_rules\n" f" - Аномалии СКУД/1С -> db_get_anomalies\n" f" - Справка -> db_get_reference\n" f"3. ЗАДАЧИ:\n" f" - При любых запросах на просмотр задач (включая опечатки вроде 'змдачи', 'таски', 'дела', 'покажи задачи') — " f"ТЫ ОБЯЗАН СРАЗУ ВЫЗВАТЬ db_get_tasks без лишних вопросов!\n" f" - КАТЕГОРИЧЕСКИ ЗАПРЕЩЕНО переспрашивать у оператора фильтры, статус или категорию задач текстом! " f"Интерактивная карточка в интерфейсе содержит все нужные фильтры.\n" f"4. ПРАВИЛА И РЕГЛАМЕНТЫ: При любых вопросах о правилах компании или арбитраже ТЫ ОБЯЗАН СРАЗУ вызвать db_get_rules.\n" f"5. Запрещено выдумывать факты и цифры по системе СКУД/1С без вызова инструментов.\n" f"6. ОБЩИЙ ДИАЛОГ: На любые отвлечённые, познавательные, научные или бытовые вопросы " f"(расстояние между планетами или городами, программирование, кругозор) отвечай полно, доброжелательно и интересно, не отказывая пользователю.\n\n" f"[СИСТЕМНЫЙ КАЛЕНДАРЬ СЕРВЕРА]\n" f"- Пользователь: {user_info}\n" f"- {calendar_context}\n" f"{active_state_context}" ) user_msg_object = {"role": "user", "content": full_user_content} try: if image_b64: user_msg_object["images"] = [image_b64] messages = [ {"role": "system", "content": "Ты — строгий модуль OCR. Перепиши весь текст с изображения буква в букву."}, user_msg_object ] msg = call_ollama_chat(messages, is_vision=True) else: clean_db_history = [dict(m) for m in db_history] for m in clean_db_history: m.pop("images", None) messages = [{"role": "system", "content": system_prompt_content}] + clean_db_history + [user_msg_object] msg = call_ollama_chat(messages, tools=TOOLS_SCHEMA, is_vision=False) raw_text_reply = msg.get("content", "") tool_calls = msg.get("tool_calls", []) if not tool_calls: tool_calls = inject_tools_if_needed(user_message, raw_text_reply, tool_calls) if tool_calls: logger.info(f"Ответ от Ollama получен. Tool calls: True (кол-во: {len(tool_calls)})") tool = tool_calls[0] fn_name = tool["function"]["name"] fn_args = tool["function"].get("arguments", {}) if isinstance(fn_args, str): try: fn_args = json.loads(fn_args) except Exception: fn_args = {} logger.info(f"Вызов функции (Tool): {fn_name} с аргументами: {fn_args}") # Ротация контекста close_tool_session_and_cleanup(session_id, close_reason=f"ACTIVATE_{fn_name}") session_state = None mark_last_user_message_ephemeral(session_id) # 1. Задачи (Просмотр) if fn_name == "db_get_tasks": raw_tasks = db_get_tasks(user_id, status=fn_args.get("status")) reply_text = "Вот интерактивный список ваших текущих задач:" db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1) return reply_text, db_get_chat_history(session_id), { "type": "TASK_INTERACTIVE_CARD", "tasks": raw_tasks } # 2. Единый диспетчер задач elif fn_name in ["db_tasks_edit", "db_add_task", "db_update_task_details", "db_delete_task"]: action = fn_args.get("action", "UPDATE").upper() if fn_name == "db_add_task": action = "ADD" elif fn_name == "db_delete_task": action = "DELETE" elif fn_name == "db_update_task_status": action = "UPDATE" if action == "DELETE": task_id_raw = str(fn_args.get("task_id", "")).replace("#", "").replace("TASK-", "").strip() db_set_session_state(session_id, "TASK_DELETE_CONFIRM", {"task_id": task_id_raw, "idle_turns": 0}) reply_text = f"Вы действительно хотите удалить задачу #{task_id_raw}?" db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1) return reply_text, db_get_chat_history(session_id), { "type": "TASK_DELETE_CONFIRM", "buttons": [ {"label": f"Удалить #{task_id_raw}", "value": f"подтверждаю удаление задачи {task_id_raw}", "style": "danger"}, {"label": "Отмена", "value": "отмена", "style": "secondary"} ] } elif action == "EXPORT": export_res = db_export_tasks_markdown( user_id=user_id, filename=fn_args.get("filename") or "ROADMAP.md", status_filter=fn_args.get("status") ) reply_text = export_res.get("message", "Отчет по задачам успешно экспортирован.") db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=0) action_payload = None if export_res.get("status") == "success": action_payload = { "type": "FILE_DOWNLOAD_CARD", "filename": export_res.get("filename"), "download_url": export_res.get("download_url"), "tasks_count": export_res.get("tasks_count") } return reply_text, db_get_chat_history(session_id), action_payload else: res = db_tasks_edit( user_id=user_id, action=action, task_id=fn_args.get("task_id"), title=fn_args.get("title"), priority=fn_args.get("priority", "MEDIUM"), status=fn_args.get("status"), module=fn_args.get("module", "general"), due_date=fn_args.get("due_date") ) raw_tasks = db_get_tasks(user_id) reply_text = res.get("message", "Операция над задачами выполнена.") db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1) return reply_text, db_get_chat_history(session_id), { "type": "TASK_INTERACTIVE_CARD", "tasks": raw_tasks } # 3. Системный промпт elif fn_name == "db_get_system_prompt": active_prompt = db_get_active_system_prompt() reply_text = ( "📋 **АКТУАЛЬНЫЙ СИСТЕМНЫЙ ПРОМПТ ИЗ БАЗЫ ДАННЫХ:**\n\n" f"```text\n{active_prompt}\n```\n\n" "Вы можете добавить, отредактировать или удалить любой пункт." ) db_set_session_state(session_id, "PROMPT_FOLLOWUP", {"idle_turns": 0}) db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1) return reply_text, db_get_chat_history(session_id), { "type": "PROMPT_VIEW", "buttons": [ {"label": "✏️ Редактировать промпт", "value": "action:open_editor", "style": "primary"}, {"label": "База знаний", "value": "покажи правила компании", "style": "secondary"} ] } # 4. База знаний и правила компании (терминальный возврат) elif fn_name == "db_get_rules": rules_data = db_get_rules() if isinstance(rules_data, dict) and "rules" in rules_data: rules_list = rules_data["rules"] elif isinstance(rules_data, list): rules_list = rules_data else: rules_list = [str(rules_data)] formatted_rules = "\n\n".join([f"{i+1}. {r.get('rule_text', r) if isinstance(r, dict) else r}" for i, r in enumerate(rules_list)]) reply_text = ( "📖 **БАЗА ЗНАНИЙ И ПРАВИЛА КОМПАНИИ (КАДРОВЫЙ АРБИТРАЖ):**\n\n" f"```text\n{formatted_rules}\n```" ) db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1) return reply_text, db_get_chat_history(session_id), { "type": "RULES_VIEW", "buttons": [ {"label": "✏️ Редактировать правила", "value": "action:open_rules_editor", "style": "primary"}, {"label": "📋 Системный промпт", "value": "покажи системный промпт", "style": "secondary"} ] } # 5. Снапшоты СКУД 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) query_date = snapshots_res.get("query_date", "выбранную дату") reply_text = f"Реестр срезов СКУД за {query_date}:" db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1) return reply_text, db_get_chat_history(session_id), { "type": "SNAPSHOTS_CARD", "data": snapshots_res } elif fn_name == "db_delete_snapshots": raw_id = fn_args.get("snapshot_id") or fn_args.get("day_str") raw_ids = fn_args.get("snapshot_ids") or [] is_confirmed = fn_args.get("confirmed", False) if raw_id and not raw_ids: if isinstance(raw_id, str) and "," in raw_id: raw_ids = [s.strip() for s in raw_id.split(",")] else: raw_ids = [raw_id] safe_ids = [s.strip() for s in raw_ids if s and not str(s).strip().startswith("Y")] if not safe_ids: reply_text = "⚠️ Итоговый срез Y защищен от удаления. Выберите дневные снапшоты." db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1) return reply_text, db_get_chat_history(session_id), None if not is_confirmed: query_date = state_data.get("query_date", "") db_set_session_state(session_id, "SNAPSHOT_DELETE_CONFIRM", { "snapshot_ids": safe_ids, "query_date": query_date, "idle_turns": 0 }) ids_str = ", ".join(safe_ids) reply_text = f"Вы действительно хотите удалить дневные снапшоты: {ids_str}?" db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1) return reply_text, db_get_chat_history(session_id), { "type": "SNAPSHOT_DELETE_CONFIRM", "buttons": [ {"label": f"Удалить ({len(safe_ids)} шт.)", "value": f"подтверждаю удаление снапшотов {ids_str}", "style": "danger"}, {"label": "Отмена", "value": "отмена", "style": "secondary"} ] } else: db_delete_snapshots(snapshot_ids=safe_ids) query_date = state_data.get("query_date", "") updated_snapshots_res = db_get_snapshots(session_id=session_id, date_str=query_date) reply_text = f"✅ Успешно удалено снапшотов: {len(safe_ids)} шт." db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1) return reply_text, db_get_chat_history(session_id), { "type": "SNAPSHOTS_CARD", "data": updated_snapshots_res } # 6. Прочие сервисные инструменты elif fn_name == "db_get_current_server_time": tool_result_content = json.dumps(db_get_current_server_time(), ensure_ascii=False) elif fn_name == "db_get_stats": tool_result_content = json.dumps(db_get_stats(), ensure_ascii=False) elif fn_name == "db_get_anomalies": tool_result_content = json.dumps(db_get_anomalies(limit=fn_args.get("limit", 100), date_str=fn_args.get("date_str")), 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) else: tool_result_content = "{}" messages.append(msg) messages.append({"role": "tool", "content": tool_result_content}) sec_msg = call_ollama_chat(messages, is_vision=False) raw_content = sec_msg.get("content", "").strip().replace("**", "").replace("*", "") final_content = clean_raw_tool_tags(clean_output(raw_content)) or "Запрос выполнен." db_save_chat_message(session_id, "assistant", final_content, is_ephemeral=0) return final_content, db_get_chat_history(session_id), None # Свободный диалог (Topic Drift) raw_str = msg.get("content", "").strip().replace("**", "").replace("*", "") final_reply = clean_raw_tool_tags(clean_output(raw_str)) or "Запрос обработан." db_save_chat_message(session_id, "assistant", final_reply, is_ephemeral=0) return final_reply, db_get_chat_history(session_id), None except Exception as ex: logger.exception(f"Непредвиденная ошибка агента: {ex}") error_reply = f"Внутренняя ошибка сервера: {ex}" return error_reply, db_get_chat_history(session_id), None