nolo-cli 0.1.23 → 0.1.29

package/server/handlers/agentRun/builtinAgents.ts

@@ -0,0 +1,347 @@
+import { APP_BUILDER_PRESET } from "../../../ai/tools/agent/presets/appBuilderPreset";
+import {
+  DEEPINFRA_KIMI_FALLBACK_MODEL,
+  FIREWORKS_KIMI_LATEST_MODEL,
+  OPENROUTER_KIMI_FALLBACK_MODEL,
+} from "../../../ai/llm/kimi";
+import { getNoloKey } from "../../../ai/llm/getNoloKey";
+import {
+  BUILTIN_AGENT_CREATOR_AGENT_KEY,
+  BUILTIN_AGENT_CREATOR_PROMPT,
+  BUILTIN_APP_BUILDER_AGENT_KEY,
+  BUILTIN_ECOMMERCE_AGENT_KEY,
+  BUILTIN_NOLO_AGENT_KEY,
+  buildBuiltinAgentCreatorRecord,
+} from "../../../core/builtinAgents";
+import { NOLO_CLUSTER_SERVERS } from "../../../database/config";
+
+export {
+  BUILTIN_AGENT_CREATOR_AGENT_KEY,
+  BUILTIN_AGENT_CREATOR_PROMPT,
+  BUILTIN_APP_BUILDER_AGENT_KEY,
+  BUILTIN_ECOMMERCE_AGENT_KEY,
+  BUILTIN_NOLO_AGENT_KEY,
+} from "../../../core/builtinAgents";
+
+export const BUILTIN_NOLO_AGENT_ID = "01NOLOAPPBLD000000019KCKT0";
+export const BUILTIN_APP_BUILDER_AGENT_ID = "01APPBUILDER00000001YAII3I";
+export const BUILTIN_ECOMMERCE_AGENT_ID = "01ECOMMERCEAG00000001PYQ2J";
+
+const DESKTOP_NOLO_LOCAL_FILE_TOOLS = [
+  "codeSearch",
+  "checkEnv",
+  "listLocalRoots",
+  "requestLocalFolderAccess",
+  "listLocalFiles",
+  "readLocalFile",
+  "proposeLocalFileChanges",
+  "executeApprovedLocalFileChanges",
+  "undoLocalFileChangeBatch",
+];
+
+const DESKTOP_NOLO_LOCAL_DEVELOPER_TOOLS = [
+  "listFiles",
+  "readFile",
+  "writeFile",
+  "editFile",
+  "searchFiles",
+  "applyPatch",
+  "runPackageScript",
+  "execShell",
+];
+
+const DESKTOP_NOLO_LOCAL_DEVELOPER_RUNTIME_POLICY = {
+  version: 1,
+  runtimeTools: DESKTOP_NOLO_LOCAL_DEVELOPER_TOOLS,
+  workspace: { mode: "task-worktree" },
+  shell: {
+    enabled: true,
+    mode: "worktree",
+    commandPolicy: "approval",
+    networkPolicy: "default-deny",
+    maxSeconds: 600,
+    maxOutputBytes: 20000,
+  },
+  git: {
+    canCommit: true,
+    canPushAlpha: false,
+    canMergeMain: false,
+  },
+  audit: {
+    logToolCalls: true,
+    logShellCommands: true,
+    writeToDialog: true,
+  },
+} as const;
+
+const DESKTOP_NOLO_LOCAL_CAPABILITY_PROMPT = [
+  "Nolo Desktop local runtime instructions:",
+  "- 你运行在 Nolo Desktop 本地环境里，具备本地文件和 shell 工具。用户要求整理桌面、整理文件、查看本机目录、执行命令、修改/创建文件时，不要默认回答无法访问本地环境。",
+  "- 对用户个人文件夹的请求，先调用 listLocalRoots 查看当前已授权 roots；如果目标文件夹还没授权，先调用 requestLocalFolderAccess 获取 rootId，再使用 listLocalFiles/readLocalFile/proposeLocalFileChanges。",
+  "- 对整理桌面这类请求，在 macOS 上优先考虑 `$HOME/Desktop`、`$HOME/Documents`、`$HOME/Downloads`；在 Windows 上优先考虑 `$env:USERPROFILE\\Desktop`、`$env:USERPROFILE\\Documents`、`$env:USERPROFILE\\Downloads` 以及 `$env:OneDrive\\Desktop`。未经用户明确确认，不要移动、删除、覆盖或批量修改用户文件。",
+  "- 如果用户选择“默认桌面路径”或类似默认系统目录，不要把“默认”原样传给 requestLocalFolderAccess。先用 execShell 解析当前机器上的真实绝对路径，再把具体 path 传给 requestLocalFolderAccess。",
+  "- 如果用户直接贴出绝对本地路径，例如 `C:\\Users\\me\\Downloads\\report.pdf` 或 `/Users/me/Desktop/note.txt`，把它视为本地文件访问请求，不要泛化回答“看不到你的磁盘”。先取父目录，申请该目录授权，再检查目标文件。",
+  "- 如果目标是 PDF、DOCX、图片或其他非纯文本文件，不要假装已经完整读懂文件内容。可以先确认文件存在、所在目录和文件名；需要读取正文时，明确说明当前本地工具主要覆盖目录/文本读取，必要时请用户上传文件或走现有上传流程。",
+  "- 对当前代码仓库、任务工作区、补丁编辑、搜索源码、运行项目命令这类 coding 任务，优先使用 listFiles/readFile/writeFile/editFile/searchFiles/applyPatch/execShell；不要把 workspace 任务误走到本地文件夹授权工具。",
+  "- You are running inside Nolo Desktop with local file and shell tools. For requests about the user's computer, Desktop folder, documents, project files, cleanup, organization, command execution, or file editing, do not say you cannot access the local environment by default.",
+  "- For user folders outside the current workspace, list authorized roots first and request folder access before using listLocalFiles/readLocalFile. For coding tasks inside the current repo/workspace, use listFiles/readFile/writeFile/editFile/searchFiles/applyPatch/execShell because those tools are workspace-scoped.",
+  "- On macOS, common user-folder paths include `$HOME/Desktop`, `$HOME/Documents`, and `$HOME/Downloads`. On Windows, common Desktop locations include `$env:USERPROFILE\\Desktop` and `$env:OneDrive\\Desktop`.",
+  "- When the user asks for the default Desktop/Documents/Downloads path, resolve the concrete absolute path with execShell first, then pass that explicit path into requestLocalFolderAccess.",
+  "- If the user pastes an absolute local path such as `C:\\Users\\me\\Downloads\\report.pdf`, treat it as a local-file access request. Derive the parent folder, request access to that folder, and then inspect the target file instead of giving a generic refusal.",
+  "- For PDF, DOCX, image, or other non-plain-text files, do not pretend you have already read the full contents. You may confirm existence and location first, then ask the user to upload the file when full content extraction is still required.",
+  "- For file organization requests, start by listing/reading enough context and present a concrete plan. Do not move, delete, overwrite, or bulk-edit user files until the user explicitly approves the proposed changes.",
+  "- If a requested path is inaccessible or ambiguous, report the exact attempted path/tool result and ask for the missing target instead of giving a generic refusal.",
+].join("\n");
+
+const BUILTIN_NOLO_BLOCKED_TOOL_NAMES = new Set([
+  "createAgent",
+  "deleteSpaces",
+]);
+
+const BUILTIN_NOLO_DRAFT_TOOL_NAMES = ["prepareAgentDraft"];
+
+const now = () => Date.now();
+
+function buildBaseAgent(overrides: Record<string, unknown>) {
+  const timestamp = now();
+  return {
+    type: "agent",
+    isPublic: true,
+    userId: "builtin",
+    createdAt: timestamp,
+    updatedAt: timestamp,
+    references: [],
+    ...overrides,
+  };
+}
+
+function buildBuiltinAppBuilderAgent() {
+  return buildBaseAgent({
+    ...APP_BUILDER_PRESET,
+    id: BUILTIN_APP_BUILDER_AGENT_ID,
+    tags: ["应用构建", "App Builder", "无代码"],
+  });
+}
+
+export function resolveBuiltinNoloModelConfig() {
+  if (process.env.FIREWORKS_API_KEY?.trim()) {
+    return {
+      provider: "fireworks",
+      model: FIREWORKS_KIMI_LATEST_MODEL,
+      inputPrice: 0.95 * 8,
+      outputPrice: 4 * 8,
+      fallbackProvider: process.env.DEEPINFRA_API_KEY?.trim()
+        ? "deepinfra"
+        : undefined,
+      fallbackModel: process.env.DEEPINFRA_API_KEY?.trim()
+        ? DEEPINFRA_KIMI_FALLBACK_MODEL
+        : undefined,
+      secondFallbackProvider: process.env.OPENROUTER_API_KEY?.trim()
+        ? "openrouter"
+        : undefined,
+      secondFallbackModel: process.env.OPENROUTER_API_KEY?.trim()
+        ? OPENROUTER_KIMI_FALLBACK_MODEL
+        : undefined,
+    };
+  }
+  if ((process.env.OPENAI_KEY || process.env.OPENAI_API_KEY)?.trim()) {
+    return {
+      provider: "openai",
+      model: "gpt-5.4-mini",
+      inputPrice: 0.75,
+      outputPrice: 4.5,
+    };
+  }
+  if (process.env.GOOGLE_API_KEY?.trim()) {
+    return {
+      provider: "google",
+      model: "gemini-2.5-flash",
+      inputPrice: 0.3 * 8,
+      outputPrice: 2.5 * 8,
+    };
+  }
+  return {
+    provider: "fireworks",
+    model: FIREWORKS_KIMI_LATEST_MODEL,
+    inputPrice: 0.95 * 8,
+    outputPrice: 4 * 8,
+  };
+}
+
+export function normalizeBuiltinPlatformAgentModelConfig(agentConfig: any) {
+  const key = agentConfig?.dbKey || agentConfig?.agentKey || agentConfig?.id;
+  const id = agentConfig?.id;
+  const isBuiltinNolo =
+    key === BUILTIN_NOLO_AGENT_KEY ||
+    id === BUILTIN_NOLO_AGENT_ID ||
+    key === BUILTIN_NOLO_AGENT_ID ||
+    (typeof key === "string" && key.endsWith(`-${BUILTIN_NOLO_AGENT_ID}`));
+  const isBuiltinEcommerce =
+    key === BUILTIN_ECOMMERCE_AGENT_KEY ||
+    id === BUILTIN_ECOMMERCE_AGENT_ID ||
+    key === BUILTIN_ECOMMERCE_AGENT_ID;
+
+  if (!isBuiltinNolo && !isBuiltinEcommerce) {
+    return agentConfig;
+  }
+
+  const modelConfig = resolveBuiltinNoloModelConfig();
+  const hasLocalProviderKey = !!getNoloKey(modelConfig.provider as any)?.trim();
+  const shouldRunBuiltinNoloInDesktop =
+    isBuiltinNolo && process.env.NOLO_DESKTOP === "1";
+  const baseTools = isBuiltinNolo
+    ? mergeToolNames(
+        (Array.isArray(agentConfig.tools) ? agentConfig.tools : []).filter(
+          (tool: string) => !BUILTIN_NOLO_BLOCKED_TOOL_NAMES.has(tool)
+        ),
+        BUILTIN_NOLO_DRAFT_TOOL_NAMES
+      )
+    : Array.isArray(agentConfig.tools) ? agentConfig.tools : undefined;
+  return {
+    ...agentConfig,
+    provider: modelConfig.provider,
+    model: modelConfig.model,
+    fallbackProvider: modelConfig.fallbackProvider,
+    fallbackModel: modelConfig.fallbackModel,
+    secondFallbackProvider: modelConfig.secondFallbackProvider,
+    secondFallbackModel: modelConfig.secondFallbackModel,
+    useServerProxy: true,
+    apiSource: "platform",
+    inputPrice: modelConfig.inputPrice,
+    outputPrice: modelConfig.outputPrice,
+    ...(isBuiltinNolo ? { tools: baseTools } : {}),
+    ...(shouldRunBuiltinNoloInDesktop
+      ? {
+          prompt: appendPrompt(
+            agentConfig.prompt,
+            DESKTOP_NOLO_LOCAL_CAPABILITY_PROMPT
+          ),
+          tools: mergeToolNames(
+            baseTools ?? [],
+            [
+              ...DESKTOP_NOLO_LOCAL_DEVELOPER_TOOLS,
+              ...DESKTOP_NOLO_LOCAL_FILE_TOOLS,
+            ]
+          ),
+          localWorkspaceMode: "task-worktree",
+          runtimeToolPolicy: DESKTOP_NOLO_LOCAL_DEVELOPER_RUNTIME_POLICY,
+        }
+      : {}),
+    ...(!hasLocalProviderKey && !shouldRunBuiltinNoloInDesktop
+      ? {
+          runtimeServerBase:
+            agentConfig.runtimeServerBase ?? NOLO_CLUSTER_SERVERS[0],
+        }
+      : {}),
+  };
+}
+
+function appendPrompt(base: unknown, addition: string) {
+  const basePrompt = typeof base === "string" ? base.trim() : "";
+  if (!basePrompt) return addition;
+  if (basePrompt.includes(addition)) return basePrompt;
+  return `${basePrompt}\n\n${addition}`;
+}
+
+function mergeToolNames(existing: string[], additions: string[]) {
+  const merged = [...existing];
+  for (const tool of additions) {
+    if (!merged.includes(tool)) merged.push(tool);
+  }
+  return merged;
+}
+
+export async function fetchBuiltinPlatformAgentRecord(agentKey: string) {
+  const isBuiltinPlatformAgent =
+    agentKey === BUILTIN_NOLO_AGENT_KEY ||
+    agentKey === BUILTIN_APP_BUILDER_AGENT_KEY ||
+    agentKey === BUILTIN_ECOMMERCE_AGENT_KEY;
+
+  if (!isBuiltinPlatformAgent) return null;
+
+  for (const server of NOLO_CLUSTER_SERVERS) {
+    try {
+      const response = await fetch(
+        `${server.replace(/\/+$/, "")}/api/v1/db/read/${encodeURIComponent(agentKey)}`
+      );
+      if (!response.ok) continue;
+      const data = await response.json();
+      if (data && typeof data === "object" && !Array.isArray(data)) {
+        return { ...data, dbKey: (data as any).dbKey ?? agentKey };
+      }
+    } catch {
+      // Try the next public server.
+    }
+  }
+
+  return null;
+}
+
+function buildBuiltinEcommerceAgent() {
+  const modelConfig = resolveBuiltinNoloModelConfig();
+  return buildBaseAgent({
+    id: BUILTIN_ECOMMERCE_AGENT_ID,
+    name: "电商商品参数助手",
+    introduction:
+      "专门获取淘宝、天猫、京东等电商商品真实参数、SKU、价格、库存和店铺信息，并做结构化对比。",
+    greeting: "发给我淘宝、天猫或京东商品链接，我会尽量获取真实商品参数。",
+    provider: modelConfig.provider,
+    model: modelConfig.model,
+    fallbackProvider: modelConfig.fallbackProvider,
+    fallbackModel: modelConfig.fallbackModel,
+    secondFallbackProvider: modelConfig.secondFallbackProvider,
+    secondFallbackModel: modelConfig.secondFallbackModel,
+    useServerProxy: true,
+    inputPrice: modelConfig.inputPrice,
+    outputPrice: modelConfig.outputPrice,
+    hasVision: false,
+    apiSource: "platform",
+    tags: ["电商", "商品参数", "数据抓取"],
+    tools: ["taobaoTmallProductScraper", "jdProductScraper"],
+    prompt: [
+      "你是电商商品参数助手，只负责获取和整理电商商品真实数据。",
+      "当用户给淘宝/天猫链接或商品 ID 时，先提取数字 itemId，再调用 taobaoTmallProductScraper。",
+      "当用户给京东链接或 SKU 时，先提取数字 skuId，再调用 jdProductScraper。",
+      "输出要结构化：商品标题、品牌、型号、店铺、价格、SKU/规格、库存、尺寸重量、质保、数据来源和缺失项。",
+      "用户问参数或详情时，必须展开工具返回的详细参数，不要只摘要基础字段。",
+      "只输出用户请求的数据结果；不要输出寒暄、确认、过渡句或结尾追问。",
+      "如果用户给多个商品，逐个调用对应工具后再做对比表。",
+      "如果工具返回缺失字段，如实说明缺失；不要编造参数，也不要要求用户手抄页面。",
+    ].join("\n"),
+  });
+}
+
+function buildBuiltinAgentCreatorAgent() {
+  const modelConfig = resolveBuiltinNoloModelConfig();
+  const hasLocalProviderKey = !!getNoloKey(modelConfig.provider as any)?.trim();
+  const baseRecord = buildBuiltinAgentCreatorRecord();
+  const { runtimeServerBase: _runtimeServerBase, ...localRecord } = baseRecord;
+  return buildBaseAgent({
+    ...(hasLocalProviderKey ? localRecord : baseRecord),
+    provider: modelConfig.provider,
+    model: modelConfig.model,
+    fallbackProvider: modelConfig.fallbackProvider,
+    fallbackModel: modelConfig.fallbackModel,
+    secondFallbackProvider: modelConfig.secondFallbackProvider,
+    secondFallbackModel: modelConfig.secondFallbackModel,
+    useServerProxy: true,
+    inputPrice: modelConfig.inputPrice,
+    outputPrice: modelConfig.outputPrice,
+  });
+}
+
+export function resolveBuiltinAgentConfig(agentKey: string) {
+  if (agentKey === BUILTIN_NOLO_AGENT_KEY) {
+    return null;
+  }
+  if (agentKey === BUILTIN_APP_BUILDER_AGENT_KEY) {
+    return buildBuiltinAppBuilderAgent();
+  }
+  if (agentKey === BUILTIN_ECOMMERCE_AGENT_KEY) {
+    return buildBuiltinEcommerceAgent();
+  }
+  if (agentKey === BUILTIN_AGENT_CREATOR_AGENT_KEY) {
+    return buildBuiltinAgentCreatorAgent();
+  }
+  return null;
+}

