tben
139b3e443d
- Resolve the crew and workspaces fresh inside each tool run, so a bot or workspace created earlier in the same reply is usable: create_bot, assign_computer, add_bot_to_workspace, delegate_task and set_bot_workspace_instructions. - Add coordinator-only update_bot, find_bot, find_workspace, delete_bot, delete_workspace and remove_bot_from_workspace tools. - Push server:workspace-removed when a workspace is deleted, and workspace_ids on server:bot-updated so a workspace's crew drops a bot that was taken out of it.
82 lines
3.4 KiB
TypeScript
82 lines
3.4 KiB
TypeScript
import { z } from "zod";
|
|
import type { BUN_SQLITE_COBBLE_BOTS } from "@/db/types/db";
|
|
import { AppData } from "@/src/data/app-data";
|
|
import type { BotHue } from "@/src/dict/bot-dict";
|
|
import grabCrewWorkers from "@/src/functions/backend/db/bots/grab-crew-workers";
|
|
import updateBot from "@/src/functions/backend/db/bots/update-bot";
|
|
import type { HarnessTool } from "../../types";
|
|
|
|
type Params = {
|
|
user_id: number | string;
|
|
/**
|
|
* The user's worker bots.
|
|
*/
|
|
workers: BUN_SQLITE_COBBLE_BOTS[];
|
|
};
|
|
|
|
type Args = {
|
|
bot_id: number;
|
|
name?: string;
|
|
role?: string;
|
|
instructions?: string;
|
|
};
|
|
|
|
/**
|
|
* Lets the coordinator change an existing crew member: its name, its
|
|
* one-line job, and its core instructions (the start of its prompt in its
|
|
* own chat and in every workspace). Its hue and picture stay as they are.
|
|
* Workspace-only instructions have their own tool
|
|
* (`set_bot_workspace_instructions`).
|
|
*/
|
|
export default function updateBotTool({ user_id, workers }: Params): HarnessTool<Args> {
|
|
const crew = workers.map((bot) => `${bot.id}: ${bot.name}`).join("; ");
|
|
|
|
return {
|
|
name: "update_bot",
|
|
description: `Change an existing crew member: name, the one-line role it's shown with, and/or its core instructions (markdown: what they do and how they work, the start of their prompt in their own chat and in every workspace). Only the fields you pass change. Instructions replace what they had, so keep what still applies, their own edits included. This is not their instructions for one workspace: use set_bot_workspace_instructions for that. Crew: ${crew || "none"}.`,
|
|
schema: z
|
|
.object({
|
|
bot_id: z.number().int(),
|
|
name: z.string().trim().min(1).max(40).optional(),
|
|
role: z.string().trim().max(120).optional(),
|
|
instructions: z.string().trim().min(1).max(AppData.PromptMaxLength).optional(),
|
|
})
|
|
.refine(
|
|
(args) =>
|
|
args.name !== undefined ||
|
|
args.role !== undefined ||
|
|
args.instructions !== undefined,
|
|
"Give the name, role or instructions to change.",
|
|
),
|
|
run: async ({ args }) => {
|
|
const fresh = await grabCrewWorkers();
|
|
const worker = fresh.find((bot) => bot.id == args.bot_id);
|
|
|
|
if (!worker?.id) {
|
|
throw new Error(
|
|
`No crew member has id ${args.bot_id}. Crew: ${fresh
|
|
.map((bot) => `${bot.id}: ${bot.name}`)
|
|
.join("; ") || "none"}.`,
|
|
);
|
|
}
|
|
|
|
const changed = [
|
|
args.name !== undefined ? "name" : "",
|
|
args.role !== undefined ? "role" : "",
|
|
args.instructions !== undefined ? "core instructions" : "",
|
|
].filter(Boolean);
|
|
|
|
const saved = await updateBot({
|
|
user_id,
|
|
bot_id: worker.id,
|
|
name: args.name ?? worker.name ?? "Bot",
|
|
role: args.role ?? worker.role ?? "",
|
|
hue: (worker.hue || "ember") as BotHue,
|
|
system_prompt: args.instructions ?? worker.system_prompt ?? "",
|
|
});
|
|
|
|
return `${saved.name}'s ${changed.join(", ")} saved.${args.instructions !== undefined ? " The instructions apply from its next run, in its own chat and in every workspace." : ""}`;
|
|
},
|
|
};
|
|
}
|