уход с работы
This commit is contained in:
+23
@@ -0,0 +1,23 @@
|
||||
# Виртуальное окружение Python
|
||||
venv/
|
||||
.venv/
|
||||
env/
|
||||
|
||||
# Кэш Python
|
||||
__pycache__/
|
||||
*.py[cod]
|
||||
*$py.class
|
||||
|
||||
# Локальная база данных SQLite
|
||||
*.db
|
||||
*.sqlite
|
||||
*.sqlite3
|
||||
|
||||
# Переменные окружения и секреты
|
||||
.env
|
||||
|
||||
# Логи и временные файлы
|
||||
*.log
|
||||
.DS_Store
|
||||
.idea/
|
||||
.vscode/
|
||||
+78
@@ -0,0 +1,78 @@
|
||||
import sqlite3
|
||||
import os
|
||||
from datetime import datetime
|
||||
|
||||
DB_NAME = "context_memory.db"
|
||||
|
||||
def init_db():
|
||||
conn = sqlite3.connect(DB_NAME)
|
||||
cursor = conn.cursor()
|
||||
|
||||
# 1. Таблица трекинга задач
|
||||
cursor.execute("""
|
||||
CREATE TABLE IF NOT EXISTS tasks (
|
||||
task_id TEXT PRIMARY KEY,
|
||||
module TEXT NOT NULL,
|
||||
title TEXT NOT NULL,
|
||||
status TEXT NOT NULL,
|
||||
priority TEXT NOT NULL,
|
||||
completed_at TEXT
|
||||
)
|
||||
""")
|
||||
|
||||
# 2. Таблица архитектурных правил и контекста
|
||||
cursor.execute("""
|
||||
CREATE TABLE IF NOT EXISTS architecture_memory (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
category TEXT NOT NULL,
|
||||
rule_text TEXT NOT NULL,
|
||||
is_active INTEGER DEFAULT 1,
|
||||
created_at TEXT DEFAULT CURRENT_TIMESTAMP
|
||||
)
|
||||
""")
|
||||
|
||||
# Первичное заполнение реестра задач [TASK-01] .. [TASK-17]
|
||||
tasks_data = [
|
||||
("TASK-01", "services/data_loader", "Исправление KeyError по ФИО в Excel экспорте 1С (переход на .iloc)", "COMPLETED", "HIGH", "2026-08-03"),
|
||||
("TASK-02", "services/zup_extractor", "Перевод подключения 1С:ЗУП на продакшн MS SQL сервер ACCOUNT-01 + гибридный фоллбэк", "COMPLETED", "HIGH", "2026-08-04"),
|
||||
("TASK-03", "services/scud_export", "Инцидент Равина В.Э.: Фиксация 100% приоритета сырых логов pLogData СКУД над расчетом УРВ Ориона", "COMPLETED", "HIGH", "2026-08-04"),
|
||||
("TASK-04", "core/database", "Реализация схемы SQLite WAL, сохранение снапшотов и маппинг колонок (department/position)", "COMPLETED", "HIGH", "2026-08-05"),
|
||||
("TASK-05", "core/database", "Исправление бага дублирования номеров снапшотов (переход на вычисление MAX() по суффиксу)", "COMPLETED", "HIGH", "2026-08-05"),
|
||||
("TASK-06", "scripts/fix_snapshots", "Создание и выполнение скрипта сквозной переиндексации снапшотов с префиксом Y", "COMPLETED", "HIGH", "2026-08-05"),
|
||||
("TASK-07", "scripts/db_cli", "Реализация CLI-инспектора БД и выравнивание номеров срезов для идеальной верстки в терминале", "COMPLETED", "HIGH", "2026-08-05"),
|
||||
("TASK-08", "main.py / logic", "Двойная проверка исключений по department_scud и 1С, удержание 100% приоритета кадровых документов", "COMPLETED", "HIGH", "2026-08-05"),
|
||||
("TASK-09", "services/text_reporter", "Лаконичный вывод неразмеченных сотрудников с короткими аббревиатурами отделов СКУД (РУК, ОИЗ, ПУ и др.)", "COMPLETED", "HIGH", "2026-08-05"),
|
||||
("TASK-10", "services/scud_export", "Внутридневной контроль: Извлечение полных цепочек проходов pLogData (входы/выходы за смену)", "BACKLOG", "HIGH", None),
|
||||
("TASK-11", "services/data_loader", "Внутридневной контроль: Расчет количества и суммарной длительности отлучек/перекуров сверх обеда", "BACKLOG", "HIGH", None),
|
||||
("TASK-12", "services/text_reporter", "Внутридневной контроль: Вывод списка системных нарушителей внутреннего распорядка в Markdown-отчет", "BACKLOG", "MEDIUM", None),
|
||||
("TASK-13", "automation", "Автоматизация сбора срезов по расписанию (Cron / Systemd Timers на 12:00, 17:00, 19:00)", "BACKLOG", "HIGH", None),
|
||||
("TASK-14", "services/notifications", "Интеграция Telegram-бота для алертов Администратору при массовых сбоях турникетов (>5%)", "BACKLOG", "MEDIUM", None),
|
||||
("TASK-15", "api / fastapi", "Создание REST API на FastAPI (/api/v1/snapshots, /api/v1/scud/logs, /api/v1/anomalies)", "BACKLOG", "HIGH", None),
|
||||
("TASK-16", "frontend / chat", "Веб-интерфейс с ИИ-чатом на базе Ollama (Qwen 2.5 14B) с поддержкой Function Calling к REST API", "BACKLOG", "HIGH", None),
|
||||
("TASK-17", "core/memory", "Создание модуля персистентной памяти проекта в SQLite и CLI-инструмента для фиксации архитектурных решений", "IN_PROGRESS", "HIGH", None),
|
||||
]
|
||||
|
||||
cursor.executemany("""
|
||||
INSERT OR REPLACE INTO tasks (task_id, module, title, status, priority, completed_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?)
|
||||
""", tasks_data)
|
||||
|
||||
# Базовые правила архитектуры
|
||||
rules_data = [
|
||||
("logic_rule", "Сырые логи pLogData СКУД имееют 100% приоритет над расчетом УРВ Ориона."),
|
||||
("logic_rule", "Официальные кадровые документы 1С (отпуска, командировки, больничные) имеют абсолютный приоритет над отсутствием в СКУД."),
|
||||
("scud_override", "Запрещено переименовывать ФИО сотрудников при 100% совпадении строк между 1С и СКУД в AI-верификаторе."),
|
||||
("formatting", "Температура указывается исключительно в градусах Цельсия, скорости в км/ч.")
|
||||
]
|
||||
|
||||
cursor.executemany("""
|
||||
INSERT INTO architecture_memory (category, rule_text)
|
||||
VALUES (?, ?)
|
||||
""", rules_data)
|
||||
|
||||
conn.commit()
|
||||
conn.close()
|
||||
print("[✓] База данных контекста и трекинга задач успешно создана и заполнена!")
|
||||
|
||||
if __name__ == "__main__":
|
||||
init_db()
|
||||
+292
@@ -0,0 +1,292 @@
|
||||
# === ANCHOR: IMPORTS_START ===
|
||||
import sqlite3
|
||||
import json
|
||||
import urllib.request
|
||||
import urllib.error
|
||||
from typing import List, Dict, Any, Tuple
|
||||
from datetime import datetime
|
||||
|
||||
DB_NAME = "context_memory.db"
|
||||
OLLAMA_URL = "http://192.168.11.3:11434/api/chat"
|
||||
MODEL_NAME = "qwen2.5:14b"
|
||||
# === ANCHOR: IMPORTS_END ===
|
||||
|
||||
|
||||
# === ANCHOR: DB_TOOLS_START ===
|
||||
def db_get_tasks(status: str = None) -> str:
|
||||
"""Извлекает список задач из реестра БД."""
|
||||
conn = sqlite3.connect(DB_NAME)
|
||||
conn.row_factory = sqlite3.Row
|
||||
cursor = conn.cursor()
|
||||
if status:
|
||||
cursor.execute("SELECT * FROM tasks WHERE status = ? ORDER BY task_id", (status.upper(),))
|
||||
else:
|
||||
cursor.execute("SELECT * FROM tasks ORDER BY task_id")
|
||||
rows = [dict(r) for r in cursor.fetchall()]
|
||||
conn.close()
|
||||
return json.dumps(rows, ensure_ascii=False)
|
||||
|
||||
|
||||
def db_update_task_status(task_id: str, status: str = None, due_date: str = None) -> str:
|
||||
"""Обновляет статус и/или срок выполнения задачи."""
|
||||
conn = sqlite3.connect(DB_NAME)
|
||||
cursor = conn.cursor()
|
||||
updates = []
|
||||
params = []
|
||||
|
||||
if status:
|
||||
status_upper = status.upper()
|
||||
updates.append("status = ?")
|
||||
params.append(status_upper)
|
||||
if status_upper == "COMPLETED":
|
||||
updates.append("completed_at = CURRENT_TIMESTAMP")
|
||||
else:
|
||||
updates.append("completed_at = NULL")
|
||||
|
||||
if due_date:
|
||||
updates.append("due_date = ?")
|
||||
params.append(due_date)
|
||||
|
||||
if not updates:
|
||||
conn.close()
|
||||
return json.dumps({"error": "Не указаны параметры для обновления"}, ensure_ascii=False)
|
||||
|
||||
params.append(task_id.upper())
|
||||
query = f"UPDATE tasks SET {', '.join(updates)} WHERE task_id = ?"
|
||||
cursor.execute(query, tuple(params))
|
||||
|
||||
if cursor.rowcount == 0:
|
||||
conn.close()
|
||||
return json.dumps({"error": f"Задача {task_id} не найдена"}, ensure_ascii=False)
|
||||
|
||||
conn.commit()
|
||||
conn.close()
|
||||
return json.dumps({"status": "success", "message": f"Задача {task_id} обновлена"}, ensure_ascii=False)
|
||||
|
||||
|
||||
def db_add_task(task_id: str, module: str, title: str, priority: str = "HIGH", due_date: str = None) -> str:
|
||||
"""Добавляет новую задачу в реестр."""
|
||||
conn = sqlite3.connect(DB_NAME)
|
||||
cursor = conn.cursor()
|
||||
try:
|
||||
cursor.execute(
|
||||
"INSERT INTO tasks (task_id, module, title, status, priority, due_date) VALUES (?, ?, ?, 'BACKLOG', ?, ?)",
|
||||
(task_id.upper(), module, title, priority.upper(), due_date)
|
||||
)
|
||||
conn.commit()
|
||||
res = {"status": "success", "message": f"Задача {task_id} создана"}
|
||||
except sqlite3.IntegrityError:
|
||||
res = {"error": f"Задача с ID {task_id} уже существует"}
|
||||
finally:
|
||||
conn.close()
|
||||
return json.dumps(res, ensure_ascii=False)
|
||||
|
||||
|
||||
def db_get_rules() -> str:
|
||||
"""Возвращает действующие архитектурные правила."""
|
||||
conn = sqlite3.connect(DB_NAME)
|
||||
conn.row_factory = sqlite3.Row
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT category, rule_text FROM architecture_memory WHERE is_active = 1")
|
||||
rows = [dict(r) for r in cursor.fetchall()]
|
||||
conn.close()
|
||||
return json.dumps(rows, ensure_ascii=False)
|
||||
# === ANCHOR: DB_TOOLS_END ===
|
||||
|
||||
|
||||
# === ANCHOR: TOOLS_SCHEMA_START ===
|
||||
TOOLS_SCHEMA = [
|
||||
{
|
||||
"type": "function",
|
||||
"function": {
|
||||
"name": "db_get_tasks",
|
||||
"description": "Получить список задач проекта с возможностью фильтрации по статусу (BACKLOG, IN_PROGRESS, COMPLETED).",
|
||||
"parameters": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"status": {
|
||||
"type": "string",
|
||||
"description": "Фильтр статуса: BACKLOG, IN_PROGRESS или COMPLETED."
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"type": "function",
|
||||
"function": {
|
||||
"name": "db_update_task_status",
|
||||
"description": "Изменить статус и/или срок выполнения (due_date) задачи в реестре.",
|
||||
"parameters": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"task_id": {"type": "string", "description": "Идентификатор задачи, например TASK-17"},
|
||||
"status": {"type": "string", "description": "Новый статус: COMPLETED, IN_PROGRESS или BACKLOG"},
|
||||
"due_date": {"type": "string", "description": "Срок выполнения задачи, например '2026-08-07 14:00'"}
|
||||
},
|
||||
"required": ["task_id"]
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"type": "function",
|
||||
"function": {
|
||||
"name": "db_add_task",
|
||||
"description": "Добавить новую задачу в бэклог проекта.",
|
||||
"parameters": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"task_id": {"type": "string", "description": "Уникальный ID задачи, например TASK-18"},
|
||||
"module": {"type": "string", "description": "Модуль проекта, например services/scud_export"},
|
||||
"title": {"type": "string", "description": "Краткое описание задачи"},
|
||||
"priority": {"type": "string", "description": "Приоритет: HIGH, MEDIUM, LOW"},
|
||||
"due_date": {"type": "string", "description": "Срок выполнения задачи, например '2026-08-07 14:00'"}
|
||||
},
|
||||
"required": ["task_id", "module", "title"]
|
||||
}
|
||||
}
|
||||
},
|
||||
{
|
||||
"type": "function",
|
||||
"function": {
|
||||
"name": "db_get_rules",
|
||||
"description": "Получить список архитектурных правил и ограничений проекта.",
|
||||
"parameters": {"type": "object", "properties": {}}
|
||||
}
|
||||
}
|
||||
]
|
||||
# === ANCHOR: TOOLS_SCHEMA_END ===
|
||||
|
||||
|
||||
# === ANCHOR: CHAT_AGENT_START ===
|
||||
def process_chat_message(user_message: str, chat_history: List[Dict[str, Any]] = None) -> Tuple[str, List[Dict[str, Any]]]:
|
||||
if chat_history is None:
|
||||
chat_history = []
|
||||
|
||||
current_now = datetime.now().strftime("%Y-%m-%d %H:%M")
|
||||
|
||||
system_prompt = {
|
||||
"role": "system",
|
||||
"content": (
|
||||
f"Текущая дата и время сервера: {current_now}.\n"
|
||||
f"Текущая дата и время сервера: {current_now}.\n"
|
||||
"Ты — ИИ-ассистент проекта SCUD Orion AI. "
|
||||
"У тебя есть доступ к функциям для просмотра, добавления и изменения статусов и сроков задач (due_date) в БД. "
|
||||
"Когда пользователь просит показать задачи, добавить новую или изменить статус — обязательно вызывай нужную функцию. "
|
||||
"После вызова функции давай пользователю понятное, лаконичное резюме результата на русском языке. "
|
||||
"При запросе списка всех задач не расписывай каждую подробно, а давай краткую сводку по количествам, "
|
||||
"так как детали отображаются в выезжающем реестре задач."
|
||||
)
|
||||
}
|
||||
|
||||
messages = [system_prompt] + chat_history + [{"role": "user", "content": user_message}]
|
||||
|
||||
payload = {
|
||||
"model": MODEL_NAME,
|
||||
"messages": messages,
|
||||
"tools": TOOLS_SCHEMA,
|
||||
"stream": False,
|
||||
"options": {
|
||||
"num_predict": 2048, # Достаточно для любых длинных списков и кода
|
||||
"num_ctx": 8192, # Расширенная память диалога без потери скорости
|
||||
"temperature": 0.2, # Низкая температура: ответы точные и без галлюцинаций
|
||||
"top_p": 0.9, # Оптимальная выборка токенов
|
||||
"num_gpu": 99 # Заставляет Ollama загрузить 100% слоев модели в VRAM
|
||||
}
|
||||
}
|
||||
|
||||
req = urllib.request.Request(
|
||||
OLLAMA_URL,
|
||||
data=json.dumps(payload).encode("utf-8"),
|
||||
headers={"Content-Type": "application/json"}
|
||||
)
|
||||
|
||||
try:
|
||||
with urllib.request.urlopen(req) as response:
|
||||
res_data = json.loads(response.read().decode("utf-8"))
|
||||
msg = res_data.get("message", {})
|
||||
|
||||
tool_calls = msg.get("tool_calls", [])
|
||||
|
||||
# Резервная проверка, если Ollama прислала вызов функции текстом
|
||||
content_str = msg.get("content", "")
|
||||
if not tool_calls and "name" in content_str and "db_" in content_str:
|
||||
try:
|
||||
parsed = json.loads(content_str)
|
||||
if "name" in parsed:
|
||||
tool_calls = [{"function": parsed}]
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
if tool_calls:
|
||||
messages.append(msg)
|
||||
for tool in tool_calls:
|
||||
fn_name = tool["function"]["name"]
|
||||
fn_args = tool["function"].get("arguments", {})
|
||||
|
||||
if fn_name == "db_get_tasks":
|
||||
result = db_get_tasks(status=fn_args.get("status"))
|
||||
elif fn_name == "db_update_task_status":
|
||||
result = db_update_task_status(
|
||||
task_id=fn_args.get("task_id"),
|
||||
status=fn_args.get("status"),
|
||||
due_date=fn_args.get("due_date")
|
||||
)
|
||||
elif fn_name == "db_add_task":
|
||||
result = db_add_task(
|
||||
task_id=fn_args.get("task_id"),
|
||||
module=fn_args.get("module"),
|
||||
title=fn_args.get("title"),
|
||||
priority=fn_args.get("priority", "HIGH"),
|
||||
due_date=fn_args.get("due_date")
|
||||
)
|
||||
elif fn_name == "db_get_rules":
|
||||
result = db_get_rules()
|
||||
else:
|
||||
result = json.dumps({"error": "Неизвестная функция"})
|
||||
|
||||
messages.append({
|
||||
"role": "tool",
|
||||
"content": result
|
||||
})
|
||||
|
||||
# Вторым запросом передаем результат обратно модели для текстового ответа
|
||||
payload = {
|
||||
"model": MODEL_NAME,
|
||||
"messages": messages,
|
||||
"tools": TOOLS_SCHEMA,
|
||||
"stream": False,
|
||||
"options": {
|
||||
"num_predict": 2048, # Достаточно для любых длинных списков и кода
|
||||
"num_ctx": 8192, # Расширенная память диалога без потери скорости
|
||||
"temperature": 0.2, # Низкая температура: ответы точные и без галлюцинаций
|
||||
"top_p": 0.9, # Оптимальная выборка токенов
|
||||
"num_gpu": 99 # Заставляет Ollama загрузить 100% слоев модели в VRAM
|
||||
}
|
||||
}
|
||||
req_second = urllib.request.Request(
|
||||
OLLAMA_URL,
|
||||
data=json.dumps(payload_second).encode("utf-8"),
|
||||
headers={"Content-Type": "application/json"}
|
||||
)
|
||||
with urllib.request.urlopen(req_second) as resp_second:
|
||||
final_res = json.loads(resp_second.read().decode("utf-8"))
|
||||
final_msg = final_res.get("message", {}).get("content", "")
|
||||
|
||||
updated_history = chat_history + [
|
||||
{"role": "user", "content": user_message},
|
||||
{"role": "assistant", "content": final_msg}
|
||||
]
|
||||
return final_msg, updated_history
|
||||
|
||||
else:
|
||||
final_msg = msg.get("content", "")
|
||||
updated_history = chat_history + [
|
||||
{"role": "user", "content": user_message},
|
||||
{"role": "assistant", "content": final_msg}
|
||||
]
|
||||
return final_msg, updated_history
|
||||
|
||||
except urllib.error.URLError as e:
|
||||
return f"Ошибка связи с Ollama ({OLLAMA_URL}): {e}", chat_history
|
||||
# === ANCHOR: CHAT_AGENT_END ===
|
||||
@@ -0,0 +1,166 @@
|
||||
# === ANCHOR: IMPORTS_START ===
|
||||
import os
|
||||
import json
|
||||
import sqlite3
|
||||
from typing import List, Optional, Dict, Any
|
||||
|
||||
from fastapi import FastAPI, HTTPException, Depends, Security
|
||||
from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
|
||||
from fastapi.staticfiles import StaticFiles
|
||||
from fastapi.responses import FileResponse, HTMLResponse
|
||||
from pydantic import BaseModel
|
||||
|
||||
from llm_agent import process_chat_message
|
||||
# === ANCHOR: IMPORTS_END ===
|
||||
|
||||
|
||||
# === ANCHOR: APP_INIT_START ===
|
||||
app = FastAPI(
|
||||
title="SCUD Orion AI Context & Task Tracker API",
|
||||
version="2.0.0",
|
||||
description="REST API локального контекста, памяти и трекинга задач для ИИ-аудитора"
|
||||
)
|
||||
|
||||
DB_NAME = "context_memory.db"
|
||||
SECURITY_TOKEN = os.getenv("API_BEARER_TOKEN", "scud_secret_token_2026")
|
||||
security = HTTPBearer()
|
||||
# === ANCHOR: APP_INIT_END ===
|
||||
|
||||
|
||||
# === ANCHOR: DATABASE_HELPERS_START ===
|
||||
def init_chat_table():
|
||||
"""Создает таблицу истории чата, если она не существует."""
|
||||
conn = sqlite3.connect(DB_NAME)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("""
|
||||
CREATE TABLE IF NOT EXISTS chat_sessions (
|
||||
session_id TEXT PRIMARY KEY,
|
||||
history_json TEXT NOT NULL,
|
||||
updated_at TEXT DEFAULT CURRENT_TIMESTAMP
|
||||
)
|
||||
""")
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
||||
init_chat_table()
|
||||
|
||||
|
||||
def verify_token(credentials: HTTPAuthorizationCredentials = Security(security)):
|
||||
if credentials.credentials != SECURITY_TOKEN:
|
||||
raise HTTPException(status_code=403, detail="Недействительный токен доступа")
|
||||
return credentials.credentials
|
||||
|
||||
|
||||
def get_db_connection():
|
||||
conn = sqlite3.connect(DB_NAME)
|
||||
conn.row_factory = sqlite3.Row
|
||||
return conn
|
||||
# === ANCHOR: DATABASE_HELPERS_END ===
|
||||
|
||||
|
||||
# === ANCHOR: SCHEMAS_START ===
|
||||
class TaskSchema(BaseModel):
|
||||
task_id: str
|
||||
module: str
|
||||
title: str
|
||||
status: str
|
||||
priority: str
|
||||
completed_at: Optional[str] = None
|
||||
|
||||
|
||||
class RuleSchema(BaseModel):
|
||||
id: int
|
||||
category: str
|
||||
rule_text: str
|
||||
is_active: int
|
||||
|
||||
|
||||
class ChatRequest(BaseModel):
|
||||
session_id: str = "default"
|
||||
message: str
|
||||
|
||||
|
||||
class ChatResponse(BaseModel):
|
||||
session_id: str
|
||||
reply: str
|
||||
# === ANCHOR: SCHEMAS_END ===
|
||||
|
||||
|
||||
# === ANCHOR: STATIC_ROUTES_START ===
|
||||
# Подключаем папку static для отдачи HTML-интерфейса
|
||||
app.mount("/static", StaticFiles(directory="static"), name="static")
|
||||
|
||||
@app.get("/", response_class=HTMLResponse)
|
||||
def read_root():
|
||||
return FileResponse("static/index.html")
|
||||
# === ANCHOR: STATIC_ROUTES_END ===
|
||||
|
||||
|
||||
# === ANCHOR: TASKS_ENDPOINTS_START ===
|
||||
@app.get("/api/v1/tasks", response_model=List[TaskSchema], dependencies=[Depends(verify_token)])
|
||||
def get_all_tasks(status: Optional[str] = None):
|
||||
conn = get_db_connection()
|
||||
cursor = conn.cursor()
|
||||
if status:
|
||||
cursor.execute("SELECT * FROM tasks WHERE status = ? ORDER BY task_id", (status.upper(),))
|
||||
else:
|
||||
cursor.execute("SELECT * FROM tasks ORDER BY task_id")
|
||||
rows = cursor.fetchall()
|
||||
conn.close()
|
||||
return [dict(r) for r in rows]
|
||||
|
||||
|
||||
@app.post("/api/v1/tasks/{task_id}/complete", dependencies=[Depends(verify_token)])
|
||||
def mark_task_completed(task_id: str):
|
||||
conn = get_db_connection()
|
||||
cursor = conn.cursor()
|
||||
cursor.execute(
|
||||
"UPDATE tasks SET status = 'COMPLETED', completed_at = CURRENT_TIMESTAMP WHERE task_id = ?",
|
||||
(task_id.upper(),)
|
||||
)
|
||||
if cursor.rowcount == 0:
|
||||
conn.close()
|
||||
raise HTTPException(status_code=404, detail=f"Задача {task_id} не найдена")
|
||||
conn.commit()
|
||||
conn.close()
|
||||
return {"status": "success", "message": f"Задача {task_id} успешно выполнена"}
|
||||
# === ANCHOR: TASKS_ENDPOINTS_END ===
|
||||
|
||||
|
||||
# === ANCHOR: RULES_AND_CHAT_ENDPOINTS_START ===
|
||||
@app.get("/api/v1/rules", response_model=List[RuleSchema], dependencies=[Depends(verify_token)])
|
||||
def get_active_rules():
|
||||
conn = get_db_connection()
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT * FROM architecture_memory WHERE is_active = 1 ORDER BY id")
|
||||
rows = cursor.fetchall()
|
||||
conn.close()
|
||||
return [dict(r) for r in rows]
|
||||
|
||||
|
||||
@app.post("/api/v1/chat", response_model=ChatResponse, dependencies=[Depends(verify_token)])
|
||||
def chat_endpoint(req: ChatRequest):
|
||||
"""Принимает сообщение, подтягивает историю из БД, выполняет вызовы функций к БД и сохраняет историю."""
|
||||
conn = get_db_connection()
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT history_json FROM chat_sessions WHERE session_id = ?", (req.session_id,))
|
||||
row = cursor.fetchone()
|
||||
|
||||
chat_history = json.loads(row["history_json"]) if row else []
|
||||
|
||||
reply, updated_history = process_chat_message(req.message, chat_history)
|
||||
|
||||
cursor.execute(
|
||||
"INSERT OR REPLACE INTO chat_sessions (session_id, history_json, updated_at) VALUES (?, ?, CURRENT_TIMESTAMP)",
|
||||
(req.session_id, json.dumps(updated_history, ensure_ascii=False))
|
||||
)
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
||||
return ChatResponse(session_id=req.session_id, reply=reply)
|
||||
|
||||
|
||||
@app.get("/health")
|
||||
def health_check():
|
||||
return {"status": "healthy", "service": "scud_context_api", "version": "2.0.0"}
|
||||
# === ANCHOR: RULES_AND_CHAT_ENDPOINTS_END ===
|
||||
@@ -0,0 +1,266 @@
|
||||
<!DOCTYPE html>
|
||||
<html lang="ru">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>SCUD Orion AI — Context Manager</title>
|
||||
<script src="https://cdn.tailwindcss.com"></script>
|
||||
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css">
|
||||
</head>
|
||||
<body class="bg-slate-50 text-slate-800 h-screen flex flex-col font-sans">
|
||||
|
||||
<!-- Хедер -->
|
||||
<header class="bg-white border-b border-slate-200 px-6 py-3 flex justify-between items-center shadow-sm">
|
||||
<div class="flex items-center space-x-3">
|
||||
<div class="bg-indigo-600 text-white p-2 rounded-lg">
|
||||
<i class="fa-solid fa-brain text-lg"></i>
|
||||
</div>
|
||||
<div>
|
||||
<h1 class="text-base font-bold leading-none text-slate-900">SCUD Orion AI</h1>
|
||||
<span class="text-xs text-slate-500">Context & Task Manager</span>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div class="flex items-center space-x-4">
|
||||
<span class="flex items-center text-xs text-emerald-600 font-medium bg-emerald-50 px-2.5 py-1 rounded-full border border-emerald-200">
|
||||
<span class="h-2 w-2 rounded-full bg-emerald-500 mr-1.5"></span> API Online
|
||||
</span>
|
||||
<button onclick="toggleDrawer()" class="bg-indigo-50 hover:bg-indigo-100 text-indigo-700 px-3.5 py-1.5 rounded-lg text-xs font-semibold flex items-center gap-2 transition border border-indigo-200">
|
||||
<i class="fa-solid fa-list-check text-indigo-600"></i>
|
||||
<span>Реестр задач</span>
|
||||
<span id="task-count-badge" class="bg-indigo-600 text-white text-[10px] px-1.5 py-0.2 rounded-full">0</span>
|
||||
</button>
|
||||
</div>
|
||||
</header>
|
||||
|
||||
<!-- Основное окно чата -->
|
||||
<main class="flex-1 flex flex-col max-w-4xl w-full mx-auto bg-white my-4 rounded-xl border border-slate-200 shadow-sm overflow-hidden">
|
||||
<div id="chat-window" class="flex-1 p-6 overflow-y-auto space-y-4 bg-slate-50/50">
|
||||
<div class="bg-white border border-slate-200 rounded-xl p-4 max-w-2xl shadow-sm">
|
||||
<p class="text-xs font-bold text-indigo-600 uppercase tracking-wider mb-1">
|
||||
<i class="fa-solid fa-robot mr-1"></i> ИИ-Ассистент
|
||||
</p>
|
||||
<p class="text-slate-700 text-sm leading-relaxed">
|
||||
Привет! Я подключен к вашей базе задач. Вы можете писать мне в чат или управлять задачами в реестре справа.
|
||||
</p>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div class="p-4 bg-white border-t border-slate-200">
|
||||
<form id="chat-form" onsubmit="sendMessage(event)" class="flex gap-2">
|
||||
<input type="text" id="user-input" placeholder="Команда или вопрос (например: 'покажи все задачи')..."
|
||||
class="flex-1 bg-slate-50 border border-slate-300 text-slate-800 rounded-lg px-4 py-2.5 text-sm focus:outline-none focus:border-indigo-600 focus:bg-white transition">
|
||||
<button type="submit" id="send-btn" class="bg-indigo-600 hover:bg-indigo-700 text-white font-medium px-5 py-2.5 rounded-lg text-sm transition flex items-center gap-2">
|
||||
<span>Отправить</span>
|
||||
<i class="fa-solid fa-paper-plane text-xs"></i>
|
||||
</button>
|
||||
</form>
|
||||
</div>
|
||||
</main>
|
||||
|
||||
<!-- Выезжающий справа реестр задач (Drawer) -->
|
||||
<div id="drawer-backdrop" onclick="toggleDrawer()" class="fixed inset-0 bg-slate-900/30 backdrop-blur-sm hidden transition-opacity z-40"></div>
|
||||
|
||||
<aside id="task-drawer" class="fixed right-0 top-0 h-full w-[420px] bg-white border-l border-slate-200 shadow-2xl transform translate-x-full transition-transform duration-300 ease-in-out z-50 flex flex-col">
|
||||
|
||||
<!-- Заголовок панели -->
|
||||
<div class="p-4 border-b border-slate-200 flex justify-between items-center bg-slate-50">
|
||||
<h2 class="font-bold text-slate-800 flex items-center gap-2 text-sm">
|
||||
<i class="fa-solid fa-list-check text-indigo-600"></i> Реестр задач
|
||||
</h2>
|
||||
<div class="flex items-center gap-2">
|
||||
<button onclick="loadTasks()" class="text-xs text-slate-500 hover:text-indigo-600 transition p-1" title="Обновить">
|
||||
<i class="fa-solid fa-rotate-right"></i>
|
||||
</button>
|
||||
<button onclick="toggleDrawer()" class="text-slate-400 hover:text-slate-700 transition p-1">
|
||||
<i class="fa-solid fa-xmark text-lg"></i>
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<!-- Фильтры по статусу -->
|
||||
<div class="flex border-b border-slate-200 bg-white px-2 pt-2 text-xs font-medium text-slate-500 gap-1">
|
||||
<button onclick="setFilter('ALL')" id="filter-ALL" class="px-3 py-1.5 rounded-t-lg border-b-2 border-indigo-600 text-indigo-600 font-bold">Все</button>
|
||||
<button onclick="setFilter('IN_PROGRESS')" id="filter-IN_PROGRESS" class="px-3 py-1.5 rounded-t-lg border-b-2 border-transparent hover:text-slate-700">В работе</button>
|
||||
<button onclick="setFilter('BACKLOG')" id="filter-BACKLOG" class="px-3 py-1.5 rounded-t-lg border-b-2 border-transparent hover:text-slate-700">Бэклог</button>
|
||||
<button onclick="setFilter('COMPLETED')" id="filter-COMPLETED" class="px-3 py-1.5 rounded-t-lg border-b-2 border-transparent hover:text-slate-700">Завершено</button>
|
||||
</div>
|
||||
|
||||
<!-- Список карточек -->
|
||||
<div id="tasks-container" class="flex-1 overflow-y-auto p-4 space-y-3 bg-slate-50/50">
|
||||
<div class="text-center text-slate-400 py-8 text-xs">Загрузка задач...</div>
|
||||
</div>
|
||||
</aside>
|
||||
|
||||
<script>
|
||||
const API_TOKEN = "scud_secret_token_2026";
|
||||
const SESSION_ID = "web_session_main";
|
||||
let currentFilter = 'ALL';
|
||||
let allTasks = [];
|
||||
|
||||
function toggleDrawer() {
|
||||
const drawer = document.getElementById('task-drawer');
|
||||
const backdrop = document.getElementById('drawer-backdrop');
|
||||
const isHidden = drawer.classList.contains('translate-x-full');
|
||||
if (isHidden) {
|
||||
drawer.classList.remove('translate-x-full');
|
||||
backdrop.classList.remove('hidden');
|
||||
} else {
|
||||
drawer.classList.add('translate-x-full');
|
||||
backdrop.classList.add('hidden');
|
||||
}
|
||||
}
|
||||
|
||||
function setFilter(status) {
|
||||
currentFilter = status;
|
||||
['ALL', 'IN_PROGRESS', 'BACKLOG', 'COMPLETED'].forEach(f => {
|
||||
const btn = document.getElementById(`filter-${f}`);
|
||||
if (f === status) {
|
||||
btn.className = "px-3 py-1.5 rounded-t-lg border-b-2 border-indigo-600 text-indigo-600 font-bold";
|
||||
} else {
|
||||
btn.className = "px-3 py-1.5 rounded-t-lg border-b-2 border-transparent hover:text-slate-700";
|
||||
}
|
||||
});
|
||||
renderTasks();
|
||||
}
|
||||
|
||||
async function loadTasks() {
|
||||
try {
|
||||
const res = await fetch('/api/v1/tasks', {
|
||||
headers: { 'Authorization': `Bearer ${API_TOKEN}` }
|
||||
});
|
||||
allTasks = await res.json();
|
||||
document.getElementById('task-count-badge').innerText = allTasks.length;
|
||||
renderTasks();
|
||||
} catch (err) {
|
||||
document.getElementById('tasks-container').innerHTML = `<div class="text-red-500 text-xs py-4 text-center">Ошибка загрузки задач</div>`;
|
||||
}
|
||||
}
|
||||
|
||||
async function quickUpdateStatus(taskId, newStatus) {
|
||||
try {
|
||||
await fetch(`/api/v1/tasks/${taskId}/complete`, {
|
||||
method: 'POST',
|
||||
headers: { 'Authorization': `Bearer ${API_TOKEN}` }
|
||||
});
|
||||
loadTasks();
|
||||
} catch (e) {
|
||||
console.error(e);
|
||||
}
|
||||
}
|
||||
|
||||
function renderTasks() {
|
||||
const container = document.getElementById('tasks-container');
|
||||
const filtered = allTasks.filter(t => currentFilter === 'ALL' || t.status === currentFilter);
|
||||
|
||||
if (filtered.length === 0) {
|
||||
container.innerHTML = `<div class="text-slate-400 text-xs py-8 text-center">Нет задач с выбранным фильтром</div>`;
|
||||
return;
|
||||
}
|
||||
|
||||
container.innerHTML = filtered.map(t => {
|
||||
let statusBadge = 'bg-slate-100 text-slate-600 border-slate-200';
|
||||
let cardBg = 'bg-white';
|
||||
if (t.status === 'COMPLETED') {
|
||||
statusBadge = 'bg-emerald-50 text-emerald-700 border-emerald-300 font-semibold';
|
||||
cardBg = 'bg-emerald-50/20';
|
||||
} else if (t.status === 'IN_PROGRESS') {
|
||||
statusBadge = 'bg-amber-50 text-amber-700 border-amber-300 font-bold';
|
||||
cardBg = 'bg-amber-50/20 border-amber-200';
|
||||
}
|
||||
|
||||
let priorityBadge = 'text-slate-500 bg-slate-100 border-slate-200';
|
||||
if (t.priority === 'HIGH') priorityBadge = 'text-red-700 bg-red-50 border-red-200 font-bold';
|
||||
|
||||
let dueDateHtml = '';
|
||||
if (t.due_date) {
|
||||
dueDateHtml = `
|
||||
<div class="mt-2 text-[11px] text-amber-800 bg-amber-50 border border-amber-200 px-2 py-0.5 rounded-md flex items-center gap-1.5 w-fit font-medium">
|
||||
<i class="fa-solid fa-clock text-amber-600"></i>
|
||||
<span>Срок: ${t.due_date}</span>
|
||||
</div>
|
||||
`;
|
||||
}
|
||||
|
||||
return `
|
||||
<div class="${cardBg} border border-slate-200 rounded-xl p-3.5 shadow-sm hover:shadow-md transition">
|
||||
<div class="flex justify-between items-center mb-1.5">
|
||||
<div class="flex items-center gap-1.5">
|
||||
<span class="font-mono text-xs font-bold text-slate-900 bg-slate-100 px-2 py-0.5 rounded border border-slate-200">${t.task_id}</span>
|
||||
<span class="text-[10px] uppercase px-1.5 py-0.5 rounded border ${priorityBadge}">${t.priority || 'HIGH'}</span>
|
||||
</div>
|
||||
<span class="text-[10px] uppercase px-2 py-0.5 rounded border ${statusBadge}">${t.status}</span>
|
||||
</div>
|
||||
|
||||
<h3 class="text-xs font-semibold text-slate-800 mb-1 leading-snug">${t.title}</h3>
|
||||
|
||||
<div class="text-[10px] text-slate-400 font-mono flex items-center gap-1">
|
||||
<i class="fa-solid fa-folder-closed text-slate-300"></i>
|
||||
<span>${t.module}</span>
|
||||
</div>
|
||||
|
||||
${dueDateHtml}
|
||||
</div>
|
||||
`;
|
||||
}).join('');
|
||||
}
|
||||
|
||||
async function sendMessage(e) {
|
||||
e.preventDefault();
|
||||
const input = document.getElementById('user-input');
|
||||
const chatWindow = document.getElementById('chat-window');
|
||||
const sendBtn = document.getElementById('send-btn');
|
||||
const text = input.value.trim();
|
||||
|
||||
if (!text) return;
|
||||
|
||||
chatWindow.innerHTML += `
|
||||
<div class="flex justify-end">
|
||||
<div class="bg-indigo-600 text-white rounded-xl px-4 py-2.5 max-w-2xl text-sm shadow-sm">
|
||||
${text}
|
||||
</div>
|
||||
</div>
|
||||
`;
|
||||
input.value = '';
|
||||
chatWindow.scrollTop = chatWindow.scrollHeight;
|
||||
|
||||
sendBtn.disabled = true;
|
||||
sendBtn.classList.add('opacity-50');
|
||||
|
||||
try {
|
||||
const res = await fetch('/api/v1/chat', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'Authorization': `Bearer ${API_TOKEN}`
|
||||
},
|
||||
body: JSON.stringify({ session_id: SESSION_ID, message: text })
|
||||
});
|
||||
|
||||
const data = await res.json();
|
||||
|
||||
chatWindow.innerHTML += `
|
||||
<div class="bg-white border border-slate-200 rounded-xl p-4 max-w-2xl shadow-sm">
|
||||
<p class="text-xs font-bold text-indigo-600 uppercase tracking-wider mb-1"><i class="fa-solid fa-robot mr-1"></i> ИИ-Ассистент</p>
|
||||
<p class="text-slate-700 text-sm whitespace-pre-wrap leading-relaxed">${data.reply}</p>
|
||||
</div>
|
||||
`;
|
||||
chatWindow.scrollTop = chatWindow.scrollHeight;
|
||||
loadTasks();
|
||||
|
||||
} catch (err) {
|
||||
chatWindow.innerHTML += `
|
||||
<div class="bg-red-50 border border-red-200 rounded-xl p-4 max-w-2xl text-red-700 text-sm">
|
||||
Ошибка связи с сервером API.
|
||||
</div>
|
||||
`;
|
||||
} finally {
|
||||
sendBtn.disabled = false;
|
||||
sendBtn.classList.remove('opacity-50');
|
||||
}
|
||||
}
|
||||
|
||||
loadTasks();
|
||||
</script>
|
||||
</body>
|
||||
</html>
|
||||
Reference in New Issue
Block a user