acontext 0.1.1__tar.gz → 0.1.2__tar.gz

{acontext-0.1.1 → acontext-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: acontext
-Version: 0.1.1
+Version: 0.1.2
 Summary: Python SDK for the Acontext API
 Keywords: acontext,sdk,client,api
 Requires-Dist: httpx>=0.28.1
@@ -10,8 +10,8 @@ Requires-Dist: pydantic>=2.12.3
 Requires-Dist: urllib3>=2.6.3
 Requires-Python: >=3.10
 Project-URL: Homepage, https://github.com/memodb-io/Acontext
-Project-URL: Issues, https://github.com/memodb-io/Acontext/issues
 Project-URL: Repository, https://github.com/memodb-io/Acontext
+Project-URL: Issues, https://github.com/memodb-io/Acontext/issues
 Description-Content-Type: text/markdown
 
 ## acontext client for python

{acontext-0.1.1 → acontext-0.1.2}/pyproject.toml

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

{acontext-0.1.1 → acontext-0.1.2}/src/acontext/resources/async_sessions.py

@@ -180,6 +180,47 @@ class AsyncSessionsAPI:
         )
         return GetTasksOutput.model_validate(data)
 
+    async def get_session_summary(
+        self,
+        session_id: str,
+        *,
+        limit: int | None = None,
+    ) -> str:
+        """Get a summary of all tasks in a session as a formatted string.
+
+        Args:
+            session_id: The UUID of the session.
+            limit: Maximum number of tasks to include in the summary. Defaults to None (all tasks).
+
+        Returns:
+            A formatted string containing the session summary with all task information.
+        """
+        tasks_output = await self.get_tasks(session_id, limit=limit, time_desc=False)
+        tasks = tasks_output.items
+
+        if not tasks:
+            return ""
+
+        parts: list[str] = []
+        for task in tasks:
+            task_lines = [
+                f'<task id="{task.order}" description="{task.data.task_description}">'
+            ]
+            if task.data.progresses:
+                task_lines.append("<progress>")
+                for i, p in enumerate(task.data.progresses, 1):
+                    task_lines.append(f"{i}. {p}")
+                task_lines.append("</progress>")
+            if task.data.user_preferences:
+                task_lines.append("<user_preference>")
+                for i, pref in enumerate(task.data.user_preferences, 1):
+                    task_lines.append(f"{i}. {pref}")
+                task_lines.append("</user_preference>")
+            task_lines.append("</task>")
+            parts.append("\n".join(task_lines))
+
+        return "\n".join(parts)
+
     async def store_message(
         self,
         session_id: str,
@@ -294,6 +335,7 @@ class AsyncSessionsAPI:
                 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}}]
+                    - Middle out: [{"type": "middle_out", "params": {"token_reduce_to": 5000}}]
                     - Token limit: [{"type": "token_limit", "params": {"limit_tokens": 20000}}]
                 Defaults to None.
             pin_editing_strategies_at_message: Message ID to pin editing strategies at.

{acontext-0.1.1 → acontext-0.1.2}/src/acontext/resources/sessions.py

@@ -180,6 +180,47 @@ class SessionsAPI:
         )
         return GetTasksOutput.model_validate(data)
 
+    def get_session_summary(
+        self,
+        session_id: str,
+        *,
+        limit: int | None = None,
+    ) -> str:
+        """Get a summary of all tasks in a session as a formatted string.
+
+        Args:
+            session_id: The UUID of the session.
+            limit: Maximum number of tasks to include in the summary. Defaults to None (all tasks).
+
+        Returns:
+            A formatted string containing the session summary with all task information.
+        """
+        tasks_output = self.get_tasks(session_id, limit=limit, time_desc=False)
+        tasks = tasks_output.items
+
+        if not tasks:
+            return ""
+
+        parts: list[str] = []
+        for task in tasks:
+            task_lines = [
+                f'<task id="{task.order}" description="{task.data.task_description}">'
+            ]
+            if task.data.progresses:
+                task_lines.append("<progress>")
+                for i, p in enumerate(task.data.progresses, 1):
+                    task_lines.append(f"{i}. {p}")
+                task_lines.append("</progress>")
+            if task.data.user_preferences:
+                task_lines.append("<user_preference>")
+                for i, pref in enumerate(task.data.user_preferences, 1):
+                    task_lines.append(f"{i}. {pref}")
+                task_lines.append("</user_preference>")
+            task_lines.append("</task>")
+            parts.append("\n".join(task_lines))
+
+        return "\n".join(parts)
+
     def store_message(
         self,
         session_id: str,
@@ -294,6 +335,7 @@ class SessionsAPI:
                 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}}]
+                    - Middle out: [{"type": "middle_out", "params": {"token_reduce_to": 5000}}]
                     - Token limit: [{"type": "token_limit", "params": {"limit_tokens": 20000}}]
                 Defaults to None.
             pin_editing_strategies_at_message: Message ID to pin editing strategies at.
@@ -320,7 +362,9 @@ class SessionsAPI:
         if edit_strategies is not None:
             params["edit_strategies"] = json.dumps(edit_strategies)
         if pin_editing_strategies_at_message is not None:
-            params["pin_editing_strategies_at_message"] = pin_editing_strategies_at_message
+            params["pin_editing_strategies_at_message"] = (
+                pin_editing_strategies_at_message
+            )
         data = self._requester.request(
             "GET", f"/session/{session_id}/messages", params=params or None
         )
@@ -366,20 +410,19 @@ class SessionsAPI:
         """
         data = self._requester.request("GET", f"/session/{session_id}/token_counts")
         return TokenCounts.model_validate(data)
+
     def messages_observing_status(self, session_id: str) -> MessageObservingStatus:
         """Get message observing status counts for a session.
-        
+
         Returns the count of messages by their observing status:
         observed, in_process, and pending.
-        
+
         Args:
             session_id: The UUID of the session.
-        
+
         Returns:
-            MessageObservingStatus object containing observed, in_process, 
+            MessageObservingStatus object containing observed, in_process,
             pending counts and updated_at timestamp.
         """
-        data = self._requester.request(
-            "GET", f"/session/{session_id}/observing_status"
-        )
+        data = self._requester.request("GET", f"/session/{session_id}/observing_status")
         return MessageObservingStatus.model_validate(data)

{acontext-0.1.1 → acontext-0.1.2}/src/acontext/types/session.py

@@ -89,10 +89,34 @@ class TokenLimitStrategy(TypedDict):
     params: TokenLimitParams
 
 
+class MiddleOutParams(TypedDict):
+    """Parameters for the middle_out edit strategy.
+
+    Attributes:
+        token_reduce_to: Target token count to reduce the prompt to. Required parameter.
+    """
+
+    token_reduce_to: int
+
+
+class MiddleOutStrategy(TypedDict):
+    """Edit strategy to reduce prompt size by removing middle messages.
+
+    Example:
+        {"type": "middle_out", "params": {"token_reduce_to": 5000}}
+    """
+
+    type: Literal["middle_out"]
+    params: MiddleOutParams
+
+
 # Union type for all edit strategies
 # When adding new strategies, add them to this Union: EditStrategy = Union[RemoveToolResultStrategy, OtherStrategy, ...]
 EditStrategy = Union[
-    RemoveToolResultStrategy, RemoveToolCallParamsStrategy, TokenLimitStrategy
+    RemoveToolResultStrategy,
+    RemoveToolCallParamsStrategy,
+    TokenLimitStrategy,
+    MiddleOutStrategy,
 ]