acontext 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

acontext/__init__.py

@@ -8,16 +8,12 @@ from .async_client import AcontextAsyncClient
 from .client import AcontextClient, FileUpload, MessagePart
 from .messages import AcontextMessage
 from .resources import (
-    AsyncBlocksAPI,
     AsyncDiskArtifactsAPI,
     AsyncDisksAPI,
     AsyncSessionsAPI,
-    AsyncSpacesAPI,
-    BlocksAPI,
     DiskArtifactsAPI,
     DisksAPI,
     SessionsAPI,
-    SpacesAPI,
 )
 from .types import Task, TaskData
 
@@ -29,14 +25,10 @@ __all__ = [
     "AcontextMessage",
     "DisksAPI",
     "DiskArtifactsAPI",
-    "BlocksAPI",
     "SessionsAPI",
-    "SpacesAPI",
     "AsyncDisksAPI",
     "AsyncDiskArtifactsAPI",
-    "AsyncBlocksAPI",
     "AsyncSessionsAPI",
-    "AsyncSpacesAPI",
     "Task",
     "TaskData",
     "__version__",

acontext/agent/__init__.py

@@ -1,9 +1,11 @@
 """Agent tools for LLM function calling."""
 
 from .disk import DISK_TOOLS
+from .sandbox import SANDBOX_TOOLS
 from .skill import SKILL_TOOLS
 
 __all__ = [
     "DISK_TOOLS",
+    "SANDBOX_TOOLS",
     "SKILL_TOOLS",
 ]

acontext/agent/base.py

@@ -1,5 +1,6 @@
 class BaseContext:
-    pass
+    def get_context_prompt(self, *args, **kwargs) -> str:
+        return ""
 
 
 class BaseConverter:

acontext/agent/disk.py

@@ -11,11 +11,18 @@ class DiskContext(BaseContext):
     client: AcontextClient
     disk_id: str
 
+    def get_context_prompt(self) -> str:
+        return """<disk>
+Consider Disk as the google drive for you and user to store and share files.
+You can use tool ends with `*_disk` to read, write, edit, and share files with users.
+Disk is only a sharable file storage, you can't use it to execute code or run commands.
+</disk>
+"""
+
 
 @dataclass
-class AsyncDiskContext(BaseContext):
+class AsyncDiskContext(DiskContext):
     client: AcontextAsyncClient
-    disk_id: str
 
 
 def _normalize_path(path: str | None) -> str:
@@ -33,7 +40,7 @@ class WriteFileTool(BaseTool):
 
     @property
     def name(self) -> str:
-        return "write_file"
+        return "write_file_disk"
 
     @property
     def description(self) -> str:
@@ -43,7 +50,7 @@ class WriteFileTool(BaseTool):
     def arguments(self) -> dict:
         return {
             "file_path": {
-                "type": "string",
+                "type": ["string", "null"],
                 "description": "Optional folder path to organize files, e.g. '/notes/' or '/documents/'. Defaults to root '/' if not specified.",
             },
             "filename": {
@@ -106,7 +113,7 @@ class ReadFileTool(BaseTool):
 
     @property
     def name(self) -> str:
-        return "read_file"
+        return "read_file_disk"
 
     @property
     def description(self) -> str:
@@ -116,7 +123,7 @@ class ReadFileTool(BaseTool):
     def arguments(self) -> dict:
         return {
             "file_path": {
-                "type": "string",
+                "type": ["string", "null"],
                 "description": "Optional directory path where the file is located, e.g. '/notes/'. Defaults to root '/' if not specified.",
             },
             "filename": {
@@ -124,11 +131,11 @@ class ReadFileTool(BaseTool):
                 "description": "Filename to read.",
             },
             "line_offset": {
-                "type": "integer",
+                "type": ["integer", "null"],
                 "description": "The line number to start reading from. Default to 0",
             },
             "line_limit": {
-                "type": "integer",
+                "type": ["integer", "null"],
                 "description": "The maximum number of lines to return. Default to 100",
             },
         }
@@ -199,7 +206,7 @@ class ReplaceStringTool(BaseTool):
 
     @property
     def name(self) -> str:
-        return "replace_string"
+        return "replace_string_disk"
 
     @property
     def description(self) -> str:
@@ -209,7 +216,7 @@ class ReplaceStringTool(BaseTool):
     def arguments(self) -> dict:
         return {
             "file_path": {
-                "type": "string",
+                "type": ["string", "null"],
                 "description": "Optional directory path where the file is located, e.g. '/notes/'. Defaults to root '/' if not specified.",
             },
             "filename": {
@@ -328,7 +335,7 @@ class ListTool(BaseTool):
 
     @property
     def name(self) -> str:
-        return "list_artifacts"
+        return "list_disk"
 
     @property
     def description(self) -> str:
@@ -403,7 +410,7 @@ class DownloadFileTool(BaseTool):
 
     @property
     def name(self) -> str:
-        return "download_file"
+        return "download_file_disk"
 
     @property
     def description(self) -> str:
@@ -413,7 +420,7 @@ class DownloadFileTool(BaseTool):
     def arguments(self) -> dict:
         return {
             "file_path": {
-                "type": "string",
+                "type": ["string", "null"],
                 "description": "Optional directory path where the file is located, e.g. '/notes/'. Defaults to root '/' if not specified.",
             },
             "filename": {
@@ -421,7 +428,7 @@ class DownloadFileTool(BaseTool):
                 "description": "Filename to get the download URL for.",
             },
             "expire": {
-                "type": "integer",
+                "type": ["integer", "null"],
                 "description": "URL expiration time in seconds. Defaults to 3600 (1 hour).",
             },
         }
@@ -482,7 +489,7 @@ class GrepArtifactsTool(BaseTool):
 
     @property
     def name(self) -> str:
-        return "grep_artifacts"
+        return "grep_disk"
 
     @property
     def description(self) -> str:
@@ -496,7 +503,7 @@ class GrepArtifactsTool(BaseTool):
                 "description": "Regex pattern to search for (e.g., 'TODO.*', 'function.*calculate', 'import.*pandas')",
             },
             "limit": {
-                "type": "integer",
+                "type": ["integer", "null"],
                 "description": "Maximum number of results to return (default 100)",
             },
         }
@@ -561,7 +568,7 @@ class GlobArtifactsTool(BaseTool):
 
     @property
     def name(self) -> str:
-        return "glob_artifacts"
+        return "glob_disk"
 
     @property
     def description(self) -> str:
@@ -575,7 +582,7 @@ class GlobArtifactsTool(BaseTool):
                 "description": "Glob pattern (e.g., '**/*.py' for all Python files, '*.txt' for text files in root, '/docs/**/*.md' for markdown in docs)",
             },
             "limit": {
-                "type": "integer",
+                "type": ["integer", "null"],
                 "description": "Maximum number of results to return (default 100)",
             },
         }

acontext/agent/prompts.py

@@ -0,0 +1,96 @@
+SKILL_REMINDER = """MANDATORY SKILL READING AND EXECUTION PROTOCOL:
+BEFORE writing ANY code or using ANY execution tools, You MUST complete ALL of
+these steps:
+
+STEP 1 - IDENTIFY ALL RELEVANT SKILLS:
+- Scan the user message for ALL trigger words from ALL skills
+- Identify EVERY skill that matches ANY trigger word
+- If multiple skills match, ALL must be processed
+- If no skills match, you can skip the following steps
+
+STEP 2 - READ ALL SKILL FILES:
+- Use the text_editor_sandbox tool to view EACH identified skill's SKILL.md file
+- READ COMPLETELY - do not skim or skip sections
+- This step is MANDATORY even if multiple skills are involved
+- DO NOT proceed until ALL relevant skill files have been read
+
+STEP 3 - EXECUTE ALL SKILL INSTRUCTIONS:
+- Follow the EXACT instructions from EACH skill file read
+- If a skill file says to execute file X, then EXECUTE file X
+- If a skill file provides code patterns, USE those patterns
+- Apply instructions from ALL skills, not just the first one
+- NEVER write generic code when skill-specific code exists
+
+CRITICAL RULES:
+- Reading the skill file is NOT sufficient - you must FOLLOW its instructions
+- Multiple skills = multiple skill files to read AND follow
+- Each skill's instructions must be executed, not just acknowledged
+- NEVER skip a skill because you already read another skill
+- The skills contain specialized, tested code that MUST be used
+
+DO NOT SKIP ANY SKILL FILES OR THEIR INSTRUCTIONS. This protocol applies to EVERY
+skill that matches the user's request, without exception."""
+
+SANDBOX_TEXT_EDITOR_REMINDER = """The text_editor_sandbox tool enables viewing, creating, and modifying text files within
+the secure sandboxed container environment.
+
+How it works:
+- All file operations occur within the sandboxed container filesystem
+
+Command guidelines:
+- Always use view before editing to understand file structure
+- For str_replace commands, ensure search strings are unique and exact
+- Include sufficient context in str_replace for accurate placement
+- Use proper escaping for special characters in search/replace strings"""
+
+SANDBOX_BASH_REMINDER = """When to use the bash_execution_sandbox tool directly:
+- File system operations requiring shell commands (moving, copying, renaming, organizing files)
+- Text processing and manipulation using standard Unix tools (grep, sed, awk, cut, sort, etc.) that
+should not be done by the text editor tool
+- Batch processing of multiple files using shell loops and wildcards
+- System inspection tasks (checking file sizes, permissions, directory structures)
+- Combining multiple command-line tools in pipelines for complex data processing
+- Archive operations (tar, unzip) and file compression/decompression
+- Converting between file formats using command-line utilities
+
+When you should write Python file and use bash tool to run it:
+- Complex data analysis or numerical computation (use file operations to write a Python script instead, and
+then the bash to run the script)
+- Tasks requiring advanced programming logic or data structures
+
+When NOT to use the bash_execution_sandbox tool:
+- Simple questions that can be answered without executing commands
+- Tasks that only require explaining shell concepts without actual execution
+
+How it works:
+- Scripts are saved to a temporary sandbox and executed with bash
+- Tool results will include stdout, stderr, and return code
+- User-uploaded files are accessible in the directory specified by the INPUT_DIR environment variable. If
+you know the file path and don't need to open the full INPUT_DIR, then just open the file directly
+
+File Operations (CRITICAL - READ CAREFULLY):
+- use text_editor_sandbox tool to view, create, and edit files.
+
+Export Your Result:
+- All the files you created kept in the sandbox, which user can't see or access.
+- If you want to export them to user, use `export_file_sandbox` tool.
+- If too many files to export(>= 6 files), zip those files and export the zip file.
+- Result files' names should be unique and descriptive, (wrong: result.md, output.md... right: 2026_us_market_trending.png)
+
+Script guidelines:
+- Write POSIX-compliant bash scripts
+- Use proper error handling and exit codes
+- Quote variables appropriately to handle spaces in filenames
+- Keep scripts clean and well-organized
+- For file operations, use text_editor_sandbox tool instead of bash commands.
+
+Never write blocking script:
+- python codes like `plt.show()` or `input()`... will block the execution of the script, don't use them. write non-blocking code instead.
+
+Container environment:
+- Filesystem persists across multiple executions within the same container
+- Standard Unix utilities available (grep, sed, awk, etc.)
+- Archive tools: tar, unzip, zip
+- Additional tools: ripgrep, fd, sqlite3, jq, imagemagick
+- Do not try to install new packages and libraries with pip as there is no internet access
+"""