File size: 9,062 Bytes
c2179b0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 | """
AgentFrame REST API v1
=======================
多会话管理: 每个 session 独立 ContextEngine 实例
端点:
GET /health 健康检查
POST /v1/sessions 创建会话
GET /v1/sessions 列出会话
DELETE /v1/sessions/<sid> 删除会话
POST /v1/sessions/<sid>/ingest 摄入知识
POST /v1/sessions/<sid>/ingest_batch 批量摄入
POST /v1/sessions/<sid>/ask 查询+生成
POST /v1/sessions/<sid>/ask_hands 查询+生成(带工具)
GET /v1/sessions/<sid>/stats 会话状态
POST /v1/sessions/<sid>/forget 触发遗忘
POST /v1/sessions/<sid>/save 保存快照
POST /v1/sessions/<sid>/load 恢复快照
"""
import os
import uuid
from flask import Flask, request, jsonify
from ..config import AgentFrameConfig
from ..core.engine import ContextEngine
class AgentFrameAPI:
"""Flask 应用工厂"""
def __init__(self, config: AgentFrameConfig = None, state_dir: str = None):
self.config = config or AgentFrameConfig.from_env()
self.state_dir = state_dir or os.environ.get(
"AGENTFRAME_STATE_DIR", "/var/lib/agentframe")
os.makedirs(self.state_dir, exist_ok=True)
self.sessions = {} # sid -> ContextEngine
self.app = Flask("agentframe")
self._register_routes()
# ============ 会话管理 ============
def _get_session(self, sid: str) -> ContextEngine:
if sid not in self.sessions:
raise KeyError(f"会话不存在: {sid}")
return self.sessions[sid]
def _create_session(self) -> str:
sid = uuid.uuid4().hex[:12]
self.sessions[sid] = ContextEngine(self.config)
return sid
def _restore_sessions(self):
"""启动时自动恢复已保存的会话快照"""
if not os.path.isdir(self.state_dir):
return
for fname in sorted(os.listdir(self.state_dir)):
if not fname.endswith(".json"):
continue
sid = fname[:-5]
if sid in self.sessions:
continue
eng = ContextEngine(self.config)
try:
if eng.load(os.path.join(self.state_dir, fname)):
self.sessions[sid] = eng
print(f" ↻ 恢复会话 {sid}")
except Exception as e:
print(f" ⚠️ 恢复失败 {sid}: {e}")
# ============ 路由 ============
def _register_routes(self):
app = self.app
self._restore_sessions()
@app.get("/health")
def health():
return jsonify({"status": "ok", "version": "1.0.0",
"sessions": len(self.sessions),
"system": "AgentFrame Context API"})
@app.post("/v1/sessions")
def create_session():
sid = self._create_session()
return jsonify({"session_id": sid, "ok": True}), 201
@app.get("/v1/sessions")
def list_sessions():
return jsonify({
"sessions": [
{"id": sid, "chunks": len(eng.agent.chunk_meta),
"turns": len(eng.chat_history) // 2}
for sid, eng in self.sessions.items()
],
"total": len(self.sessions),
})
@app.delete("/v1/sessions/<sid>")
def delete_session(sid):
if sid in self.sessions:
del self.sessions[sid]
return jsonify({"ok": True, "deleted": sid})
return jsonify({"error": "会话不存在"}), 404
@app.post("/v1/sessions/<sid>/ingest")
def ingest(sid):
try:
eng = self._get_session(sid)
except KeyError as e:
return jsonify({"error": str(e)}), 404
data = request.get_json(force=True, silent=True) or {}
text = data.get("text", "")
if not text:
return jsonify({"error": "缺少 text 字段"}), 400
cid = eng.ingest(text, data.get("tags", []), data.get("source"))
return jsonify({"ok": True, "chunk_id": cid, "state": eng.stats()})
@app.post("/v1/sessions/<sid>/ingest_batch")
def ingest_batch(sid):
try:
eng = self._get_session(sid)
except KeyError as e:
return jsonify({"error": str(e)}), 404
data = request.get_json(force=True, silent=True) or {}
chunks = data.get("chunks", [])
if not chunks:
return jsonify({"error": "缺少 chunks 字段"}), 400
ids = eng.ingest_batch(chunks)
return jsonify({"ok": True, "chunk_ids": ids})
@app.post("/v1/sessions/<sid>/ask")
def ask(sid):
try:
eng = self._get_session(sid)
except KeyError as e:
return jsonify({"error": str(e)}), 404
data = request.get_json(force=True, silent=True) or {}
q = data.get("query", "")
if not q:
return jsonify({"error": "缺少 query 字段"}), 400
result = eng.ask(q, chat=data.get("chat", True),
use_fast=data.get("use_fast", False))
return jsonify({"ok": True, **result.to_dict()})
@app.post("/v1/sessions/<sid>/ask_hands")
def ask_hands(sid):
try:
eng = self._get_session(sid)
except KeyError as e:
return jsonify({"error": str(e)}), 404
data = request.get_json(force=True, silent=True) or {}
q = data.get("query", "")
if not q:
return jsonify({"error": "缺少 query 字段"}), 400
result = eng.ask_with_hands(q)
return jsonify({"ok": True, **result.to_dict()})
@app.post("/v1/sessions/<sid>/autopilot")
def autopilot(sid):
"""自主记忆管理: 分析对话, 自动记/忘"""
try:
eng = self._get_session(sid)
except KeyError as e:
return jsonify({"error": str(e)}), 404
data = request.get_json(force=True, silent=True) or {}
q = data.get("query", "")
if not q:
return jsonify({"error": "缺少 query 字段"}), 400
# 用最近一次回答做记忆分析
answer = ""
if eng.chat_history and eng.chat_history[-1].get("role") == "assistant":
answer = eng.chat_history[-1]["content"]
decision = eng.autopilot_memory(q, answer)
return jsonify({"ok": True, **decision})
@app.get("/v1/sessions/<sid>/stats")
def stats(sid):
try:
eng = self._get_session(sid)
except KeyError as e:
return jsonify({"error": str(e)}), 404
return jsonify(eng.stats())
@app.post("/v1/sessions/<sid>/forget")
def forget(sid):
try:
eng = self._get_session(sid)
except KeyError as e:
return jsonify({"error": str(e)}), 404
data = request.get_json(force=True, silent=True) or {}
threshold = float(data.get("threshold", 0.15))
victims = eng.forget(threshold)
return jsonify({"ok": True, "forgotten": victims,
"remaining": len(eng.agent.chunk_meta)})
@app.post("/v1/sessions/<sid>/save")
def save(sid):
try:
eng = self._get_session(sid)
except KeyError as e:
return jsonify({"error": str(e)}), 404
path = os.path.join(self.state_dir, f"{sid}.json")
eng.save(path)
return jsonify({"ok": True, "path": path})
@app.post("/v1/sessions/<sid>/load")
def load(sid):
try:
eng = self._get_session(sid)
except KeyError as e:
return jsonify({"error": str(e)}), 404
path = os.path.join(self.state_dir, f"{sid}.json")
ok = eng.load(path)
return jsonify({"ok": ok, "path": path})
def create_app(config: AgentFrameConfig = None, state_dir: str = None) -> Flask:
"""应用工厂 (供 gunicorn/flask run 使用)"""
return AgentFrameAPI(config, state_dir).app
def main():
"""直接运行: python -m agentframe.api.server [port]"""
import sys
config = AgentFrameConfig.from_env()
port = int(sys.argv[1]) if len(sys.argv) > 1 else config.api.port
api = AgentFrameAPI(config)
print(f"🚀 AgentFrame API v1.0.0")
print(f" 四层上下文保持: 认知×路由×存储×物理")
print(f" 会话: POST /v1/sessions | 摄入: /ingest | 查询: /ask")
print(f" 监听: {config.api.host}:{port}")
api.app.run(host=config.api.host, port=port, debug=config.api.debug)
if __name__ == "__main__":
main()
|