package/server/handlers/agentRun/continueDialogContext.ts

@@ -0,0 +1,90 @@
+import type { ExistingDialogRef } from "./dialogLookup";
+import type { ChatMessage } from "./types";
+
+type ResolveContinueDialogContextOptions = {
+  continueDialogId?: string | null;
+  history: ChatMessage[];
+  userId: string | null;
+  resolveExistingDialogRef(args: {
+    dialogId: string;
+    preferredUserId: string | null;
+  }): Promise<ExistingDialogRef | null>;
+  fetchConvMsgs(args: { dialogId: string; limit: number }): Promise<any[]>;
+  summarizeMessagesForDebug?: (messages: any[]) => unknown;
+  log?: (...args: unknown[]) => void;
+  warn?: (...args: unknown[]) => void;
+};
+
+export type ContinueDialogContext = {
+  contextHistory: ChatMessage[];
+  existingDialogRecord: any | null;
+  existingDialogKey: string | null;
+  continueDialogUserId: string | null;
+};
+
+export async function resolveContinueDialogContext(
+  options: ResolveContinueDialogContextOptions,
+): Promise<ContinueDialogContext> {
+  const {
+    continueDialogId,
+    history,
+    userId,
+    resolveExistingDialogRef,
+    fetchConvMsgs,
+    summarizeMessagesForDebug,
+    log = console.log,
+    warn = console.warn,
+  } = options;
+
+  let contextHistory: ChatMessage[] = history;
+  let existingDialogRecord: any = null;
+  let existingDialogKey: string | null = null;
+  let continueDialogUserId: string | null = userId;
+
+  if (!continueDialogId) {
+    return {
+      contextHistory,
+      existingDialogRecord,
+      existingDialogKey,
+      continueDialogUserId,
+    };
+  }
+
+  try {
+    const existingDialogRef = await resolveExistingDialogRef({
+      dialogId: continueDialogId,
+      preferredUserId: userId,
+    });
+    existingDialogRecord = existingDialogRef?.dialogRecord ?? null;
+    existingDialogKey = existingDialogRef?.dialogKey ?? null;
+    continueDialogUserId = existingDialogRef?.dialogUserId ?? userId;
+  } catch (_) { }
+
+  try {
+    const prevMsgs = await fetchConvMsgs({ dialogId: continueDialogId, limit: 50 });
+    contextHistory = (prevMsgs as any[])
+      .reverse()
+      .filter((message: any) => message.role !== "system")
+      .map((message: any): ChatMessage => ({
+        role: message.role,
+        content: message.content ?? null,
+        ...(message.tool_calls ? { tool_calls: message.tool_calls } : {}),
+        ...(message.toolCallId ? { tool_call_id: message.toolCallId } : {}),
+      }));
+    log("[agentRun] continue-history-loaded", {
+      dialogId: continueDialogId,
+      contextHistory: summarizeMessagesForDebug
+        ? summarizeMessagesForDebug(contextHistory as any[])
+        : contextHistory,
+    });
+  } catch (error) {
+    warn(`[agentRun] continueDialogId=${continueDialogId} load failed:`, error);
+  }
+
+  return {
+    contextHistory,
+    existingDialogRecord,
+    existingDialogKey,
+    continueDialogUserId,
+  };
+}

