chore: mini-openclaw 初始版本
Next.js 前端 + FastAPI/LangChain 后端透明 agent平台 - 含 .gitignore, 敏感文件(sessions数据/.env/venv等)已排除 - 待后续添加 trajectory 自进化机制
This commit is contained in:
@@ -0,0 +1,71 @@
|
||||
"""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)}")
|
||||
Reference in New Issue
Block a user