nolo-cli 0.1.30 → 0.1.31

package/machineWsSession.ts

@@ -9,6 +9,7 @@ export type ConnectorWebSocketOptions = {
   headers: Record<string, string>;
   onMessage: (message: string) => void | Promise<void>;
   sentMessages: string[];
+  signal?: AbortSignal;
 };
 
 const CONNECTOR_WS_KEEPALIVE_MS = 25_000;
@@ -35,6 +36,7 @@ export async function defaultConnectWebSocket(
       headers: options.headers,
     } as any);
     let opened = false;
+    let aborted = options.signal?.aborted ?? false;
     let keepalive: ReturnType<typeof setInterval> | null = null;
     const clearKeepalive = () => {
       if (keepalive) {
@@ -42,6 +44,16 @@ export async function defaultConnectWebSocket(
         keepalive = null;
       }
     };
+    const handleAbort = () => {
+      aborted = true;
+      clearKeepalive();
+      try {
+        ws.close();
+      } catch {
+        resolve();
+      }
+    };
+    options.signal?.addEventListener("abort", handleAbort, { once: true });
     ws.addEventListener(
       "open",
       () => {
@@ -58,13 +70,17 @@ export async function defaultConnectWebSocket(
     );
     ws.addEventListener("error", () => {
       clearKeepalive();
+      if (aborted) {
+        resolve();
+        return;
+      }
       reject(new Error("connector websocket failed"));
     });
     ws.addEventListener(
       "close",
       () => {
         clearKeepalive();
-        if (opened) resolve();
+        if (opened || aborted) resolve();
         else reject(new Error("connector websocket closed before opening"));
       },
       { once: true },
@@ -88,11 +104,13 @@ export async function runMachineWsSession(options: {
   machine: MachineHeartbeat;
   serverUrl: string;
   authToken: string;
+  signal?: AbortSignal;
   sendHeartbeat: () => Promise<void>;
   runHeartbeatLoop: (options: HeartbeatLoopOptions) => Promise<void>;
   resolveConnectorWebSocketTarget: (args: {
     serverUrl: string;
     machineId: string;
+    connectorSurface?: string;
     headers: Record<string, string>;
   }) => Promise<string>;
   connectWebSocket: (url: string, options: ConnectorWebSocketOptions) => Promise<void>;
@@ -103,6 +121,9 @@ export async function runMachineWsSession(options: {
   let heartbeatLoopPromise: Promise<void> | null = null;
   let websocketPromise: Promise<void> | null = null;
   try {
+    if (options.signal?.aborted) {
+      return 0;
+    }
     await options.sendHeartbeat();
     options.output.write(
       `Connector websocket connected: ${options.machine.name} (${options.machine.machineId})\n`
@@ -120,19 +141,35 @@ export async function runMachineWsSession(options: {
     const wsTarget = await options.resolveConnectorWebSocketTarget({
       serverUrl: options.serverUrl,
       machineId: options.machine.machineId,
+      connectorSurface: "cli",
       headers: { Authorization: `Bearer ${options.authToken}` },
     });
     websocketPromise = options.connectWebSocket(wsTarget, {
       headers: { Authorization: `Bearer ${options.authToken}` },
+      signal: options.signal,
       sentMessages,
       onMessage: (message) =>
         options.onMessage(message, (response) => {
           sentMessages.push(response);
         }),
     });
-    await Promise.race([websocketPromise, heartbeatMonitorPromise]);
+    const abortPromise = options.signal
+      ? new Promise<"aborted">((resolve) => {
+          if (options.signal?.aborted) {
+            resolve("aborted");
+            return;
+          }
+          options.signal?.addEventListener("abort", () => resolve("aborted"), { once: true });
+        })
+      : null;
+    const raceResult = await Promise.race(
+      [websocketPromise, heartbeatMonitorPromise, abortPromise].filter(Boolean) as Promise<void | "aborted">[]
+    );
     heartbeatAbort.abort();
     await Promise.allSettled([websocketPromise, heartbeatLoopPromise]);
+    if (raceResult === "aborted") {
+      return 0;
+    }
     return 0;
   } catch (error) {
     heartbeatAbort.abort();

package/noloReadBridgePrompt.ts

@@ -0,0 +1,16 @@
+const REPO_CLI_PREFIX = "bun packages/cli/index.ts";
+
+export function buildNoloReadOnlyBridgePrompt() {
+  return [
+    "--- Nolo read-only CLI bridge ---",
+    "This CLI runtime does not receive server-side function tools directly.",
+    "Use these repo-local read commands to fetch authoritative Nolo docs and records on demand instead of trusting prompt snapshots or summaries alone.",
+    "Run commands from the repository root. The runner already provides server URL and auth in the environment.",
+    `Read doc/page truth: ${REPO_CLI_PREFIX} doc read "<pageKey>"`,
+    `Read skill doc truth: ${REPO_CLI_PREFIX} skill-doc read "<pageKey>"`,
+    `Read dialog truth: ${REPO_CLI_PREFIX} dialog read "<dialogId|dialogUrl>"`,
+    `Read shared space truth: ${REPO_CLI_PREFIX} space read "<spaceId|spaceKey|spaceUrl>"`,
+    `Read agent truth: ${REPO_CLI_PREFIX} agent read "<agentKey|agentAlias|agentUrl>"`,
+    "If a read command fails, report the exact command and error.",
+  ].join("\n");
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "nolo-cli",
-  "version": "0.1.30",
+  "version": "0.1.31",
   "type": "module",
   "description": "Agent-first terminal workspace for Nolo",
   "bin": {
@@ -97,11 +97,104 @@
     "runtimeModeArgs.ts",
     "tableCommands.ts",
     "updateCommands.ts",
-    "client/**/*.ts",
-    "tui/**/*.ts",
+    "client/compactDialog.test.ts",
+    "client/localAgentRecords.test.ts",
+    "client/localRuntimeDryRun.test.ts",
+    "client/agentRunTaskRunRecorder.ts",
+    "client/workspaceSession.test.ts",
+    "client/localProviderResolver.test.ts",
+    "client/localDialogRecords.ts",
+    "client/agentRunLocalExecutor.ts",
+    "client/localDialogRecords.test.ts",
+    "client/profileConfig.test.ts",
+    "client/profileConfig.ts",
+    "client/agentRunRuntimeSelection.ts",
+    "client/localAgentRecords.ts",
+    "client/taskWorktree.ts",
+    "client/localToolPolicy.test.ts",
+    "client/streamingOutput.test.ts",
+    "client/streamingOutput.ts",
+    "client/localToolPolicy.ts",
+    "client/compactDialog.ts",
+    "client/hybridRecordStore.test.ts",
+    "client/agentRun.ts",
+    "client/localRuntimeAdapter.test.ts",
+    "client/agentRunRuntimeSelection.test.ts",
+    "client/agentRunLocalExecution.test.ts",
+    "client/agentRunInput.ts",
+    "client/localProviderResolver.ts",
+    "client/taskRunPrompt.ts",
+    "client/hybridRecordStore.ts",
+    "client/agentConfigResolver.test.ts",
+    "client/agentRunServerTransport.ts",
+    "client/workspaceSession.ts",
+    "client/agentRun.test.ts",
+    "client/agentConfigResolver.ts",
+    "client/agentRunLocalExecution.ts",
+    "client/localRuntimeAdapter.ts",
+    "tui/tuiPresentation.ts",
+    "tui/readlineWorkspace.ts",
+    "tui/session.ts",
+    "tui/tuiPresentation.test.ts",
+    "tui/tuiCommandInterpreter.test.ts",
+    "tui/tuiCommandInterpreter.ts",
+    "tui/tuiCatalog.ts",
     "README.md",
-    "ai/**/*.ts",
-    "connector-experimental/**/*.ts",
+    "ai/agent.ts",
+    "ai/tools/agent/taskRunTool.ts",
+    "ai/agent/cliExecutor.ts",
+    "ai/agent/agentSlice.ts",
+    "ai/agent/machineRunPermissions.ts",
+    "ai/agent/agentRunDialogLineage.ts",
+    "ai/agent/cliPrompt.ts",
+    "ai/index.ts",
+    "connector-experimental/protocol.ts",
+    "connector-experimental/machineInfo.ts",
+    "../agent-runtime/index.ts",
+    "../ai/tools/agent/taskRunTool.ts",
+    "../ai/agent/cliExecutor.ts",
+    "../ai/agent/cliPrompt.ts",
+    "../agent-runtime/hostAdapter.ts",
+    "../agent-runtime/localLoop.ts",
+    "../agent-runtime/agentConfigOptions.ts",
+    "../agent-runtime/openAiCompatibleProvider.ts",
+    "../agent-runtime/openAiCompatibleProviderConfig.ts",
+    "../agent-runtime/platformChatProvider.ts",
+    "../agent-runtime/runtimeDecision.ts",
+    "../agent-runtime/runtimeFacts.ts",
+    "../agent-runtime/agentRecordKeys.ts",
+    "../agent-runtime/agentRecordConfig.ts",
+    "../agent-runtime/dialogMessageRecord.ts",
+    "../agent-runtime/dialogStatus.ts",
+    "../agent-runtime/dialogWritePlan.ts",
+    "../agent-runtime/hybridRecordStore.ts",
+    "../agent-runtime/workspaceSession.ts",
+    "../agent-runtime/taskWorkspace.ts",
+    "../agent-runtime/hostRuntimeConfig.ts",
+    "../agent-runtime/localToolPolicy.ts",
+    "../agent-runtime/runtimeToolPolicy.ts",
+    "../agent-runtime/localWorkspaceTools.ts",
+    "../agent-runtime/types.ts",
+    "../database/config.ts",
+    "../app/types.ts",
+    "../create/space/spaceKeys.ts",
+    "spaceAttachCommands.ts",
+    "cliRuntimeDoctorReport.ts",
+    "../ai/agent/agentRunDialogLineage.ts",
+    "../ai/agent/machineRunPermissions.ts",
+    "noloReadBridgePrompt.ts",
+    "../chat/messages/fetchMessages.ts",
+    "../chat/messages/types.ts",
+    "../database/server/dbPath.ts",
+    "../database/server/db.ts",
+    "../database/server/ensureDbOpen.ts",
+    "../database/server/legacyServerDb.ts",
+    "../database/server/authorityStoreTypes.ts",
+    "../database/server/serverStoreFactory.ts",
+    "../database/server/levelAuthorityStore.ts",
+    "../database/server/memoryAuthorityStore.ts",
+    "../database/keys.ts",
+    "../database/utils/ulid.ts",
     "agent-runtime/**/*.ts",
     "connector-experimental/**/*.ts",
     "app/**/*.ts",
@@ -110,8 +203,8 @@
     "ai/**/*.ts",
     "chat/**/*.ts",
     "render/**/*.ts",
-    "auth/**/*.ts",
     "share/**/*.ts",
+    "auth/**/*.ts",
     "life/**/*.ts",
     "core/**/*.ts",
     "server/**/*.ts",

package/render/layout/mainLayoutViewMode.ts

@@ -12,6 +12,15 @@ export interface SpaceRouteContext {
   isSpaceRoute: boolean;
   isSpaceRootRoute: boolean;
   spaceId: string | null;
+  routeSection:
+    | "none"
+    | "root"
+    | "files"
+    | "ai"
+    | "scheduled"
+    | "members"
+    | "settings"
+    | "content";
 }
 
 export function isAllViewRoutePath(pathname: string): boolean {
@@ -25,6 +34,7 @@ export function getSpaceRouteContext(pathname: string): SpaceRouteContext {
       isSpaceRoute: false,
       isSpaceRootRoute: false,
       spaceId: null,
+      routeSection: "none",
     };
   }
 
@@ -37,10 +47,25 @@ export function getSpaceRouteContext(pathname: string): SpaceRouteContext {
     // Keep the encoded segment so route classification never fails on bad input.
   }
 
+  const routeSection = !descendantPath
+    ? "root"
+    : descendantPath === "files" || descendantPath.startsWith("files/")
+      ? "files"
+      : descendantPath === "ai" || descendantPath.startsWith("ai/")
+        ? "ai"
+        : descendantPath === "scheduled" || descendantPath.startsWith("scheduled/")
+          ? "scheduled"
+          : descendantPath === "members" || descendantPath.startsWith("members/")
+            ? "members"
+            : descendantPath === "settings" || descendantPath.startsWith("settings/")
+              ? "settings"
+              : "content";
+
   return {
     isSpaceRoute: true,
     isSpaceRootRoute: !descendantPath,
     spaceId,
+    routeSection,
   };
 }
 

package/render/page/server/createPage.ts

@@ -5,7 +5,7 @@ import { format } from "date-fns";
 import { createPageKey } from "../../../database/keys";
 import { DataType } from "../../../create/types";
 import { markdownToSlate } from "../../../create/editor/transforms/markdownToSlate";
-import serverDb, { ensureServerDbOpen } from "../../../database/server/db";
+import { ensureServerDbOpen, getServerAuthorityStore } from "../../../database/server/db";
 import type { PageData } from "../types";
 import {
     createEmptyParagraph,
@@ -15,6 +15,8 @@ import {
 import { slateToRenderMarkdown } from "../../../create/editor/transforms/slateToRenderMarkdown";
 import { parseSkillDocProtocol } from "../../../ai/skills/skillDocProtocol";
 
+const serverDb = getServerAuthorityStore();
+
 export interface CreatePageServerArgs {
     userId: string;
     title?: string;

package/render/page/server/handleCreatePage.ts

@@ -2,7 +2,7 @@
 // HTTP handler for createPage (Bun serve 兼容)
 
 import { createPageServer } from "./createPage";
-import { handleToken } from "../../../auth/server/token";
+import { readUserFromRequest } from "../../../auth/server/token";
 import { logger } from "../../../auth/server/shared";
 
 const CORS_HEADERS = {
@@ -27,15 +27,7 @@ const CORS_HEADERS = {
  */
 export const handleCreatePage = async (req: Request): Promise<Response> => {
     try {
-        // 鉴权：创建一个简化的 req 对象用于 handleToken
-        const authReq: any = {
-            headers: {
-                get: (name: string) => req.headers.get(name),
-            },
-        };
-        const authRes: any = {}; // handleToken 可能不需要 res
-
-        const user = await handleToken(authReq, authRes);
+        const user = await readUserFromRequest(req);
         const userId = user?.userId;
 
         if (!userId) {
@@ -88,4 +80,3 @@ export const handleCreatePage = async (req: Request): Promise<Response> => {
         );
     }
 };
-

package/runtimeDoctorCommands.ts

@@ -1,41 +1,22 @@
-import { resolveAgentRuntimeDecision } from "./agent-runtime/runtimeDecision";
 import {
   type AgentCommandDeps,
   type LocalRuntimeProbeResult,
 } from "./agentCommandSupport";
+import {
+  buildCliRuntimeDoctorProbe,
+  buildCliRuntimeDoctorReport,
+  formatCliRuntimeDoctorReport,
+} from "./cliRuntimeDoctorReport";
 import {
   resolveAuthToken,
   resolveServerUrl,
   type EnvLike,
 } from "./cliEnvHelpers";
 
-function detectLocalAgentConfig(env: EnvLike) {
-  return Boolean(readLocalAgentKey(env) || env.NOLO_AGENT_CACHE_READY);
-}
-
 function readLocalAgentKey(env: EnvLike) {
   return env.NOLO_LOCAL_AGENT_KEY || env.NOLO_AGENT || "";
 }
 
-function detectLocalProvider(env: EnvLike) {
-  return Boolean(
-    env.OPENAI_API_KEY ||
-    env.ANTHROPIC_API_KEY ||
-    env.GOOGLE_API_KEY ||
-    env.NOLO_LOCAL_OPENAI_BASE_URL ||
-    env.OLLAMA_BASE_URL
-  );
-}
-
-function detectProviderLabel(env: EnvLike) {
-  if (env.NOLO_LOCAL_OPENAI_BASE_URL) return `openai-compatible endpoint ${env.NOLO_LOCAL_OPENAI_BASE_URL}`;
-  if (env.OLLAMA_BASE_URL) return `ollama-compatible endpoint ${env.OLLAMA_BASE_URL}`;
-  if (env.OPENAI_API_KEY) return "openai via env OPENAI_API_KEY";
-  if (env.ANTHROPIC_API_KEY) return "anthropic via env ANTHROPIC_API_KEY";
-  if (env.GOOGLE_API_KEY) return "google via env GOOGLE_API_KEY";
-  return "missing";
-}
-
 async function defaultLocalRuntimeProbe(env: EnvLike): Promise<LocalRuntimeProbeResult> {
   const { resolveCliLocalRuntimeDbPath } = await import("./localRuntimeDb");
   const dbPath = resolveCliLocalRuntimeDbPath({ env });
@@ -72,47 +53,15 @@ export async function runDoctorRuntimeCommand(
   const authToken = resolveAuthToken(args, env);
   const serverUrl = resolveServerUrl(args, env);
   const localProbe = await (deps.localRuntimeProbe ?? defaultLocalRuntimeProbe)(env);
-  const hasLocalAgentConfig = detectLocalAgentConfig(env) && localProbe.ok && localProbe.agentFound;
-  const hasLocalProvider = detectLocalProvider(env);
-  const missingLocalCapabilities = [
-    ...(localProbe.ok ? [] : ["leveldb"]),
-    ...(hasLocalAgentConfig ? [] : ["agent-config"]),
-    ...(hasLocalProvider ? [] : ["provider"]),
-  ];
-  const decision = resolveAgentRuntimeDecision({
-    requestedMode: "auto",
-    syncRequested: false,
-    host: "cli",
-    hasLocalAgentConfig,
-    hasLocalProvider,
-    hasLocalPersistence: true,
-    missingLocalCapabilities,
-    requiresServer: false,
-    serverFallbackAvailable: Boolean(authToken || env.NOLO_SERVER || env.BASE_URL),
-  });
-
-  output.write(`Runtime: ${decision.mode}\n`);
-  output.write(`Reason: ${decision.reason}\n`);
-  output.write(`LevelDB: ${localProbe.ok ? "ok" : "failed"}\n`);
-  output.write(`DB path: ${localProbe.dbPath}\n`);
-  if (!localProbe.ok && localProbe.error) {
-    output.write(`DB error: ${localProbe.error}\n`);
-  }
-  if (localProbe.agentKey) {
-    output.write(`Agent config: ${localProbe.agentFound ? "found" : "missing"} (${localProbe.agentKey})\n`);
-  } else {
-    output.write("Agent config: missing (no local agent key)\n");
-  }
-  output.write(`Provider: ${hasLocalProvider ? "available" : "missing"} (${detectProviderLabel(env)})\n`);
-  output.write("Persistence: LevelDB local store\n");
-  output.write(`Sync: ${authToken ? "available" : "unavailable"}${authToken ? "" : " (not authenticated)"}\n`);
-  output.write(`Server fallback: ${decision.runnable && decision.mode === "server" ? serverUrl : authToken || env.NOLO_SERVER || env.BASE_URL ? serverUrl : "unavailable"}\n`);
-  if (decision.missingLocalCapabilities.length > 0) {
-    output.write("Missing local capabilities:\n");
-    for (const capability of decision.missingLocalCapabilities) {
-      output.write(`- ${capability}\n`);
-    }
-  }
+  const report = buildCliRuntimeDoctorReport(
+    buildCliRuntimeDoctorProbe({
+      env,
+      localProbe,
+      authToken,
+      serverUrl,
+    })
+  );
+  output.write(formatCliRuntimeDoctorReport(report));
 
-  return decision.runnable ? 0 : 1;
+  return report.decision.runnable ? 0 : 1;
 }

package/server/api/methods.ts

@@ -1,6 +1,6 @@
 // ai/server/apiMethods.ts
 
-import serverDb from "../../database/server/db";
+import { getServerAuthorityStore } from "../../database/server/db";
 import { MemberRole, SpaceVisibility, SpaceMemberWithSpaceInfo } from "../../app/types";
 import { fetchMessages, MessageWithKey } from "../../chat/messages/fetchMessages";
 import type { Message } from "../../chat/messages/types";
@@ -23,14 +23,20 @@ import {
   updateEmailTags,
 } from "../../database/server/emailRepository";
 import { isTombstoneRecord } from "../../database/tombstones";
+
+type ApiMethodsStore = ReturnType<typeof getServerAuthorityStore>;
+
+const getApiMethodsStore = () => getServerAuthorityStore();
+
 // DB 列表迭代 (保留，因为 fetchUserSpaceMemberships 仍在使用)
 async function dbListEntries<T>(
+  store: ApiMethodsStore,
   gte: string,
   lte: string,
   filter?: (key: string, value: T) => boolean
 ): Promise<Array<{ key: string; value: T }>> {
   const res: Array<{ key: string; value: T }> = [];
-  for await (const [key, value] of serverDb.iterator({ gte, lte })) {
+  for await (const [key, value] of store.iterator({ gte, lte })) {
     if (typeof key !== "string") continue;
     if (!filter || filter(key, value as T)) res.push({ key, value: value as T });
   }
@@ -88,6 +94,7 @@ const membershipFromSpace = (
 };
 
 const repairMissingMembershipsFromSpaces = async (
+  store: ApiMethodsStore,
   userId: string,
   existing: SpaceMemberWithSpaceInfo[],
   indexedSpaceIds: Set<string>
@@ -100,6 +107,7 @@ const repairMissingMembershipsFromSpaces = async (
   );
   const repaired: SpaceMemberWithSpaceInfo[] = [];
   const spaces = await dbListEntries<any>(
+    store,
     "space-",
     "space-\xff",
     (key, space) =>
@@ -116,13 +124,14 @@ const repairMissingMembershipsFromSpaces = async (
     if (bySpaceId.has(normalizedSpaceId)) continue;
     bySpaceId.set(normalizedSpaceId, membership);
     repaired.push(membership);
-    await serverDb.put(membershipKey(userId, normalizedSpaceId), membership);
+    await store.put(membershipKey(userId, normalizedSpaceId), membership);
   }
 
   return [...existing, ...repaired];
 };
 
 const hasActiveSpaceMembership = async (
+  store: ApiMethodsStore,
   userId: string,
   memberKey: string,
   membership: SpaceMemberWithSpaceInfo
@@ -149,7 +158,7 @@ const hasActiveSpaceMembership = async (
   }
 
   try {
-    const space = await serverDb.get(`space-${keySpaceId}`);
+    const space = await store.get(`space-${keySpaceId}`);
     if (!space || isTombstoneRecord(space)) return false;
     if (space.ownerId === userId) return true;
     return Array.isArray(space.members) && space.members.includes(userId);
@@ -162,11 +171,13 @@ export async function fetchUserSpaceMemberships(
   params: string | { userId: string },
   context?: ApiContext
 ): Promise<SpaceMemberWithSpaceInfo[]> {
+  const store = getApiMethodsStore();
   const userId =
     context?.userId || (typeof params === "string" ? params : params.userId);
   if (!userId) throw new Error("userId 必传");
   const prefix = `space-member-${userId}-`;
   const memberships = await dbListEntries<SpaceMemberWithSpaceInfo>(
+    store,
     prefix,
     prefix + "\xff",
     (_, membership) => !isTombstoneRecord(membership)
@@ -179,7 +190,7 @@ export async function fetchUserSpaceMemberships(
   );
   const activeMemberships = await Promise.all(
     memberships.map(async ({ key, value }) =>
-      await hasActiveSpaceMembership(userId, key, value)
+      await hasActiveSpaceMembership(store, userId, key, value)
         ? value
         : null
     )
@@ -187,7 +198,12 @@ export async function fetchUserSpaceMemberships(
   const verifiedMemberships = activeMemberships.filter(
     (membership): membership is SpaceMemberWithSpaceInfo => !!membership
   );
-  return repairMissingMembershipsFromSpaces(userId, verifiedMemberships, indexedSpaceIds);
+  return repairMissingMembershipsFromSpaces(
+    store,
+    userId,
+    verifiedMemberships,
+    indexedSpaceIds
+  );
 }
 
 interface FetchConvMsgsParams {
@@ -201,6 +217,7 @@ export async function fetchConvMsgs(
   params: string | FetchConvMsgsParams,
   context?: ApiContext
 ): Promise<Message[]> {
+  const store = getApiMethodsStore();
   let dialogId: string,
     dialogKey: string | undefined,
     limit: number | undefined,
@@ -226,7 +243,7 @@ export async function fetchConvMsgs(
     };
   }
 
-  const dialogRecord = await findDialogRecord(dialogId, dialogKey);
+  const dialogRecord = await findDialogRecord(store, dialogId, dialogKey);
   if (!dialogRecord) {
     throw {
       code: "not_found",
@@ -247,7 +264,7 @@ export async function fetchConvMsgs(
     };
   }
 
-  const msgs: MessageWithKey[] = await fetchMessages(serverDb, dialogId, {
+  const msgs: MessageWithKey[] = await fetchMessages(store, dialogId, {
     limit,
     beforeKey,
     throwOnError: true,
@@ -275,6 +292,7 @@ function isDialogRecordForId(record: any, dialogId: string): boolean {
 }
 
 async function readDialogRecordByKey(
+  store: ApiMethodsStore,
   dialogKey: string | undefined,
   dialogId: string
 ): Promise<{ dbKey: string; record: any } | null> {
@@ -282,7 +300,7 @@ async function readDialogRecordByKey(
     return null;
   }
   try {
-    const record = await serverDb.get(dialogKey);
+    const record = await store.get(dialogKey);
     if (!isDialogRecordForId(record, dialogId)) return null;
     return { dbKey: dialogKey, record };
   } catch (error) {
@@ -291,15 +309,19 @@ async function readDialogRecordByKey(
   }
 }
 
-async function findDialogRecord(dialogId: string, dialogKey?: string): Promise<{
+async function findDialogRecord(
+  store: ApiMethodsStore,
+  dialogId: string,
+  dialogKey?: string
+): Promise<{
   dbKey: string;
   record: any;
 } | null> {
-  const directRecord = await readDialogRecordByKey(dialogKey, dialogId);
+  const directRecord = await readDialogRecordByKey(store, dialogKey, dialogId);
   if (directRecord) return directRecord;
 
   const suffix = `-${dialogId}`;
-  for await (const [key, value] of serverDb.iterator({
+  for await (const [key, value] of store.iterator({
     gte: "dialog-",
     lte: "dialog-\uffff",
   })) {

package/server/api/tableHandlers.ts

@@ -1,10 +1,12 @@
 // server/api/tableHandlers.ts
-import serverDb from "../../database/server/db";
+import { getServerAuthorityStore } from "../../database/server/db";
 import { metaKey, rowKey } from "../../database/keys";
 import { DataType } from "../../create/types";
 import type { ApiContext, ApiError } from "./types";
 import { assertCanReadTableRows } from "../handlers/tableAccess";
 
+const getTableHandlersStore = () => getServerAuthorityStore();
+
 interface ListTableRowsParams {
   tenantId: string;
   tableId: string;
@@ -14,6 +16,7 @@ export async function listTableRows(
   params: ListTableRowsParams,
   context?: ApiContext
 ) {
+  const store = getTableHandlersStore();
   const { tenantId, tableId } = params || {};
   if (!tenantId || !tableId) {
     throw new Error("tenantId 和 tableId 必传");
@@ -24,7 +27,7 @@ export async function listTableRows(
       message: "Cannot list table rows for another tenant",
     } as ApiError;
   }
-  const tableMeta = await serverDb.get(metaKey(tenantId, tableId)).catch(() => null);
+  const tableMeta = await store.get(metaKey(tenantId, tableId)).catch(() => null);
   try {
     await assertCanReadTableRows({
       userId: context.userId,
@@ -44,7 +47,7 @@ export async function listTableRows(
   const MAX_ROWS = 5000;
 
   const rows: any[] = [];
-  for await (const [, value] of serverDb.iterator({ gte, lte })) {
+  for await (const [, value] of store.iterator({ gte, lte })) {
     if (value?.type === DataType.TABLE_ROW && !value?.deletedAt) {
       rows.push(value);
       if (rows.length >= MAX_ROWS) break;