acontext 0.0.7__tar.gz → 0.0.8__tar.gz

{acontext-0.0.7 → acontext-0.0.8}/PKG-INFO

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

{acontext-0.0.7 → acontext-0.0.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "acontext"
-version = "0.0.7"
+version = "0.0.8"
 description = "Python SDK for the Acontext API"
 readme = "README.md"
 requires-python = ">=3.10"

{acontext-0.0.7 → acontext-0.0.8}/src/acontext/messages.py

@@ -31,22 +31,22 @@ class AcontextMessage:
     Represents an Acontext-format message payload.
     """
 
-    role: Literal["user", "assistant", "system"]
+    role: Literal["user", "assistant"]
     parts: list[MessagePart]
     meta: Mapping[str, Any] | None = None
 
 
 def build_acontext_message(
     *,
-    role: Literal["user", "assistant", "system"],
+    role: Literal["user", "assistant"],
     parts: Sequence[MessagePart | str | Mapping[str, Any]],
     meta: Mapping[str, Any] | None = None,
 ) -> AcontextMessage:
     """
     Construct an Acontext-format message blob and associated multipart files.
     """
-    if role not in {"user", "assistant", "system"}:
-        raise ValueError("role must be one of {'user', 'assistant', 'system'}")
+    if role not in {"user", "assistant"}:
+        raise ValueError("role must be one of {'user', 'assistant'}")
 
     normalized_parts = [normalize_message_part(part) for part in parts]
 

{acontext-0.0.7 → acontext-0.0.8}/src/acontext/resources/async_sessions.py

@@ -3,12 +3,13 @@
 import json
 from collections.abc import Mapping
 from dataclasses import asdict
-from typing import Any, BinaryIO, Literal
+from typing import Any, BinaryIO, Literal, Optional, List
 
 from .._utils import build_params
 from ..client_types import AsyncRequesterProtocol
 from ..messages import AcontextMessage
 from ..types.session import (
+    EditStrategy,
     GetMessagesOutput,
     GetTasksOutput,
     LearningStatus,
@@ -264,6 +265,7 @@ class AsyncSessionsAPI:
         with_asset_public_url: bool | None = None,
         format: Literal["acontext", "openai", "anthropic"] = "openai",
         time_desc: bool | None = None,
+        edit_strategies: Optional[List[EditStrategy]] = None,
     ) -> GetMessagesOutput:
         """Get messages for a session.
 
@@ -272,8 +274,14 @@ class AsyncSessionsAPI:
             limit: Maximum number of messages to return. Defaults to None.
             cursor: Cursor for pagination. Defaults to None.
             with_asset_public_url: Whether to include presigned URLs for assets. Defaults to None.
-            format: The format of the messages. Defaults to "acontext".
+            format: The format of the messages. Defaults to "openai".
             time_desc: Order by created_at descending if True, ascending if False. Defaults to None.
+            edit_strategies: Optional list of edit strategies to apply before format conversion.
+                Each strategy is a dict with 'type' and 'params' keys.
+                Examples:
+                    - Remove tool results: [{"type": "remove_tool_result", "params": {"keep_recent_n_tool_results": 3}}]
+                    - Token limit: [{"type": "token_limit", "params": {"limit_tokens": 20000}}]
+                Defaults to None.
 
         Returns:
             GetMessagesOutput containing the list of messages and pagination information.
@@ -289,6 +297,8 @@ class AsyncSessionsAPI:
                 time_desc=time_desc,
             )
         )
+        if edit_strategies is not None:
+            params["edit_strategies"] = json.dumps(edit_strategies)
         data = await self._requester.request(
             "GET", f"/session/{session_id}/messages", params=params or None
         )

{acontext-0.0.7 → acontext-0.0.8}/src/acontext/resources/async_spaces.py

@@ -181,7 +181,7 @@ class AsyncSpacesAPI:
         """
         payload = {"save": save}
         data = await self._requester.request(
-            "PATCH",
+            "PUT",
             f"/space/{space_id}/experience_confirmations/{experience_id}",
             json_data=payload,
         )

{acontext-0.0.7 → acontext-0.0.8}/src/acontext/resources/sessions.py

@@ -3,12 +3,13 @@
 import json
 from collections.abc import Mapping
 from dataclasses import asdict
-from typing import Any, BinaryIO, Literal
+from typing import Any, BinaryIO, Literal, Optional, List
 
 from .._utils import build_params
 from ..client_types import RequesterProtocol
 from ..messages import AcontextMessage
 from ..types.session import (
+    EditStrategy,
     GetMessagesOutput,
     GetTasksOutput,
     LearningStatus,
@@ -71,12 +72,14 @@ class SessionsAPI:
         self,
         *,
         space_id: str | None = None,
+        disable_task_tracking: bool | None = None,
         configs: Mapping[str, Any] | None = None,
     ) -> Session:
         """Create a new session.
 
         Args:
             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.
 
         Returns:
@@ -85,6 +88,8 @@ class SessionsAPI:
         payload: dict[str, Any] = {}
         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:
             payload["configs"] = configs
         data = self._requester.request("POST", "/session", json_data=payload)
@@ -264,6 +269,7 @@ class SessionsAPI:
         with_asset_public_url: bool | None = None,
         format: Literal["acontext", "openai", "anthropic"] = "openai",
         time_desc: bool | None = None,
