feat(code): multi-repo support for cloud tasks

adboio · adboio · commit 09d5cbd61c98 · 2026-07-06T11:06:29.000-04:00
diff --git a/packages/agent/src/server/agent-server.ts b/packages/agent/src/server/agent-server.ts
@@ -16,6 +16,7 @@ import {
 import { type ServerType, serve } from "@hono/node-server";
 import { execGh } from "@posthog/git/gh";
 import { getCurrentBranch } from "@posthog/git/queries";
+import { buildAdditionalDirectoriesPrompt } from "@posthog/shared";
 import { unzipSync } from "fflate";
 import { Hono } from "hono";
 import { z } from "zod";
@@ -2471,11 +2472,14 @@ export class AgentServer {
     slackThreadUrl?: string | null,
     inboxReportUrl?: string | null,
   ): string | { append: string } {
-    const cloudAppend = this.buildCloudSystemPrompt(
-      prUrl,
-      slackThreadUrl,
-      inboxReportUrl,
-    );
+    // Extra cloned repos are surfaced to the model exactly like the local
+    // agent's additional directories, so it knows it may work across them.
+    const cloudAppend = [
+      this.buildCloudSystemPrompt(prUrl, slackThreadUrl, inboxReportUrl),
+      buildAdditionalDirectoriesPrompt(this.config.additionalDirectories),
+    ]
+      .filter(Boolean)
+      .join("\n\n");
     const userPrompt = this.config.claudeCode?.systemPrompt;
 
     // String override: combine user prompt with cloud instructions
@@ -2513,8 +2517,14 @@ export class AgentServer {
     const plugins = this.config.claudeCode?.plugins;
     const effort =
       runtimeAdapter === "claude" ? this.config.reasoningEffort : undefined;
+    // Claude-only: the SDK enforces directory access. Codex relies on the
+    // <additional_directories> prompt block plus the sandbox's open filesystem.
+    const additionalDirectories =
+      runtimeAdapter === "claude"
+        ? this.config.additionalDirectories
+        : undefined;
 
-    if (!plugins?.length && !effort) {
+    if (!plugins?.length && !effort && !additionalDirectories?.length) {
       return undefined;
     }
 
@@ -2525,6 +2535,9 @@ export class AgentServer {
     if (effort) {
       options.effort = effort;
     }
+    if (additionalDirectories?.length) {
+      options.additionalDirectories = additionalDirectories;
+    }
     return { claudeCode: { options } };
   }
 
diff --git a/packages/agent/src/server/bin.ts b/packages/agent/src/server/bin.ts
@@ -115,6 +115,10 @@ program
     "--allowedDomains <domains>",
     "Comma-separated list of domains allowed for web tools (WebFetch, WebSearch)",
   )
+  .option(
+    "--additionalDirectories <paths>",
+    "Comma-separated extra repo checkout paths the agent may read/edit beyond --repositoryPath",
+  )
   .action(async (options) => {
     const envResult = envSchema.safeParse(process.env);
 
@@ -149,6 +153,13 @@ program
           .filter(Boolean)
       : undefined;
 
+    const additionalDirectories = options.additionalDirectories
+      ? options.additionalDirectories
+          .split(",")
+          .map((d: string) => d.trim())
+          .filter(Boolean)
+      : undefined;
+
     if (
       env.POSTHOG_CODE_RUNTIME_ADAPTER &&
       env.POSTHOG_CODE_MODEL &&
@@ -186,6 +197,7 @@ program
       baseBranch: options.baseBranch,
       claudeCode,
       allowedDomains,
+      additionalDirectories,
       runtimeAdapter: env.POSTHOG_CODE_RUNTIME_ADAPTER,
       model: env.POSTHOG_CODE_MODEL,
       reasoningEffort: env.POSTHOG_CODE_REASONING_EFFORT,
diff --git a/packages/agent/src/server/types.ts b/packages/agent/src/server/types.ts
@@ -29,6 +29,13 @@ export interface AgentServerConfig {
   mcpServers?: RemoteMcpServer[];
   baseBranch?: string;
   claudeCode?: ClaudeCodeConfig;
+  /**
+   * Extra repo checkout paths (e.g. /tmp/workspace/repos/org/repo) the agent may
+   * read and edit beyond the primary `repositoryPath`. Surfaced to the model as
+   * an `<additional_directories>` prompt block and passed to the Claude SDK's
+   * `additionalDirectories`.
+   */
+  additionalDirectories?: string[];
   allowedDomains?: string[];
   runtimeAdapter?: "claude" | "codex";
   model?: string;
diff --git a/packages/api-client/src/posthog-client.ts b/packages/api-client/src/posthog-client.ts
@@ -2218,6 +2218,7 @@ export class PostHogAPIClient {
         model?: string | null;
         reasoning_effort?: string | null;
         channel?: string | null;
+        additional_repositories?: string[];
       },
   ) {
     const teamId = await this.getTeamId();
diff --git a/packages/core/src/task-detail/taskCreationSaga.ts b/packages/core/src/task-detail/taskCreationSaga.ts
@@ -644,6 +644,11 @@ export class TaskCreationSaga extends Saga<
             input.workspaceMode === "cloud"
               ? (input.reasoningLevel ?? null)
               : undefined,
+          additional_repositories:
+            input.workspaceMode === "cloud" &&
+            input.additionalRepositories?.length
+              ? input.additionalRepositories
+              : undefined,
           signal_report: input.signalReportId ?? undefined,
           channel: input.channelId ?? undefined,
         });
diff --git a/packages/core/src/task-detail/taskInput.ts b/packages/core/src/task-detail/taskInput.ts
@@ -19,6 +19,7 @@ export interface PrepareTaskInputOptions {
   sandboxEnvironmentId?: string;
   signalReportId?: string;
   additionalDirectories?: string[];
+  additionalRepositories?: string[];
   channelContext?: string;
   channelName?: string;
   channelId?: string;
@@ -58,6 +59,9 @@ export function prepareTaskInput(
       options.signalReportId && isCloud ? "signal_report" : undefined,
     signalReportId: options.signalReportId,
     additionalDirectories: isCloud ? undefined : options.additionalDirectories,
+    additionalRepositories: isCloud
+      ? options.additionalRepositories
+      : undefined,
     channelContext: options.channelContext,
     channelName: options.channelName,
     channelId: options.channelId,
diff --git a/packages/shared/src/agent-prompts.ts b/packages/shared/src/agent-prompts.ts
@@ -0,0 +1,18 @@
+/**
+ * Builds the `<additional_directories>` system-prompt block shared by the local
+ * agent (workspace-server) and the cloud agent-server, so both describe extra
+ * working directories to the model identically. The block itself is returned
+ * without surrounding whitespace; callers join it into their prompt. Returns an
+ * empty string when there are no additional directories.
+ */
+export function buildAdditionalDirectoriesPrompt(
+  directories: readonly string[] | undefined,
+): string {
+  if (!directories?.length) return "";
+  const escapeXml = (value: string) =>
+    value.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;");
+  const dirs = directories
+    .map((directory) => `  <directory>${escapeXml(directory)}</directory>`)
+    .join("\n");
+  return `The user has granted you access to additional directories outside the working directory. You may read and edit files in these paths just like the working directory:\n<additional_directories>\n${dirs}\n</additional_directories>`;
+}
diff --git a/packages/shared/src/domain-types.ts b/packages/shared/src/domain-types.ts
@@ -48,6 +48,8 @@ export interface Task {
   created_by?: UserBasic | null;
   origin_product: string;
   repository?: string | null; // Format: "organization/repository" (e.g., "posthog/posthog-js")
+  /** Extra "org/repo" repos cloned into the cloud sandbox alongside `repository`. */
+  additional_repositories?: string[];
   github_integration?: number | null;
   github_user_integration?: string | null;
   json_schema?: Record<string, unknown> | null;
diff --git a/packages/shared/src/index.ts b/packages/shared/src/index.ts
@@ -1,3 +1,4 @@
+export { buildAdditionalDirectoriesPrompt } from "./agent-prompts";
 export * from "./analytics-events";
 export { type ArchivedTask, archivedTaskSchema } from "./archive-domain";
 export { withTimeout } from "./async";
diff --git a/packages/shared/src/task-creation-domain.ts b/packages/shared/src/task-creation-domain.ts
@@ -18,6 +18,13 @@ export interface TaskCreationInput {
   filePaths?: string[];
   repoPath?: string;
   repository?: string | null;
+  /**
+   * Cloud-only. Extra "org/repo" repositories cloned into the sandbox alongside
+   * the primary `repository`, so the agent can read/work across them. The PR
+   * still targets the primary repo. Local tasks use `additionalDirectories`
+   * (absolute paths) instead — the two are mutually exclusive by workspaceMode.
+   */
+  additionalRepositories?: string[];
   workspaceMode?: WorkspaceMode;
   branch?: string | null;
   // When the branch exists only on the remote, opt in to fetching and checking
diff --git a/packages/ui/src/features/folder-picker/AdditionalRepositoriesButton.tsx b/packages/ui/src/features/folder-picker/AdditionalRepositoriesButton.tsx
@@ -0,0 +1,108 @@
+import { GithubLogo, Plus } from "@phosphor-icons/react";
+import {
+  Button,
+  Combobox,
+  ComboboxContent,
+  ComboboxEmpty,
+  ComboboxInput,
+  ComboboxItem,
+  ComboboxList,
+  ComboboxTrigger,
+} from "@posthog/quill";
+import { type RefObject, useMemo, useRef } from "react";
+
+interface AdditionalRepositoriesButtonProps {
+  values: string[];
+  onChange: (values: string[]) => void;
+  /** Full "org/repo" list the user can pick from (useUserRepositoryIntegration). */
+  repositories: string[];
+  /** The primary repo, excluded from the pickable list. */
+  primaryRepository?: string | null;
+  disabled?: boolean;
+  anchor?: RefObject<HTMLElement | null>;
+}
+
+const VISIBLE_LIMIT = 50;
+
+/**
+ * Cloud analog of {@link AdditionalDirectoriesButton}: a compact multi-select
+ * for adding extra GitHub repos to a cloud task. The backend clones each into
+ * the sandbox alongside the primary repo so the agent can read/work across
+ * them. Values are lowercased "org/repo" strings; the primary repo is never
+ * selectable here.
+ */
+export function AdditionalRepositoriesButton({
+  values,
+  onChange,
+  repositories,
+  primaryRepository,
+  disabled = false,
+  anchor,
+}: AdditionalRepositoriesButtonProps) {
+  const triggerRef = useRef<HTMLButtonElement>(null);
+
+  const selectableRepos = useMemo(
+    () => repositories.filter((repo) => repo !== primaryRepository),
+    [repositories, primaryRepository],
+  );
+
+  // Never keep the primary repo in the selection (it may have been added as an
+  // extra and then promoted to primary).
+  const selectedValues = useMemo(
+    () => values.filter((repo) => repo !== primaryRepository),
+    [values, primaryRepository],
+  );
+  const count = selectedValues.length;
+
+  // Nothing to add when the user has no other repos than the primary.
+  if (selectableRepos.length === 0) return null;
+
+  return (
+    <Combobox
+      items={selectableRepos}
+      limit={VISIBLE_LIMIT}
+      multiple
+      value={selectedValues}
+      onValueChange={(next: string[]) =>
+        onChange(next.filter((repo) => repo !== primaryRepository))
+      }
+      disabled={disabled}
+    >
+      <ComboboxTrigger
+        render={
+          <Button
+            ref={triggerRef}
+            variant="outline"
+            size="sm"
+            disabled={disabled}
+            aria-label="Additional repositories"
+          >
+            <GithubLogo size={14} weight="regular" className="shrink-0" />
+            {count > 0 ? (
+              <span className="font-medium tabular-nums">+{count}</span>
+            ) : (
+              <Plus size={12} weight="bold" className="shrink-0" />
+            )}
+          </Button>
+        }
+      />
+      <ComboboxContent
+        anchor={anchor ?? triggerRef}
+        side="bottom"
+        sideOffset={6}
+        align="start"
+        className="min-w-[280px]"
+      >
+        <ComboboxInput placeholder="Add repositories..." />
+        <ComboboxEmpty>No repositories found.</ComboboxEmpty>
+        <ComboboxList>
+          {(repo: string) => (
+            <ComboboxItem key={repo} value={repo}>
+              {repo}
+            </ComboboxItem>
+          )}
+        </ComboboxList>
+      </ComboboxContent>
+    </Combobox>
+  );
+}
diff --git a/packages/ui/src/features/task-detail/components/AdditionalReposBadge.stories.tsx b/packages/ui/src/features/task-detail/components/AdditionalReposBadge.stories.tsx
@@ -0,0 +1,48 @@
+import type { Task } from "@posthog/shared/domain-types";
+import type { Meta, StoryObj } from "@storybook/react-vite";
+import { AdditionalReposBadge } from "./AdditionalReposBadge";
+
+const task = (overrides: Partial<Task> = {}): Task => ({
+  id: "task-1",
+  task_number: 42,
+  slug: "PH-42",
+  title: "Add multi-repo support",
+  description: "Wire the SDK change through both repos",
+  created_at: "2026-07-01T09:00:00Z",
+  updated_at: "2026-07-01T09:00:00Z",
+  origin_product: "user_created",
+  repository: "posthog/posthog",
+  ...overrides,
+});
+
+const meta = {
+  title: "Task Detail/AdditionalReposBadge",
+  component: AdditionalReposBadge,
+} satisfies Meta<typeof AdditionalReposBadge>;
+
+export default meta;
+type Story = StoryObj<typeof meta>;
+
+export const OneExtraRepo: Story = {
+  args: {
+    task: task({ additional_repositories: ["posthog/posthog-js"] }),
+  },
+};
+
+export const SeveralExtraRepos: Story = {
+  args: {
+    task: task({
+      additional_repositories: [
+        "posthog/posthog-js",
+        "posthog/posthog.com",
+        "posthog/posthog-python",
+      ],
+    }),
+  },
+};
+
+export const NoExtras_RendersNothing: Story = {
+  args: {
+    task: task(),
+  },
+};
diff --git a/packages/ui/src/features/task-detail/components/AdditionalReposBadge.tsx b/packages/ui/src/features/task-detail/components/AdditionalReposBadge.tsx
@@ -0,0 +1,37 @@
+import { GithubLogo } from "@phosphor-icons/react";
+import type { Task } from "@posthog/shared/domain-types";
+import { Tooltip } from "../../../primitives/Tooltip";
+
+/**
+ * Header badge for multi-repo cloud tasks: shows how many extra repositories
+ * were cloned into the sandbox alongside the primary, with the full list in a
+ * tooltip. Renders nothing for single-repo tasks, so it's invisible everywhere
+ * until a task actually has `additional_repositories`.
+ */
+export function AdditionalReposBadge({ task }: { task: Task }) {
+  const extras = task.additional_repositories ?? [];
+  if (extras.length === 0) return null;
+
+  return (
+    <Tooltip
+      side="bottom"
+      content={
+        <div className="flex flex-col gap-1">
+          <span className="text-(--gray-11)">Repositories in this task</span>
+          {task.repository && (
+            <span className="font-mono">{task.repository}</span>
+          )}
+          {extras.map((repo) => (
+            <span key={repo} className="font-mono">
+              {repo}
+            </span>
+          ))}
+        </div>
+      }
+    >
+      <span className="no-drag flex h-[24px] shrink-0 items-center gap-1 rounded-md border border-(--gray-6) px-2 font-mono text-(--gray-11) text-[11px]">
+        <GithubLogo size={12} className="shrink-0" />+{extras.length}
+      </span>
+    </Tooltip>
+  );
+}
diff --git a/packages/ui/src/features/task-detail/components/TaskInput.tsx b/packages/ui/src/features/task-detail/components/TaskInput.tsx
diff --git a/packages/ui/src/features/task-detail/hooks/useTaskCreation.ts b/packages/ui/src/features/task-detail/hooks/useTaskCreation.ts
diff --git a/packages/ui/src/shell/HeaderRow.tsx b/packages/ui/src/shell/HeaderRow.tsx
diff --git a/packages/workspace-server/src/services/agent/agent.ts b/packages/workspace-server/src/services/agent/agent.ts

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
	`1`	`+export { buildAdditionalDirectoriesPrompt } from "./agent-prompts";`
`1`	`2`	`export * from "./analytics-events";`
`2`	`3`	`export { type ArchivedTask, archivedTaskSchema } from "./archive-domain";`
`3`	`4`	`export { withTimeout } from "./async";`