luckytri 1.0.0

package/server/studio/readiness.js

@@ -0,0 +1,190 @@
+import { createHash } from "node:crypto";
+import { defaultModel, pickModel } from "../core/model-manager.js";
+
+// The health panel and connection test must describe the same saved profile.
+export function savedConnectionSettings(store, modelId = "") {
+  const hasProfiles = store.db
+    .prepare(
+      "SELECT 1 FROM sqlite_master WHERE type='table' AND name='core_config'",
+    )
+    .get();
+  const row =
+    hasProfiles &&
+    store.db.prepare("SELECT value FROM core_config WHERE id='models'").get();
+  const models = row ? JSON.parse(row.value) : [];
+  const profile = Array.isArray(models)
+    ? modelId
+      ? models.find((m) => m.id === modelId)
+      : pickModel(models, "default")
+    : null;
+  if (modelId && !profile) return null;
+  const settings = store.settings();
+  const resolved = profile || defaultModel(settings);
+  return {
+    isDefault: !profile || !!profile.isDefault,
+    profile: resolved,
+    settings: profile
+      ? {
+          ...settings,
+          baseUrl: profile.baseUrl,
+          model: profile.model,
+          apiKey: profile.apiKey || "",
+          providerPreset: profile.provider || settings.providerPreset,
+          reasoningEffort: profile.reasoningEffort || "none",
+          temperature: profile.temperature ?? settings.temperature,
+          topP: profile.topP ?? settings.topP,
+          maxTokens: profile.maxOutputTokens || settings.maxTokens || 256,
+        }
+      : settings,
+  };
+}
+export function modelSignature(settings) {
+  return createHash("sha256")
+    .update(
+      JSON.stringify([
+        settings.baseUrl,
+        settings.model,
+        process.env.LLM_API_KEY || settings.apiKey,
+        settings.providerPreset,
+        settings.reasoningEffort,
+        settings.temperature,
+        settings.topP,
+        settings.maxTokens,
+      ]),
+    )
+    .digest("hex");
+}
+export function recordModelCheck(store, settings, ok, latency, error = "") {
+  store.db
+    .prepare(
+      "INSERT INTO model_checks(id,signature,ok,latency,time,error) VALUES (1,?,?,?,?,?) ON CONFLICT(id) DO UPDATE SET signature=excluded.signature,ok=excluded.ok,latency=excluded.latency,time=excluded.time,error=excluded.error",
+    )
+    .run(modelSignature(settings), Number(ok), latency, Date.now(), error);
+}
+// The steps to get a channel talking. Which two come first depends on how she
+// is connected: a OneBot client needs the shared token, the official bot needs
+// its AppID and AppSecret.
+function connectionChecks({ channel, online, tokenConfigured, qqbot }) {
+  if (channel === "qqbot") {
+    const configured = !!qqbot?.configured;
+    const groups = qqbot?.groupMessages;
+    return [
+      {
+        id: "credentials",
+        name: "QQ 官方机器人凭据",
+        done: configured,
+        detail: configured
+          ? `AppID ${qqbot.appId}`
+          : "在「系统 → 连接 QQ」填写机器人的 AppID 和 AppSecret",
+        tab: "setup",
+      },
+      {
+        id: "qq",
+        name: "QQ 连接",
+        done: online,
+        detail: online
+          ? groups === "mentions"
+            ? "已连接 QQ 开放平台；群里目前只收到 @ 她的消息，在开放平台开启「接收所有消息」后她才能看到整个群"
+            : "已连接 QQ 开放平台"
+          : qqbot?.error ||
+            (configured ? "正在连接 QQ 开放平台" : "填写凭据后自动连接"),
+        tab: "settings",
+      },
+    ];
+  }
+  return [
+    {
+      id: "token",
+      name: "QQ 接入令牌",
+      done: tokenConfigured,
+      detail: tokenConfigured
+        ? "已配置"
+        : "在本机 .env 中填写 ONEBOT_TOKEN，然后重启 LuckyTri",
+      tab: "setup",
+    },
+    {
+      id: "qq",
+      name: "QQ 连接",
+      done: online,
+      detail: online
+        ? "OneBot 11 反向 WebSocket 已连接"
+        : "在接入端配置 OneBot 11 反向 WebSocket 客户端",
+      tab: "settings",
+    },
+  ];
+}
+
+export function readiness(
+  store,
+  {
+    online = false,
+    tokenConfigured = !!process.env.ONEBOT_TOKEN,
+    channel = "onebot",
+    qqbot = null,
+  } = {},
+) {
+  const s = store.settings(),
+    record = store.db.prepare("SELECT * FROM model_checks WHERE id=1").get();
+  const effective = savedConnectionSettings(store).settings;
+  // Previous connection tests signed their temporary 256-token test budget.
+  // Accept that exact historical signature, without marking untested profiles ready.
+  const current =
+    !!record &&
+    [
+      effective,
+      { ...effective, maxTokens: Math.min(effective.maxTokens || 256, 256) },
+    ].some((value) => record.signature === modelSignature(value));
+  const sessions = store.db
+    .prepare("SELECT COUNT(*) n FROM sessions WHERE enabled=1")
+    .get().n;
+  const checks = [
+    ...connectionChecks({ channel, online, tokenConfigured, qqbot }),
+    {
+      id: "model",
+      name: "当前模型已测试",
+      done: current && !!record.ok,
+      detail: current
+        ? record.ok
+          ? `测试成功 · ${record.latency} ms`
+          : record.error
+        : record
+          ? "模型配置已变更，请重新测试"
+          : "保存模型配置后，点击测试模型连接",
+      tab: "settings",
+    },
+    {
+      id: "mode",
+      name: "开启真实回复",
+      done: !s.demo && s.enabled,
+      detail: s.demo
+        ? "当前是模拟模式，不向 QQ 发言"
+        : !s.enabled
+          ? "总开关已关闭"
+          : "真实模式与总开关已开启",
+      tab: "settings",
+    },
+    {
+      id: "session",
+      name: "选择参与的会话",
+      done: sessions > 0,
+      detail: sessions
+        ? `${sessions} 个会话已开启`
+        : "到「对话」里开启一个群聊或私聊",
+      tab: "sessions",
+    },
+  ];
+  return {
+    checks,
+    completed: checks.filter((x) => x.done).length,
+    total: checks.length,
+    ready: checks.every((x) => x.done),
+    modelTest: record
+      ? {
+          current,
+          ok: current && !!record.ok,
+          time: record.time,
+          latency: record.latency,
+        }
+      : null,
+  };
+}

