From f9841ebdb290ca597c53230cfe81be0ca73d63ec Mon Sep 17 00:00:00 2001
From: Brace Sproul <braceasproul@gmail.com>
Date: Fri, 20 Jun 2025 11:00:49 -0700
Subject: [PATCH] fix: Pass stream mode to subagent calls in LG (#279)

* fix: Pass stream mode to subagent calls in LG

* cr
---
 apps/open-swe/src/graphs/manager/nodes/create-new-session.ts | 1 +
 apps/open-swe/src/graphs/manager/nodes/start-planner.ts      | 1 +
 apps/open-swe/src/graphs/planner/nodes/proposed-plan.ts      | 1 +
 3 files changed, 3 insertions(+)

diff --git a/apps/open-swe/src/graphs/manager/nodes/create-new-session.ts b/apps/open-swe/src/graphs/manager/nodes/create-new-session.ts
index 9907ab36..4e877eec 100644
--- a/apps/open-swe/src/graphs/manager/nodes/create-new-session.ts
+++ b/apps/open-swe/src/graphs/manager/nodes/create-new-session.ts
@@ -96,6 +96,7 @@ ${ISSUE_CONTENT_CLOSE_TAG}`,
     },
     ifNotExists: "create",
     streamResumable: true,
+    streamMode: ["values", "messages", "custom"],
   });
 
   return {
diff --git a/apps/open-swe/src/graphs/manager/nodes/start-planner.ts b/apps/open-swe/src/graphs/manager/nodes/start-planner.ts
index 6e2b0f7e..83980422 100644
--- a/apps/open-swe/src/graphs/manager/nodes/start-planner.ts
+++ b/apps/open-swe/src/graphs/manager/nodes/start-planner.ts
@@ -47,6 +47,7 @@ export async function startPlanner(
       ifNotExists: "create",
       multitaskStrategy: "enqueue",
       streamResumable: true,
+      streamMode: ["values", "messages", "custom"],
     });
 
     return {
diff --git a/apps/open-swe/src/graphs/planner/nodes/proposed-plan.ts b/apps/open-swe/src/graphs/planner/nodes/proposed-plan.ts
index 9ab9fdd6..043c891d 100644
--- a/apps/open-swe/src/graphs/planner/nodes/proposed-plan.ts
+++ b/apps/open-swe/src/graphs/planner/nodes/proposed-plan.ts
@@ -122,6 +122,7 @@ export async function interruptProposedPlan(
       },
       ifNotExists: "create",
       streamResumable: true,
+      streamMode: ["values", "messages", "custom"],
     },
   );