Files
cobble/src/harness/tools/bot/tasks/grab-bot-task-tools.ts
T
tbenandClaude Opus 5.5 864dcce30a Scope workspace tasks to the bot that has them
A workspace's task now carries the bot that has it there (bot_id with
workspace_id), or none while it's shared. A bot reaches its own list and,
in its workspaces, its own tasks plus the shared ones; a workspace admin
every task there. Bots create workspace tasks for themselves by default,
take a shared one by starting it, and hand-overs record the new bot.
Removing or deleting a bot makes its workspace tasks shared again. The
conversation's Tasks modal shows that bot's tasks in the workspace.

Co-Authored-By: Claude Opus 5.5 <[email protected]>
2026-10-27 19:19:01 +01:00

81 lines
3.0 KiB
TypeScript

import type { BUN_SQLITE_COBBLE_BOTS, BUN_SQLITE_COBBLE_WORKSPACES } from "@/db/types/db";
import grabCrewWorkers from "@/src/functions/backend/db/bots/grab-crew-workers";
import grabUserWorkspaces from "@/src/functions/backend/db/workspaces/grab-user-workspaces";
import grabStorageScopes from "@/src/utils/storage-access";
import type { HarnessTool } from "../../../types";
import createTaskTool from "./create-task-tool";
import listTasksTool from "./list-tasks-tool";
import updateTaskTool from "./update-task-tool";
type Params = {
user_id: number | string;
bot_id: number | string;
/**
* Whether the run is the coordinator, which reaches every task, can add
* to any list and reaches every folder.
*/
is_coordinator: boolean;
/**
* The workspaces the bot is admin of (`grabBotRunPowers`): in one of
* them, create_task can also give tasks there to the other members.
*/
admin_workspace_ids?: number[];
workspace_id?: number | string | null;
};
/**
* The task tools every run gets (`src/harness/tools/bot/tasks/`): the
* to-do list (a bot's own, and in each workspace its tasks there and the
* shared ones), so a chat reply, a task and a scheduled run all read it,
* move it along and file what they produced. The coordinator's create_task
* may target any worker or workspace; its update_task may link any file it
* can reach. A workspace admin's may give tasks there to the other members.
*/
export default async function grabBotTaskTools({
user_id,
bot_id,
is_coordinator,
admin_workspace_ids = [],
workspace_id,
}: Params): Promise<HarnessTool[]> {
let workers: BUN_SQLITE_COBBLE_BOTS[] = [];
let members: BUN_SQLITE_COBBLE_BOTS[] = [];
let workspaces: BUN_SQLITE_COBBLE_WORKSPACES[] = [];
let all_workspace_ids: number[] = [];
if (is_coordinator) {
workers = await grabCrewWorkers({ workspace_id });
workspaces = (await grabUserWorkspaces({ user_id })).filter((workspace) => !workspace.archived);
all_workspace_ids = workspaces.map((workspace) => Number(workspace.id));
} else if (workspace_id && admin_workspace_ids.includes(Number(workspace_id))) {
members = (await grabCrewWorkers({ workspace_id })).filter((member) => member.id != Number(bot_id));
}
/**
* `grabStorageScopes` wants a numeric bot id.
*/
const numeric_bot_id = Number(bot_id);
const scopes = grabStorageScopes({
bot_id: numeric_bot_id,
bot_kind: is_coordinator ? "coordinator" : "worker",
workspace_id: workspace_id ? Number(workspace_id) : null,
all_workspace_ids,
});
return [
listTasksTool({ user_id, bot_id, is_coordinator, admin_workspace_ids, workspace_id }),
createTaskTool({
user_id,
bot_id,
workspace_id,
is_coordinator,
workers,
workspaces,
admin_workspace_ids,
members,
}),
updateTaskTool({ user_id, bot_id, is_coordinator, admin_workspace_ids, scopes }),
];
}