Skip to content

Commit 3831af3

Browse files
committed
fix: task_await bulk scope check fallback
Change-Id: Ie853bc24cc969052136abad09f41025d2101abbc Signed-off-by: Thomas Kosiewski <tk@coder.com>
1 parent 58689b6 commit 3831af3

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

src/node/services/tools/task_await.ts

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,18 @@ export const createTaskAwaitTool: ToolFactory = (config: ToolConfiguration) => {
3434
requestedIds ?? taskService.listActiveDescendantAgentTaskIds(workspaceId);
3535

3636
const uniqueTaskIds = dedupeStrings(candidateTaskIds);
37+
const bulkFilter = (
38+
taskService as unknown as {
39+
filterDescendantAgentTaskIds?: (
40+
ancestorWorkspaceId: string,
41+
taskIds: string[]
42+
) => string[];
43+
}
44+
).filterDescendantAgentTaskIds;
3745
const descendantTaskIdSet = new Set(
38-
taskService.filterDescendantAgentTaskIds(workspaceId, uniqueTaskIds)
46+
typeof bulkFilter === "function"
47+
? bulkFilter(workspaceId, uniqueTaskIds)
48+
: uniqueTaskIds.filter((taskId) => taskService.isDescendantAgentTask(workspaceId, taskId))
3949
);
4050

4151
const results = await Promise.all(

0 commit comments

Comments
 (0)