JOJO
478d24c4dd
本次变更是对版本控制功能的大范围重构，核心目标：
- 默认使用浅备份（只追踪 write_file/edit_file 实际修改的文件），避免大工作区首次初始化耗时十几秒；
- 保留可选的完全备份（git write-tree），不污染 git log；
- 回溯时统一选择范围（仅对话 / 对话+工作区）和模式（覆盖 / 复制）；
- 修复复制对话、侧边栏同步、新建对话延迟、diff 统计与渲染等连锁问题。
后端改动：
- modules/shallow_versioning.py（新增）
- 实现 ShallowVersioningManager：track_edit、make_snapshot、rewind、list_snapshots。
- 基于 difflib.SequenceMatcher 计算真实 insertions/deletions，replace 场景正确显示 +N/-M。
- diff 详情使用结构化 add/remove/context 行，不再解析 unified_diff 文本。
- modules/versioning_manager.py
- 完全备份从 git commit 改为 git write-tree，字段 commit/parent_commit 改为 tree_hash/parent_tree_hash。
- 恢复改用 read-tree + checkout-index；create_checkpoint 支持 diff_summary 参数。
- get_checkpoint_detail 保留 API 字段名 last_commit/restored_commit 以兼容前端。
- core/main_terminal_parts/tools_execution.py
- write_file/edit_file 在执行前调用浅备份 track_edit，保留编辑前状态。
- server/chat_flow_task_main.py
- 消息轮次结束时根据 backup_mode 选择完全备份 create_checkpoint 或浅备份 make_snapshot。
- 浅备份模式下同时写入 ConversationVersioningManager 检查点行，让前端能统一展示。
- server/conversation.py
- restore/detail/list API 支持 shallow 模式；list/detail 对浅备份实时重新计算统计。
- 创建对话时读取 personalization 的 versioning_backup_mode，shallow 模式固定 tracking_mode 为 conversation_only。
- core/web_terminal.py
- _ensure_conversation_versioning_enabled 读取默认备份模式并写入 versioning meta。
- modules/file_manager/crud_mixin.py / replace_mixin.py
- write_file / edit_file 返回 original_file / new_file / replacement_details（真实旧行/新行）。
- modules/personalization_manager.py
- 新增 versioning_enabled_by_default、versioning_backup_mode（shallow/full）。
- utils/perf_log.py（新增）
- 创建新对话全链路性能日志，用于定位延迟瓶颈。
- utils/conversation_manager/{index,list_search,crud}_mixin.py
- 加性能日志；index_mixin 主动检查新增对话文件，修复侧边栏不同步。
前端改动：
- static/src/components/overlay/VersioningDialog.vue
- 去掉顶部管理范围下拉；底部新增回溯范围与回溯模式选择。
- diff 详情改为 div.diff-line 网格布局（marker + content），支持横向滚动，状态显示缩写 A/M/D。
- 修复 pre 标签空白黑行、容器被挤窄导致折行等问题。
- static/src/app/methods/versioning.ts
- 移除 mismatch 弹窗逻辑；支持 copy 模式；selectVersioningCheckpoint 增强错误捕获。
- static/src/components/personalization/PersonalizationDrawer.vue
- 「工作区与权限」新增版本控制默认开关与备份方式选择。
- static/src/stores/personalization.ts
- 新增 versioning_enabled_by_default、versioning_backup_mode 字段。
- static/src/components/chat/actions/ToolAction.vue / toolRenderers.ts
- edit_file / write_file 工具展示基于 result.details / result.new_file 的真实替换结果。
- static/src/app/state.ts / watchers.ts / App.vue
- 新增 versioningInitializingBackupToastId，完全备份初始化时显示 toast。
- static/src/app/methods/conversation/action.ts / message/send.ts
- host + 完全备份 + 默认开启时，创建新对话/从 /new 发送首条消息显示「正在初始化备份」toast。
验证：
- /Library/Developer/CommandLineTools/Library/Frameworks/Python3.framework/Versions/3.9/bin/python3.9 -m unittest test.test_server_refactor_smoke 通过。
- npm run build 通过。
- 宿主机与 docker 模式新建对话、编辑文件、回溯、复制对话均验证通过。
2153 lines
91 KiB
Python
2153 lines
91 KiB
Python
from __future__ import annotations
|
||
import sys, os
|
||
PROJECT_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
|
||
if PROJECT_ROOT not in sys.path:
|
||
sys.path.insert(0, PROJECT_ROOT)
|
||
|
||
import asyncio, json, time, re, os, shutil
|
||
from datetime import datetime, timedelta
|
||
from pathlib import Path
|
||
from collections import defaultdict, Counter, deque
|
||
from io import BytesIO
|
||
from typing import Dict, Any, Optional, List, Tuple
|
||
|
||
from flask import Blueprint, request, jsonify, session, send_file
|
||
from flask_socketio import emit
|
||
from werkzeug.utils import secure_filename
|
||
import zipfile
|
||
|
||
from config import (
|
||
OUTPUT_FORMATS,
|
||
AUTO_FIX_TOOL_CALL,
|
||
AUTO_FIX_MAX_ATTEMPTS,
|
||
MAX_ITERATIONS_PER_TASK,
|
||
MAX_CONSECUTIVE_SAME_TOOL,
|
||
MAX_TOTAL_TOOL_CALLS,
|
||
TOOL_CALL_COOLDOWN,
|
||
MAX_UPLOAD_SIZE,
|
||
DEFAULT_CONVERSATIONS_LIMIT,
|
||
MAX_CONVERSATIONS_LIMIT,
|
||
CONVERSATIONS_DIR,
|
||
DEFAULT_RESPONSE_MAX_TOKENS,
|
||
DEFAULT_PROJECT_PATH,
|
||
LOGS_DIR,
|
||
AGENT_VERSION,
|
||
THINKING_FAST_INTERVAL,
|
||
PROJECT_MAX_STORAGE_MB,
|
||
PROJECT_MAX_STORAGE_BYTES,
|
||
UPLOAD_SCAN_LOG_SUBDIR,
|
||
)
|
||
from modules.personalization_manager import (
|
||
load_personalization_config,
|
||
save_personalization_config,
|
||
THINKING_INTERVAL_MIN,
|
||
THINKING_INTERVAL_MAX,
|
||
)
|
||
from modules.upload_security import UploadSecurityError
|
||
from modules.user_manager import UserWorkspace
|
||
from modules.usage_tracker import QUOTA_DEFAULTS
|
||
from modules.sub_agent import TERMINAL_STATUSES
|
||
from modules.versioning_manager import ConversationVersioningManager, VersioningError
|
||
from modules.shallow_versioning import ShallowVersioningManager
|
||
from utils.perf_log import perf_log, PerfTimer
|
||
from core.web_terminal import WebTerminal
|
||
from utils.tool_result_formatter import format_tool_result_for_context
|
||
from utils.conversation_manager import ConversationManager
|
||
from utils.api_client import DeepSeekClient
|
||
|
||
from .auth_helpers import api_login_required, resolve_admin_policy, get_current_user_record, get_current_username
|
||
from .context import with_terminal, get_terminal_for_sid, get_gui_manager, get_upload_guard, build_upload_error_response, ensure_conversation_loaded, reset_system_state, get_user_resources, get_or_create_usage_tracker
|
||
from .utils_common import (
|
||
build_review_lines,
|
||
debug_log,
|
||
_sanitize_filename_component,
|
||
log_backend_chunk,
|
||
log_frontend_chunk,
|
||
log_streaming_debug_entry,
|
||
brief_log,
|
||
DEBUG_LOG_FILE,
|
||
CHUNK_BACKEND_LOG_FILE,
|
||
CHUNK_FRONTEND_LOG_FILE,
|
||
STREAMING_DEBUG_LOG_FILE,
|
||
)
|
||
from .extensions import socketio
|
||
from .state import (
|
||
RECENT_UPLOAD_EVENT_LIMIT,
|
||
RECENT_UPLOAD_FEED_LIMIT,
|
||
user_manager,
|
||
container_manager,
|
||
get_last_active_ts,
|
||
)
|
||
from .usage import record_user_activity
|
||
from .conversation_stats import (
|
||
build_admin_dashboard_snapshot,
|
||
compute_workspace_storage,
|
||
collect_user_token_statistics,
|
||
collect_upload_events,
|
||
summarize_upload_events,
|
||
)
|
||
from .deep_compression import run_deep_compression
|
||
|
||
conversation_bp = Blueprint('conversation', __name__)
|
||
|
||
|
||
def _terminate_running_sub_agents(terminal: WebTerminal, reason: str = "") -> int:
|
||
"""切换/新建对话时，强制终止当前对话仍在运行的子智能体，并记录系统消息。"""
|
||
manager = getattr(terminal, "sub_agent_manager", None)
|
||
if not manager:
|
||
return 0
|
||
current_conv_id = getattr(getattr(terminal, "context_manager", None), "current_conversation_id", None)
|
||
if not current_conv_id:
|
||
return 0
|
||
try:
|
||
manager.reconcile_task_states(conversation_id=current_conv_id)
|
||
except Exception:
|
||
pass
|
||
running_tasks = [
|
||
task for task in manager.tasks.values()
|
||
if task.get("status") not in TERMINAL_STATUSES.union({"terminated"})
|
||
and task.get("run_in_background")
|
||
and task.get("conversation_id") == current_conv_id
|
||
]
|
||
if not running_tasks:
|
||
return 0
|
||
stopped_count = 0
|
||
for task in running_tasks:
|
||
task_id = task.get("task_id")
|
||
manager.terminate_sub_agent(task_id=task_id)
|
||
stopped_count += 1
|
||
return stopped_count
|
||
|
||
|
||
def _cancel_running_tasks(username: str, workspace_id: str, timeout_seconds: float = 4.0) -> Tuple[int, bool]:
|
||
"""取消当前工作区运行中的主任务，并等待其停止，避免切换对话后串写。"""
|
||
try:
|
||
from .tasks import task_manager
|
||
except Exception as exc:
|
||
debug_log(f"[TaskCancel] 导入 task_manager 失败: {exc}")
|
||
return 0, True
|
||
|
||
active_statuses = {"pending", "running", "cancel_requested"}
|
||
|
||
def _active_tasks():
|
||
try:
|
||
recs = task_manager.list_tasks(username, workspace_id)
|
||
except Exception:
|
||
recs = task_manager.list_tasks(username)
|
||
return [rec for rec in recs if getattr(rec, "status", None) in active_statuses]
|
||
|
||
running = _active_tasks()
|
||
if not running:
|
||
return 0, True
|
||
|
||
canceled = 0
|
||
for rec in running:
|
||
task_id = getattr(rec, "task_id", None)
|
||
if task_id and task_manager.cancel_task(username, task_id):
|
||
canceled += 1
|
||
|
||
deadline = time.time() + max(timeout_seconds, 0.5)
|
||
while time.time() < deadline:
|
||
if not _active_tasks():
|
||
return canceled, True
|
||
time.sleep(0.1)
|
||
|
||
return canceled, False
|
||
|
||
|
||
def _get_active_workspace_task(username: str, workspace_id: str):
|
||
"""返回当前工作区仍在运行/停止中的主任务。用于避免视图导航改写运行任务的 terminal 上下文。"""
|
||
try:
|
||
from .tasks import task_manager
|
||
active_statuses = {"pending", "running", "cancel_requested"}
|
||
tasks = [
|
||
rec for rec in task_manager.list_tasks(username, workspace_id)
|
||
if getattr(rec, "status", None) in active_statuses
|
||
]
|
||
tasks.sort(key=lambda rec: getattr(rec, "created_at", 0), reverse=True)
|
||
return tasks[0] if tasks else None
|
||
except Exception as exc:
|
||
debug_log(f"[ConversationSafeNav] 查询运行任务失败: {exc}")
|
||
return None
|
||
|
||
|
||
def _build_safe_load_result(terminal: WebTerminal, conversation_id: str) -> Dict[str, Any]:
|
||
"""只读取对话元数据，不调用 terminal.load_conversation，避免修改运行任务正在使用的上下文。"""
|
||
normalized_id = _normalize_conv_id(conversation_id)
|
||
cm = getattr(getattr(terminal, "context_manager", None), "conversation_manager", None)
|
||
data = cm.load_conversation(normalized_id) if cm else None
|
||
if not data:
|
||
return {
|
||
"success": False,
|
||
"error": "对话不存在或加载失败",
|
||
"message": f"对话加载失败: {normalized_id}",
|
||
}
|
||
meta = data.get("metadata", {}) or {}
|
||
run_mode = meta.get("run_mode") or getattr(terminal, "run_mode", "fast")
|
||
thinking_mode = bool(meta.get("thinking_mode", run_mode != "fast"))
|
||
return {
|
||
"success": True,
|
||
"conversation_id": normalized_id,
|
||
"title": data.get("title", "未知对话"),
|
||
"messages_count": len(data.get("messages", []) or []),
|
||
"run_mode": run_mode,
|
||
"thinking_mode": thinking_mode,
|
||
"model_key": meta.get("model_key") or getattr(terminal, "model_key", None),
|
||
"message": f"对话已加载: {normalized_id}",
|
||
"safe_navigation": True,
|
||
}
|
||
|
||
|
||
def _normalize_conv_id(conversation_id: str) -> str:
|
||
conv = (conversation_id or "").strip()
|
||
if not conv:
|
||
return conv
|
||
return conv if conv.startswith("conv_") else f"conv_{conv}"
|
||
|
||
|
||
def _normalize_versioning_tracking_mode(value: Optional[str]) -> str:
|
||
return ConversationVersioningManager.normalize_tracking_mode(value)
|
||
|
||
|
||
def _can_use_versioning_scope(username: str, tracking_mode: str) -> bool:
|
||
normalized_tracking_mode = _normalize_versioning_tracking_mode(tracking_mode)
|
||
return _is_host_mode_request(username) or (
|
||
normalized_tracking_mode == ConversationVersioningManager.TRACKING_MODE_CONVERSATION_ONLY
|
||
)
|
||
|
||
|
||
def _is_host_mode_request(username: str) -> bool:
|
||
return bool(session.get("host_mode")) or (username == "host")
|
||
|
||
|
||
def _get_conv_versioning_manager(workspace: UserWorkspace, conversation_id: str) -> ConversationVersioningManager:
|
||
normalized = _normalize_conv_id(conversation_id)
|
||
return ConversationVersioningManager(
|
||
project_path=workspace.project_path,
|
||
data_dir=workspace.data_dir,
|
||
conversation_id=normalized,
|
||
)
|
||
|
||
|
||
def _sanitize_scope_component(value: Any, default: str = "default") -> str:
|
||
raw = str(value or "").strip()
|
||
if not raw:
|
||
return default
|
||
sanitized = re.sub(r"[^0-9A-Za-z._-]+", "_", raw).strip("._-")
|
||
if not sanitized:
|
||
sanitized = default
|
||
return sanitized[:120]
|
||
|
||
|
||
def _resolve_input_draft_path(workspace: UserWorkspace, username: str) -> Tuple[Path, str]:
|
||
base_dir = Path(workspace.data_dir).expanduser().resolve() / "composer_drafts"
|
||
if _is_host_mode_request(username):
|
||
workspace_id = session.get("host_workspace_id") or session.get("workspace_id") or "default"
|
||
scope = f"host_workspace:{workspace_id}"
|
||
filename = f"{_sanitize_scope_component(workspace_id)}.json"
|
||
return (base_dir / "host" / filename).resolve(), scope
|
||
safe_user = _sanitize_scope_component(username or "user", default="user")
|
||
workspace_id = getattr(workspace, "workspace_id", None) or session.get("workspace_id") or "default"
|
||
safe_workspace = _sanitize_scope_component(workspace_id)
|
||
scope = f"user:{safe_user}:project:{safe_workspace}"
|
||
return (base_dir / "docker" / f"{safe_workspace}.json").resolve(), scope
|
||
|
||
|
||
def _read_input_draft_payload(path: Path) -> Dict[str, Any]:
|
||
if not path.exists():
|
||
return {}
|
||
try:
|
||
raw = json.loads(path.read_text(encoding="utf-8")) or {}
|
||
return raw if isinstance(raw, dict) else {}
|
||
except Exception:
|
||
return {}
|
||
|
||
|
||
def _atomic_write_input_draft(path: Path, payload: Dict[str, Any]) -> None:
|
||
path.parent.mkdir(parents=True, exist_ok=True)
|
||
tmp = path.with_suffix(path.suffix + ".tmp")
|
||
tmp.write_text(json.dumps(payload, ensure_ascii=False, indent=2), encoding="utf-8")
|
||
tmp.replace(path)
|
||
|
||
|
||
def _get_conversation_versioning_meta(terminal: WebTerminal, conversation_id: str) -> Dict[str, Any]:
|
||
normalized = _normalize_conv_id(conversation_id)
|
||
data = terminal.context_manager.conversation_manager.load_conversation(normalized) or {}
|
||
meta = data.get("metadata") or {}
|
||
versioning = meta.get("versioning") or {}
|
||
if not isinstance(versioning, dict):
|
||
versioning = {}
|
||
enabled = bool(versioning.get("enabled", False))
|
||
tracking_mode = _normalize_versioning_tracking_mode(versioning.get("tracking_mode"))
|
||
mode = "overwrite"
|
||
return {
|
||
"enabled": enabled,
|
||
"mode": mode,
|
||
"tracking_mode": tracking_mode,
|
||
"conversation_id": normalized,
|
||
"metadata": meta,
|
||
}
|
||
|
||
|
||
def _ensure_conversation_versioning_enabled(
|
||
terminal: WebTerminal,
|
||
workspace: UserWorkspace,
|
||
conversation_id: str,
|
||
tracking_mode: Optional[str] = None,
|
||
) -> None:
|
||
"""为指定对话启用版本控制并创建初始 checkpoint（用于默认开启场景）。"""
|
||
normalized_id = _normalize_conv_id(conversation_id)
|
||
host_mode = _is_host_mode_request(get_current_username())
|
||
try:
|
||
prefs = load_personalization_config(workspace.data_dir)
|
||
except Exception:
|
||
prefs = {}
|
||
backup_mode = str(prefs.get("versioning_backup_mode") or "shallow").strip().lower()
|
||
backup_mode = "full" if backup_mode == "full" else "shallow"
|
||
if backup_mode == "shallow":
|
||
default_mode = ConversationVersioningManager.TRACKING_MODE_CONVERSATION_ONLY
|
||
elif host_mode:
|
||
default_mode = ConversationVersioningManager.TRACKING_MODE_WORKSPACE_AND_CONVERSATION
|
||
else:
|
||
default_mode = ConversationVersioningManager.TRACKING_MODE_CONVERSATION_ONLY
|
||
manager = _get_conv_versioning_manager(workspace, normalized_id)
|
||
normalized_tracking_mode = _normalize_versioning_tracking_mode(tracking_mode or default_mode)
|
||
meta = manager.set_enabled(enabled=True, mode="overwrite", tracking_mode=normalized_tracking_mode)
|
||
conv_data = terminal.context_manager.conversation_manager.load_conversation(normalized_id) or {}
|
||
snapshot_payload = {
|
||
"conversation_id": normalized_id,
|
||
"title": conv_data.get("title"),
|
||
"metadata": conv_data.get("metadata") or {},
|
||
"messages": conv_data.get("messages") or [],
|
||
"message_index": -1,
|
||
"run_status": "initial",
|
||
}
|
||
init_result = manager.ensure_initial_checkpoint(
|
||
workspace_path=str(workspace.project_path),
|
||
conversation_snapshot=snapshot_payload,
|
||
tracking_mode=normalized_tracking_mode,
|
||
)
|
||
init_row = init_result.get("row") or {}
|
||
if init_row.get("tree_hash"):
|
||
meta["last_tree_hash"] = init_row.get("tree_hash")
|
||
_update_conversation_versioning_meta(
|
||
terminal,
|
||
normalized_id,
|
||
enabled=True,
|
||
mode="overwrite",
|
||
tracking_mode=normalized_tracking_mode,
|
||
backup_mode=backup_mode,
|
||
last_commit=meta.get("last_commit"),
|
||
last_input_seq=int(meta.get("last_input_seq") or 0),
|
||
)
|
||
|
||
|
||
def _update_conversation_versioning_meta(
|
||
terminal: WebTerminal,
|
||
conversation_id: str,
|
||
*,
|
||
enabled: bool,
|
||
mode: str,
|
||
tracking_mode: Optional[str] = None,
|
||
backup_mode: Optional[str] = None,
|
||
last_commit: Optional[str] = None,
|
||
last_input_seq: Optional[int] = None,
|
||
) -> bool:
|
||
normalized = _normalize_conv_id(conversation_id)
|
||
normalized_tracking_mode = _normalize_versioning_tracking_mode(tracking_mode)
|
||
payload: Dict[str, Any] = {
|
||
"versioning": {
|
||
"enabled": bool(enabled),
|
||
"mode": "overwrite",
|
||
"tracking_mode": normalized_tracking_mode,
|
||
"updated_at": datetime.now().isoformat(),
|
||
}
|
||
}
|
||
if backup_mode is not None:
|
||
payload["versioning"]["backup_mode"] = "full" if backup_mode == "full" else "shallow"
|
||
if last_commit is not None:
|
||
payload["versioning"]["last_commit"] = last_commit
|
||
if last_input_seq is not None:
|
||
payload["versioning"]["last_input_seq"] = int(last_input_seq)
|
||
return terminal.context_manager.conversation_manager.update_conversation_metadata(normalized, payload)
|
||
|
||
|
||
@conversation_bp.route('/api/input-draft', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def get_input_draft(terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
del terminal
|
||
try:
|
||
path, scope = _resolve_input_draft_path(workspace, username)
|
||
payload = _read_input_draft_payload(path)
|
||
content = payload.get("content")
|
||
updated_at = payload.get("updated_at")
|
||
skill_refs = payload.get("skill_refs")
|
||
editor_json = payload.get("editor_json")
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"content": content if isinstance(content, str) else "",
|
||
"skill_refs": skill_refs if isinstance(skill_refs, list) else [],
|
||
"editor_json": editor_json if isinstance(editor_json, dict) else None,
|
||
"updated_at": str(updated_at or ""),
|
||
"scope": scope,
|
||
}
|
||
})
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": f"读取输入草稿失败: {exc}"}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/input-draft', methods=['POST', 'PUT'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def upsert_input_draft(terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
del terminal
|
||
try:
|
||
body = request.get_json(silent=True) or {}
|
||
content = body.get("content") if isinstance(body, dict) else ""
|
||
if content is None:
|
||
content = ""
|
||
if not isinstance(content, str):
|
||
content = str(content)
|
||
if len(content) > 40000:
|
||
return jsonify({"success": False, "error": "输入草稿过长（最大 40000 字符）"}), 400
|
||
raw_skill_refs = body.get("skill_refs") if isinstance(body, dict) else []
|
||
skill_refs = []
|
||
if isinstance(raw_skill_refs, list):
|
||
for item in raw_skill_refs[:100]:
|
||
if not isinstance(item, dict):
|
||
continue
|
||
name = str(item.get("name") or "")[:200]
|
||
description = str(item.get("description") or "")[:1000]
|
||
path_value = str(item.get("path") or "")[:4000]
|
||
if name and path_value:
|
||
skill_refs.append({
|
||
"name": name,
|
||
"description": description,
|
||
"path": path_value,
|
||
})
|
||
editor_json = body.get("editor_json") if isinstance(body, dict) else None
|
||
if editor_json is not None and not isinstance(editor_json, dict):
|
||
editor_json = None
|
||
|
||
path, scope = _resolve_input_draft_path(workspace, username)
|
||
if content == "":
|
||
if path.exists():
|
||
try:
|
||
path.unlink()
|
||
except Exception:
|
||
pass
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"saved": True,
|
||
"cleared": True,
|
||
"scope": scope,
|
||
"updated_at": datetime.now().isoformat(),
|
||
}
|
||
})
|
||
|
||
payload = {
|
||
"content": content,
|
||
"skill_refs": skill_refs,
|
||
"editor_json": editor_json,
|
||
"updated_at": datetime.now().isoformat(),
|
||
"scope": scope,
|
||
}
|
||
_atomic_write_input_draft(path, payload)
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"saved": True,
|
||
"cleared": False,
|
||
"scope": scope,
|
||
"updated_at": payload["updated_at"],
|
||
}
|
||
})
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": f"保存输入草稿失败: {exc}"}), 500
|
||
|
||
|
||
# === 背景生成对话标题（从 app_legacy 拆分） ===
|
||
@conversation_bp.route('/api/conversations', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def get_conversations(terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""获取对话列表"""
|
||
try:
|
||
# 获取查询参数
|
||
limit = request.args.get('limit', 20, type=int)
|
||
offset = request.args.get('offset', 0, type=int)
|
||
non_empty = request.args.get('non_empty', '0') in ('1', 'true', 'True')
|
||
|
||
# 限制参数范围
|
||
limit = max(1, min(limit, 10000)) # 限制在1-10000之间
|
||
offset = max(0, offset)
|
||
|
||
result = terminal.get_conversations_list(limit=limit, offset=offset, non_empty=non_empty)
|
||
|
||
if result["success"]:
|
||
return jsonify({
|
||
"success": True,
|
||
"data": result["data"]
|
||
})
|
||
else:
|
||
return jsonify({
|
||
"success": False,
|
||
"error": result.get("error", "Unknown error"),
|
||
"message": result.get("message", "获取对话列表失败")
|
||
}), 500
|
||
|
||
except Exception as e:
|
||
print(f"[API] 获取对话列表错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e),
|
||
"message": "获取对话列表时发生异常"
|
||
}), 500
|
||
|
||
@conversation_bp.route('/api/conversations', methods=['POST'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def create_conversation(terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""创建新对话"""
|
||
perf_log("create_conversation route enter", extra={"username": username})
|
||
t0 = time.perf_counter()
|
||
try:
|
||
data = request.get_json() or {}
|
||
# 前端现在期望"新建对话"回到用户配置的默认模型/模式，
|
||
# 只有当客户端显式要求保留当前模式时才使用传入值。
|
||
preserve_mode = bool(data.get('preserve_mode'))
|
||
thinking_mode = data.get('thinking_mode') if preserve_mode and 'thinking_mode' in data else None
|
||
run_mode = data.get('mode') if preserve_mode and 'mode' in data else None
|
||
|
||
# 多工作区并行后，新建/切换对话只是前端视图导航，不应停止同工作区正在运行的主任务。
|
||
# 同工作区单任务互斥由 /api/tasks 创建任务时兜底限制；输入框禁用由前端根据任务状态处理。
|
||
workspace_id = session.get("workspace_id") or "default"
|
||
active_task = _get_active_workspace_task(username=username, workspace_id=workspace_id)
|
||
if active_task:
|
||
# 运行中时只能创建“视图用”的新对话文件，不能调用 terminal.create_new_conversation。
|
||
# 后者会切换 context_manager.current_conversation_id，导致运行任务后续内容串写到新对话。
|
||
cm = getattr(getattr(terminal, "context_manager", None), "conversation_manager", None)
|
||
if not cm:
|
||
return jsonify({"success": False, "error": "对话管理器未初始化"}), 500
|
||
try:
|
||
prefs = load_personalization_config(workspace.data_dir)
|
||
except Exception:
|
||
prefs = {}
|
||
safe_run_mode = run_mode
|
||
if safe_run_mode not in {"fast", "thinking", "deep"}:
|
||
candidate = (prefs or {}).get("default_run_mode")
|
||
safe_run_mode = candidate if candidate in {"fast", "thinking", "deep"} else "fast"
|
||
safe_thinking = bool(thinking_mode) if thinking_mode is not None else safe_run_mode != "fast"
|
||
previous_cm_current = getattr(cm, "current_conversation_id", None)
|
||
default_permission_mode = (prefs or {}).get("default_permission_mode")
|
||
if default_permission_mode not in ("readonly", "approval", "auto_approval", "unrestricted"):
|
||
default_permission_mode = None
|
||
conversation_id = cm.create_conversation(
|
||
project_path=str(workspace.project_path),
|
||
thinking_mode=safe_thinking,
|
||
run_mode=safe_run_mode,
|
||
initial_messages=[],
|
||
model_key=(prefs or {}).get("default_model") or getattr(terminal, "model_key", None),
|
||
metadata_overrides={
|
||
"permission_mode": default_permission_mode or getattr(terminal, "get_permission_mode", lambda: "unrestricted")(),
|
||
"execution_mode": getattr(terminal, "get_execution_mode", lambda: "sandbox")(),
|
||
},
|
||
)
|
||
try:
|
||
cm.current_conversation_id = previous_cm_current
|
||
except Exception:
|
||
pass
|
||
result = {
|
||
"success": True,
|
||
"conversation_id": conversation_id,
|
||
"message": f"已创建新对话: {conversation_id}",
|
||
"safe_navigation": True,
|
||
}
|
||
else:
|
||
result = terminal.create_new_conversation(thinking_mode=thinking_mode, run_mode=run_mode)
|
||
|
||
if result["success"]:
|
||
session['run_mode'] = terminal.run_mode
|
||
session['thinking_mode'] = terminal.thinking_mode
|
||
perf_log("create_conversation before default versioning", elapsed_ms=(time.perf_counter() - t0) * 1000, extra={"conv_id": result.get("conversation_id")})
|
||
# 根据个性化设置，为新对话默认开启版本控制（安全导航路径也需要）。
|
||
try:
|
||
prefs = load_personalization_config(workspace.data_dir)
|
||
if bool(prefs.get("versioning_enabled_by_default", True)):
|
||
_ensure_conversation_versioning_enabled(terminal, workspace, result["conversation_id"])
|
||
except Exception as exc:
|
||
debug_log(f"[Versioning] create_conversation apply default failed: {exc}")
|
||
perf_log("create_conversation after default versioning", elapsed_ms=(time.perf_counter() - t0) * 1000, extra={"conv_id": result.get("conversation_id")})
|
||
# 广播对话列表更新事件
|
||
socketio.emit('conversation_list_update', {
|
||
'action': 'created',
|
||
'conversation_id': result["conversation_id"]
|
||
}, room=f"user_{username}")
|
||
|
||
if not result.get("safe_navigation"):
|
||
# 安全导航创建不代表后端 terminal 当前对话已切换，因此不广播 conversation_changed。
|
||
socketio.emit('conversation_changed', {
|
||
'conversation_id': result["conversation_id"],
|
||
'title': "新对话"
|
||
}, room=f"user_{username}")
|
||
|
||
perf_log("create_conversation route done", elapsed_ms=(time.perf_counter() - t0) * 1000, extra={"conv_id": result.get("conversation_id")})
|
||
return jsonify(result), 201
|
||
else:
|
||
perf_log("create_conversation route failed", elapsed_ms=(time.perf_counter() - t0) * 1000)
|
||
return jsonify(result), 500
|
||
|
||
except Exception as e:
|
||
print(f"[API] 创建对话错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e),
|
||
"message": "创建对话时发生异常"
|
||
}), 500
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def get_conversation_info(terminal: WebTerminal, workspace: UserWorkspace, username: str, conversation_id):
|
||
"""获取特定对话信息"""
|
||
try:
|
||
# 通过ConversationManager直接获取对话数据
|
||
conversation_data = terminal.context_manager.conversation_manager.load_conversation(conversation_id)
|
||
|
||
if conversation_data:
|
||
# 提取关键信息，不返回完整消息内容（避免数据量过大）
|
||
info = {
|
||
"id": conversation_data["id"],
|
||
"title": conversation_data["title"],
|
||
"created_at": conversation_data["created_at"],
|
||
"updated_at": conversation_data["updated_at"],
|
||
"metadata": conversation_data["metadata"],
|
||
"messages_count": len(conversation_data.get("messages", []))
|
||
}
|
||
|
||
return jsonify({
|
||
"success": True,
|
||
"data": info
|
||
})
|
||
else:
|
||
return jsonify({
|
||
"success": False,
|
||
"error": "Conversation not found",
|
||
"message": f"对话 {conversation_id} 不存在"
|
||
}), 404
|
||
|
||
except Exception as e:
|
||
print(f"[API] 获取对话信息错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e),
|
||
"message": "获取对话信息时发生异常"
|
||
}), 500
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/load', methods=['PUT'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def load_conversation(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""加载特定对话"""
|
||
try:
|
||
workspace_id = session.get("workspace_id") or "default"
|
||
active_task = _get_active_workspace_task(username=username, workspace_id=workspace_id)
|
||
if active_task:
|
||
# 同工作区有任务运行时，加载/查看其他对话不能改后端 terminal 当前上下文。
|
||
# 否则运行任务后续保存与事件归属会被切到当前查看的对话。
|
||
result = _build_safe_load_result(terminal, conversation_id)
|
||
else:
|
||
result = terminal.load_conversation(conversation_id)
|
||
|
||
if result["success"]:
|
||
session['run_mode'] = terminal.run_mode
|
||
session['thinking_mode'] = terminal.thinking_mode
|
||
session['model_key'] = getattr(terminal, "model_key", None)
|
||
normalized_id = _normalize_conv_id(conversation_id)
|
||
try:
|
||
vm = _get_conv_versioning_manager(workspace, normalized_id)
|
||
vmeta = _get_conversation_versioning_meta(terminal, normalized_id)
|
||
tracking_mode = _normalize_versioning_tracking_mode(vmeta.get("tracking_mode"))
|
||
enabled = bool(vmeta.get("enabled")) and _can_use_versioning_scope(username, tracking_mode)
|
||
latest_checkpoint = vm.get_latest_checkpoint() if enabled else None
|
||
latest_commit = (latest_checkpoint or {}).get("tree_hash") if latest_checkpoint else None
|
||
result["versioning"] = {
|
||
"host_mode": _is_host_mode_request(username),
|
||
"enabled": enabled,
|
||
"mode": "overwrite",
|
||
"tracking_mode": tracking_mode,
|
||
"latest_seq": int((latest_checkpoint or {}).get("seq") or 0) if latest_checkpoint else None,
|
||
"latest_commit": latest_commit,
|
||
}
|
||
except Exception as exc:
|
||
debug_log(f"[Versioning] load status 读取失败: {exc}")
|
||
result["versioning"] = {
|
||
"host_mode": _is_host_mode_request(username),
|
||
"enabled": False,
|
||
"mode": "overwrite",
|
||
"tracking_mode": ConversationVersioningManager.TRACKING_MODE_WORKSPACE_AND_CONVERSATION,
|
||
"error": str(exc),
|
||
}
|
||
|
||
if not result.get("safe_navigation"):
|
||
# 安全导航只改变前端查看对象，不代表后端 terminal 当前上下文已切换。
|
||
socketio.emit('conversation_changed', {
|
||
'conversation_id': conversation_id,
|
||
'title': result.get("title", "未知对话"),
|
||
'messages_count': result.get("messages_count", 0)
|
||
}, room=f"user_{username}")
|
||
|
||
# 广播系统状态更新（因为当前对话改变了）
|
||
status = terminal.get_status()
|
||
socketio.emit('status_update', status, room=f"user_{username}")
|
||
|
||
# 清理和重置相关UI状态
|
||
socketio.emit('conversation_loaded', {
|
||
'conversation_id': conversation_id,
|
||
'clear_ui': True # 提示前端清理当前UI状态
|
||
}, room=f"user_{username}")
|
||
|
||
return jsonify(result)
|
||
else:
|
||
return jsonify(result), 404 if "不存在" in result.get("message", "") else 500
|
||
|
||
except Exception as e:
|
||
print(f"[API] 加载对话错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e),
|
||
"message": "加载对话时发生异常"
|
||
}), 500
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>', methods=['DELETE'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def delete_conversation(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""删除特定对话"""
|
||
try:
|
||
# 检查是否是当前对话
|
||
is_current = (terminal.context_manager.current_conversation_id == conversation_id)
|
||
|
||
result = terminal.delete_conversation(conversation_id)
|
||
|
||
if result["success"]:
|
||
# 广播对话列表更新事件
|
||
socketio.emit('conversation_list_update', {
|
||
'action': 'deleted',
|
||
'conversation_id': conversation_id
|
||
}, room=f"user_{username}")
|
||
|
||
# 如果删除的是当前对话，广播对话清空事件
|
||
if is_current:
|
||
socketio.emit('conversation_changed', {
|
||
'conversation_id': None,
|
||
'title': None,
|
||
'cleared': True
|
||
}, room=f"user_{username}")
|
||
|
||
# 更新系统状态
|
||
status = terminal.get_status()
|
||
socketio.emit('status_update', status, room=f"user_{username}")
|
||
|
||
return jsonify(result)
|
||
else:
|
||
return jsonify(result), 404 if "不存在" in result.get("message", "") else 500
|
||
|
||
except Exception as e:
|
||
print(f"[API] 删除对话错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e),
|
||
"message": "删除对话时发生异常"
|
||
}), 500
|
||
|
||
@conversation_bp.route('/api/conversations/search', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def search_conversations(terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""搜索对话"""
|
||
try:
|
||
query = request.args.get('q', '').strip()
|
||
limit = request.args.get('limit', 20, type=int)
|
||
|
||
if not query:
|
||
return jsonify({
|
||
"success": False,
|
||
"error": "Missing query parameter",
|
||
"message": "请提供搜索关键词"
|
||
}), 400
|
||
|
||
# 限制参数范围
|
||
limit = max(1, min(limit, 50))
|
||
|
||
result = terminal.search_conversations(query, limit)
|
||
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"results": result["results"],
|
||
"count": result["count"],
|
||
"query": query
|
||
}
|
||
})
|
||
|
||
except Exception as e:
|
||
print(f"[API] 搜索对话错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e),
|
||
"message": "搜索对话时发生异常"
|
||
}), 500
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/messages', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def get_conversation_messages(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""获取对话的消息历史（可选功能，用于调试或详细查看）"""
|
||
try:
|
||
# 获取完整对话数据
|
||
conversation_data = terminal.context_manager.conversation_manager.load_conversation(conversation_id)
|
||
|
||
if conversation_data:
|
||
messages = conversation_data.get("messages", [])
|
||
|
||
# 可选：限制消息数量，避免返回过多数据
|
||
limit = request.args.get('limit', type=int)
|
||
if limit:
|
||
messages = messages[-limit:] # 获取最后N条消息
|
||
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"conversation_id": conversation_id,
|
||
"messages": messages,
|
||
"total_count": len(conversation_data.get("messages", []))
|
||
}
|
||
})
|
||
else:
|
||
return jsonify({
|
||
"success": False,
|
||
"error": "Conversation not found",
|
||
"message": f"对话 {conversation_id} 不存在"
|
||
}), 404
|
||
|
||
except Exception as e:
|
||
print(f"[API] 获取对话消息错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e),
|
||
"message": "获取对话消息时发生异常"
|
||
}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/conversations/media/<path:media_id>', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def download_conversation_media(media_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""按 media_store 中的 media_id 下载会话媒体。"""
|
||
try:
|
||
ctx = getattr(terminal, "context_manager", None)
|
||
media_store = getattr(ctx, "media_store", None)
|
||
if media_store is None:
|
||
return jsonify({"success": False, "error": "media_store 不可用"}), 503
|
||
|
||
target_id = str(media_id or "").strip()
|
||
if not target_id:
|
||
return jsonify({"success": False, "error": "media_id 不能为空"}), 400
|
||
|
||
entry = media_store.get_media_entry(target_id)
|
||
if not isinstance(entry, dict):
|
||
return jsonify({"success": False, "error": "媒体不存在"}), 404
|
||
payload = media_store.load_bytes_by_media_id(target_id)
|
||
if payload is None:
|
||
return jsonify({"success": False, "error": "媒体文件不存在"}), 404
|
||
|
||
mime_type = str(entry.get("mime_type") or "application/octet-stream").strip() or "application/octet-stream"
|
||
blob_name = str(entry.get("blob_rel_path") or "")
|
||
filename = Path(blob_name).name if blob_name else target_id.replace(":", "_")
|
||
return send_file(
|
||
BytesIO(payload),
|
||
mimetype=mime_type,
|
||
as_attachment=False,
|
||
download_name=filename,
|
||
conditional=True,
|
||
etag=True,
|
||
max_age=86400,
|
||
)
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/versioning', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def get_conversation_versioning_status(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
try:
|
||
normalized_id = _normalize_conv_id(conversation_id)
|
||
host_mode = _is_host_mode_request(username)
|
||
versioning_meta = _get_conversation_versioning_meta(terminal, normalized_id)
|
||
tracking_mode = _normalize_versioning_tracking_mode(versioning_meta.get("tracking_mode"))
|
||
enabled = bool(versioning_meta.get("enabled")) and _can_use_versioning_scope(username, tracking_mode)
|
||
manager = _get_conv_versioning_manager(workspace, normalized_id)
|
||
latest = manager.get_latest_checkpoint() if enabled else None
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"conversation_id": normalized_id,
|
||
"host_mode": host_mode,
|
||
"supports_workspace_tracking": host_mode,
|
||
"supports_conversation_only": True,
|
||
"enabled": enabled,
|
||
"mode": "overwrite",
|
||
"tracking_mode": tracking_mode,
|
||
"latest_seq": int((latest or {}).get("seq") or 0) if latest else None,
|
||
"latest_commit": (latest or {}).get("tree_hash"),
|
||
}
|
||
})
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/versioning', methods=['POST'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def update_conversation_versioning(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
try:
|
||
normalized_id = _normalize_conv_id(conversation_id)
|
||
host_mode = _is_host_mode_request(username)
|
||
current_meta = _get_conversation_versioning_meta(terminal, normalized_id)
|
||
payload = request.get_json(silent=True) or {}
|
||
enabled = bool(payload.get("enabled"))
|
||
mode = "overwrite"
|
||
tracking_mode = _normalize_versioning_tracking_mode(
|
||
payload.get("tracking_mode") if "tracking_mode" in payload else current_meta.get("tracking_mode")
|
||
)
|
||
if enabled and tracking_mode == ConversationVersioningManager.TRACKING_MODE_WORKSPACE_AND_CONVERSATION and not host_mode:
|
||
tracking_mode = ConversationVersioningManager.TRACKING_MODE_CONVERSATION_ONLY
|
||
|
||
manager = _get_conv_versioning_manager(workspace, normalized_id)
|
||
meta = manager.set_enabled(enabled=enabled, mode=mode, tracking_mode=tracking_mode)
|
||
if enabled:
|
||
conv_data = terminal.context_manager.conversation_manager.load_conversation(normalized_id) or {}
|
||
snapshot_payload = {
|
||
"conversation_id": normalized_id,
|
||
"title": conv_data.get("title"),
|
||
"metadata": conv_data.get("metadata") or {},
|
||
"messages": conv_data.get("messages") or [],
|
||
"message_index": -1,
|
||
"run_status": "initial",
|
||
}
|
||
init_result = manager.ensure_initial_checkpoint(
|
||
workspace_path=str(workspace.project_path),
|
||
conversation_snapshot=snapshot_payload,
|
||
tracking_mode=tracking_mode,
|
||
)
|
||
init_row = init_result.get("row") or {}
|
||
debug_log(
|
||
f"[Versioning][Init] conv={normalized_id} enabled={enabled} "
|
||
f"tracking_mode={tracking_mode} "
|
||
f"created={init_result.get('created')} reason={init_result.get('reason')} "
|
||
f"seq={init_row.get('seq')} commit={init_row.get('commit')}"
|
||
)
|
||
if init_row.get("tree_hash"):
|
||
meta["last_tree_hash"] = init_row.get("tree_hash")
|
||
ok = _update_conversation_versioning_meta(
|
||
terminal,
|
||
normalized_id,
|
||
enabled=enabled,
|
||
mode=mode,
|
||
tracking_mode=tracking_mode,
|
||
last_commit=meta.get("last_commit"),
|
||
last_input_seq=int(meta.get("last_input_seq") or 0),
|
||
)
|
||
if not ok:
|
||
return jsonify({"success": False, "error": "更新对话版本配置失败"}), 404
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"conversation_id": normalized_id,
|
||
"enabled": bool(meta.get("enabled")),
|
||
"mode": "overwrite",
|
||
"tracking_mode": tracking_mode,
|
||
"last_commit": meta.get("last_commit"),
|
||
"last_input_seq": int(meta.get("last_input_seq") or 0),
|
||
}
|
||
})
|
||
except VersioningError as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 400
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/versioning/checkpoints', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def list_conversation_versioning_checkpoints(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
try:
|
||
normalized_id = _normalize_conv_id(conversation_id)
|
||
host_mode = _is_host_mode_request(username)
|
||
versioning_meta = _get_conversation_versioning_meta(terminal, normalized_id)
|
||
tracking_mode = _normalize_versioning_tracking_mode(versioning_meta.get("tracking_mode"))
|
||
if tracking_mode == ConversationVersioningManager.TRACKING_MODE_WORKSPACE_AND_CONVERSATION and not host_mode:
|
||
return jsonify({"success": False, "error": "当前模式不支持工作区版本点，请切换到宿主机模式"}), 400
|
||
if not versioning_meta.get("enabled"):
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {"enabled": False, "mode": "overwrite", "tracking_mode": tracking_mode, "items": []}
|
||
})
|
||
manager = _get_conv_versioning_manager(workspace, normalized_id)
|
||
rows = manager.list_checkpoints()
|
||
backup_mode = str(versioning_meta.get("backup_mode") or "shallow").strip().lower()
|
||
if backup_mode == "shallow":
|
||
shallow_manager = ShallowVersioningManager(
|
||
project_path=workspace.project_path,
|
||
data_dir=workspace.data_dir,
|
||
conversation_id=normalized_id,
|
||
)
|
||
normalized_rows: List[Dict[str, Any]] = []
|
||
for row in rows:
|
||
row = dict(row)
|
||
shallow_message_id = row.get("shallow_message_id")
|
||
if not shallow_message_id:
|
||
snapshot_file = row.get("snapshot_file")
|
||
if snapshot_file:
|
||
try:
|
||
snapshot_path = (Path(workspace.data_dir) / "save" / normalized_id / str(snapshot_file)).resolve()
|
||
if snapshot_path.exists():
|
||
snapshot_data = json.loads(snapshot_path.read_text(encoding="utf-8", errors="ignore"))
|
||
shallow_message_id = snapshot_data.get("shallow_message_id")
|
||
except Exception:
|
||
pass
|
||
if shallow_message_id:
|
||
stats = shallow_manager.get_diff_stats(str(shallow_message_id)) or {}
|
||
files = shallow_manager.get_file_diff_stats(str(shallow_message_id)) or []
|
||
row["insertions"] = int(stats.get("insertions") or 0)
|
||
row["deletions"] = int(stats.get("deletions") or 0)
|
||
row["files_changed"] = len(files)
|
||
row["files"] = files
|
||
normalized_rows.append(row)
|
||
rows = normalized_rows
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"enabled": True,
|
||
"mode": "overwrite",
|
||
"tracking_mode": tracking_mode,
|
||
"items": rows,
|
||
}
|
||
})
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/versioning/checkpoints/<int:seq>', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def get_conversation_versioning_checkpoint_detail(conversation_id, seq: int, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
try:
|
||
normalized_id = _normalize_conv_id(conversation_id)
|
||
host_mode = _is_host_mode_request(username)
|
||
vmeta = _get_conversation_versioning_meta(terminal, normalized_id)
|
||
tracking_mode = _normalize_versioning_tracking_mode(vmeta.get("tracking_mode"))
|
||
if tracking_mode == ConversationVersioningManager.TRACKING_MODE_WORKSPACE_AND_CONVERSATION and not host_mode:
|
||
return jsonify({"success": False, "error": "当前模式不支持工作区版本详情，请切换到宿主机模式"}), 400
|
||
manager = _get_conv_versioning_manager(workspace, normalized_id)
|
||
row = manager.get_checkpoint_detail(seq, include_patch=True)
|
||
if not row:
|
||
return jsonify({"success": False, "error": "未找到对应版本点"}), 404
|
||
|
||
backup_mode = str(vmeta.get("backup_mode") or "shallow").strip().lower()
|
||
debug_log(f"[Versioning][Detail] conv={normalized_id} seq={seq} backup_mode={backup_mode} row_shallow_msg_id={row.get('shallow_message_id')} files_count={len(row.get('files') or [])}")
|
||
if backup_mode == "shallow":
|
||
shallow_message_id = row.get("shallow_message_id")
|
||
# 兼容旧检查点：shallow_message_id 之前只写在 snapshot_file 里
|
||
if not shallow_message_id:
|
||
snapshot_file = row.get("snapshot_file")
|
||
if snapshot_file:
|
||
try:
|
||
snapshot_path = (Path(workspace.data_dir) / "save" / normalized_id / str(snapshot_file)).resolve()
|
||
if snapshot_path.exists():
|
||
snapshot_data = json.loads(snapshot_path.read_text(encoding="utf-8", errors="ignore"))
|
||
shallow_message_id = snapshot_data.get("shallow_message_id")
|
||
except Exception:
|
||
pass
|
||
files = row.get("files") or []
|
||
debug_log(f"[Versioning][Detail] resolved shallow_message_id={shallow_message_id} files={files}")
|
||
if shallow_message_id and isinstance(files, list):
|
||
shallow_manager = ShallowVersioningManager(
|
||
project_path=workspace.project_path,
|
||
data_dir=workspace.data_dir,
|
||
conversation_id=normalized_id,
|
||
)
|
||
stats = shallow_manager.get_diff_stats(str(shallow_message_id)) or {}
|
||
normalized_files: List[Dict[str, Any]] = []
|
||
for file_item in files:
|
||
if not isinstance(file_item, dict):
|
||
continue
|
||
item = dict(file_item)
|
||
path = str(item.get("path") or "")
|
||
if path:
|
||
patch = shallow_manager.get_file_patch_lines(path, str(shallow_message_id))
|
||
debug_log(f"[Versioning][Detail] path={path} patch={patch}")
|
||
item["patch_lines"] = patch.get("lines") or []
|
||
item["patch_truncated"] = bool(patch.get("truncated"))
|
||
normalized_files.append(item)
|
||
row["files"] = normalized_files
|
||
row["insertions"] = int(stats.get("insertions") or 0)
|
||
row["deletions"] = int(stats.get("deletions") or 0)
|
||
row["files_changed"] = len(normalized_files)
|
||
debug_log(f"[Versioning][Detail] response files={row.get('files')}")
|
||
return jsonify({"success": True, "data": row})
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 500
|
||
|
||
|
||
def _copy_versioning_records(source_conversation_id: str, target_conversation_id: str, data_dir: Path) -> bool:
|
||
"""把源对话的版本控制记录（checkpoints / snapshots / git / meta）复制到目标对话。"""
|
||
source_dir = (Path(data_dir) / "save" / source_conversation_id).resolve()
|
||
target_dir = (Path(data_dir) / "save" / target_conversation_id).resolve()
|
||
if not source_dir.exists():
|
||
return False
|
||
try:
|
||
if target_dir.exists():
|
||
shutil.rmtree(target_dir)
|
||
shutil.copytree(source_dir, target_dir)
|
||
# 复制后保持目标 meta 与源一致，无需修改 conversation_id（manager 按路径构造）。
|
||
return True
|
||
except Exception as exc:
|
||
debug_log(f"[Versioning] copy records failed {source_conversation_id} -> {target_conversation_id}: {exc}")
|
||
return False
|
||
|
||
|
||
def _restore_checkpoint_to_conversation(
|
||
terminal: WebTerminal,
|
||
workspace: UserWorkspace,
|
||
source_conversation_id: str,
|
||
target_conversation_id: str,
|
||
seq: int,
|
||
tracking_mode: str,
|
||
host_mode: bool,
|
||
backup_mode: Optional[str] = None,
|
||
) -> Dict[str, Any]:
|
||
"""把源对话的指定 checkpoint 恢复到目标对话，覆盖目标对话内容。"""
|
||
source_manager = _get_conv_versioning_manager(workspace, source_conversation_id)
|
||
target_manager = _get_conv_versioning_manager(workspace, target_conversation_id)
|
||
checkpoint = source_manager.get_checkpoint(seq)
|
||
if not checkpoint:
|
||
raise VersioningError("未找到对应版本点")
|
||
|
||
normalized_backup_mode = "full" if backup_mode == "full" else "shallow"
|
||
if normalized_backup_mode == "shallow":
|
||
# 浅备份模式：根据检查点行中的 shallow_message_id 恢复被跟踪文件
|
||
shallow_message_id = checkpoint.get("shallow_message_id")
|
||
if shallow_message_id:
|
||
shallow_manager = ShallowVersioningManager(
|
||
project_path=workspace.project_path,
|
||
data_dir=workspace.data_dir,
|
||
conversation_id=source_conversation_id,
|
||
)
|
||
rewind_result = shallow_manager.rewind(str(shallow_message_id))
|
||
debug_log(
|
||
f"[Versioning][Restore] shallow rewind conv={target_conversation_id} seq={seq} "
|
||
f"msg_id={shallow_message_id} files_changed={len(rewind_result.get('files_changed') or [])}"
|
||
)
|
||
else:
|
||
debug_log(
|
||
f"[Versioning][Restore] shallow checkpoint missing message_id "
|
||
f"conv={target_conversation_id} seq={seq}"
|
||
)
|
||
elif tracking_mode == ConversationVersioningManager.TRACKING_MODE_WORKSPACE_AND_CONVERSATION:
|
||
if not host_mode:
|
||
raise VersioningError("当前模式不支持工作区回溯，请切换到宿主机模式")
|
||
target_manager.restore_to_tree(checkpoint.get("tree_hash"))
|
||
debug_log(
|
||
f"[Versioning][Restore] git restored conv={target_conversation_id} seq={seq} "
|
||
f"commit={checkpoint.get('commit')}"
|
||
)
|
||
else:
|
||
debug_log(
|
||
f"[Versioning][Restore] conversation-only mode, skip workspace restore "
|
||
f"conv={target_conversation_id} seq={seq}"
|
||
)
|
||
|
||
cm = terminal.context_manager.conversation_manager
|
||
conv_data = cm.load_conversation(source_conversation_id) or {}
|
||
conv_meta = conv_data.get("metadata") or {}
|
||
|
||
snapshot = source_manager.get_checkpoint_snapshot(seq)
|
||
snapshot_messages = (snapshot or {}).get("messages")
|
||
if not isinstance(snapshot_messages, list):
|
||
snapshot_messages = None
|
||
snapshot_meta = (snapshot or {}).get("metadata")
|
||
if not isinstance(snapshot_meta, dict):
|
||
snapshot_meta = {}
|
||
|
||
if snapshot_messages is None:
|
||
all_messages = conv_data.get("messages") or []
|
||
msg_index = int(checkpoint.get("message_index") or 0)
|
||
snapshot_messages = all_messages[: max(0, msg_index + 1)]
|
||
debug_log(
|
||
f"[Versioning][Restore] fallback truncation conv={target_conversation_id} seq={seq} "
|
||
f"msg_index={msg_index} restored_messages={len(snapshot_messages)}"
|
||
)
|
||
else:
|
||
debug_log(
|
||
f"[Versioning][Restore] snapshot loaded conv={target_conversation_id} seq={seq} "
|
||
f"messages={len(snapshot_messages)} meta_keys={list(snapshot_meta.keys())[:8]}"
|
||
)
|
||
|
||
restore_meta = dict(conv_meta or {})
|
||
if snapshot_meta:
|
||
restore_meta.update(snapshot_meta)
|
||
|
||
restore_project_path = restore_meta.get("project_path") or str(workspace.project_path)
|
||
restore_thinking_mode = bool(restore_meta.get("thinking_mode", False))
|
||
restore_run_mode = restore_meta.get("run_mode") or ("thinking" if restore_thinking_mode else "fast")
|
||
restore_model_key = restore_meta.get("model_key")
|
||
restore_has_images = bool(restore_meta.get("has_images", False))
|
||
restore_has_videos = bool(restore_meta.get("has_videos", False))
|
||
|
||
ok = cm.save_conversation(
|
||
conversation_id=target_conversation_id,
|
||
messages=snapshot_messages,
|
||
project_path=restore_project_path,
|
||
thinking_mode=restore_thinking_mode,
|
||
run_mode=restore_run_mode,
|
||
model_key=restore_model_key,
|
||
has_images=restore_has_images,
|
||
has_videos=restore_has_videos,
|
||
)
|
||
if not ok:
|
||
raise VersioningError("保存恢复后的对话失败")
|
||
|
||
prune_info = target_manager.prune_checkpoints_after(seq)
|
||
resolved_last_commit = prune_info.get("last_tree_hash") or checkpoint.get("tree_hash")
|
||
if not resolved_last_commit:
|
||
resolved_last_commit = (target_manager.load_meta() or {}).get("last_tree_hash")
|
||
|
||
_update_conversation_versioning_meta(
|
||
terminal,
|
||
target_conversation_id,
|
||
enabled=True,
|
||
mode="overwrite",
|
||
tracking_mode=tracking_mode,
|
||
last_commit=resolved_last_commit,
|
||
last_input_seq=int(prune_info.get("max_seq") or checkpoint.get("seq") or 0),
|
||
)
|
||
cm.update_conversation_metadata(
|
||
target_conversation_id,
|
||
{
|
||
**restore_meta,
|
||
"versioning": {
|
||
"enabled": True,
|
||
"mode": "overwrite",
|
||
"tracking_mode": tracking_mode,
|
||
"last_commit": resolved_last_commit,
|
||
"last_input_seq": int(prune_info.get("max_seq") or checkpoint.get("seq") or 0),
|
||
"updated_at": datetime.now().isoformat(),
|
||
},
|
||
},
|
||
)
|
||
debug_log(
|
||
f"[Versioning][Restore] saved conv={target_conversation_id} seq={seq} "
|
||
f"messages={len(snapshot_messages)}"
|
||
)
|
||
return {
|
||
"restored_seq": int(checkpoint.get("seq") or 0),
|
||
"restored_commit": checkpoint.get("tree_hash"),
|
||
"tracking_mode": tracking_mode,
|
||
}
|
||
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/versioning/restore', methods=['POST'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def restore_conversation_versioning_checkpoint(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
try:
|
||
normalized_id = _normalize_conv_id(conversation_id)
|
||
host_mode = _is_host_mode_request(username)
|
||
payload = request.get_json(silent=True) or {}
|
||
seq = int(payload.get("seq") or 0)
|
||
if seq < 0:
|
||
return jsonify({"success": False, "error": "缺少有效 seq"}), 400
|
||
|
||
vmeta = _get_conversation_versioning_meta(terminal, normalized_id)
|
||
if not vmeta.get("enabled"):
|
||
return jsonify({"success": False, "error": "当前对话未开启版本管理"}), 400
|
||
|
||
restore_mode = str(payload.get("mode") or "overwrite").lower()
|
||
if restore_mode not in {"overwrite", "copy"}:
|
||
restore_mode = "overwrite"
|
||
|
||
requested_tracking_mode = _normalize_versioning_tracking_mode(payload.get("tracking_mode"))
|
||
|
||
manager = _get_conv_versioning_manager(workspace, normalized_id)
|
||
checkpoint = manager.get_checkpoint(seq)
|
||
if not checkpoint:
|
||
return jsonify({"success": False, "error": "未找到对应版本点"}), 404
|
||
|
||
# 默认使用 checkpoint 创建时的 tracking_mode；若前端显式指定则优先采用。
|
||
tracking_mode = _normalize_versioning_tracking_mode(
|
||
requested_tracking_mode or checkpoint.get("tracking_mode") or vmeta.get("tracking_mode")
|
||
)
|
||
if tracking_mode == ConversationVersioningManager.TRACKING_MODE_WORKSPACE_AND_CONVERSATION and not host_mode:
|
||
return jsonify({"success": False, "error": "当前模式不支持工作区回溯，请切换到宿主机模式"}), 400
|
||
|
||
cm = terminal.context_manager.conversation_manager
|
||
target_conversation_id = normalized_id
|
||
|
||
if restore_mode == "copy":
|
||
duplicate_result = terminal.context_manager.duplicate_conversation(normalized_id)
|
||
if not duplicate_result.get("success"):
|
||
return jsonify({"success": False, "error": duplicate_result.get("error", "复制对话失败")}), 500
|
||
target_conversation_id = _normalize_conv_id(duplicate_result["duplicate_conversation_id"])
|
||
copied = _copy_versioning_records(normalized_id, target_conversation_id, workspace.data_dir)
|
||
if not copied:
|
||
# 复制记录失败时继续执行，只是版本控制记录无法继承；后续仍可恢复对话内容。
|
||
debug_log(f"[Versioning][Restore] copy records skipped for {target_conversation_id}")
|
||
|
||
debug_log(
|
||
f"[Versioning][Restore] start conv={normalized_id} seq={seq} mode={restore_mode} "
|
||
f"tracking_mode={tracking_mode} target={target_conversation_id} "
|
||
f"target_commit={checkpoint.get('commit')} workspace={workspace.project_path}"
|
||
)
|
||
|
||
backup_mode = str(vmeta.get("backup_mode") or "shallow").strip().lower()
|
||
restore_info = _restore_checkpoint_to_conversation(
|
||
terminal, workspace, normalized_id, target_conversation_id, seq, tracking_mode, host_mode, backup_mode
|
||
)
|
||
|
||
# overwrite 场景需要清空内存历史避免反向覆写；copy 场景目标是新对话，无需处理。
|
||
if restore_mode == "overwrite":
|
||
try:
|
||
current_loaded_id = getattr(terminal.context_manager, "current_conversation_id", None)
|
||
if current_loaded_id == normalized_id:
|
||
terminal.context_manager.conversation_history = []
|
||
debug_log(f"[Versioning][Restore] cleared in-memory history before reload conv={normalized_id}")
|
||
except Exception as exc:
|
||
debug_log(f"[Versioning][Restore] clear in-memory history failed: {exc}")
|
||
|
||
terminal.load_conversation(target_conversation_id)
|
||
debug_log(
|
||
f"[Versioning][Restore] reload done conv={target_conversation_id} "
|
||
f"messages={len(getattr(terminal.context_manager, 'conversation_history', []) or [])}"
|
||
)
|
||
session['run_mode'] = terminal.run_mode
|
||
session['thinking_mode'] = terminal.thinking_mode
|
||
|
||
socketio.emit('conversation_list_update', {
|
||
'action': 'version_restored',
|
||
'conversation_id': target_conversation_id
|
||
}, room=f"user_{username}")
|
||
socketio.emit('conversation_changed', {
|
||
'conversation_id': target_conversation_id,
|
||
'title': (cm.load_conversation(target_conversation_id) or {}).get("title", "版本回溯对话"),
|
||
}, room=f"user_{username}")
|
||
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"restore_mode": restore_mode,
|
||
"conversation_id": target_conversation_id,
|
||
"restored_seq": restore_info["restored_seq"],
|
||
"restored_commit": restore_info["restored_commit"],
|
||
"tracking_mode": restore_info["tracking_mode"],
|
||
"source_conversation_id": normalized_id,
|
||
}
|
||
})
|
||
except VersioningError as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 400
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/compress', methods=['POST'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def compress_conversation(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""深层压缩指定对话（in-place）：生成 compact 文件、标记历史前缀为已压缩，按设置决定是否续接。"""
|
||
try:
|
||
policy = resolve_admin_policy(get_current_user_record())
|
||
if policy.get("ui_blocks", {}).get("block_compress_conversation"):
|
||
return jsonify({"success": False, "error": "压缩对话已被管理员禁用"}), 403
|
||
normalized_id = conversation_id if conversation_id.startswith('conv_') else f"conv_{conversation_id}"
|
||
result = asyncio.run(
|
||
run_deep_compression(
|
||
web_terminal=terminal,
|
||
workspace=workspace,
|
||
conversation_id=normalized_id,
|
||
mode="manual",
|
||
sender=None,
|
||
)
|
||
)
|
||
|
||
if not result.get("success"):
|
||
status_code = 404 if "不存在" in result.get("error", "") else (409 if result.get("in_progress") else 400)
|
||
return jsonify(result), status_code
|
||
|
||
# in-place 压缩：对话 id 不变。通知前端当前对话内容已变化（历史前缀被标记），刷新展示。
|
||
load_result = terminal.load_conversation(normalized_id)
|
||
if load_result.get("success"):
|
||
socketio.emit('conversation_list_update', {
|
||
'action': 'compressed',
|
||
'conversation_id': normalized_id
|
||
}, room=f"user_{username}")
|
||
socketio.emit('conversation_loaded', {
|
||
'conversation_id': normalized_id,
|
||
'clear_ui': True
|
||
}, room=f"user_{username}")
|
||
|
||
response_payload = {
|
||
"success": True,
|
||
"in_place": True,
|
||
"compressed_conversation_id": normalized_id,
|
||
"compact_file": result.get("compact_file"),
|
||
"summary_failed": result.get("summary_failed", False),
|
||
"guide_message": result.get("guide_message"),
|
||
"compress_form": result.get("compress_form"),
|
||
# 手动压缩只有一种行为：生成压缩消息（引导语），不自动续接，等待用户继续发送消息才工作。
|
||
"compress_behavior": "wait",
|
||
"load_result": load_result
|
||
}
|
||
|
||
guide_message = (result.get("guide_message") or "").strip()
|
||
if guide_message:
|
||
# 只把引导语作为 user 消息追加进历史，不触发请求。
|
||
try:
|
||
terminal.context_manager.add_conversation(
|
||
role="user",
|
||
content=guide_message,
|
||
metadata={"message_source": "compression_handoff"},
|
||
)
|
||
response_payload["auto_task_started"] = False
|
||
response_payload["guide_inserted"] = True
|
||
# 通知前端实时显示这条 compact 消息（覆盖 socket 与 in-place 未刷新场景）
|
||
try:
|
||
emit(
|
||
"user_message",
|
||
{
|
||
"message": guide_message,
|
||
"images": [],
|
||
"videos": [],
|
||
"media_refs": [],
|
||
"message_source": "compression_handoff",
|
||
"visibility": "compact",
|
||
"starts_work": False,
|
||
"metadata": {
|
||
"message_source": "compression_handoff",
|
||
"visibility": "compact",
|
||
"starts_work": False,
|
||
},
|
||
"conversation_id": normalized_id,
|
||
},
|
||
room=f"user_{username}",
|
||
)
|
||
except Exception as emit_exc:
|
||
debug_log(f"[Compression] 发送 user_message 事件失败: {emit_exc}")
|
||
except Exception as exc:
|
||
debug_log(f"[Compression] 追加引导语消息失败: {exc}")
|
||
response_payload["auto_task_started"] = False
|
||
response_payload["guide_inserted"] = False
|
||
response_payload["guide_insert_error"] = str(exc)
|
||
else:
|
||
response_payload["auto_task_started"] = False
|
||
|
||
return jsonify(response_payload)
|
||
|
||
except Exception as e:
|
||
print(f"[API] 压缩对话错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e),
|
||
"message": "压缩对话时发生异常"
|
||
}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/compression_status', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def get_conversation_compression_status(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
try:
|
||
normalized_id = conversation_id if conversation_id.startswith('conv_') else f"conv_{conversation_id}"
|
||
data = terminal.context_manager.conversation_manager.load_conversation(normalized_id) or {}
|
||
meta = data.get("metadata", {}) or {}
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"conversation_id": normalized_id,
|
||
"compression_in_progress": bool(meta.get("compression_in_progress", False)),
|
||
"compression_mode": meta.get("compression_mode"),
|
||
"compression_stage": meta.get("compression_stage"),
|
||
"compression_error": meta.get("compression_error"),
|
||
"compression_count": int(meta.get("compression_count", 0) or 0),
|
||
"compression_job_id": meta.get("compression_job_id"),
|
||
}
|
||
})
|
||
except Exception as e:
|
||
return jsonify({"success": False, "error": str(e)}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/compression_cancel', methods=['POST'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def cancel_conversation_compression(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
try:
|
||
normalized_id = conversation_id if conversation_id.startswith('conv_') else f"conv_{conversation_id}"
|
||
ok = terminal.context_manager.conversation_manager.update_conversation_metadata(
|
||
normalized_id,
|
||
{
|
||
"compression_in_progress": False,
|
||
"compression_mode": None,
|
||
"compression_stage": None,
|
||
"compression_resume_payload": None,
|
||
"compression_error": "用户切换对话导致压缩取消",
|
||
}
|
||
)
|
||
if not ok:
|
||
return jsonify({"success": False, "error": "对话不存在或取消失败"}), 404
|
||
return jsonify({"success": True, "conversation_id": normalized_id})
|
||
except Exception as e:
|
||
return jsonify({"success": False, "error": str(e)}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/sub_agents', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def list_sub_agents(terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""返回当前对话的子智能体任务列表。"""
|
||
manager = getattr(terminal, "sub_agent_manager", None)
|
||
if not manager:
|
||
return jsonify({"success": True, "data": []})
|
||
try:
|
||
try:
|
||
# 防止不同进程创建的子智能体未被当前进程感知
|
||
manager._load_state()
|
||
except Exception:
|
||
pass
|
||
conversation_id = terminal.context_manager.current_conversation_id
|
||
data = manager.get_overview(conversation_id=conversation_id)
|
||
if not data:
|
||
# 若当前对话暂无数据但存在运行中子智能体，回退为全局运行态，避免面板空白
|
||
all_overview = manager.get_overview(conversation_id=None)
|
||
if all_overview:
|
||
terminal_statuses = TERMINAL_STATUSES.union({"terminated"})
|
||
running_only = [item for item in all_overview if item.get("status") not in terminal_statuses]
|
||
if running_only:
|
||
data = running_only
|
||
debug_log(
|
||
"[SubAgent] /api/sub_agents overview "
|
||
+ json.dumps({
|
||
"conversation_id": conversation_id,
|
||
"count": len(data),
|
||
"tasks": [
|
||
{
|
||
"task_id": item.get("task_id"),
|
||
"status": item.get("status"),
|
||
"run_in_background": item.get("run_in_background"),
|
||
"conversation_id": item.get("conversation_id"),
|
||
} for item in data
|
||
],
|
||
}, ensure_ascii=False)
|
||
)
|
||
if not hasattr(terminal, "_announced_sub_agent_tasks"):
|
||
terminal._announced_sub_agent_tasks = set()
|
||
announced = terminal._announced_sub_agent_tasks
|
||
notified_from_history = set()
|
||
try:
|
||
history = getattr(terminal.context_manager, "conversation_history", []) or []
|
||
for msg in history:
|
||
meta = msg.get("metadata") or {}
|
||
task_id = meta.get("task_id")
|
||
if meta.get("sub_agent_notice") and task_id:
|
||
notified_from_history.add(task_id)
|
||
except Exception:
|
||
notified_from_history = set()
|
||
for item in data:
|
||
task_id = item.get("task_id")
|
||
raw_task = manager.tasks.get(task_id) if task_id else None
|
||
run_in_background = bool(raw_task.get("run_in_background")) if isinstance(raw_task, dict) else False
|
||
item["run_in_background"] = run_in_background
|
||
status = item.get("status")
|
||
notified_flag = bool(raw_task.get("notified")) if isinstance(raw_task, dict) else False
|
||
already_notified = (
|
||
(task_id in announced) or
|
||
(task_id in notified_from_history) or
|
||
notified_flag
|
||
)
|
||
notice_pending = (
|
||
run_in_background
|
||
and task_id
|
||
and not already_notified
|
||
and (status in TERMINAL_STATUSES or status == "terminated")
|
||
)
|
||
item["notice_pending"] = notice_pending
|
||
debug_log(
|
||
"[SubAgent] /api/sub_agents notice_pending computed "
|
||
+ json.dumps({
|
||
"conversation_id": conversation_id,
|
||
"tasks": [
|
||
{
|
||
"task_id": item.get("task_id"),
|
||
"status": item.get("status"),
|
||
"run_in_background": item.get("run_in_background"),
|
||
"notice_pending": item.get("notice_pending"),
|
||
} for item in data
|
||
],
|
||
}, ensure_ascii=False)
|
||
)
|
||
return jsonify({"success": True, "data": data})
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/sub_agents/<task_id>/activity', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def get_sub_agent_activity(task_id: str, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""返回指定子智能体的活动记录（进度）。"""
|
||
manager = getattr(terminal, "sub_agent_manager", None)
|
||
if not manager:
|
||
return jsonify({"success": False, "error": "子智能体管理器不可用"}), 404
|
||
try:
|
||
try:
|
||
manager._load_state()
|
||
except Exception:
|
||
pass
|
||
|
||
task = manager.tasks.get(task_id)
|
||
if not task:
|
||
return jsonify({"success": False, "error": "未找到对应子智能体任务"}), 404
|
||
|
||
progress_file = task.get("progress_file")
|
||
if not progress_file:
|
||
task_root = task.get("task_root")
|
||
if task_root:
|
||
progress_file = str(Path(task_root) / "progress.jsonl")
|
||
|
||
entries: List[Dict[str, Any]] = []
|
||
# 默认返回足够多的记录，确保历史工具调用完整展示；同时保留上限防止异常大文件拖垮响应。
|
||
limit = request.args.get("limit", "100000")
|
||
try:
|
||
limit_num = max(1, min(int(limit), 100000))
|
||
except Exception:
|
||
limit_num = 100000
|
||
|
||
if progress_file and Path(progress_file).exists():
|
||
try:
|
||
lines = Path(progress_file).read_text(encoding="utf-8").splitlines()
|
||
if limit_num:
|
||
lines = lines[-limit_num:]
|
||
for line in lines:
|
||
line = line.strip()
|
||
if not line:
|
||
continue
|
||
try:
|
||
entries.append(json.loads(line))
|
||
except Exception:
|
||
continue
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": f"读取进度失败: {exc}"}), 500
|
||
|
||
payload = {
|
||
"task_id": task_id,
|
||
"status": task.get("status"),
|
||
"entries": entries,
|
||
}
|
||
return jsonify({"success": True, "data": payload})
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/sub_agents/<task_id>/terminate', methods=['POST'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def terminate_sub_agent(task_id: str, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""手动停止指定子智能体。"""
|
||
manager = getattr(terminal, "sub_agent_manager", None)
|
||
if not manager:
|
||
return jsonify({"success": False, "error": "子智能体管理器不可用"}), 404
|
||
try:
|
||
try:
|
||
manager._load_state()
|
||
except Exception:
|
||
pass
|
||
|
||
task = manager.tasks.get(task_id)
|
||
if not task:
|
||
return jsonify({"success": False, "error": "未找到对应子智能体任务"}), 404
|
||
|
||
current_conv_id = terminal.context_manager.current_conversation_id
|
||
task_conv_id = task.get("conversation_id")
|
||
if current_conv_id and task_conv_id and task_conv_id != current_conv_id:
|
||
return jsonify({"success": False, "error": "无权限停止该子智能体任务"}), 403
|
||
|
||
result = manager.terminate_sub_agent(task_id=task_id)
|
||
if not result.get("success"):
|
||
return jsonify({"success": False, "error": result.get("error") or "停止子智能体失败"}), 400
|
||
return jsonify({"success": True, "data": result})
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/background_commands', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def list_background_commands(terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""返回当前对话的后台 run_command 列表。"""
|
||
manager = getattr(terminal, "background_command_manager", None)
|
||
if not manager:
|
||
return jsonify({"success": True, "data": []})
|
||
try:
|
||
conversation_id = terminal.context_manager.current_conversation_id
|
||
limit_raw = request.args.get("limit", "200")
|
||
try:
|
||
limit_num = max(1, min(int(limit_raw), 1000))
|
||
except Exception:
|
||
limit_num = 200
|
||
|
||
records = manager.list_records(conversation_id=conversation_id, limit=limit_num)
|
||
data = []
|
||
terminal_statuses = {"completed", "failed", "timeout", "cancelled"}
|
||
for rec in records:
|
||
result = rec.get("result") if isinstance(rec.get("result"), dict) else {}
|
||
status = rec.get("status")
|
||
notice_pending = bool(
|
||
status in terminal_statuses
|
||
and not rec.get("notified")
|
||
and not rec.get("claimed_by_sleep")
|
||
)
|
||
data.append({
|
||
"command_id": rec.get("command_id"),
|
||
"status": status,
|
||
"command": rec.get("command"),
|
||
"conversation_id": rec.get("conversation_id"),
|
||
"created_at": rec.get("created_at"),
|
||
"updated_at": rec.get("updated_at"),
|
||
"finished_at": rec.get("finished_at"),
|
||
"timeout": rec.get("timeout"),
|
||
"return_code": result.get("return_code"),
|
||
"run_in_background": True,
|
||
"notice_pending": notice_pending,
|
||
})
|
||
return jsonify({"success": True, "data": data})
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/background_commands/<command_id>', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def get_background_command_detail(command_id: str, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""返回指定后台 run_command 的详情（含实时输出）。"""
|
||
manager = getattr(terminal, "background_command_manager", None)
|
||
if not manager:
|
||
return jsonify({"success": False, "error": "后台命令管理器不可用"}), 404
|
||
try:
|
||
rec = manager.get_record_with_output(command_id)
|
||
if not rec:
|
||
return jsonify({"success": False, "error": "未找到对应后台命令"}), 404
|
||
|
||
current_conv_id = terminal.context_manager.current_conversation_id
|
||
rec_conv_id = rec.get("conversation_id")
|
||
if current_conv_id and rec_conv_id and rec_conv_id != current_conv_id:
|
||
return jsonify({"success": False, "error": "无权限访问该后台命令"}), 403
|
||
|
||
result = rec.get("result") if isinstance(rec.get("result"), dict) else {}
|
||
payload = {
|
||
"command_id": rec.get("command_id"),
|
||
"status": rec.get("status"),
|
||
"command": rec.get("command"),
|
||
"conversation_id": rec.get("conversation_id"),
|
||
"created_at": rec.get("created_at"),
|
||
"updated_at": rec.get("updated_at"),
|
||
"finished_at": rec.get("finished_at"),
|
||
"timeout": rec.get("timeout"),
|
||
"return_code": result.get("return_code"),
|
||
"message": result.get("message"),
|
||
"output": rec.get("output") or "",
|
||
"run_in_background": True,
|
||
}
|
||
return jsonify({"success": True, "data": payload})
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/background_commands/<command_id>/cancel', methods=['POST'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def cancel_background_command(command_id: str, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""手动停止指定后台 run_command。"""
|
||
manager = getattr(terminal, "background_command_manager", None)
|
||
if not manager:
|
||
return jsonify({"success": False, "error": "后台命令管理器不可用"}), 404
|
||
try:
|
||
rec = manager.get_record(command_id)
|
||
if not rec:
|
||
return jsonify({"success": False, "error": "未找到对应后台命令"}), 404
|
||
current_conv_id = terminal.context_manager.current_conversation_id
|
||
rec_conv_id = rec.get("conversation_id")
|
||
if current_conv_id and rec_conv_id and rec_conv_id != current_conv_id:
|
||
return jsonify({"success": False, "error": "无权限停止该后台命令"}), 403
|
||
|
||
result = manager.cancel_command(command_id)
|
||
if not result.get("status"):
|
||
return jsonify({"success": False, "error": result.get("error") or "停止后台命令失败"}), 400
|
||
return jsonify({"success": True, "data": result})
|
||
except Exception as exc:
|
||
return jsonify({"success": False, "error": str(exc)}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/duplicate', methods=['POST'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def duplicate_conversation(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""复制指定对话，生成新的对话副本"""
|
||
|
||
try:
|
||
result = terminal.context_manager.duplicate_conversation(conversation_id)
|
||
|
||
if not result.get("success"):
|
||
status_code = 404 if "不存在" in result.get("error", "") else 400
|
||
return jsonify(result), status_code
|
||
|
||
new_conversation_id = result["duplicate_conversation_id"]
|
||
load_result = terminal.load_conversation(new_conversation_id)
|
||
|
||
if load_result.get("success"):
|
||
socketio.emit('conversation_list_update', {
|
||
'action': 'duplicated',
|
||
'conversation_id': new_conversation_id
|
||
}, room=f"user_{username}")
|
||
socketio.emit('conversation_changed', {
|
||
'conversation_id': new_conversation_id,
|
||
'title': load_result.get('title', '复制的对话'),
|
||
'messages_count': load_result.get('messages_count', 0)
|
||
}, room=f"user_{username}")
|
||
socketio.emit('conversation_loaded', {
|
||
'conversation_id': new_conversation_id,
|
||
'clear_ui': True
|
||
}, room=f"user_{username}")
|
||
|
||
response_payload = {
|
||
"success": True,
|
||
"duplicate_conversation_id": new_conversation_id,
|
||
"load_result": load_result
|
||
}
|
||
|
||
return jsonify(response_payload)
|
||
|
||
except Exception as e:
|
||
print(f"[API] 复制对话错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e),
|
||
"message": "复制对话时发生异常"
|
||
}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/review_preview', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def review_conversation_preview(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""生成对话回顾预览（不落盘，只返回前若干行文本）"""
|
||
policy = resolve_admin_policy(get_current_user_record())
|
||
if policy.get("ui_blocks", {}).get("block_conversation_review"):
|
||
return jsonify({"success": False, "error": "对话引用已被管理员禁用"}), 403
|
||
try:
|
||
current_id = terminal.context_manager.current_conversation_id
|
||
if conversation_id == current_id:
|
||
return jsonify({
|
||
"success": False,
|
||
"message": "无法引用当前对话"
|
||
}), 400
|
||
|
||
conversation_data = terminal.context_manager.conversation_manager.load_conversation(conversation_id)
|
||
if not conversation_data:
|
||
return jsonify({
|
||
"success": False,
|
||
"error": "Conversation not found",
|
||
"message": f"对话 {conversation_id} 不存在"
|
||
}), 404
|
||
|
||
limit = request.args.get('limit', default=20, type=int) or 20
|
||
lines = build_review_lines(conversation_data.get("messages", []), limit=limit)
|
||
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"preview": lines,
|
||
"count": len(lines)
|
||
}
|
||
})
|
||
except Exception as e:
|
||
print(f"[API] 对话回顾预览错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e),
|
||
"message": "生成预览时发生异常"
|
||
}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/review', methods=['POST'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def review_conversation(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""生成完整对话回顾 Markdown 文件"""
|
||
policy = resolve_admin_policy(get_current_user_record())
|
||
if policy.get("ui_blocks", {}).get("block_conversation_review"):
|
||
return jsonify({"success": False, "error": "对话引用已被管理员禁用"}), 403
|
||
try:
|
||
current_id = terminal.context_manager.current_conversation_id
|
||
if conversation_id == current_id:
|
||
return jsonify({
|
||
"success": False,
|
||
"message": "无法引用当前对话"
|
||
}), 400
|
||
|
||
conversation_data = terminal.context_manager.conversation_manager.load_conversation(conversation_id)
|
||
if not conversation_data:
|
||
return jsonify({
|
||
"success": False,
|
||
"error": "Conversation not found",
|
||
"message": f"对话 {conversation_id} 不存在"
|
||
}), 404
|
||
|
||
messages = conversation_data.get("messages", [])
|
||
lines = build_review_lines(messages)
|
||
content = "\n".join(lines) + "\n"
|
||
char_count = len(content)
|
||
|
||
review_dir = workspace.project_path / ".agents" / "review"
|
||
review_dir.mkdir(parents=True, exist_ok=True)
|
||
|
||
title = conversation_data.get("title") or "untitled"
|
||
safe_title = _sanitize_filename_component(title)
|
||
timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
|
||
filename = f"review_{safe_title}_{timestamp}.md"
|
||
target = review_dir / filename
|
||
|
||
target.write_text(content, encoding='utf-8')
|
||
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"path": f".agents/review/{filename}",
|
||
"char_count": char_count
|
||
}
|
||
})
|
||
except Exception as e:
|
||
print(f"[API] 对话回顾生成错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e),
|
||
"message": "生成对话回顾时发生异常"
|
||
}), 500
|
||
|
||
@conversation_bp.route('/api/conversations/statistics', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def get_conversations_statistics(terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""获取对话统计信息"""
|
||
try:
|
||
stats = terminal.context_manager.get_conversation_statistics()
|
||
|
||
return jsonify({
|
||
"success": True,
|
||
"data": stats
|
||
})
|
||
|
||
except Exception as e:
|
||
print(f"[API] 获取对话统计错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e),
|
||
"message": "获取对话统计时发生异常"
|
||
}), 500
|
||
|
||
@conversation_bp.route('/api/conversations/current', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def get_current_conversation(terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""获取当前对话信息"""
|
||
current_id = terminal.context_manager.current_conversation_id
|
||
|
||
# 如果是临时ID，返回空的对话信息
|
||
if not current_id or current_id.startswith('temp_'):
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"id": current_id,
|
||
"title": "新对话",
|
||
"messages_count": 0,
|
||
"is_temporary": True,
|
||
"title_locked": False
|
||
}
|
||
})
|
||
|
||
# 如果是真实的对话ID，查找对话数据
|
||
try:
|
||
conversation_data = terminal.context_manager.conversation_manager.load_conversation(current_id)
|
||
if conversation_data:
|
||
metadata = conversation_data.get("metadata", {}) or {}
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"id": current_id,
|
||
"title": conversation_data.get("title", "未知对话"),
|
||
"messages_count": len(conversation_data.get("messages", [])),
|
||
"is_temporary": False,
|
||
"title_locked": bool(metadata.get("title_locked", False))
|
||
}
|
||
})
|
||
else:
|
||
return jsonify({
|
||
"success": False,
|
||
"error": "对话不存在"
|
||
}), 404
|
||
|
||
except Exception as e:
|
||
print(f"[API] 获取当前对话错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e)
|
||
}), 500
|
||
|
||
@socketio.on('send_command')
|
||
def handle_command(data):
|
||
"""处理系统命令"""
|
||
command = data.get('command', '')
|
||
|
||
username, terminal, _ = get_terminal_for_sid(request.sid)
|
||
if not terminal:
|
||
emit('error', {'message': 'System not initialized'})
|
||
return
|
||
record_user_activity(username)
|
||
|
||
result = _execute_system_command(terminal, command)
|
||
emit('command_result', result)
|
||
|
||
|
||
def _execute_system_command(terminal: WebTerminal, command: str) -> Dict[str, Any]:
|
||
"""执行系统命令，供 WebSocket 与 REST API 复用。"""
|
||
command = (command or '').strip()
|
||
if command.startswith('/'):
|
||
command = command[1:]
|
||
|
||
parts = command.split(maxsplit=1)
|
||
cmd = parts[0].lower() if parts else ''
|
||
|
||
if cmd == "clear":
|
||
terminal.context_manager.conversation_history.clear()
|
||
if terminal.thinking_mode:
|
||
terminal.api_client.start_new_task(force_deep=terminal.deep_thinking_mode)
|
||
return {
|
||
'command': cmd,
|
||
'success': True,
|
||
'message': '对话已清除'
|
||
}
|
||
|
||
if cmd == "status":
|
||
status = terminal.get_status()
|
||
if terminal.terminal_manager:
|
||
status['terminals'] = terminal.terminal_manager.list_terminals()
|
||
return {
|
||
'command': cmd,
|
||
'success': True,
|
||
'data': status
|
||
}
|
||
|
||
if cmd == "terminals":
|
||
if terminal.terminal_manager:
|
||
return {
|
||
'command': cmd,
|
||
'success': True,
|
||
'data': terminal.terminal_manager.list_terminals()
|
||
}
|
||
return {
|
||
'command': cmd,
|
||
'success': False,
|
||
'message': '终端系统未初始化'
|
||
}
|
||
|
||
return {
|
||
'command': cmd or command,
|
||
'success': False,
|
||
'message': f'未知命令: {cmd or command}'
|
||
}
|
||
|
||
|
||
@conversation_bp.route('/api/commands', methods=['POST'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def execute_command_api(terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""通过 REST API 执行系统命令（用于无 WebSocket 前端）。"""
|
||
try:
|
||
payload = request.get_json(silent=True) or {}
|
||
command = (payload.get('command') or '').strip()
|
||
if not command:
|
||
return jsonify({
|
||
"success": False,
|
||
"message": "命令不能为空"
|
||
}), 400
|
||
|
||
record_user_activity(username)
|
||
result = _execute_system_command(terminal, command)
|
||
status_code = 200 if result.get('success') else 400
|
||
return jsonify(result), status_code
|
||
except Exception as exc:
|
||
return jsonify({
|
||
"success": False,
|
||
"message": f"命令执行异常: {exc}"
|
||
}), 500
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/token-statistics', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def get_conversation_token_statistics(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""获取特定对话的token统计"""
|
||
try:
|
||
stats = terminal.context_manager.get_conversation_token_statistics(conversation_id)
|
||
|
||
if stats:
|
||
return jsonify({
|
||
"success": True,
|
||
"data": stats
|
||
})
|
||
else:
|
||
return jsonify({
|
||
"success": False,
|
||
"error": "Conversation not found",
|
||
"message": f"对话 {conversation_id} 不存在"
|
||
}), 404
|
||
|
||
except Exception as e:
|
||
print(f"[API] 获取token统计错误: {e}")
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e),
|
||
"message": "获取token统计时发生异常"
|
||
}), 500
|
||
|
||
|
||
@conversation_bp.route('/api/conversations/<conversation_id>/tokens', methods=['GET'])
|
||
@api_login_required
|
||
@with_terminal
|
||
def get_conversation_tokens(conversation_id, terminal: WebTerminal, workspace: UserWorkspace, username: str):
|
||
"""获取对话的当前完整上下文token数（包含所有动态内容）"""
|
||
try:
|
||
current_tokens = terminal.context_manager.get_current_context_tokens(conversation_id)
|
||
return jsonify({
|
||
"success": True,
|
||
"data": {
|
||
"total_tokens": current_tokens
|
||
}
|
||
})
|
||
except Exception as e:
|
||
return jsonify({
|
||
"success": False,
|
||
"error": str(e)
|
||
}), 500
|