@townco/agent 0.1.98 → 0.1.101

package/dist/runner/langchain/tools/e2b.d.ts

@@ -45,4 +45,10 @@ export declare function makeTownE2BTools(): readonly [import("langchain").Dynami
 }, {
     sandboxPath: string;
     fileName?: string | undefined;
+}, string>, import("langchain").DynamicStructuredTool<z.ZodObject<{
+    document_id: z.ZodString;
+}, z.core.$strip>, {
+    document_id: string;
+}, {
+    document_id: string;
 }, string>];

package/dist/runner/langchain/tools/e2b.js

@@ -284,12 +284,67 @@ function makeE2BToolsInternal(getSandbox) {
     downloadFromSandbox.prettyName = "Download from Sandbox";
     // biome-ignore lint/suspicious/noExplicitAny: Need to add custom properties to LangChain tool
     downloadFromSandbox.icon = "Download";
+    // Tool 6: Upload Library Document to Sandbox
+    const uploadLibraryDocument = tool(async ({ document_id }) => {
+        const sandbox = await getSandbox();
+        try {
+            const libraryApiUrl = process.env.LIBRARY_API_URL;
+            const libraryApiKey = process.env.LIBRARY_ROOT_API_KEY;
+            if (!libraryApiUrl || !libraryApiKey) {
+                throw new Error("LIBRARY_API_URL and LIBRARY_ROOT_API_KEY environment variables are required");
+            }
+            const response = await fetch(`${libraryApiUrl}/sandbox/upload_document_to_sandbox`, {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json",
+                    Authorization: `Bearer ${libraryApiKey}`,
+                },
+                body: JSON.stringify({
+                    document_id,
+                    sandbox_id: sandbox.sandboxId,
+                }),
+            });
+            if (!response.ok) {
+                const text = await response.text();
+                throw new Error(`Library API error: ${response.status} - ${text}`);
+            }
+            const result = await response.json();
+            return result.path;
+        }
+        catch (error) {
+            logger.error("Error uploading library document to sandbox", {
+                error,
+                document_id,
+            });
+            return `Error uploading document: ${error instanceof Error ? error.message : String(error)}`;
+        }
+    }, {
+        name: "E2B_UploadLibraryDocument",
+        description: "Upload a document from the library to the cloud sandbox. " +
+            "Use this to make library documents available for processing in the sandbox environment.",
+        schema: z.object({
+            document_id: z
+                .string()
+                .describe("The ID of the document to upload from the library"),
+        }),
+    });
+    // biome-ignore lint/suspicious/noExplicitAny: Need to add custom properties to LangChain tool
+    uploadLibraryDocument.prettyName = "Upload Library Document";
+    // biome-ignore lint/suspicious/noExplicitAny: Need to add custom properties to LangChain tool
+    uploadLibraryDocument.icon = "Upload";
+    // biome-ignore lint/suspicious/noExplicitAny: Need to add custom properties to LangChain tool
+    uploadLibraryDocument.verbiage = {
+        active: "Uploading document {document_id}",
+        past: "Uploaded document {document_id}",
+        paramKey: "document_id",
+    };
     return [
         runCode,
         runBash,
         readSandboxFile,
         writeSandboxFile,
         downloadFromSandbox,
+        uploadLibraryDocument,
     ];
 }
 /**