feat: add prompt attachment support across sessions

2026-04-11 16:58:39 +07:00
parent aa1d33f0a0
commit 28f45a41a5
21 changed files with 915 additions and 70 deletions
@@ -0,0 +1,118 @@
+import { describe, expect, it } from "vitest";
+
+import {
+  CreateAgentRequestMessageSchema,
+  CreatePaseoWorktreeRequestSchema,
+  SendAgentMessageRequestSchema,
+} from "./messages.js";
+
+describe("shared messages attachments", () => {
+  it("keeps known attachments and drops unknown create-agent attachments", () => {
+    const parsed = CreateAgentRequestMessageSchema.parse({
+      type: "create_agent_request",
+      requestId: "req-1",
+      config: {
+        provider: "codex",
+        cwd: "/tmp/repo",
+      },
+      initialPrompt: "Review this PR",
+      attachments: [
+        {
+          type: "github_pr",
+          mimeType: "application/github-pr",
+          number: 123,
+          title: "Fix race in worktree setup",
+          url: "https://github.com/getpaseo/paseo/pull/123",
+          body: "Body",
+          baseRefName: "main",
+          headRefName: "fix/worktree-race",
+        },
+        {
+          type: "future_attachment",
+          mimeType: "application/future",
+          foo: "bar",
+        },
+      ],
+    });
+
+    expect(parsed.attachments).toEqual([
+      {
+        type: "github_pr",
+        mimeType: "application/github-pr",
+        number: 123,
+        title: "Fix race in worktree setup",
+        url: "https://github.com/getpaseo/paseo/pull/123",
+        body: "Body",
+        baseRefName: "main",
+        headRefName: "fix/worktree-race",
+      },
+    ]);
+  });
+
+  it("keeps known attachments and drops unknown send-message attachments", () => {
+    const parsed = SendAgentMessageRequestSchema.parse({
+      type: "send_agent_message_request",
+      requestId: "req-2",
+      agentId: "agent-1",
+      text: "Review",
+      attachments: [
+        {
+          type: "github_issue",
+          mimeType: "application/github-issue",
+          number: 55,
+          title: "Improve startup error details",
+          url: "https://github.com/getpaseo/paseo/issues/55",
+          body: "Body",
+        },
+        {
+          type: "future_attachment",
+          mimeType: "application/future",
+          foo: "bar",
+        },
+      ],
+    });
+
+    expect(parsed.attachments).toEqual([
+      {
+        type: "github_issue",
+        mimeType: "application/github-issue",
+        number: 55,
+        title: "Improve startup error details",
+        url: "https://github.com/getpaseo/paseo/issues/55",
+        body: "Body",
+      },
+    ]);
+  });
+
+  it("keeps known attachments and drops unknown worktree-create attachments", () => {
+    const parsed = CreatePaseoWorktreeRequestSchema.parse({
+      type: "create_paseo_worktree_request",
+      requestId: "req-3",
+      cwd: "/tmp/repo",
+      attachments: [
+        {
+          type: "github_pr",
+          mimeType: "application/github-pr",
+          number: 99,
+          title: "Fork-safe PR checkout",
+          url: "https://github.com/getpaseo/paseo/pull/99",
+        },
+        {
+          type: "future_attachment",
+          mimeType: "application/future",
+          foo: "bar",
+        },
+      ],
+    });
+
+    expect(parsed.attachments).toEqual([
+      {
+        type: "github_pr",
+        mimeType: "application/github-pr",
+        number: 99,
+        title: "Fork-safe PR checkout",
+        url: "https://github.com/getpaseo/paseo/pull/99",
+      },
+    ]);
+  });
+});