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

acontext/resources/async_sandboxes.py

@@ -0,0 +1,85 @@
+"""
+Sandboxes endpoints (async).
+"""
+
+from .._utils import build_params
+from ..client_types import AsyncRequesterProtocol
+from ..types.sandbox import (
+    GetSandboxLogsOutput,
+    SandboxCommandOutput,
+    SandboxRuntimeInfo,
+)
+from ..types.tool import FlagResponse
+
+
+class AsyncSandboxesAPI:
+    def __init__(self, requester: AsyncRequesterProtocol) -> None:
+        self._requester = requester
+
+    async def create(self) -> SandboxRuntimeInfo:
+        """Create and start a new sandbox.
+
+        Returns:
+            SandboxRuntimeInfo containing the sandbox ID, status, and timestamps.
+        """
+        data = await self._requester.request("POST", "/sandbox")
+        return SandboxRuntimeInfo.model_validate(data)
+
+    async def exec_command(
+        self,
+        *,
+        sandbox_id: str,
+        command: str,
+        timeout: float | None = None,
+    ) -> SandboxCommandOutput:
+        """Execute a shell command in the sandbox.
+
+        Args:
+            sandbox_id: The UUID of the sandbox.
+            command: The shell command to execute.
+            timeout: Optional timeout in seconds for this command.
+                    If not provided, uses the client's default timeout.
+
+        Returns:
+            SandboxCommandOutput containing stdout, stderr, and exit code.
+        """
+        data = await self._requester.request(
+            "POST",
+            f"/sandbox/{sandbox_id}/exec",
+            json_data={"command": command},
+            timeout=timeout,
+        )
+        return SandboxCommandOutput.model_validate(data)
+
+    async def kill(self, sandbox_id: str) -> FlagResponse:
+        """Kill a running sandbox.
+
+        Args:
+            sandbox_id: The UUID of the sandbox to kill.
+
+        Returns:
+            FlagResponse with status and error message.
+        """
+        data = await self._requester.request("DELETE", f"/sandbox/{sandbox_id}")
+        return FlagResponse.model_validate(data)
+
+    async def get_logs(
+        self,
+        *,
+        limit: int | None = None,
+        cursor: str | None = None,
+        time_desc: bool | None = None,
+    ) -> GetSandboxLogsOutput:
+        """Get sandbox logs for the project with cursor-based pagination.
+
+        Args:
+            limit: Maximum number of logs to return (default 20, max 200).
+            cursor: Cursor for pagination. Use the cursor from the previous response to get the next page.
+            time_desc: Order by created_at descending if True, ascending if False (default False).
+
+        Returns:
+            GetSandboxLogsOutput containing the list of sandbox logs and pagination information.
+        """
+        params = build_params(limit=limit, cursor=cursor, time_desc=time_desc)
+        data = await self._requester.request("GET", "/sandbox/logs", params=params or None)
+        return GetSandboxLogsOutput.model_validate(data)

acontext/resources/async_sessions.py

