acontext 0.0.13__py3-none-any.whl → 0.0.14__py3-none-any.whl

acontext/agent/disk.py

@@ -261,15 +261,71 @@ class ListTool(BaseTool):
         if not artifacts_list and not result.directories:
             return f"No files or directories found in '{normalized_path}'"
 
-        output_parts = []
-        if artifacts_list:
-            output_parts.append(f"Files: {', '.join(artifacts_list)}")
-        if result.directories:
-            output_parts.append(f"Directories: {', '.join(result.directories)}")
-
-        ls_sect = "\n".join(output_parts)
+        file_sect = "\n".join(artifacts_list) or "[NO FILE]"
+        dir_sect = (
+            "\n".join([d.rstrip("/") + "/" for d in result.directories]) or "[NO DIR]"
+        )
         return f"""[Listing in {normalized_path}]
-{ls_sect}"""
+Directories:
+{dir_sect}
+Files:
+{file_sect}"""
+
+
+class DownloadFileTool(BaseTool):
+    """Tool for getting a public download URL for a file on the Acontext disk."""
+
+    @property
+    def name(self) -> str:
+        return "download_file"
+
+    @property
+    def description(self) -> str:
+        return "Get a public URL to download a file. Returns a presigned URL that can be shared or used to access the file."
+
+    @property
+    def arguments(self) -> dict:
+        return {
+            "file_path": {
+                "type": "string",
+                "description": "Optional directory path where the file is located, e.g. '/notes/'. Defaults to root '/' if not specified.",
+            },
+            "filename": {
+                "type": "string",
+                "description": "Filename to get the download URL for.",
+            },
+            "expire": {
+                "type": "integer",
+                "description": "URL expiration time in seconds. Defaults to 3600 (1 hour).",
+            },
+        }
+
+    @property
+    def required_arguments(self) -> list[str]:
+        return ["filename"]
+
+    def execute(self, ctx: DiskContext, llm_arguments: dict) -> str:
+        """Get a public download URL for a file."""
+        filename = llm_arguments.get("filename")
+        file_path = llm_arguments.get("file_path")
+        expire = llm_arguments.get("expire", 3600)
+
+        if not filename:
+            raise ValueError("filename is required")
+
+        normalized_path = _normalize_path(file_path)
+        result = ctx.client.disks.artifacts.get(
+            ctx.disk_id,
+            file_path=normalized_path,
+            filename=filename,
+            with_public_url=True,
+            expire=expire,
+        )
+
+        if not result.public_url:
+            raise RuntimeError("Failed to get public URL: server did not return a URL.")
+
+        return f"Public download URL for '{normalized_path}{filename}' (expires in {expire}s):\n{result.public_url}"
 
 
 class DiskToolPool(BaseToolPool):
@@ -284,6 +340,7 @@ DISK_TOOLS.add_tool(WriteFileTool())
 DISK_TOOLS.add_tool(ReadFileTool())
 DISK_TOOLS.add_tool(ReplaceStringTool())
 DISK_TOOLS.add_tool(ListTool())
+DISK_TOOLS.add_tool(DownloadFileTool())
 
 
 if __name__ == "__main__":
@@ -323,3 +380,9 @@ if __name__ == "__main__":
         ctx, "read_file", {"filename": "test.txt", "file_path": "/try/"}
     )
     print(r)
+    r = DISK_TOOLS.execute_tool(
+        ctx,
+        "download_file",
+        {"filename": "test.txt", "file_path": "/try/", "expire": 300},
+    )
+    print(r)

{acontext-0.0.13.dist-info → acontext-0.0.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: acontext
-Version: 0.0.13
+Version: 0.0.14
 Summary: Python SDK for the Acontext API
 Keywords: acontext,sdk,client,api
 Requires-Dist: httpx>=0.28.1

{acontext-0.0.13.dist-info → acontext-0.0.14.dist-info}/RECORD

@@ -3,7 +3,7 @@ acontext/_constants.py,sha256=Ikuy_Wz3CPmXjKPLXb4Y580-fe54o1hZ2ZB1Bpw3sFE,363
 acontext/_utils.py,sha256=GKQH45arKh0sDu64u-5jwrII_ctnU_oChYlgR5lRkfE,1250
 acontext/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 acontext/agent/base.py,sha256=kN2KzV6fOlvmMSi_0yu2QqwVLtHh09TznAy_SDtC8iM,2998
-acontext/agent/disk.py,sha256=tYSvdWKjJ6KIvAk3UzJaQABB4rOL7GAMvEVmI715-hw,10566
+acontext/agent/disk.py,sha256=d9sOggIs4da7niB9TaE0ab82_7l0O5n08eOcQxcaYRs,12623
 acontext/async_client.py,sha256=mfyR-WVy_LEOXjrkTcXLNaPDC5euaZ9zFI_5e1gapGs,8153
 acontext/client.py,sha256=jFr5-Q8obdcZBamUNtY3fCklpOJiNiIKLzEApTmbnTg,7920
 acontext/client_types.py,sha256=uVBWzLbZyXrqkljG49ojdQL_xX6N3n_HGt4Bs_TEE48,987
@@ -28,6 +28,6 @@ acontext/types/session.py,sha256=H_n4vZJ02ZO5flqGZ8R5AYVSScxScf2r7UgsN5LRIa0,996
 acontext/types/space.py,sha256=uxbPrOHYpsntPHqhMCLQ2KovM7BngHC5Q2j7qexVrN8,2537
 acontext/types/tool.py,sha256=-mVn-vgk2SENK0Ubt-ZgWFZxKa-ddABqcAgXQ69YY-E,805
 acontext/uploads.py,sha256=6twnqQOY_eerNuEjeSKsE_3S0IfJUiczXtAy4aXqDl8,1379
-acontext-0.0.13.dist-info/WHEEL,sha256=RRVLqVugUmFOqBedBFAmA4bsgFcROUBiSUKlERi0Hcg,79
-acontext-0.0.13.dist-info/METADATA,sha256=xB8jBuzFtOiOMcXpOTvFgb-pvVPx7nwxtdd8cXhaMaE,859
-acontext-0.0.13.dist-info/RECORD,,
+acontext-0.0.14.dist-info/WHEEL,sha256=KSLUh82mDPEPk0Bx0ScXlWL64bc8KmzIPNcpQZFV-6E,79
+acontext-0.0.14.dist-info/METADATA,sha256=sm1g1hhGbW15feWubCtcNkjgK5Ja1S_-xr1lIbXvmjU,859
+acontext-0.0.14.dist-info/RECORD,,

{acontext-0.0.13.dist-info → acontext-0.0.14.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: uv 0.9.21
+Generator: uv 0.9.22
 Root-Is-Purelib: true
 Tag: py3-none-any