acontext 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl

acontext/__init__.py

@@ -19,6 +19,7 @@ from .resources import (
     SessionsAPI,
     SpacesAPI,
 )
+from .types import Task, TaskData
 
 __all__ = [
     "AcontextClient",
@@ -36,6 +37,8 @@ __all__ = [
     "AsyncBlocksAPI",
     "AsyncSessionsAPI",
     "AsyncSpacesAPI",
+    "Task",
+    "TaskData",
     "__version__",
 ]
 

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/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/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/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/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/types/__init__.py

@@ -20,6 +20,7 @@ from .session import (
     PublicURL,
     Session,
     Task,
+    TaskData,
     TokenCounts,
 )
 from .block import Block
@@ -58,6 +59,7 @@ __all__ = [
     "PublicURL",
     "Session",
     "Task",
+    "TaskData",
     "TokenCounts",
     # Space types
     "ExperienceConfirmation",

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"
@@ -60,6 +120,25 @@ class Session(BaseModel):
     updated_at: str = Field(..., description="ISO 8601 formatted update timestamp")
 
 
+class TaskData(BaseModel):
+    """Task data model representing the structured data stored in a task.
+
+    This schema matches the TaskData model in acontext_core/schema/session/task.py
+    and the Go API TaskData struct.
+    """
+
+    task_description: str = Field(..., description="Description of the task")
+    progresses: list[str] | None = Field(
+        None, description="List of progress updates for the task"
+    )
+    user_preferences: list[str] | None = Field(
+        None, description="List of user preferences related to the task"
+    )
+    sop_thinking: str | None = Field(
+        None, description="Standard Operating Procedure thinking notes"
+    )
+
+
 class Task(BaseModel):
     """Task model representing a task in a session."""
 
@@ -67,7 +146,7 @@ class Task(BaseModel):
     session_id: str = Field(..., description="Session UUID")
     project_id: str = Field(..., description="Project UUID")
     order: int = Field(..., description="Task order")
-    data: dict[str, Any] = Field(..., description="Task data dictionary")
+    data: TaskData = Field(..., description="Structured task data")
     status: str = Field(
         ...,
         description="Task status: 'success', 'failed', 'running', or 'pending'",
@@ -138,4 +217,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",
+    )

{acontext-0.0.7.dist-info → acontext-0.0.9.dist-info}/METADATA

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

{acontext-0.0.7.dist-info → acontext-0.0.9.dist-info}/RECORD

@@ -1,4 +1,4 @@
-acontext/__init__.py,sha256=CZzHIZ2VEtNQFj0MkH3eDx-A78fSCYsUklDvhYUsD2Y,965
+acontext/__init__.py,sha256=jAgRawWIjyMd6g3gq7Xm_3vNB31cPj8rMFPO4LGQKdM,1027
 acontext/_constants.py,sha256=-OxfFwn4UdkQiNkyWhBmpM7KnZv6SgR-gMPEkjLKtDA,362
 acontext/_utils.py,sha256=GKQH45arKh0sDu64u-5jwrII_ctnU_oChYlgR5lRkfE,1250
 acontext/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -8,26 +8,26 @@ 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
 acontext/errors.py,sha256=W9i7_t46q9kbci3XAyZiyQhS154d4R9rSon3na-gjgs,1296
-acontext/messages.py,sha256=L-fCnnZ9OIzK3zmumAhHhQ4YdBgoFdDXrb91yz5hGXk,2350
+acontext/messages.py,sha256=oNRZStfhcPFt6DPOfs_-Q7R1Xui6dOMr3wmpmC8pzvE,2310
 acontext/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 acontext/resources/__init__.py,sha256=KSVQ3YJ-wuU6OWGWJ47gDGbjjetP102aEsVSR8JUb_4,721
 acontext/resources/async_blocks.py,sha256=e_iJpgcAdwS2tXQ16MMCnySlddp5JV3BadN_EyqZSF4,5555
 acontext/resources/async_disks.py,sha256=2JjLpUkz5YZEkLt_jCts_khTG_b7lvf4cUMfoaJcnI8,6471
-acontext/resources/async_sessions.py,sha256=b5oha4np1-RmNf2eRsnuJ2UJjdPyztExYQF-7_1pIAc,11894
-acontext/resources/async_spaces.py,sha256=s3CrIDmabh2tOOPtuDn40aAsFaG2oAVFonjoNPHtiV0,6410
+acontext/resources/async_sessions.py,sha256=oFVcU0rBGD5Mhue7HdKxRQermmSHSlI05eNRgPV-8J4,12541
+acontext/resources/async_spaces.py,sha256=5lIZjI8GmgoMBcftAMD2vDj2FmbLZ6pi1-yNCWKv4cM,6408
 acontext/resources/async_tools.py,sha256=RbGaF2kX65Mun-q-Fp5H1J8waWTLIdCOfbdY19jpn1o,1091
 acontext/resources/blocks.py,sha256=HJdAy5HdyTcHCYCPmqNdvApYKZ6aWs-ORIi_wQt3TUM,5447
 acontext/resources/disks.py,sha256=BjVhVXoujHWhg6L0TG9GmW9HLTTldJYEPxCbuppRkc4,6336
-acontext/resources/sessions.py,sha256=9iVd_2ZCxwuUFIfdsp1lRSTHb9JoFTHFkhzT7065kwY,11715
-acontext/resources/spaces.py,sha256=krPC836R9FDFrgdxZ-hmcF-mMwrGhI0Cac3zszG0URA,6269
+acontext/resources/sessions.py,sha256=mETp0dtGBi9e20L5f_KbCzw-6oP0pjtUmx-9Vlb8vYA,12634
+acontext/resources/spaces.py,sha256=LJP9krygZGdktMDtYGudCpk7yBxe_4v34QLk3SdejwQ,6267
 acontext/resources/tools.py,sha256=II_185B0HYKSP43hizE6C1zs7kjkkPLKihuEG8s-DRY,1046
-acontext/types/__init__.py,sha256=vrKXYUO8zZ4uIBFqRdnp-DRY0tzTDk7rNSUn0bdpTA8,1418
+acontext/types/__init__.py,sha256=QWNLmdSnLE2b5XRXxnI2OKnHVdKQPJH0nJvvOYwhB1o,1448
 acontext/types/block.py,sha256=CzKByunk642rWXNNnh8cx67UzKLKDAxODmC_whwjP90,1078
 acontext/types/disk.py,sha256=g9u3rgCh05rK-gay19NUi-WyzI5Vt1GQwcyKv-5TSC4,2361
-acontext/types/session.py,sha256=OprkkCsEbDHrQ_Kpd6yv8lcyoCfV54IcTmNft-gAyPI,5656
+acontext/types/session.py,sha256=HU8i8zxHGfGkHS9YRlh4JDst3aWZLfqSIUVq-HX4GYg,8444
 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.7.dist-info/WHEEL,sha256=z-mOpxbJHqy3cq6SvUThBZdaLGFZzdZPtgWLcP2NKjQ,79
-acontext-0.0.7.dist-info/METADATA,sha256=pGwq18SKhhmwrXpVWKB7FHuTplaAmkxEEK4QwidULQI,14395
-acontext-0.0.7.dist-info/RECORD,,
+acontext-0.0.9.dist-info/WHEEL,sha256=xDCZ-UyfvkGuEHPeI7BcJzYKIZzdqN8A8o1M5Om8IyA,79
+acontext-0.0.9.dist-info/METADATA,sha256=O1dmjGIvaW5EisSVf5E9huYMe2jo_SmrYUUhy6JVv_Q,14395
+acontext-0.0.9.dist-info/RECORD,,

{acontext-0.0.7.dist-info → acontext-0.0.9.dist-info}/WHEEL

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