@@ -12,7 +12,6 @@ from ..types.session import (
     EditStrategy,
     GetMessagesOutput,
     GetTasksOutput,
-    LearningStatus,
     ListSessionsOutput,
     Message,
     MessageObservingStatus,
@@ -38,8 +37,6 @@ class AsyncSessionsAPI:
         self,
         *,
         user: str | None = None,
-        space_id: str | None = None,
-        not_connected: bool | None = None,
         limit: int | None = None,
         cursor: str | None = None,
         time_desc: bool | None = None,
@@ -48,8 +45,6 @@ class AsyncSessionsAPI:
 
         Args:
             user: Filter by user identifier. Defaults to None.
-            space_id: Filter sessions by space ID. Defaults to None.
-            not_connected: Filter sessions that are not connected to a space. Defaults to None.
             limit: Maximum number of sessions to return. Defaults to None.
             cursor: Cursor for pagination. Defaults to None.
             time_desc: Order by created_at descending if True, ascending if False. Defaults to None.
@@ -60,11 +55,8 @@ class AsyncSessionsAPI:
         params: dict[str, Any] = {}
         if user:
             params["user"] = user
-        if space_id:
-            params["space_id"] = space_id
         params.update(
             build_params(
-                not_connected=not_connected,
                 limit=limit,
                 cursor=cursor,
                 time_desc=time_desc,
@@ -77,7 +69,6 @@ class AsyncSessionsAPI:
         self,
         *,
         user: str | None = None,
-        space_id: str | None = None,
         disable_task_tracking: bool | None = None,
         configs: Mapping[str, Any] | None = None,
     ) -> Session:
@@ -85,7 +76,6 @@ class AsyncSessionsAPI:
 
         Args:
             user: Optional user identifier string. Defaults to None.
-            space_id: Optional space ID to associate with the session. Defaults to None.
             disable_task_tracking: Whether to disable task tracking for this session. Defaults to None (server default: False).
             configs: Optional session configuration dictionary. Defaults to None.
 
@@ -95,8 +85,6 @@ class AsyncSessionsAPI:
         payload: dict[str, Any] = {}
         if user:
             payload["user"] = user
-        if space_id:
-            payload["space_id"] = space_id
         if disable_task_tracking is not None:
             payload["disable_task_tracking"] = disable_task_tracking
         if configs is not None:
@@ -141,18 +129,6 @@ class AsyncSessionsAPI:
         data = await self._requester.request("GET", f"/session/{session_id}/configs")
         return Session.model_validate(data)
 
-    async def connect_to_space(self, session_id: str, *, space_id: str) -> None:
-        """Connect a session to a space.
-
-        Args:
-            session_id: The UUID of the session.
-            space_id: The UUID of the space to connect to.
-        """
-        payload = {"space_id": space_id}
-        await self._requester.request(
-            "POST", f"/session/{session_id}/connect_to_space", json_data=payload
-        )
-
     async def get_tasks(
         self,
         session_id: str,
@@ -382,23 +358,6 @@ class AsyncSessionsAPI:
         data = await self._requester.request("POST", f"/session/{session_id}/flush")
         return data  # type: ignore
 
-    async def get_learning_status(self, session_id: str) -> LearningStatus:
-        """Get learning status for a session.
-
-        Returns the count of space digested tasks and not space digested tasks.
-        If the session is not connected to a space, returns 0 and 0.
-
-        Args:
-            session_id: The UUID of the session.
-
-        Returns:
-            LearningStatus object containing space_digested_count and not_space_digested_count.
-        """
-        data = await self._requester.request(
-            "GET", f"/session/{session_id}/get_learning_status"
-        )
-        return LearningStatus.model_validate(data)
-
     async def get_token_counts(self, session_id: str) -> TokenCounts:
         """Get total token counts for all text and tool-call parts in a session.
 

acontext/resources/async_skills.py

@@ -9,6 +9,7 @@ from typing import Any, BinaryIO, cast
 from .._utils import build_params
 from ..client_types import AsyncRequesterProtocol
 from ..types.skill import (
+    DownloadSkillToSandboxResp,
     GetSkillFileResp,
     ListSkillsOutput,
     Skill,
@@ -133,3 +134,42 @@ class AsyncSkillsAPI:
 
         data = await self._requester.request("GET", endpoint, params=params)
         return GetSkillFileResp.model_validate(data)
+
+    async def download_to_sandbox(
+        self,
+        *,
+        skill_id: str,
+        sandbox_id: str,
+    ) -> DownloadSkillToSandboxResp:
+        """Download all files from a skill to a sandbox environment.
+
+        Files are placed at /skills/{skill_name}/.
+
+        Args:
+            skill_id: The UUID of the skill to download.
+            sandbox_id: The UUID of the target sandbox.
+
+        Returns:
+            DownloadSkillToSandboxResp containing success status, the directory path
+            where the skill was installed, and the skill's name and description.
+
+        Example:
+        ```python
+            result = await client.skills.download_to_sandbox(
+                skill_id="skill-uuid",
+                sandbox_id="sandbox-uuid"
+            )
+            print(f"Success: {result.success}")
+            print(f"Skill installed at: {result.dir_path}")
+            print(f"Skill name: {result.name}")
+            print(f"Description: {result.description}")
+        ```
+        """
+        payload: dict[str, Any] = {"sandbox_id": sandbox_id}
+
+        data = await self._requester.request(
+            "POST",
+            f"/agent_skills/{skill_id}/download_to_sandbox",
+            json_data=payload,
+        )
+        return DownloadSkillToSandboxResp.model_validate(data)

acontext/resources/async_users.py

@@ -41,7 +41,7 @@ class AsyncUsersAPI:
             identifier: The user identifier string.
 
         Returns:
-            GetUserResourcesOutput containing counts for Spaces, Sessions, Disks, and Skills.
+            GetUserResourcesOutput containing counts for Sessions, Disks, and Skills.
         """
         data = await self._requester.request(
             "GET", f"/user/{quote(identifier, safe='')}/resources"
@@ -49,7 +49,7 @@ class AsyncUsersAPI:
         return GetUserResourcesOutput.model_validate(data)
 
     async def delete(self, identifier: str) -> None:
-        """Delete a user and cascade delete all associated resources (Space, Session, Disk, Skill).
+        """Delete a user and cascade delete all associated resources (Session, Disk, Skill).
 
         Args:
             identifier: The user identifier string.

acontext/resources/disks.py

@@ -37,26 +37,28 @@ class DisksAPI:
         time_desc: bool | None = None,
     ) -> ListDisksOutput:
         """List all disks in the project.
-        
+
         Args:
             user: Filter by user identifier. Defaults to None.
             limit: Maximum number of disks to return. Defaults to None.
             cursor: Cursor for pagination. Defaults to None.
             time_desc: Order by created_at descending if True, ascending if False. Defaults to None.
-            
+
         Returns:
             ListDisksOutput containing the list of disks and pagination information.
         """
-        params = build_params(user=user, limit=limit, cursor=cursor, time_desc=time_desc)
+        params = build_params(
+            user=user, limit=limit, cursor=cursor, time_desc=time_desc
+        )
         data = self._requester.request("GET", "/disk", params=params or None)
         return ListDisksOutput.model_validate(data)
 
     def create(self, *, user: str | None = None) -> Disk:
         """Create a new disk.
-        
+
         Args:
             user: Optional user identifier string. Defaults to None.
-            
+
         Returns:
             The created Disk object.
         """
@@ -68,7 +70,7 @@ class DisksAPI:
 
     def delete(self, disk_id: str) -> None:
         """Delete a disk by its ID.
-        
+
         Args:
             disk_id: The UUID of the disk to delete.
         """
@@ -83,20 +85,22 @@ class DiskArtifactsAPI:
         self,
         disk_id: str,
         *,
-        file: FileUpload
-        | tuple[str, BinaryIO | bytes]
-        | tuple[str, BinaryIO | bytes, str],
+        file: (
+            FileUpload
+            | tuple[str, BinaryIO | bytes]
+            | tuple[str, BinaryIO | bytes, str]
+        ),
         file_path: str | None = None,
         meta: Mapping[str, Any] | None = None,
     ) -> Artifact:
         """Upload a file to create or update an artifact.
-        
+
         Args:
             disk_id: The UUID of the disk.
             file: The file to upload (FileUpload object or tuple format).
             file_path: Directory path (not including filename), defaults to "/".
             meta: Custom metadata as JSON-serializable dict, defaults to None.
-            
+
         Returns:
             Artifact containing the created/updated artifact information.
         """
@@ -126,7 +130,7 @@ class DiskArtifactsAPI:
         expire: int | None = None,
     ) -> GetArtifactResp:
         """Get an artifact by disk ID, file path, and filename.
-        
+
         Args:
             disk_id: The UUID of the disk.
             file_path: Directory path (not including filename).
@@ -134,7 +138,7 @@ class DiskArtifactsAPI:
             with_public_url: Whether to include a presigned public URL. Defaults to None.
             with_content: Whether to include file content. Defaults to None.
             expire: URL expiration time in seconds. Defaults to None.
-            
+
         Returns:
             GetArtifactResp containing the artifact and optionally public URL and content.
         """
@@ -145,7 +149,9 @@ class DiskArtifactsAPI:
             with_content=with_content,
             expire=expire,
         )
