qiaoqichaoandClaude Code
d56c699ce6
按 docs/v2/05-implementation-plan.md §1 候选迁移表与 claude_use/T00 白名单， 从 55e2cdb 抽取 224 个文件（2.48 MB）到独立目录 F:/mini-OpenClaw-v2。 - 排除：凭据(.env/server-deploy.conf/旧DELIVERY.md)、运行数据(xingbang_data 等 4 个 SQLite)、依赖(venv/node_modules 符号链接)、归档(_backup*/test-evidence/ ori_file)、Office 临时文件 - 目标目录独立 git 仓库，初始恢复点；Gitea 远程与分支待 OQ-06 Q2 定案 - 未部署、未启动服务、未调用模型 Co-Authored-By: Claude Code <noreply@anthropic.com>
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
"""POST /api/sessions/{session_id}/compress — Compress conversation history."""
|
|
|
|
import os
|
|
import traceback
|
|
from typing import Any
|
|
|
|
from fastapi import APIRouter, HTTPException
|
|
from langchain_core.messages import HumanMessage
|
|
|
|
from graph.session_manager import session_manager
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
async def _generate_summary(messages: list[dict[str, Any]]) -> str:
|
|
"""Use DeepSeek to generate a compressed summary of messages."""
|
|
from langchain_deepseek import ChatDeepSeek
|
|
|
|
llm = ChatDeepSeek(
|
|
model=os.getenv("DEEPSEEK_MODEL", "deepseek-chat"),
|
|
api_key=os.getenv("DEEPSEEK_API_KEY"),
|
|
api_base=os.getenv("DEEPSEEK_BASE_URL", "https://api.deepseek.com"),
|
|
temperature=0.3,
|
|
)
|
|
|
|
# Format messages for summary
|
|
formatted = []
|
|
for msg in messages:
|
|
role = msg.get("role", "unknown")
|
|
content = msg.get("content", "")
|
|
if content:
|
|
formatted.append(f"{role}: {content[:500]}")
|
|
|
|
conversation_text = "\n".join(formatted)
|
|
|
|
prompt = (
|
|
"请将以下对话历史压缩为简洁的中文摘要，保留关键信息、决策和结论。"
|
|
"摘要不超过500字。只输出摘要内容，不要添加额外说明。\n\n"
|
|
f"{conversation_text}"
|
|
)
|
|
|
|
result = await llm.ainvoke([HumanMessage(content=prompt)])
|
|
return result.content.strip()
|
|
|
|
|
|
@router.post("/sessions/{session_id}/compress")
|
|
async def compress_session(session_id: str) -> dict[str, Any]:
|
|
"""Compress the first 50% of conversation history into a summary."""
|
|
messages = session_manager.load_session(session_id)
|
|
if len(messages) < 4:
|
|
raise HTTPException(
|
|
status_code=400,
|
|
detail="Not enough messages to compress (need at least 4)",
|
|
)
|
|
|
|
# Take the first 50% of messages
|
|
num_to_remove = max(4, len(messages) // 2)
|
|
|
|
messages_to_compress = messages[:num_to_remove]
|
|
|
|
try:
|
|
summary = await _generate_summary(messages_to_compress)
|
|
session_manager.compress_history(session_id, summary, num_to_remove)
|
|
remaining = len(messages) - num_to_remove
|
|
return {
|
|
"archived_count": num_to_remove,
|
|
"remaining_count": remaining,
|
|
}
|
|
except Exception as e:
|
|
traceback.print_exc()
|
|
raise HTTPException(status_code=500, detail=f"Compression failed: {str(e)}")
|