package/server/studio/settings.js

@@ -0,0 +1,236 @@
+import { runtimePaths } from "../paths.js";
+import { VERSION } from "../version.js";
+import { publicSession } from "../channels/session-key.js";
+import { CHANNEL_TYPES } from "../channels/hub.js";
+import { readiness } from "./readiness.js";
+import { FEEDBACK_LABELS } from "../core/feedback.js";
+import {
+  EFFORT_LABELS,
+  MODEL_CATALOG,
+  MODEL_PRESETS,
+  REASONING_EFFORTS,
+} from "../core/model-presets.js";
+import { applySpeakerNames, speakerNames } from "../core/speaker-names.js";
+import { displayNames, displaySession } from "./display-names.js";
+
+export function mountStudio(app, store, runtime, chat) {
+  app.get("/api/storage/cleanup", (req, res) => {
+    if (!runtime.cleanup)
+      return res.status(501).json({ error: "备份整理不可用" });
+    res.json(runtime.cleanup.info());
+  });
+  app.patch("/api/storage/cleanup", (req, res) => {
+    if (!runtime.cleanup)
+      return res.status(501).json({ error: "备份整理不可用" });
+    try {
+      res.json(runtime.cleanup.save(req.body));
+    } catch (error) {
+      res.status(400).json({ error: error.message });
+    }
+  });
+  app.post("/api/storage/cleanup/manual", async (req, res) => {
+    if (!runtime.cleanup)
+      return res.status(501).json({ error: "备份整理不可用" });
+    try {
+      res.json(await runtime.cleanup.manual(req.body.items));
+    } catch (error) {
+      res.status(409).json({ error: error.message });
+    }
+  });
+  app.get("/api/service/status", (req, res) =>
+    res.json({
+      app: "luckybot",
+      workspace: runtimePaths().package,
+      home: runtimePaths().home,
+      version: VERSION,
+      pid: process.pid,
+      schemaVersion: store.db.prepare("PRAGMA user_version").get().user_version,
+    }),
+  );
+  app.post("/api/service/stop", (req, res) => {
+    res.json({ ok: true });
+    setTimeout(runtime.shutdown, 150);
+  });
+  app.get("/api/state", (req, res) => {
+    // Secrets are written from the studio and never read back into it.
+    const {
+      apiKey,
+      qqbotSecret,
+      onebotToken: _onebotToken,
+      ...settings
+    } = store.settings();
+    const demo = Number(settings.demo);
+    const connection = runtime.connection();
+    const namesForDisplay = displayNames(store.db, chat?.now() ?? Date.now());
+    res.json({
+      settings: {
+        ...settings,
+        hasApiKey: !!(apiKey || process.env.LLM_API_KEY),
+        hasQqbotSecret: !!(qqbotSecret || process.env.QQBOT_APP_SECRET),
+      },
+      readiness: readiness(store, {
+        online: connection.online,
+        tokenConfigured: connection.onebot?.tokenConfigured,
+        channel: connection.channel,
+        qqbot: connection.qqbot,
+      }),
+      feedbackLabels: FEEDBACK_LABELS,
+      modelPresets: MODEL_PRESETS,
+      modelCatalog: MODEL_CATALOG,
+      reasoningEfforts: REASONING_EFFORTS,
+      effortLabels: EFFORT_LABELS,
+      connection,
+      backup: runtime.backup?.() ?? null,
+      sessions: store.db
+        .prepare(
+          "SELECT s.*, (SELECT text FROM messages WHERE session_id=s.id AND is_demo=? ORDER BY id DESC LIMIT 1) preview,(SELECT COUNT(*) FROM messages WHERE session_id=s.id AND is_demo=?) message_count FROM sessions s ORDER BY s.id",
+        )
+        .all(demo, demo)
+        .map((session) => ({
+          ...displaySession(publicSession(session), namesForDisplay),
+          style: store.groupStyle(session.id, demo),
+        })),
+      decisions: (() => {
+        const rows = store.db
+          .prepare(
+            "SELECT d.*,f.tag AS feedback FROM decisions d LEFT JOIN reply_feedback f ON f.decision_id=d.id WHERE d.is_demo=? ORDER BY d.id DESC LIMIT 80",
+          )
+          .all(demo);
+        const names = speakerNames(
+          store.db,
+          rows.map((row) => row.session_id),
+        );
+        return rows.map((row) => ({
+          ...row,
+          reason: applySpeakerNames(row.reason, names),
+        }));
+      })(),
+      stats: {
+        messages: store.db
+          .prepare("SELECT COUNT(*) n FROM messages WHERE is_demo=?")
+          .get(demo).n,
+        replies: store.db
+          .prepare(
+            "SELECT COUNT(*) n FROM decisions WHERE reply!='' AND is_demo=?",
+          )
+          .get(demo).n,
+      },
+    });
+  });
+  // Connection and runtime switches only. Who she is lives in her nature;
+  // whether she speaks is hers to decide.
+  app.patch("/api/settings", (req, res) => {
+    const v = req.body;
+    const allowed = {};
+    for (const k of [
+      "aliases",
+      "baseUrl",
+      "model",
+      "apiKey",
+      "providerPreset",
+      "reasoningEffort",
+      "channel",
+      "qqbotAppId",
+      "qqbotSecret",
+    ])
+      if (k in v) {
+        if (typeof v[k] !== "string" || v[k].length > 4000)
+          return res.status(400).json({ error: "文本配置无效" });
+        allowed[k] = v[k];
+      }
+    if ("channel" in allowed) {
+      if (!CHANNEL_TYPES.includes(allowed.channel))
+        return res.status(400).json({ error: "接入方式无效" });
+      const forced = String(process.env.LUCKYTRI_CHANNEL || "").trim();
+      if (forced && forced !== allowed.channel)
+        return res
+          .status(409)
+          .json({ error: "接入方式已由环境变量 LUCKYTRI_CHANNEL 固定" });
+    }
+    if ("qqbotAppId" in allowed) {
+      allowed.qqbotAppId = allowed.qqbotAppId.trim();
+      if (
+        allowed.qqbotAppId &&
+        !/^[0-9A-Za-z_-]{1,40}$/.test(allowed.qqbotAppId)
+      )
+        return res.status(400).json({ error: "AppID 格式无效" });
+    }
+    if ("qqbotSecret" in allowed) {
+      allowed.qqbotSecret = allowed.qqbotSecret.trim();
+      if (allowed.qqbotSecret.length > 200)
+        return res.status(400).json({ error: "AppSecret 格式无效" });
+    }
+    for (const k of ["enabled", "demo", "memoryEnabled", "memoryCandidates"])
+      if (k in v) {
+        if (typeof v[k] !== "boolean")
+          return res.status(400).json({ error: "开关配置无效" });
+        allowed[k] = v[k];
+      }
+    for (const [k, min, max] of [
+      ["temperature", 0, 2],
+      ["topP", 0, 1],
+      ["maxTokens", 64, 2000],
+    ])
+      if (k in v) {
+        if (
+          !Number.isFinite(v[k]) ||
+          v[k] < min ||
+          v[k] > max ||
+          (k === "maxTokens" && !Number.isInteger(v[k]))
+        )
+          return res.status(400).json({ error: "数值配置超出范围" });
+        allowed[k] = v[k];
+      }
+    const retired = [
+      "probability",
+      "cooldown",
+      "voicePreset",
+      "slangLevel",
+      "allowMildProfanity",
+      "qualityRewrite",
+      "adaptGroupStyle",
+      "persona",
+      "name",
+    ].filter((k) => k in v);
+    if (retired.length)
+      return res.status(400).json({
+        error: `这些设置已经不存在：${retired.join("、")}。名字和天性在「TA → 天性」里修改`,
+      });
+    if (
+      "providerPreset" in allowed &&
+      !Object.hasOwn(MODEL_PRESETS, allowed.providerPreset)
+    )
+      return res.status(400).json({ error: "模型服务商预设无效" });
+    if (
+      "reasoningEffort" in allowed &&
+      !REASONING_EFFORTS.some((item) => item.id === allowed.reasoningEffort)
+    )
+      return res.status(400).json({ error: "思考强度无效" });
+    if ("baseUrl" in allowed) {
+      try {
+        const u = new URL(allowed.baseUrl);
+        if (
+          !["http:", "https:"].includes(u.protocol) ||
+          u.username ||
+          u.password ||
+          u.search ||
+          u.hash
+        )
+          throw 0;
+      } catch {
+        return res.status(400).json({ error: "API 地址无效" });
+      }
+    }
+    if ("model" in allowed && !allowed.model.trim())
+      return res.status(400).json({ error: "模型名称不能为空" });
+    store.save(allowed);
+    // A new channel or new credentials take effect now, without a restart.
+    if (
+      "channel" in allowed ||
+      "qqbotAppId" in allowed ||
+      "qqbotSecret" in allowed
+    )
+      runtime.syncChannel?.();
+    res.json({ ok: true });
+  });
+}

package/server/version.js

@@ -0,0 +1,10 @@
+// The one place the running version is read: `package.json`. Everything that
+// needs to say which LuckyTri it is (request headers, database archives)
+// imports it from here, so a release only changes one number.
+import { readFileSync } from "node:fs";
+
+export const VERSION = JSON.parse(
+  readFileSync(new URL("../package.json", import.meta.url), "utf8"),
+).version;
+
+export const USER_AGENT = `LuckyTri/${VERSION}`;