-        data = self._requester.request("GET", f"/disk/{disk_id}/artifact", params=params)
+        data = self._requester.request(
+            "GET", f"/disk/{disk_id}/artifact", params=params
+        )
         return GetArtifactResp.model_validate(data)
 
     def update(
@@ -157,13 +163,13 @@ class DiskArtifactsAPI:
         meta: Mapping[str, Any],
     ) -> UpdateArtifactResp:
         """Update an artifact's metadata.
-        
+
         Args:
             disk_id: The UUID of the disk.
             file_path: Directory path (not including filename).
             filename: The filename of the artifact.
             meta: Custom metadata as JSON-serializable dict.
-            
+
         Returns:
             UpdateArtifactResp containing the updated artifact information.
         """
@@ -172,7 +178,9 @@ class DiskArtifactsAPI:
             "file_path": full_path,
             "meta": json.dumps(cast(Mapping[str, Any], meta)),
         }
-        data = self._requester.request("PUT", f"/disk/{disk_id}/artifact", json_data=payload)
+        data = self._requester.request(
+            "PUT", f"/disk/{disk_id}/artifact", json_data=payload
+        )
         return UpdateArtifactResp.model_validate(data)
 
     def delete(
@@ -183,7 +191,7 @@ class DiskArtifactsAPI:
         filename: str,
     ) -> None:
         """Delete an artifact by disk ID, file path, and filename.
-        
+
         Args:
             disk_id: The UUID of the disk.
             file_path: Directory path (not including filename).
@@ -200,18 +208,20 @@ class DiskArtifactsAPI:
         path: str | None = None,
     ) -> ListArtifactsResp:
         """List artifacts in a disk at a specific path.
