docker sandbox working locally

Author: joshsny

apps/twig/src/api/posthogClient.ts

@@ -194,15 +194,39 @@ export class PostHogAPIClient {
     return await response.json();
   }
 
-  async createTaskRun(taskId: string): Promise<TaskRun> {
+  async getTaskRunConnectionToken(
+    taskId: string,
+    runId: string,
+  ): Promise<{ token: string }> {
+    const teamId = await this.getTeamId();
+    const url = new URL(
+      `${this.api.baseUrl}/api/projects/${teamId}/tasks/${taskId}/runs/${runId}/connection_token/`,
+    );
+    const response = await this.api.fetcher.fetch({
+      method: "get",
+      url,
+      path: `/api/projects/${teamId}/tasks/${taskId}/runs/${runId}/connection_token/`,
+    });
+
+    if (!response.ok) {
+      throw new Error(`Failed to fetch connection token: ${response.statusText}`);
+    }
+
+    return await response.json();
+  }
+
+  async createTaskRun(
+    taskId: string,
+    environment: "local" | "cloud" = "local",
+  ): Promise<TaskRun> {
     const teamId = await this.getTeamId();
     const data = await this.api.post(
       `/api/projects/{project_id}/tasks/{task_id}/runs/`,
       {
         path: { project_id: teamId.toString(), task_id: taskId },
         //@ts-expect-error the generated client does not infer the request type unless explicitly specified on the viewset
         body: {
-          environment: "local" as const,
+          environment,
         },
       },
     );

apps/twig/src/main/services/agent/schemas.ts

@@ -19,13 +19,17 @@ export const sessionConfigSchema = z.object({
   taskId: z.string(),
   taskRunId: z.string(),
   repoPath: z.string(),
-  credentials: credentialsSchema,
+  credentials: credentialsSchema.optional(),
   logUrl: z.string().optional(),
   sdkSessionId: z.string().optional(),
   model: z.string().optional(),
   executionMode: executionModeSchema.optional(),
   /** Additional directories Claude can access beyond cwd (for worktree support) */
   additionalDirectories: z.array(z.string()).optional(),
+  runMode: z.enum(["local", "cloud"]).optional(),
+  /** Cloud transport config - required when runMode is 'cloud' */
+  sandboxUrl: z.string().optional(),
+  connectionToken: z.string().optional(),
 });
 
 export type SessionConfig = z.infer<typeof sessionConfigSchema>;
@@ -48,6 +52,9 @@ export const startSessionInput = z.object({
   runMode: z.enum(["local", "cloud"]).optional(),
   /** Additional directories Claude can access beyond cwd (for worktree support) */
   additionalDirectories: z.array(z.string()).optional(),
+  /** Cloud transport config - required when runMode is 'cloud' */
+  sandboxUrl: z.string().optional(),
+  connectionToken: z.string().optional(),
 });
 
 export type StartSessionInput = z.infer<typeof startSessionInput>;