tui.js:进厂补恢复出厂(格式化)入口、整厂停用改单条 disable_all、flush 改常驻轮询热重载
This commit is contained in:
@@ -474,61 +474,40 @@ function mmReadCatalog() {
|
||||
})
|
||||
}
|
||||
|
||||
// ── 热重载接力:等 daemon 把 outbox 排空并写完 flush 标记,再由 TUI 执行 reload ──
|
||||
// 触发方:每次 mmEnqueue 批次入队后调用(since = 入队时刻,只认之后落盘的 flush,
|
||||
// 避免被上一批的旧标记骗到)。同一时间只跑一个 watcher,后续批次合并进来。
|
||||
// 幂等:flush.serverReload=true 说明 server 进程内通道已成功(将来宿主补通道时),TUI 免重复触发。
|
||||
let mmWatch = null
|
||||
function mmReloadAfterFlush(context, tag) {
|
||||
const since = Date.now()
|
||||
if (mmWatch) {
|
||||
if (since > mmWatch.since) mmWatch.since = since
|
||||
mmWatch.tag = mmWatch.tag + "+" + tag
|
||||
log("[oc-session-all] reload watcher 已在跑,合并批次: " + mmWatch.tag)
|
||||
return
|
||||
}
|
||||
const w = { since, tag }
|
||||
mmWatch = w
|
||||
;(async () => {
|
||||
const t0 = Date.now()
|
||||
// ── 常驻 flush 轮询:daemon(server 插件)每次把队列/对账写完会落 flush 标记,
|
||||
// 这里每 2s 看一眼,发现新标记就执行 location.reload(TUI ctx 的 client.location
|
||||
// 是本环境唯一实测可用的 reload 通道;server 插件 ctx 无 client/serverUrl)。
|
||||
// 覆盖两类来源:① /manage 入队操作消费完 ② 后台对账压制 catalog 回潮后。
|
||||
// 启动时以当前 flush.at 为基线,不重放历史标记。
|
||||
let mmFlushSeenAt = 0
|
||||
function mmStartFlushPoller(context) {
|
||||
try {
|
||||
if (existsSync(MM_FLUSH)) {
|
||||
try {
|
||||
mmFlushSeenAt = JSON.parse(readFileSync(MM_FLUSH, "utf-8")).at || 0
|
||||
} catch {}
|
||||
}
|
||||
} catch {}
|
||||
log("[oc-session-all] flush 常驻轮询已启动(2s,基线 at=" + mmFlushSeenAt + ")")
|
||||
setInterval(async () => {
|
||||
try {
|
||||
let waiting = false
|
||||
for (;;) {
|
||||
if (Date.now() - t0 > 180000) {
|
||||
log("[oc-session-all] 等 daemon flush 超时(180s),放弃热重载 tag=" + w.tag)
|
||||
try {
|
||||
context.ui.toast.show({ message: "热重载等待超时(daemon 未 flush),详见日志", variant: "error", duration: 5000 })
|
||||
} catch {}
|
||||
return
|
||||
}
|
||||
let flush = null
|
||||
try {
|
||||
if (existsSync(MM_FLUSH)) flush = JSON.parse(readFileSync(MM_FLUSH, "utf-8"))
|
||||
} catch {}
|
||||
if (flush && typeof flush.at === "number" && flush.at >= w.since) {
|
||||
if (flush.serverReload) {
|
||||
log("[oc-session-all] server 侧进程内 reload 已成功,TUI 免重复触发 tag=" + w.tag)
|
||||
return
|
||||
}
|
||||
break // DB 已全部写完 → 出去执行 TUI 侧 reload
|
||||
}
|
||||
if (!waiting) {
|
||||
log("[oc-session-all] 等 daemon 排空队列并落 flush… tag=" + w.tag)
|
||||
waiting = true
|
||||
}
|
||||
await delay(700)
|
||||
if (!existsSync(MM_FLUSH)) return
|
||||
let f = null
|
||||
try {
|
||||
f = JSON.parse(readFileSync(MM_FLUSH, "utf-8"))
|
||||
} catch {
|
||||
return
|
||||
}
|
||||
await delay(500) // 最后一条 DB 写的可见性缓冲
|
||||
if (!f || typeof f.at !== "number" || f.at <= mmFlushSeenAt) return
|
||||
mmFlushSeenAt = f.at
|
||||
if (f.serverReload) {
|
||||
log("[oc-session-all] flush 新标记 tag=" + f.lastOp + "(server 侧已 reload,TUI 免重复触发)")
|
||||
return
|
||||
}
|
||||
log("[oc-session-all] flush 新标记 tag=" + f.lastOp + " → 执行 TUI 热重载")
|
||||
await delay(500) // DB 写可见性缓冲
|
||||
const ok = await doReload(context)
|
||||
log(
|
||||
"[oc-session-all] 模型目录热重载 " +
|
||||
(ok ? "OK" : "ERROR") +
|
||||
" tag=" +
|
||||
w.tag +
|
||||
" 耗时=" +
|
||||
((Date.now() - t0) / 1000).toFixed(1) +
|
||||
"s",
|
||||
)
|
||||
log("[oc-session-all] 模型目录热重载 " + (ok ? "OK" : "ERROR") + " tag=" + f.lastOp)
|
||||
try {
|
||||
context.ui.toast.show({
|
||||
message: ok ? "模型目录已热重载,/models 立即生效" : "热重载失败(详见 oc-session-all.log)",
|
||||
@@ -537,11 +516,9 @@ function mmReloadAfterFlush(context, tag) {
|
||||
})
|
||||
} catch {}
|
||||
} catch (e) {
|
||||
log("[oc-session-all] mmReloadAfterFlush ERROR: " + (e.stack || e.message))
|
||||
} finally {
|
||||
if (mmWatch === w) mmWatch = null
|
||||
log("[oc-session-all] flush poller ERROR: " + (e.stack || e.message))
|
||||
}
|
||||
})()
|
||||
}, 2000)
|
||||
}
|
||||
|
||||
async function openModelManager(context) {
|
||||
@@ -658,7 +635,6 @@ async function openModelManager(context) {
|
||||
if (!pickProvider) return
|
||||
if (pickProvider === "__reset_all__") {
|
||||
for (const p of connectedProviders) mmEnqueue("reset", { provider: p.id })
|
||||
mmReloadAfterFlush(context, "reset_all")
|
||||
context.ui.toast.show({
|
||||
message: "恢复出厂已入队(" + connectedProviders.length + " 个 provider),daemon 写完自动热重载,/models 随即生效",
|
||||
variant: "info",
|
||||
@@ -673,24 +649,35 @@ async function openModelManager(context) {
|
||||
const dispName = target.name
|
||||
const models = target.models || []
|
||||
|
||||
// 第二层:管理粒度 —— 单个模型 vs 整个厂商
|
||||
// 第二层:管理粒度 —— 单个模型 vs 整个厂商 vs 恢复出厂(格式化)
|
||||
const scope = await context.ui.dialog.select({
|
||||
title: "模型管理 · " + dispName + " · 选择粒度",
|
||||
placeholder: "管单个模型,还是管整个厂商?",
|
||||
placeholder: "管单个模型、整个厂商,还是格式化恢复出厂?",
|
||||
options: [
|
||||
{ title: "🧩 单个模型(停用/启用其中一个)", value: "model" },
|
||||
{ title: "🏭 整个厂商(全部停用 / 全部启用)", value: "provider" },
|
||||
{ title: "♻️ 恢复出厂(格式化该厂商 → 出厂模型集)", value: "factory" },
|
||||
],
|
||||
})
|
||||
if (!scope) return
|
||||
|
||||
if (scope === "factory") {
|
||||
// 格式化 = 整厂重置回出厂 pristine(含把已停用的全部拉回),并从 cfg 停用列表摘除
|
||||
mmEnqueue("reset", { provider: target.id })
|
||||
context.ui.toast.show({
|
||||
message: "已入队:格式化 " + dispName + "(恢复出厂模型集),daemon 写完自动热重载",
|
||||
variant: "info",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
if (scope === "provider") {
|
||||
// 厂商粒度:停用=逐个下架现存模型;启用=恢复该厂商全部模型(reset 拉回 pristine)
|
||||
// 厂商粒度:停用=整厂一条 disable_all 下架(持久化进 cfg 停用列表);启用=reset 拉回 pristine
|
||||
const pAct = await context.ui.dialog.select({
|
||||
title: "对整个 " + dispName + " 执行",
|
||||
placeholder: "停用整个厂商,还是恢复全部模型?",
|
||||
options: [
|
||||
{ title: "⛔ 停用整个厂商(全部模型下架)", value: "disable_all" },
|
||||
{ title: "⛔ 停用整个厂商(全部模型下架 · 持久化进 cfg 停用列表)", value: "disable_all" },
|
||||
{ title: "✅ 启用整个厂商(恢复全部模型)", value: "enable_all" },
|
||||
],
|
||||
})
|
||||
@@ -700,15 +687,13 @@ async function openModelManager(context) {
|
||||
context.ui.toast.show({ message: dispName + " 的模型已全部停用", variant: "warning" })
|
||||
return
|
||||
}
|
||||
for (const m of models) mmEnqueue("disable", { provider: target.id, modelID: m })
|
||||
mmReloadAfterFlush(context, "disable_all:" + target.id)
|
||||
mmEnqueue("disable_all", { provider: target.id })
|
||||
context.ui.toast.show({
|
||||
message: "已入队:停用整个 " + dispName + "(" + models.length + " 个模型),daemon 写完自动热重载",
|
||||
message: "已入队:停用整个 " + dispName + "(" + models.length + " 个模型 · 一条搞定 · 持久化),daemon 写完自动热重载",
|
||||
variant: "info",
|
||||
})
|
||||
} else {
|
||||
mmEnqueue("reset", { provider: target.id })
|
||||
mmReloadAfterFlush(context, "enable_all:" + target.id)
|
||||
context.ui.toast.show({
|
||||
message: "已入队:启用整个 " + dispName + "(恢复全部模型),daemon 写完自动热重载",
|
||||
variant: "info",
|
||||
@@ -746,7 +731,6 @@ async function openModelManager(context) {
|
||||
if (!action) return
|
||||
if (action === "disable") mmEnqueue("disable", { provider: target.id, modelID: pickModel })
|
||||
else mmEnqueue("enable", { provider: target.id, modelID: pickModel })
|
||||
mmReloadAfterFlush(context, action + ":" + target.id + "/" + pickModel)
|
||||
context.ui.toast.show({
|
||||
message: "已入队:" + action + " " + dispName + "/" + pickModel + ",daemon 写完自动热重载,/models 随即更新",
|
||||
variant: "info",
|
||||
@@ -762,7 +746,8 @@ async function openModelManager(context) {
|
||||
export default Plugin.define({
|
||||
id: "oc-session-all",
|
||||
async setup(context) {
|
||||
log("[oc-session-all] v10 loaded — 会话选择器 + MCP配置对调(mcpcfg.txt) + 推送兜底(GET+merge补shell) + 热重载 + skill热加载")
|
||||
log("[oc-session-all] v11 loaded — 会话选择器 + MCP配置对调(mcpcfg.txt) + 推送兜底(GET+merge补shell) + flush常驻热重载 + 模型停用持久化(cfg) + skill热加载")
|
||||
mmStartFlushPoller(context) // 常驻轮询 flush 标记:/manage 操作与后台对账回潮都靠它热重载
|
||||
|
||||
// ── 读取 mcpcfg.txt(键值小白配置)────────────────────────────────
|
||||
const mcfg = readMcpCfg()
|
||||
|
||||
Reference in New Issue
Block a user