-        
+
         Args:
             disk_id: The UUID of the disk.
             path: Directory path to list. Defaults to None (root).
-            
+
         Returns:
             ListArtifactsResp containing the list of artifacts.
         """
         params: dict[str, Any] = {}
         if path is not None:
             params["path"] = path
-        data = self._requester.request("GET", f"/disk/{disk_id}/artifact/ls", params=params or None)
+        data = self._requester.request(
+            "GET", f"/disk/{disk_id}/artifact/ls", params=params or None
+        )
         return ListArtifactsResp.model_validate(data)
 
     def grep_artifacts(
@@ -222,15 +232,15 @@ class DiskArtifactsAPI:
         limit: int = 100,
     ) -> list[Artifact]:
         """Search artifact content using regex pattern.
-        
+
         Args:
             disk_id: The disk ID to search in
             query: Regex pattern to search for in file content
             limit: Maximum number of results (default 100, max 1000)
-            
+
         Returns:
             List of matching artifacts
-            
+
         Example:
         ```python
             # Search for TODO comments in code
@@ -242,9 +252,7 @@ class DiskArtifactsAPI:
         """
         params = build_params(query=query, limit=limit)
         data = self._requester.request(
-            "GET",
-            f"/disk/{disk_id}/artifact/grep",
-            params=params
+            "GET", f"/disk/{disk_id}/artifact/grep", params=params
         )
         return TypeAdapter(list[Artifact]).validate_python(data)
 
@@ -256,15 +264,15 @@ class DiskArtifactsAPI:
         limit: int = 100,
     ) -> list[Artifact]:
         """Search artifact paths using glob pattern.
-        
+
         Args:
             disk_id: The disk ID to search in
             query: Glob pattern (e.g., '**/*.py', '*.txt')
             limit: Maximum number of results (default 100, max 1000)
-            
+
         Returns:
             List of matching artifacts
-            
+
         Example:
         ```python
             # Find all Python files
