chore: checkpoint pi lab rust integration work

2026-07-11 20:39:18 +08:00
parent d47f6447fd
commit d16eccfe10
39 changed files with 4843 additions and 455 deletions
@@ -0,0 +1,162 @@
+#!/usr/bin/env node
+"use strict";
+
+const assert = require("node:assert");
+const fs = require("node:fs");
+const os = require("node:os");
+const path = require("node:path");
+const { chromium } = require("playwright");
+const {
+  setupWorkspaceAccess,
+  seedAiPolicy,
+  seedAiRuntime,
+} = require("./lib/control-plane-dev-seed");
+
+const BASE = process.env.MNOTE_UI_BASE_URL || "http://127.0.0.1:3000";
+const STAMP = Date.now();
+const OUT = process.env.MNOTE_PI_HISTORY_TAIL_OUT || path.join(os.tmpdir(), `mnote-pi-history-tail-${STAMP}`);
+const TIMEOUT = Number.parseInt(process.env.UI_TIMEOUT_MS || "90000", 10);
+const ACTOR_ID = process.env.MNOTE_E2E_ACTOR_ID || "mnote-e2e";
+const WORKSPACE_ID = `local-ws:${ACTOR_ID}:pi-history-tail`;
+const ROOT_PATH = path.join(OUT, "workspace");
+const ROOT_URI = `file://${ROOT_PATH}`;
+const PAGE_PATH = `pi-history-tail-${STAMP}.md`;
+const MODEL_PROVIDER = "omniroute";
+const MODEL_ID = "gpt-5.4-mini";
+const CHROMIUM_EXECUTABLE = process.env.PLAYWRIGHT_CHROMIUM_EXECUTABLE
+  || (fs.existsSync("/usr/bin/chromium-browser") ? "/usr/bin/chromium-browser" : "")
+  || (fs.existsSync("/usr/bin/chromium") ? "/usr/bin/chromium" : "")
+  || (fs.existsSync("/usr/bin/google-chrome") ? "/usr/bin/google-chrome" : "");
+
+async function quickLogin(page) {
+  await page.goto(`${BASE}/auth`, { waitUntil: "commit", timeout: TIMEOUT });
+  const quickLoginButton = page.getByRole("button", { name: "测试账号快速登录" });
+  await quickLoginButton.waitFor({ state: "visible", timeout: TIMEOUT });
+  await Promise.all([
+    page.waitForURL((url) => !url.toString().includes("/auth"), { timeout: TIMEOUT }).catch(() => null),
+    quickLoginButton.click(),
+  ]);
+}
+
+async function main() {
+  fs.mkdirSync(ROOT_PATH, { recursive: true });
+  fs.writeFileSync(path.join(ROOT_PATH, PAGE_PATH), "# Pi history tail smoke\n", "utf8");
+  fs.mkdirSync(OUT, { recursive: true });
+  const sessionId = `pi-history-tail-${STAMP}`;
+  const runId = `pi_run_${sessionId}`;
+  const duplicateText = `DUPLICATE_TAIL_REPLY_${STAMP}`;
+  const result = { ok: false, outputDir: OUT, checks: {}, screenshots: {} };
+  const browser = await chromium.launch({
+    headless: process.env.MNOTE_PI_HISTORY_TAIL_HEADED === "1" ? false : true,
+    executablePath: CHROMIUM_EXECUTABLE || undefined,
+  });
+  const context = await browser.newContext({ viewport: { width: 1440, height: 980 } });
+  await context.addInitScript(() => {
+    window.__MNOTE_PI_LAB_TEST__ = true;
+  });
+  const page = await context.newPage();
+  try {
+    await quickLogin(page);
+    const piSessionDir = path.join(OUT, "pi-session");
+    const piSessionFile = path.join(piSessionDir, `${STAMP}_session.jsonl`);
+    fs.mkdirSync(piSessionDir, { recursive: true });
+    fs.writeFileSync(
+      piSessionFile,
+      [
+        JSON.stringify({ id: "u1", type: "message", message: { role: "user", content: "first duplicate history turn" }, seq: 1 }),
+        JSON.stringify({ id: "a1", parentId: "u1", type: "message", message: { role: "assistant", content: [{ type: "text", text: duplicateText }] }, seq: 2 }),
+      ].join("\n") + "\n",
+      "utf8",
+    );
+    await setupWorkspaceAccess(page.request, BASE, {
+      actorId: ACTOR_ID,
+      email: "mnote.e2e@example.com",
+      username: ACTOR_ID,
+      displayName: ACTOR_ID,
+      role: "admin",
+      workspaceId: WORKSPACE_ID,
+      workspaceName: "Pi history tail smoke",
+      rootPath: ROOT_PATH,
+      rootUri: ROOT_URI,
+      permission: "write",
+      capabilities: ["ai", "read", "write"],
+      timeoutMs: TIMEOUT,
+    });
+    await seedAiPolicy(page.request, BASE, {
+      id: `pi-history-tail-policy-${STAMP}`,
+      userId: ACTOR_ID,
+      workspaceId: WORKSPACE_ID,
+      allowedRootsJson: [{ rootUri: ROOT_URI, rootPath: ROOT_PATH, permission: "write" }],
+      modelPolicyJson: {
+        defaultModel: `${MODEL_PROVIDER}/${MODEL_ID}`,
+        allowedModels: [`${MODEL_PROVIDER}/${MODEL_ID}`],
+        tools: {},
+        skills: {},
+        mcpServers: {},
+      },
+      quotaJson: { daily: 20 },
+      timeoutMs: TIMEOUT,
+    });
+    await seedAiRuntime(page.request, BASE, {
+      userId: ACTOR_ID,
+      workspaceId: WORKSPACE_ID,
+      documentId: PAGE_PATH,
+      sessionId,
+      runId,
+      title: "Pi history duplicate tail smoke",
+      profile: "pi_lab",
+      acpRuntime: "pi",
+      status: "runtime_running",
+      runtimeJson: {
+        runtimeMode: "rpc",
+        rootUri: ROOT_URI,
+        workspaceId: WORKSPACE_ID,
+        pagePath: PAGE_PATH,
+        pageTitle: "Pi history duplicate tail smoke",
+        modelProvider: MODEL_PROVIDER,
+        modelId: MODEL_ID,
+        thinkingLevel: "high",
+        piSessionDir,
+        piSessionFile,
+      },
+      payloadJson: { message: "history duplicate tail smoke" },
+      events: [
+        { eventType: "user_prompt", payloadJson: { message: "first duplicate history turn" } },
+        { eventType: "pi_rpc_event", payloadJson: { type: "message_update", assistantMessageEvent: { type: "text_delta", delta: duplicateText } } },
+        { eventType: "pi_rpc_event", payloadJson: { type: "message_update", assistantMessageEvent: { type: "text_end" } } },
+        { eventType: "pi_rpc_event", payloadJson: { type: "message_update", assistantMessageEvent: { type: "text_delta", delta: duplicateText } } },
+        { eventType: "pi_rpc_event", payloadJson: { type: "message_update", assistantMessageEvent: { type: "text_end" } } },
+        { eventType: "pi_rpc_event", payloadJson: { type: "agent_end", messages: [{ role: "assistant", content: [{ type: "text", text: duplicateText }] }] } },
+      ],
+      timeoutMs: TIMEOUT,
+    });
+    await page.goto(`${BASE}/page-ai/pi`, { waitUntil: "commit", timeout: TIMEOUT });
+    await page.locator('[data-page-ai-pi-lab="drawer"]').waitFor({ state: "visible", timeout: TIMEOUT });
+    await page.waitForFunction(() => !!window.__mnotePiLabTest, null, { timeout: TIMEOUT });
+    await page.locator("[data-page-ai-pi-lab-history]").click();
+    await page.locator(`[data-page-ai-pi-lab-history-row="${sessionId}"]`).waitFor({ state: "visible", timeout: TIMEOUT });
+    await page.locator(`[data-page-ai-pi-lab-history-session="${sessionId}"]`).first().click();
+    await page.waitForFunction(
+      ({ text }) => Array.from(document.querySelectorAll('[data-page-ai-pi-lab-message-role="assistant"]'))
+        .filter((node) => (node.textContent || "").includes(text)).length === 1,
+      { text: duplicateText },
+      { timeout: TIMEOUT },
+    );
+    result.checks.duplicateAssistantCount = await page.locator('[data-page-ai-pi-lab-message-role="assistant"]', { hasText: duplicateText }).count();
+    assert.equal(result.checks.duplicateAssistantCount, 1, "history replay should use Pi JSONL tree as the single message source");
+    result.screenshots.history = path.join(OUT, "history-tail.png");
+    await page.screenshot({ path: result.screenshots.history, fullPage: false });
+    result.ok = true;
+  } catch (error) {
+    result.error = error && error.stack ? error.stack : String(error);
+    result.screenshots.failure = path.join(OUT, "failure.png");
+    await page.screenshot({ path: result.screenshots.failure, fullPage: true }).catch(() => {});
+    process.exitCode = 1;
+  } finally {
+    fs.writeFileSync(path.join(OUT, "result.json"), JSON.stringify(result, null, 2), "utf8");
+    await browser.close();
+    console.log(JSON.stringify({ ok: result.ok, outputDir: OUT, result: path.join(OUT, "result.json") }, null, 2));
+  }
+}
+
+main();