tbenandClaude Opus 5.5
623112fd19
A workspace task with no bot was seen by every bot working there. Now a task can't be saved without a bot, a worker reaches only its own (the workspace admin and the coordinator still reach all), the workspace page picks who has a new task, and a bot leaving a workspace or deleted passes its tasks there to the workspace's admin bot, else the coordinator (also at boot for rows without a bot). Co-Authored-By: Claude Opus 5.5 <[email protected]>
81 lines
3.1 KiB
TypeScript
81 lines
3.1 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 (every task belongs to one bot: its own list and its tasks in
|
|
* each workspace; an admin's whole workspace), 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 }),
|
|
];
|
|
}
|