nolo-cli 0.1.40 → 0.1.41

package/ai/tools/readTool.ts

@@ -0,0 +1,194 @@
+// 文件路径: ai/tools/readTool.ts
+
+import { readAction } from "../../database/actions/read";
+import { readAndWaitAction } from "../../database/actions/readAndWait";
+import { slateToSimplifiedMarkdown } from "../../create/editor/transforms/slateToSimplifiedMarkdown";
+import type { PageData } from "../../render/page/types";
+import { getRuntimeServerContext } from "../../database/runtimeServerContext";
+import { isTableMetaKey, rowKey, isAgentKey } from "../../database/keys";
+import { DataType } from "../../create/types";
+import { TableMeta } from "../../render/table/types";
+import { fetchAndSerializeTable } from "../../render/table/utils/tableSerialization";
+import { readFileFunc } from "./readFileTool";
+
+// ---- Types ----
+
+export type ReadArgs = {
+  /**
+   * 对应 database 的 dbKey
+   */
+  dbKey: string;
+
+  /**
+   * 是否等待远程结果：
+   * - false（默认）：使用 readAction，本地优先，远程在后台同步
+   * - true：使用 readAndWaitAction，等待远程与本地决策后返回“权威”结果
+   */
+  waitRemote?: boolean;
+
+  /**
+   * Backward-compatible escape hatch for models that call read with a local
+   * file path even when the readFile tool is available.
+   */
+  filePath?: string;
+  path?: string;
+};
+
+// ---- 工具 Schema，供 LLM 调用 ----
+
+export const readFunctionSchema = {
+  name: "read",
+  description: [
+    "根据指定的 dbKey 从本地/远程数据库读取一条记录。",
+    "支持所有内容类型：",
+    "- page-xxx：页面内容（返回 Markdown）",
+    "- dialog-xxx：对话历史（返回消息列表）",
+    "- agent-xxx：Agent 配置",
+    "- table-xxx：表格数据",
+    "- space-xxx：Space 完整数据（含分类和内容目录）",
+    "行为说明：",
+    "- 默认本地优先：若本地存在则立即返回，后台同步远程；",
+    "- waitRemote=true：等待远程权威结果后返回（适合需要最新数据的场景）。",
+  ].join("\n"),
+  parameters: {
+    type: "object",
+    properties: {
+      dbKey: {
+        type: "string",
+        description: "要读取的数据键，例如 PAGE-xxx、dialog-xxx、agent-xxx、space-xxx。",
+      },
+      waitRemote: {
+        type: "boolean",
+        description:
+          "是否等待远程结果。false=本地优先并在后台同步远程；true=等待远程与本地决策后返回。",
+        default: false,
+      },
+    },
+    required: ["dbKey"],
+  },
+};
+
+// ---- 执行函数 ----
+
+export async function readFunc(
+  args: ReadArgs,
+  thunkApi: any,
+  context?: { parentMessageId?: string; signal?: AbortSignal }
+): Promise<{ rawData: any; displayData?: string }> {
+  const { dbKey, waitRemote = false } = args || {};
+  const localFilePath =
+    typeof args?.filePath === "string"
+      ? args.filePath
+      : typeof args?.path === "string"
+        ? args.path
+        : "";
+
+  if (!dbKey && localFilePath) {
+    const result = await readFileFunc(
+      {
+        filePath: localFilePath,
+        startLine: (args as any).startLine,
+        endLine: (args as any).endLine,
+      },
+      thunkApi,
+      context
+    );
+    return {
+      rawData: result,
+      displayData:
+        typeof result === "string" ? result : JSON.stringify(result, null, 2),
+    };
+  }
+
+  if (!dbKey || typeof dbKey !== "string") {
+    throw new Error("read 工具需要提供字符串类型的 id（dbKey）。");
+  }
+
+  try {
+    const signal = context?.signal;
+
+    const result = waitRemote
+      ? await readAndWaitAction(dbKey, thunkApi)
+      : await readAction({ dbKey, signal }, thunkApi);
+
+    if (!result) {
+      throw new Error(`未找到 dbKey 为 ${dbKey} 的数据。`);
+    }
+
+    // 智能化处理：如果是页面，自动转 Markdown
+    if (dbKey.startsWith("PAGE-") || (result as any).type === DataType.DOC) {
+      const pageData = result as PageData;
+      const markdownContent = slateToSimplifiedMarkdown(pageData.slateData || []);
+
+      return {
+        rawData: {
+          ...pageData,
+          content: markdownContent, // 附加转换后的内容
+        },
+        displayData: `已成功读取页面《${pageData.title}》。\n\n内容预览：\n\n${markdownContent}`,
+      };
+    }
+
+    // 智能化处理：如果是表格，自动转 Markdown
+    if (isTableMetaKey(dbKey) || (result as any).type === DataType.TABLE) {
+      const tableMeta = result as TableMeta;
+      const title = tableMeta.displayName || `Table (${tableMeta.tableId})`;
+
+      // Use shared utility
+      const { rows, markdown: tableMd } = await thunkApi.dispatch(
+        async (_dispatch: any, getState: any, { db }: any) => {
+          const state = getState();
+          const { currentToken: token, remoteServers } =
+            getRuntimeServerContext(state);
+
+          return await fetchAndSerializeTable(tableMeta, db, {
+            token,
+            remoteServers,
+          });
+        }
+      );
+
+      return {
+        rawData: {
+          ...tableMeta,
+          rows,
+          markdown: tableMd,
+        },
+        displayData: `已成功读取表格《${title}》。共 ${rows.length} 行数据。\n\n内容预览：\n\n${tableMd}`,
+      };
+    }
+
+    // 智能化处理：如果是 Agent/Cybot，显示基本信息
+    if (
+      isAgentKey(dbKey) ||
+      (result as any).type === DataType.AGENT ||
+      (result as any).type === DataType.CYBOT
+    ) {
+      const agent = result as any;
+      const name = agent.name || "未命名 Agent";
+      const desc = agent.introduction || agent.description || "无描述";
+      const modelInfo = agent.model ? ` (模型: ${agent.model})` : "";
+
+      return {
+        rawData: agent,
+        displayData: `已成功读取 Agent《${name}》${modelInfo}。\n\n描述：${desc}\n提示词预览：\n${agent.prompt?.slice(0, 200) || "无"
+          }...`,
+      };
+    }
+
+    const sourceLabel = waitRemote
+      ? "已等待远程与本地完成后返回权威结果"
+      : "本地优先，已触发后台与远程同步";
+
+
+    return {
+      rawData: result,
+      displayData: `已读取数据: "${result.dbKey}"（${sourceLabel}）`,
+    };
+  } catch (error: any) {
+    console.error("执行 read 工具时发生错误:", error);
+    throw new Error(
+      `读取数据 (${dbKey}) 失败：${error?.message || String(error)}`
+    );
+  }
+}

