panda
067880bf2e
新增: - backend/logger.py — 集中日志模块 (JSON格式 + trace_id + 独立llm.log) - @log_node / @_log_route 装饰器覆盖17个节点和8个路由 改进: - backend/llm.py — _LLMLoggingWrapper 自动记录LLM输入输出 - backend/llm.py — API Key优先读ANTHROPIC_API_KEY，模型名改为MiniMax-M2.7 - backend/llm.py — get_llm() 新增caller参数标识调用来源 - backend/validation.py — 新增验证结果/连接失败日志 - backend/session.py — 新增会话创建/删除日志 - app.py — 新增用户交互日志 (输入/执行/异常/会话操作) - app.py — 提前导入torchvision抑制transformers懒加载报错 - .env.example — 新增LOG_DIR/LOG_LEVEL/ANTHROPIC_API_KEY等配置项 - .gitignore — 新增logs/和db/忽略规则 文档: - ROADMAP.md — 新增阶段四: 可观测性 - README.md — 补充日志架构/LLM配置/项目结构 - CLAUDE.md — 同步最新配置/日志/MAX_RETRY(3) - CODE_GUIDE.md — 新增第15章日志系统，更新架构图/LLM/配置
119 lines
3.4 KiB
Python
119 lines
3.4 KiB
Python
"""多会话持久化管理模块。
|
|
|
|
每个会话对应一个独立的 JSON 文件存储在 ./sessions/ 目录下。
|
|
"""
|
|
|
|
import json
|
|
import os
|
|
import uuid
|
|
from datetime import datetime, timezone
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
from backend.logger import get_logger
|
|
|
|
load_dotenv()
|
|
|
|
_session_log = get_logger("session")
|
|
|
|
SESSIONS_DIR = Path(os.getenv("SESSIONS_DIR", "./sessions"))
|
|
|
|
|
|
def _ensure_dir():
|
|
SESSIONS_DIR.mkdir(parents=True, exist_ok=True)
|
|
|
|
|
|
def _session_path(session_id: str) -> Path:
|
|
return SESSIONS_DIR / f"{session_id}.json"
|
|
|
|
|
|
def generate_session_id() -> str:
|
|
return uuid.uuid4().hex[:12]
|
|
|
|
|
|
def create_session(name: str = "", agent_state: Optional[dict] = None) -> dict:
|
|
"""创建新会话，返回会话元数据。"""
|
|
_ensure_dir()
|
|
sid = generate_session_id()
|
|
now = _now_iso()
|
|
data = {
|
|
"session_id": sid,
|
|
"session_name": name or f"新建报表 {now[:10]}",
|
|
"created_at": now,
|
|
"updated_at": now,
|
|
"agent_state": agent_state or {},
|
|
}
|
|
with open(_session_path(sid), "w", encoding="utf-8") as f:
|
|
json.dump(data, f, ensure_ascii=False, indent=2)
|
|
_session_log.info("创建会话", extra={"session_id": sid, "session_name": data["session_name"]})
|
|
return data
|
|
|
|
|
|
def load_session(session_id: str) -> Optional[dict]:
|
|
"""按 ID 加载会话数据。未找到则返回 None。"""
|
|
_ensure_dir()
|
|
fp = _session_path(session_id)
|
|
if not fp.exists():
|
|
return None
|
|
with open(fp, "r", encoding="utf-8") as f:
|
|
return json.load(f)
|
|
|
|
|
|
def save_session(session_id: str, agent_state: dict, session_name: str = ""):
|
|
"""将会话状态保存（更新）至磁盘。"""
|
|
_ensure_dir()
|
|
fp = _session_path(session_id)
|
|
data = {}
|
|
if fp.exists():
|
|
with open(fp, "r", encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
|
|
data["session_id"] = session_id
|
|
if session_name:
|
|
data["session_name"] = session_name
|
|
if not data.get("session_name"):
|
|
data["session_name"] = f"报表 {data.get('created_at', _now_iso())[:10]}"
|
|
data["updated_at"] = _now_iso()
|
|
if not data.get("created_at"):
|
|
data["created_at"] = data["updated_at"]
|
|
data["agent_state"] = agent_state
|
|
|
|
with open(fp, "w", encoding="utf-8") as f:
|
|
json.dump(data, f, ensure_ascii=False, indent=2)
|
|
|
|
|
|
def list_all_sessions() -> list[dict]:
|
|
"""列出所有历史会话（仅摘要，不含完整 agent_state）。"""
|
|
_ensure_dir()
|
|
sessions = []
|
|
for fp in sorted(SESSIONS_DIR.glob("*.json"), key=os.path.getmtime, reverse=True):
|
|
try:
|
|
with open(fp, "r", encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
sessions.append({
|
|
"session_id": data.get("session_id", fp.stem),
|
|
"session_name": data.get("session_name", fp.stem),
|
|
"created_at": data.get("created_at", ""),
|
|
"updated_at": data.get("updated_at", ""),
|
|
})
|
|
except (json.JSONDecodeError, KeyError):
|
|
continue
|
|
return sessions
|
|
|
|
|
|
def delete_session(session_id: str) -> bool:
|
|
"""按 ID 删除会话文件。"""
|
|
_ensure_dir()
|
|
fp = _session_path(session_id)
|
|
if fp.exists():
|
|
fp.unlink()
|
|
_session_log.info("删除会话", extra={"session_id": session_id})
|
|
return True
|
|
return False
|
|
|
|
|
|
def _now_iso() -> str:
|
|
return datetime.now(timezone.utc).isoformat()
|