@@ -276,8 +284,98 @@ class DiskArtifactsAPI:
         """
         params = build_params(query=query, limit=limit)
         data = self._requester.request(
-            "GET",
-            f"/disk/{disk_id}/artifact/glob",
-            params=params
+            "GET", f"/disk/{disk_id}/artifact/glob", params=params
         )
         return TypeAdapter(list[Artifact]).validate_python(data)
+
+    def download_to_sandbox(
+        self,
+        disk_id: str,
+        *,
+        file_path: str,
+        filename: str,
+        sandbox_id: str,
+        sandbox_path: str,
+    ) -> bool:
+        """Download an artifact from disk storage to a sandbox environment.
+
+        Args:
+            disk_id: The UUID of the disk containing the artifact.
+            file_path: Directory path of the artifact (not including filename).
+            filename: The filename of the artifact.
+            sandbox_id: The UUID of the target sandbox.
+            sandbox_path: Destination directory in the sandbox.
+
+        Returns:
+            True if the download was successful.
+
+        Example:
+        ```python
+            success = client.disks.artifacts.download_to_sandbox(
+                disk_id="disk-uuid",
+                file_path="/documents/",
+                filename="report.pdf",
+                sandbox_id="sandbox-uuid",
+                sandbox_path="/home/user/"
+            )
+            print(f"Success: {success}")
+        ```
+        """
+        payload = {
+            "file_path": file_path,
+            "filename": filename,
+            "sandbox_id": sandbox_id,
+            "sandbox_path": sandbox_path,
+        }
+        data = self._requester.request(
+            "POST",
+            f"/disk/{disk_id}/artifact/download_to_sandbox",
+            json_data=payload,
+        )
+        return bool(data.get("success", False))
+
+    def upload_from_sandbox(
+        self,
+        disk_id: str,
+        *,
+        sandbox_id: str,
+        sandbox_path: str,
+        sandbox_filename: str,
+        file_path: str,
+    ) -> Artifact:
+        """Upload a file from a sandbox environment to disk storage as an artifact.
+
+        Args:
+            disk_id: The UUID of the target disk.
+            sandbox_id: The UUID of the source sandbox.
+            sandbox_path: Source directory in the sandbox (not including filename).
+            sandbox_filename: Filename in the sandbox.
+            file_path: Destination directory path on the disk.
+
+        Returns:
+            Artifact containing the created artifact information.
+
+        Example:
+        ```python
+            artifact = client.disks.artifacts.upload_from_sandbox(
+                disk_id="disk-uuid",
+                sandbox_id="sandbox-uuid",
+                sandbox_path="/home/user/",
+                sandbox_filename="output.txt",
+                file_path="/results/"
+            )
+            print(f"Created: {artifact.path}{artifact.filename}")
+        ```
+        """
+        payload = {
+            "sandbox_id": sandbox_id,
+            "sandbox_path": sandbox_path,
+            "sandbox_filename": sandbox_filename,
+            "file_path": file_path,
+        }
+        data = self._requester.request(
+            "POST",
+            f"/disk/{disk_id}/artifact/upload_from_sandbox",
+            json_data=payload,
+        )
+        return Artifact.model_validate(data)

acontext/resources/sandboxes.py

@@ -0,0 +1,85 @@
+"""
+Sandboxes endpoints.
+"""
+
+from .._utils import build_params
+from ..client_types import RequesterProtocol
+from ..types.sandbox import (
+    GetSandboxLogsOutput,
+    SandboxCommandOutput,
+    SandboxRuntimeInfo,
+)
+from ..types.tool import FlagResponse
+
+
+class SandboxesAPI:
+    def __init__(self, requester: RequesterProtocol) -> None:
+        self._requester = requester
+
+    def create(self) -> SandboxRuntimeInfo:
+        """Create and start a new sandbox.
+
+        Returns:
+            SandboxRuntimeInfo containing the sandbox ID, status, and timestamps.
+        """
+        data = self._requester.request("POST", "/sandbox")
+        return SandboxRuntimeInfo.model_validate(data)
+
+    def exec_command(
+        self,
+        *,
+        sandbox_id: str,
+        command: str,
+        timeout: float | None = None,
+    ) -> SandboxCommandOutput:
+        """Execute a shell command in the sandbox.
+
+        Args:
+            sandbox_id: The UUID of the sandbox.
+            command: The shell command to execute.
+            timeout: Optional timeout in seconds for this command.
+                    If not provided, uses the client's default timeout.
+
+        Returns:
+            SandboxCommandOutput containing stdout, stderr, and exit code.
+        """
+        data = self._requester.request(
+            "POST",
+            f"/sandbox/{sandbox_id}/exec",
+            json_data={"command": command},
+            timeout=timeout,
+        )
+        return SandboxCommandOutput.model_validate(data)
+
+    def kill(self, sandbox_id: str) -> FlagResponse:
+        """Kill a running sandbox.
+
+        Args:
+            sandbox_id: The UUID of the sandbox to kill.
+
+        Returns:
+            FlagResponse with status and error message.
+        """
+        data = self._requester.request("DELETE", f"/sandbox/{sandbox_id}")
+        return FlagResponse.model_validate(data)
+
+    def get_logs(
+        self,
+        *,
+        limit: int | None = None,
+        cursor: str | None = None,
+        time_desc: bool | None = None,
+    ) -> GetSandboxLogsOutput:
+        """Get sandbox logs for the project with cursor-based pagination.
+
+        Args:
+            limit: Maximum number of logs to return (default 20, max 200).
+            cursor: Cursor for pagination. Use the cursor from the previous response to get the next page.
+            time_desc: Order by created_at descending if True, ascending if False (default False).
+
+        Returns:
+            GetSandboxLogsOutput containing the list of sandbox logs and pagination information.
+        """
+        params = build_params(limit=limit, cursor=cursor, time_desc=time_desc)
+        data = self._requester.request("GET", "/sandbox/logs", params=params or None)
+        return GetSandboxLogsOutput.model_validate(data)