package/ai/tools/readXPostTool.ts

@@ -0,0 +1,229 @@
+import { createXReadFailure } from "../../integrations/x-reader/types";
+import type { XPost, XReadResult } from "../../integrations/x-reader/types";
+import { getRequestConfig, ToolApiError } from "./toolApiClient";
+
+export const readXPostFunctionSchema = {
+  name: "read_x_post",
+  description:
+    "读取 X/Twitter status 链接的可见帖子正文、作者和结构化数据。适合用户给出 x.com/twitter.com 帖子链接并要求查看、总结、解释或抽取信息的场景。默认通过桌面本地 Chrome/CDP bridge 读取，不要求用户粘贴 cookie 或 token。",
+  parameters: {
+    type: "object",
+    properties: {
+      url: {
+        type: "string",
+        description:
+          "要读取的 X/Twitter status URL，例如 https://x.com/user/status/123。",
+      },
+      keepOpen: {
+        type: "boolean",
+        description: "调试时是否保留临时 Chrome bridge。默认 false。",
+        default: false,
+      },
+      profileDir: {
+        type: "string",
+        description:
+          "桌面端本地 Chrome 专用 profile 目录。用于让用户在本机登录一次 X 后复用本地账号状态；不要传入用户日常 Chrome profile。",
+      },
+      headless: {
+        type: "boolean",
+        description:
+          "是否用 headless 模式启动本地 Chrome。需要用户首次登录本地 X profile 时设为 false 并配合 keepOpen。",
+        default: true,
+      },
+    },
+    required: ["url"],
+  },
+};
+
+type ReadXPostToolContext = {
+  reader?: (
+    url: string,
+    args: { keepOpen?: boolean; profileDir?: string; headless?: boolean },
+  ) => Promise<XReadResult<XPost>>;
+};
+
+async function callLocalReadXPostApi(
+  thunkApi: any,
+  body: object,
+): Promise<XReadResult<XPost>> {
+  const { currentServer, token } = getRequestConfig(thunkApi);
+  const browserOrigin = (globalThis as any).window?.location?.origin;
+  const baseUrl = typeof browserOrigin === "string" && browserOrigin
+    ? browserOrigin
+    : currentServer.replace(/\/+$/, "");
+  const headers: Record<string, string> = { "Content-Type": "application/json" };
+  if (token) headers.Authorization = `Bearer ${token}`;
+
+  const response = await fetch(`${baseUrl}/api/read-x-post`, {
+    method: "POST",
+    headers,
+    body: JSON.stringify(body),
+  });
+  const text = await response.text();
+  let data: any = null;
+  try {
+    data = text ? JSON.parse(text) : null;
+  } catch {
+    data = null;
+  }
+  if (!response.ok) {
+    throw new ToolApiError(
+      data?.error?.message ?? `read_x_post API 请求失败，状态码: ${response.status}`,
+      {
+        status: response.status,
+        code: data?.error?.code,
+        details: data,
+      },
+    );
+  }
+  return data as XReadResult<XPost>;
+}
+
+async function callDesktopReadXPostApi(
+  thunkApi: any,
+  body: object,
+): Promise<XReadResult<XPost>> {
+  const { token } = getRequestConfig(thunkApi);
+  const port = Number(process.env.NOLO_DESKTOP_SERVER_PORT ?? 3233);
+  const headers: Record<string, string> = { "Content-Type": "application/json" };
+  if (token) headers.Authorization = `Bearer ${token}`;
+
+  const response = await fetch(`http://127.0.0.1:${port}/api/read-x-post`, {
+    method: "POST",
+    headers,
+    body: JSON.stringify(body),
+  });
+  const text = await response.text();
+  let data: any = null;
+  try {
+    data = text ? JSON.parse(text) : null;
+  } catch {
+    data = null;
+  }
+  if (!response.ok) {
+    throw new ToolApiError(
+      data?.error?.message ?? `desktop read_x_post API 请求失败，状态码: ${response.status}`,
+      {
+        status: response.status,
+        code: data?.error?.code,
+        details: data,
+      },
+    );
+  }
+  return data as XReadResult<XPost>;
+}
+
+async function readWithDefaultBridge(
+  url: string,
+  args: { keepOpen?: boolean; profileDir?: string; headless?: boolean },
+  thunkApi?: any,
+): Promise<XReadResult<XPost>> {
+  if (process.env.PLATFORM === "web") {
+    if (thunkApi?.getState) {
+      return callLocalReadXPostApi(thunkApi, { url, ...args });
+    }
+
+    if (typeof window !== "undefined" && (window as any).__NOLO_DESKTOP__) {
+      try {
+        const res = await fetch("/api/read-x-post", {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify({ url, ...args }),
+        });
+        const data = await res.json().catch(() => null);
+        if (res.ok && data) {
+          return data as XReadResult<XPost>;
+        }
+        return createXReadFailure({
+          code: "network_error",
+          message: `desktop read_x_post endpoint failed: HTTP ${res.status}`,
+          nextStep: "请确认桌面端本地服务仍在运行，然后重试。",
+          backend: "desktop_local_browser",
+        });
+      } catch (error) {
+        return createXReadFailure({
+          code: "network_error",
+          message:
+            error instanceof Error
+              ? error.message
+              : "desktop read_x_post endpoint request failed",
+          nextStep: "请确认桌面端本地服务仍在运行，然后重试。",
+          backend: "desktop_local_browser",
+        });
+      }
+    }
+
+    return createXReadFailure({
+      code: "not_connected",
+      message: "read_x_post 需要通过服务器或桌面本地 bridge 执行，不能在普通浏览器 bundle 内直接启动 Chrome/CDP。",
+      nextStep: "请使用服务器 agent run 或桌面端本地 bridge 路径执行该工具。",
+      backend: "desktop_local_browser",
+    });
+  }
+
+  if (process.env.NOLO_DESKTOP === "1" && thunkApi?.getState) {
+    return callDesktopReadXPostApi(thunkApi, { url, ...args });
+  }
+
+  const importBridge = new Function("specifier", "return import(specifier)") as <
+    T = any,
+  >(
+    specifier: string,
+  ) => Promise<T>;
+  const { readXPostWithBridge } = await importBridge<{
+    readXPostWithBridge: (
+      url: string,
+      args: { keepOpen?: boolean; profileDir?: string; headless?: boolean },
+    ) => Promise<XReadResult<XPost>>;
+  }>("../../integrations/x-reader/bridge/readXPostWithBridge.ts");
+  return readXPostWithBridge(url, args);
+}
+
+function assertXStatusUrl(url: string) {
+  if (!/^https?:\/\/(x|twitter)\.com\/[^/]+\/status\/\d+/i.test(url)) {
+    throw new Error("read_x_post 需要一个有效的 X/Twitter status URL。");
+  }
+}
+
+function formatDisplay(result: XReadResult<XPost>) {
+  if (!result.ok) {
+    return [
+      `读取 X 帖子失败：${result.message}`,
+      `失败代码：${result.code}`,
+      result.nextStep ? `下一步：${result.nextStep}` : "",
+      `后端：${result.backend}`,
+    ]
+      .filter(Boolean)
+      .join("\n");
+  }
+
+  const post = result.data;
+  return [
+    `已读取 X 帖子：@${post.author.handle}${post.author.displayName ? `（${post.author.displayName}）` : ""}`,
+    `URL: ${post.url}`,
+    `后端：${result.backend}`,
+    "",
+    post.text,
+  ].join("\n");
+}
+
+export async function readXPostFunc(
+  args: { url: string; keepOpen?: boolean; profileDir?: string; headless?: boolean },
+  thunkApi: any,
+  context: ReadXPostToolContext = {},
+): Promise<{ rawData: XReadResult<XPost>; displayData: string }> {
+  const url = String(args?.url ?? "").trim();
+  assertXStatusUrl(url);
+
+  const keepOpen = Boolean(args?.keepOpen);
+  const profileDir = String(args?.profileDir ?? "").trim() || undefined;
+  const headless = args?.headless;
+  const rawData =
+    (await context.reader?.(url, { keepOpen, profileDir, headless })) ??
+    (await readWithDefaultBridge(url, { keepOpen, profileDir, headless }, thunkApi));
+
+  return {
+    rawData,
+    displayData: formatDisplay(rawData),
+  };
+}