package/server/handlers/agentRun/debugContext.ts

@@ -0,0 +1,58 @@
+import type { MemoryItem } from "../../../ai/memory/types";
+
+const clipDebugText = (value: string, maxLength = 180) => {
+  const normalized = value.replace(/\s+/g, " ").trim();
+  if (normalized.length <= maxLength) return normalized;
+  return `${normalized.slice(0, maxLength - 1)}…`;
+};
+
+const summarizeMemoryRuntime = (selectedItems: MemoryItem[]) => ({
+  selectedCount: selectedItems.length,
+  selectedItems: selectedItems.map((item) => ({
+    id: item.id,
+    ownerType: item.ownerType,
+    ownerId: item.ownerId,
+    visibility: item.visibility,
+    subjectType: item.subjectType,
+    subjectId: item.subjectId,
+    kind: item.kind,
+    importance: item.importance,
+    confidence: item.confidence,
+    activationCount: item.activationCount,
+    tags: item.tags ?? [],
+    facet: item.facet ?? null,
+    patternKey: item.patternKey ?? null,
+    contentPreview: clipDebugText(item.content),
+  })),
+});
+
+export function buildAgentRunDebugContext({
+  enabled,
+  runtimeSystemMessageContext,
+  selectedMemoryItems,
+}: {
+  enabled: boolean;
+  runtimeSystemMessageContext: {
+    layers: Array<Record<string, any>>;
+    cacheProfile: unknown;
+  };
+  selectedMemoryItems: MemoryItem[];
+}) {
+  if (!enabled) return undefined;
+
+  return {
+    contextLayers: runtimeSystemMessageContext.layers.map((layer) => ({
+      id: layer.id,
+      owner: layer.owner,
+      cacheScope: layer.cacheScope,
+      charCount: layer.charCount,
+      estimatedTokens: layer.estimatedTokens,
+      ...(typeof layer.tokenBudget === "number"
+        ? { tokenBudget: layer.tokenBudget }
+        : {}),
+      ...(layer.budgetStatus ? { budgetStatus: layer.budgetStatus } : {}),
+    })),
+    cacheProfile: runtimeSystemMessageContext.cacheProfile,
+    memoryRuntime: summarizeMemoryRuntime(selectedMemoryItems),
+  };
+}

