tzt
ddef8bec1c
- OpenAICompatChat 默认 stream=True：httpx 流式解析 OpenAI chunk， tool_calls 碎片按 index 组装（不在正文展示），include_usage 计量； 失败自动回退非流式一次（已有部分增量输出则如实抛出）； 补 raise_for_status（修复流式 4xx 误入回退的缺陷） - ToolLoop 增 on_delta（签名探测兼容 2/3 参 chat_fn）；_loads_json_object 宽松解析 - DeltaThrottle >=48 字符节流落 delta 事件；单模型与两级模式（含规划者）接线 - 前端：streamText 打字机渲染 + 光标动画；工具/阶段事件到达时清空归档 - 配套修复：AgentView 闭包持有 push 前原始对象导致响应式丢失、过程事件不渲染 - 测试 +7（SSE 解析/碎片组装/回退/部分失败抛出/on_delta/节流/service delta 事件）， 全量 296 passed
128 lines
4.9 KiB
Python
128 lines
4.9 KiB
Python
"""可调整的运行设置（SettingsStore）—— 让用户自定义内置小模型 / 大模型 / 管线。
|
||
|
||
用户可在 Web 界面"模型设置"里调整并持久化到 config/settings.json（gitignore），
|
||
重启后保留。调整会触发 v2 管线重建（gateway.api.build_v2_pipeline 重新读取）。
|
||
|
||
默认值与 config/config.yaml 的 v2 段一致；settings.json 只存用户改动覆盖项。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
from pathlib import Path
|
||
from typing import Any, Dict, Optional
|
||
|
||
_SETTINGS_PATH = Path(__file__).resolve().parent.parent / "config" / "settings.json"
|
||
|
||
# 可调整项（含默认值）；用户改动存这里
|
||
DEFAULTS: Dict[str, Any] = {
|
||
"worker": {
|
||
"backend": "llama_server", # mock | openai/api | llama_server
|
||
"model": "models/qwen3.5-4b-q4_k_m.gguf",
|
||
"base_url": "", # openai 后端填 http://127.0.0.1:11434/v1 等
|
||
"port": 8901, # llama_server 端口
|
||
"temperature": 0.3,
|
||
"max_fix_attempts": 2,
|
||
"per_step_timeout_s": 15,
|
||
"code_timeout_s": 10,
|
||
},
|
||
"architect": {
|
||
"model": "deepseek-v4-flash",
|
||
"base_url": "https://api.deepseek.com",
|
||
"api_key": "",
|
||
},
|
||
"pipeline": {
|
||
"fast_path": True,
|
||
"rounds_cap": 6,
|
||
"api_token_cap": 8000,
|
||
"breach_policy": "architect_do",
|
||
},
|
||
"agent": {
|
||
"workspace_dir": "agent_workspace", # 智能体工作区根目录（越界即拒）
|
||
"recent_workspaces": [], # 最近打开的工作区（供快速切换）
|
||
"max_rounds": 8, # 工具循环轮数上限
|
||
"token_cap": 20000, # 单次智能体任务 token 熔断
|
||
"allow_shell": False, # 允许 run_command 执行 shell（默认关）
|
||
"shell_timeout_s": 20, # shell 命令超时
|
||
"allow_net": True, # 允许 web_fetch 抓取公网页面（SSRF 防护内置）
|
||
"max_handoffs": 2, # 两级模式：规划者<->执行者交接轮数上限
|
||
"approval_policy": "dangerous", # 审批策略：off | dangerous（写/编辑/命令询问）| all
|
||
"approval_timeout_s": 120, # 审批等待超时（超时自动拒绝）
|
||
},
|
||
}
|
||
|
||
|
||
class SettingsStore:
|
||
"""用户可调整设置（内存 + settings.json 持久化）。"""
|
||
|
||
def __init__(self, path: Optional[Path] = None):
|
||
self._path = Path(path) if path else _SETTINGS_PATH
|
||
self._data: Dict[str, Any] = {}
|
||
self.load()
|
||
|
||
# ---------- 持久化 ----------
|
||
def load(self) -> None:
|
||
if self._path.exists():
|
||
try:
|
||
self._data = json.loads(self._path.read_text(encoding="utf-8"))
|
||
except Exception:
|
||
self._data = {}
|
||
else:
|
||
self._data = {}
|
||
|
||
def save(self) -> None:
|
||
self._path.parent.mkdir(parents=True, exist_ok=True)
|
||
self._path.write_text(
|
||
json.dumps(self._data, ensure_ascii=False, indent=2), encoding="utf-8")
|
||
|
||
# ---------- 访问 ----------
|
||
def to_dict(self) -> Dict[str, Any]:
|
||
"""返回 默认值 + 用户覆盖 合并后的完整设置。"""
|
||
merged: Dict[str, Any] = {}
|
||
for section, defaults in DEFAULTS.items():
|
||
ov = self._data.get(section, {})
|
||
merged[section] = {**defaults, **(ov if isinstance(ov, dict) else {})}
|
||
return merged
|
||
|
||
def get(self, section: str, key: str, default: Any = None) -> Any:
|
||
merged = self.to_dict()
|
||
return merged.get(section, {}).get(key, default)
|
||
|
||
def update(self, patch: Dict[str, Any]) -> Dict[str, Any]:
|
||
"""应用部分更新（可只传改动的 section/key）。返回合并后的完整设置。"""
|
||
for section, values in patch.items():
|
||
if section not in DEFAULTS or not isinstance(values, dict):
|
||
continue
|
||
cur = self._data.setdefault(section, {})
|
||
for k, v in values.items():
|
||
if k in DEFAULTS[section]:
|
||
cur[k] = _coerce(v, DEFAULTS[section][k])
|
||
self.save()
|
||
return self.to_dict()
|
||
|
||
def reset(self) -> Dict[str, Any]:
|
||
"""恢复默认。"""
|
||
self._data = {}
|
||
self.save()
|
||
return self.to_dict()
|
||
|
||
|
||
def _coerce(value: Any, template: Any) -> Any:
|
||
"""按默认值的类型把输入转成一致类型（数值容错）。"""
|
||
if isinstance(template, bool):
|
||
return bool(value)
|
||
if isinstance(template, int):
|
||
try:
|
||
return int(float(value))
|
||
except (TypeError, ValueError):
|
||
return template
|
||
if isinstance(template, float):
|
||
try:
|
||
return float(value)
|
||
except (TypeError, ValueError):
|
||
return template
|
||
return value
|
||
|
||
|
||
def load_settings(path: Optional[Path] = None) -> SettingsStore:
|
||
return SettingsStore(path)
|