JOJO
5459d82637
CLI（bun 跑源码即生效）： - boot：loadBootData 一次性加载模型清单（含 context_window）/个性化默认 （含压缩设置）/path 授权/会话列表；新会话 unshift 标 current - /session：Enter 真实加载（timeline.reset + getSessionHistory 渲染历史文本） - /new：对齐 web /new 页面语义——空对话草稿不立即建会话，首条消息 createTask 不带 cid 由服务端惰性创建，响应回 id 后插入列表标 current - /context + 状态栏：token_update 事件（运行期事件流）+ 打开面板时查询 新端点 GET /api/runtime/sessions/<cid>/token-stats；百分比算法对齐 web InputComposer（自动压缩阈值优先于模型窗口） - /model：Enter = 本地生效 + createTask 带 model_key/run_mode/thinking_mode 覆盖（会话级即时）+ savePersonalization patch 存默认（effort default 存 null） - /path：增删后 savePathAuths 全量提交两组（服务端 deny 校验保留） - /agents /tasks /workflow /rewind：打开时拉真实列表（sub_agents/ background_commands/workflows/versioning-checkpoints） - 发送链路统一 sendWithSettings 携带 /model 生效值（含排队队列） 后端： - 8 端点装饰器 api_login_required → api_login_or_host_token_required （personalization×2 / path-authorization×2 / sub_agents / background_commands / workflows / versioning-checkpoints），web session 行为不变 - headless_app 用 add_url_rule 共享上述视图函数（URL 与 full 一致）； sub_agents/background_commands 支持 query 显式传 conversation_id （Bearer 装配的 terminal 无当前对话概念） - RuntimeService 新增 get_session_token_stats；gateway_api 挂 token-stats 路由 - headless 404 兜底：非 /api/ 的 GET 路径回落告知页（对齐 SPA fallback） 验证：tsc ✅ py_compile ✅ 冒烟 6/6 ✅ headless 装配 54 路由 ✅ localconfig 无头实测读到真实配置 ✅；真实服务端到端待用户重启 8091 实机验收 Co-authored-by: Astrion powered by Kimi-K3 <astrion-agent@users.noreply.github.com>
238 lines
9.5 KiB
Python
238 lines
9.5 KiB
Python
"""Astrion headless 后端入口（Gateway + Runtime，无 Web 站点职责）。
|
||
|
||
与 `server.app`（full 形态）是**同一后端进程的两种启动方式**：同一 RuntimeService
|
||
单例、同一数据目录、同一端口（默认 8091）。区别仅在启动时挂载的路由面——
|
||
headless 只挂 CLI / 桌面 / API 客户端依赖的 API 蓝图：
|
||
|
||
gateway_bp : /api/runtime/sessions（CLI 主通道：创建会话 / 拉历史）
|
||
tasks : run.* 任务域（发消息 / 事件轮询 / 取消）
|
||
status_bp : /api/health 探活、/api/status、host_workspace 工作区管理
|
||
approval_bp : 工具 / 计划 / 提问三组审批（Runtime 人机交互契约）
|
||
usage_bp : /api/usage 配额查询
|
||
|
||
被砍的「web 站点职责」（静态页面、web 登录、旧版通道等）在本形态下访问返回 404；
|
||
GET / 返回形态告知页（避免浏览器打开时误判为服务故障）。
|
||
|
||
注意：本入口通过 import server.app_legacy 复用 initialize_system / start_background_jobs，
|
||
因此 import 链与 full 形态一致（app_legacy 模块级创建的 app 对象不会被 serve）。
|
||
headless 精简的是**路由面**（攻击面收敛 + CLI 依赖契约显式化），不是 import 链。
|
||
|
||
启动：python -m server.headless_app [--path PATH] [--port 8091] [--thinking-mode] [--debug]
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import argparse
|
||
import os
|
||
import secrets
|
||
from datetime import timedelta
|
||
from pathlib import Path
|
||
|
||
from flask import Flask, jsonify, request
|
||
|
||
from config import (
|
||
DEFAULT_PROJECT_PATH,
|
||
MAX_UPLOAD_SIZE,
|
||
OUTPUT_FORMATS,
|
||
WEB_SERVER_HOST,
|
||
WEB_SERVER_PORT,
|
||
)
|
||
|
||
DEFAULT_PORT = WEB_SERVER_PORT
|
||
|
||
# 复用 full 形态的初始化与后台任务（模块级 app 对象仅被创建，不会被 serve）
|
||
from server.app_legacy import initialize_system, start_background_jobs # noqa: F401
|
||
from server.gateway_api import gateway_bp
|
||
from server.tasks.web import get_tasks_blueprint
|
||
from server.status import status_bp
|
||
from server.chat.approval import approval_bp
|
||
from server.usage import usage_bp
|
||
from server.security import attach_security_hooks
|
||
|
||
HEADLESS_BLUEPRINTS = (
|
||
("gateway", gateway_bp),
|
||
("tasks", get_tasks_blueprint()),
|
||
("status", status_bp),
|
||
("approval", approval_bp),
|
||
("usage", usage_bp),
|
||
)
|
||
|
||
_HEADLESS_LANDING_HTML = """<!DOCTYPE html>
|
||
<html lang="zh-CN">
|
||
<head>
|
||
<meta charset="utf-8">
|
||
<meta name="viewport" content="width=device-width, initial-scale=1">
|
||
<title>Astrion · API 后端实例</title>
|
||
<style>
|
||
* { margin: 0; padding: 0; box-sizing: border-box; }
|
||
body {
|
||
min-height: 100vh;
|
||
display: flex;
|
||
align-items: center;
|
||
justify-content: center;
|
||
background: #faf9f5;
|
||
color: #1f1f1f;
|
||
font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "PingFang SC",
|
||
"Hiragino Sans GB", "Microsoft YaHei", sans-serif;
|
||
}
|
||
.card { max-width: 560px; padding: 40px 44px; }
|
||
.brand { display: flex; align-items: center; gap: 12px; margin-bottom: 20px; }
|
||
.brand-name { font-size: 22px; font-weight: 650; letter-spacing: 0.2px; }
|
||
.badge {
|
||
font-size: 12px;
|
||
font-weight: 600;
|
||
padding: 3px 10px;
|
||
border-radius: 999px;
|
||
background: rgba(204, 120, 92, 0.14);
|
||
color: #b05c3f;
|
||
letter-spacing: 0.4px;
|
||
}
|
||
h1 { font-size: 17px; font-weight: 600; margin-bottom: 10px; }
|
||
p { font-size: 14px; line-height: 1.75; color: #55534e; margin-bottom: 10px; }
|
||
code {
|
||
font-family: "JetBrains Mono", "SF Mono", Menlo, Consolas, monospace;
|
||
font-size: 12.5px;
|
||
background: #efe9de;
|
||
padding: 2px 6px;
|
||
border-radius: 5px;
|
||
}
|
||
.cmd {
|
||
display: block;
|
||
margin-top: 16px;
|
||
padding: 12px 14px;
|
||
background: #f5f0e8;
|
||
border: 1px solid #e8e0d2;
|
||
border-radius: 8px;
|
||
font-family: "JetBrains Mono", "SF Mono", Menlo, Consolas, monospace;
|
||
font-size: 12.5px;
|
||
color: #3d3929;
|
||
}
|
||
.muted { font-size: 12px; color: #8a8578; margin-top: 18px; }
|
||
</style>
|
||
</head>
|
||
<body>
|
||
<div class="card">
|
||
<div class="brand">
|
||
<span class="brand-name">Astrion</span>
|
||
<span class="badge">HEADLESS</span>
|
||
</div>
|
||
<h1>这是 Astrion 的 API 后端实例，运行正常。</h1>
|
||
<p>
|
||
本实例以 <strong>headless 形态</strong>启动：只提供 Gateway / Runtime API
|
||
（供 CLI、桌面端、Android 等客户端使用），不包含 Web 界面。
|
||
</p>
|
||
<p>如需 Web 界面，请改用 full 形态启动：</p>
|
||
<span class="cmd">python -m server.app</span>
|
||
<p class="muted">Astrion headless backend · same runtime, API-only surface</p>
|
||
</div>
|
||
</body>
|
||
</html>
|
||
"""
|
||
|
||
|
||
def create_headless_app() -> Flask:
|
||
"""创建 headless 形态的 Flask app（最小 API 路由面）。"""
|
||
app = Flask(__name__)
|
||
app.config['MAX_CONTENT_LENGTH'] = MAX_UPLOAD_SIZE
|
||
# Flask session 仍被 Bearer 通道的身份注入使用（请求内 session 对象），
|
||
# 必须配置 SECRET_KEY；解析规则与 full 形态一致。
|
||
_secret_key = os.environ.get("WEB_SECRET_KEY") or os.environ.get("SECRET_KEY")
|
||
if not _secret_key:
|
||
_secret_key = secrets.token_hex(32)
|
||
print(f"{OUTPUT_FORMATS['warning']} WEB_SECRET_KEY 未设置，已生成临时密钥（重启后所有会话将失效）。")
|
||
app.config['SECRET_KEY'] = _secret_key
|
||
app.config['PERMANENT_SESSION_LIFETIME'] = timedelta(hours=12)
|
||
app.config['SESSION_COOKIE_SAMESITE'] = os.environ.get("WEB_COOKIE_SAMESITE", "Strict")
|
||
app.config['SESSION_COOKIE_HTTPONLY'] = True
|
||
_cookie_secure_env = (os.environ.get("WEB_COOKIE_SECURE") or "").strip().lower()
|
||
app.config['SESSION_COOKIE_SECURE'] = _cookie_secure_env in {"1", "true", "yes"}
|
||
|
||
attach_security_hooks(app)
|
||
|
||
for _name, bp in HEADLESS_BLUEPRINTS:
|
||
app.register_blueprint(bp)
|
||
|
||
# host 全局设置面（personalization / path-authorization）：视图函数定义在 chat_bp 模块内，
|
||
# 装饰器已双通道化（web session 或 host Bearer 均可）；此处直接注册到 headless app，
|
||
# URL 与 full 形态一致（Blueprint.route 只登记不包装，函数对象可安全复用）。
|
||
from server.chat.settings import (
|
||
get_personalization_settings,
|
||
update_personalization_settings,
|
||
)
|
||
from server.chat.permission import (
|
||
get_path_authorization,
|
||
update_path_authorization,
|
||
)
|
||
from server.conversation import list_background_commands, list_sub_agents, list_conversation_versioning_checkpoints
|
||
from server.workflow_page import api_list_workflows
|
||
|
||
app.add_url_rule('/api/personalization', view_func=get_personalization_settings, methods=['GET'])
|
||
app.add_url_rule('/api/personalization', view_func=update_personalization_settings, methods=['POST'])
|
||
app.add_url_rule('/api/path-authorization', view_func=get_path_authorization, methods=['GET'])
|
||
app.add_url_rule('/api/path-authorization', view_func=update_path_authorization, methods=['POST'])
|
||
# 子智能体/后台指令列表（/agents /tasks 面板数据源；conversation_id 走 query 显式指定）
|
||
app.add_url_rule('/api/sub_agents', view_func=list_sub_agents, methods=['GET'])
|
||
app.add_url_rule('/api/background_commands', view_func=list_background_commands, methods=['GET'])
|
||
# 工作流库列表（/workflow）与版本回溯检查点（/rewind）
|
||
app.add_url_rule('/api/workflows', view_func=api_list_workflows, methods=['GET'])
|
||
app.add_url_rule(
|
||
'/api/conversations/<conversation_id>/versioning/checkpoints',
|
||
view_func=list_conversation_versioning_checkpoints,
|
||
methods=['GET'],
|
||
)
|
||
|
||
@app.route('/')
|
||
def headless_landing():
|
||
return _HEADLESS_LANDING_HTML
|
||
|
||
@app.errorhandler(404)
|
||
def headless_not_found(_err):
|
||
# 非 API 的 GET 路径（/new、/<conv_id> 等前端路由）统一回落告知页——
|
||
# 对齐 full 形态的 SPA fallback 行为，避免浏览器打开收藏链接看到裸 404；
|
||
# API 路径保持 JSON 404。
|
||
if request.method == 'GET' and not request.path.startswith('/api/'):
|
||
return _HEADLESS_LANDING_HTML
|
||
return jsonify({"success": False, "error": "not found"}), 404
|
||
|
||
return app
|
||
|
||
|
||
def run_headless(path: str, thinking_mode: bool = False, port: int = DEFAULT_PORT, debug: bool = False):
|
||
"""初始化并以 headless 形态运行（与 full 形态共享初始化与后台任务）。"""
|
||
app = create_headless_app()
|
||
# 与 full 形态一致：同机多实例时 Cookie 按端口隔离，避免互相覆盖
|
||
app.config['SESSION_COOKIE_NAME'] = os.environ.get(
|
||
"WEB_SESSION_COOKIE_NAME", f"agents_session_{port}"
|
||
)
|
||
initialize_system(path, thinking_mode)
|
||
start_background_jobs()
|
||
app.run(
|
||
host=WEB_SERVER_HOST,
|
||
port=port,
|
||
debug=debug,
|
||
use_reloader=debug,
|
||
threaded=True,
|
||
)
|
||
|
||
|
||
def parse_arguments():
|
||
parser = argparse.ArgumentParser(description="Astrion headless backend (Gateway + Runtime, API-only)")
|
||
parser.add_argument(
|
||
"--path",
|
||
default=str(Path(DEFAULT_PROJECT_PATH).resolve()),
|
||
help="默认工作区路径（仅作兜底，工作区可在客户端中管理）",
|
||
)
|
||
parser.add_argument("--port", type=int, default=DEFAULT_PORT, help="监听端口")
|
||
parser.add_argument("--thinking-mode", action="store_true", help="启用思考模式")
|
||
parser.add_argument("--debug", action="store_true", help="调试模式")
|
||
return parser.parse_args()
|
||
|
||
|
||
def main():
|
||
args = parse_arguments()
|
||
print(f"{OUTPUT_FORMATS['info']} 以 headless 形态启动（仅 API 路由面，无 Web 界面）...")
|
||
run_headless(path=args.path, thinking_mode=args.thinking_mode, port=args.port, debug=args.debug)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|