package/server/handlers/agentRun/dialogLookup.ts

@@ -0,0 +1,67 @@
+import serverDb from "../../../database/server/db";
+import { createKey } from "../../../database/keys";
+import { DataType } from "../../../create/types";
+
+export interface ExistingDialogRef {
+  dialogKey: string;
+  dialogRecord: any;
+  dialogUserId: string | null;
+}
+
+const buildDialogKey = (userId: string, dialogId: string) =>
+  createKey(DataType.DIALOG, userId, dialogId);
+
+export const extractDialogUserId = (
+  dialogKey: string,
+  dialogId: string,
+): string | null => {
+  const prefix = `${DataType.DIALOG}-`;
+  const suffix = `-${dialogId}`;
+  if (!dialogKey.startsWith(prefix) || !dialogKey.endsWith(suffix)) {
+    return null;
+  }
+
+  const userId = dialogKey.slice(prefix.length, dialogKey.length - suffix.length);
+  return userId || null;
+};
+
+export const resolveExistingDialogRef = async (params: {
+  dialogId: string;
+  preferredUserId?: string | null;
+}): Promise<ExistingDialogRef | null> => {
+  const { dialogId, preferredUserId } = params;
+  if (!dialogId) return null;
+
+  if (preferredUserId) {
+    const preferredDialogKey = buildDialogKey(preferredUserId, dialogId);
+    try {
+      const dialogRecord = await serverDb.get(preferredDialogKey);
+      return {
+        dialogKey: preferredDialogKey,
+        dialogRecord,
+        dialogUserId: preferredUserId,
+      };
+    } catch (_) {
+      // Fall through to a suffix scan when the caller cannot prove ownership.
+    }
+  }
+
+  const gte = `${DataType.DIALOG}-`;
+  const lte = `${DataType.DIALOG}-\uffff`;
+  for await (const [key, value] of serverDb.iterator({ gte, lte })) {
+    if (
+      typeof key === "string" &&
+      key.endsWith(`-${dialogId}`) &&
+      !key.includes(`-${dialogId}-msg-`) &&
+      value?.type === DataType.DIALOG
+    ) {
+      return {
+        dialogKey: key,
+        dialogRecord: value,
+        dialogUserId: extractDialogUserId(key, dialogId),
+      };
+    }
+  }
+
+  return null;
+};

package/server/handlers/agentRun/foregroundStreamDialog.ts

@@ -0,0 +1,13 @@
+import {
+  type PreparedRunDialog,
+  type PrepareRunDialogOptions,
+  prepareRunDialog,
+} from "./runDialogPreparation";
+
+export type ForegroundStreamDialog = PreparedRunDialog;
+
+export function prepareForegroundStreamDialog(
+  options: PrepareRunDialogOptions,
+): Promise<ForegroundStreamDialog> {
+  return prepareRunDialog(options);
+}