+        edit_strategies: Optional[List[EditStrategy]] = None,
     ) -> GetMessagesOutput:
         """Get messages for a session.
 
@@ -272,8 +278,14 @@ class SessionsAPI:
             limit: Maximum number of messages to return. Defaults to None.
             cursor: Cursor for pagination. Defaults to None.
             with_asset_public_url: Whether to include presigned URLs for assets. Defaults to None.
-            format: The format of the messages. Defaults to "acontext".
+            format: The format of the messages. Defaults to "openai".
             time_desc: Order by created_at descending if True, ascending if False. Defaults to None.
+            edit_strategies: Optional list of edit strategies to apply before format conversion.
+                Each strategy is a dict with 'type' and 'params' keys.
+                Examples:
+                    - Remove tool results: [{"type": "remove_tool_result", "params": {"keep_recent_n_tool_results": 3}}]
+                    - Token limit: [{"type": "token_limit", "params": {"limit_tokens": 20000}}]
+                Defaults to None.
 
         Returns:
             GetMessagesOutput containing the list of messages and pagination information.
@@ -289,6 +301,8 @@ class SessionsAPI:
                 time_desc=time_desc,
             )
         )
+        if edit_strategies is not None:
+            params["edit_strategies"] = json.dumps(edit_strategies)
         data = self._requester.request(
             "GET", f"/session/{session_id}/messages", params=params or None
         )
@@ -332,7 +346,5 @@ class SessionsAPI:
         Returns:
             TokenCounts object containing total_tokens.
         """
-        data = self._requester.request(
-            "GET", f"/session/{session_id}/token_counts"
-        )
+        data = self._requester.request("GET", f"/session/{session_id}/token_counts")
         return TokenCounts.model_validate(data)

{acontext-0.0.7 → acontext-0.0.8}/src/acontext/resources/spaces.py

@@ -179,7 +179,7 @@ class SpacesAPI:
         """
         payload = {"save": save}
         data = self._requester.request(
-            "PATCH",
+            "PUT",
             f"/space/{space_id}/experience_confirmations/{experience_id}",
             json_data=payload,
         )

{acontext-0.0.7 → acontext-0.0.8}/src/acontext/types/session.py

@@ -1,10 +1,67 @@
 """Type definitions for session, message, and task resources."""
 
-from typing import Any
+from typing import Any, Literal, NotRequired, TypedDict, Union
 
 from pydantic import BaseModel, Field
 
 
+class RemoveToolResultParams(TypedDict, total=False):
+    """Parameters for the remove_tool_result edit strategy.
+
+    Attributes:
+        keep_recent_n_tool_results: Number of most recent tool results to keep with original content.
+            Defaults to 3 if not specified.
+        tool_result_placeholder: Custom text to replace old tool results with.
+            Defaults to "Done" if not specified.
+    """
+
+    keep_recent_n_tool_results: NotRequired[int]
+    tool_result_placeholder: NotRequired[str]
+
+
+class RemoveToolResultStrategy(TypedDict):
+    """Edit strategy to replace old tool results with placeholder text.
+
+    Example:
+        {"type": "remove_tool_result", "params": {"keep_recent_n_tool_results": 5, "tool_result_placeholder": "Cleared"}}
+    """
+
+    type: Literal["remove_tool_result"]
+    params: RemoveToolResultParams
+
+
+class TokenLimitParams(TypedDict):
+    """Parameters for the token_limit edit strategy.
+
+    Attributes:
+        limit_tokens: Maximum number of tokens to keep. Required parameter.
+            Messages will be removed from oldest to newest until total tokens <= limit_tokens.
+            Tool-call and tool-result pairs are always removed together.
+    """
+
+    limit_tokens: int
+
+
+class TokenLimitStrategy(TypedDict):
+    """Edit strategy to truncate messages based on token count.
+
+    Removes oldest messages until the total token count is within the specified limit.
+    Maintains tool-call/tool-result pairing - when removing a message with tool-calls,
+    the corresponding tool-result messages are also removed.
+
+    Example:
+        {"type": "token_limit", "params": {"limit_tokens": 20000}}
+    """
+
+    type: Literal["token_limit"]
+    params: TokenLimitParams
+
+
+# Union type for all edit strategies
+# When adding new strategies, add them to this Union: EditStrategy = Union[RemoveToolResultStrategy, OtherStrategy, ...]
+EditStrategy = Union[RemoveToolResultStrategy, TokenLimitStrategy]
+
+
 class Asset(BaseModel):
     """Asset model representing a file asset."""
 
@@ -35,7 +92,7 @@ class Message(BaseModel):
     id: str = Field(..., description="Message UUID")
     session_id: str = Field(..., description="Session UUID")
     parent_id: str | None = Field(None, description="Parent message UUID")
-    role: str = Field(..., description="Message role: 'user', 'assistant', or 'system'")
+    role: str = Field(..., description="Message role: 'user' or 'assistant'")
     meta: dict[str, Any] = Field(..., description="Message metadata")
     parts: list[Part] = Field(..., description="List of message parts")
     task_id: str | None = Field(None, description="Task UUID if associated with a task")
@@ -52,6 +109,9 @@ class Session(BaseModel):
 
     id: str = Field(..., description="Session UUID")
     project_id: str = Field(..., description="Project UUID")
+    disable_task_tracking: bool = Field(
+        False, description="Whether task tracking is disabled for this session"
+    )
     space_id: str | None = Field(None, description="Space UUID, optional")
     configs: dict[str, Any] | None = Field(
         None, description="Session configuration dictionary"
@@ -138,4 +198,7 @@ class LearningStatus(BaseModel):
 class TokenCounts(BaseModel):
     """Response model for token counts."""
 
-    total_tokens: int = Field(..., description="Total token count for all text and tool-call parts in a session")
+    total_tokens: int = Field(
+        ...,
+        description="Total token count for all text and tool-call parts in a session",
+    )