Add bots files handling logic

2026-10-21 17:56:09 +01:00
parent b0af4db1bf
commit d17a4a7bdd
97 changed files with 5843 additions and 41 deletions
@@ -0,0 +1,50 @@
+import type { BUN_SQLITE_COBBLE_BOTS } from "@/db/types/db";
+import grabUserWorkspaces from "@/src/functions/backend/db/workspaces/grab-user-workspaces";
+import type { HarnessTool } from "../../types";
+import generateImageTool from "./generate-image-tool";
+import grabStorageTools from "./grab-storage-tools";
+import makePdfTool from "./make-pdf-tool";
+import storageToolContext from "./storage-tool-context";
+
+type Params = {
+    bot: Pick<BUN_SQLITE_COBBLE_BOTS, "id" | "kind">;
+    user_id: number | string;
+    workspace_id?: number | string | null;
+    session_id?: number | string;
+    /**
+     * Called for a file the bot wrote with "attach" set, so the reply
+     * carries it.
+     */
+    onAttach?: (params: { media_id: number; name: string }) => void;
+};
+
+/**
+ * The tools a bot uses to keep and search files (`src/harness/tools/
+ * storage/`): listing, reading, writing, moving, deleting and searching
+ * its own directory and, in a workspace or as the coordinator, the shared
+ * ones, plus making images and PDFs into them. Every run gets them, so a
+ * chat reply, a task and a scheduled run all work with the same files.
+ */
+export default async function grabBotFileTools({
+    bot,
+    user_id,
+    workspace_id,
+    session_id,
+    onAttach,
+}: Params): Promise<HarnessTool[]> {
+    const all_workspace_ids =
+        bot.kind == "coordinator"
+            ? (await grabUserWorkspaces({ user_id })).filter((w) => !w.archived).map((w) => Number(w.id))
+            : [];
+
+    const context = storageToolContext({
+        bot,
+        user_id,
+        workspace_id,
+        session_id,
+        all_workspace_ids,
+        onAttach,
+    });
+
+    return [...grabStorageTools(context), generateImageTool(context), makePdfTool(context)];
+}