feat(sdk): add learning status api

Author: gusye1234

src/client/acontext-py/src/acontext/resources/async_sessions.py

@@ -11,6 +11,7 @@
 from ..types.session import (
     GetMessagesOutput,
     GetTasksOutput,
+    LearningStatus,
     ListSessionsOutput,
     Message,
     Session,
@@ -303,3 +304,20 @@ async def flush(self, session_id: str) -> dict[str, Any]:
         """
         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)

src/client/acontext-py/src/acontext/resources/sessions.py

@@ -11,6 +11,7 @@
 from ..types.session import (
     GetMessagesOutput,
     GetTasksOutput,
+    LearningStatus,
     ListSessionsOutput,
     Message,
     Session,
@@ -303,3 +304,20 @@ def flush(self, session_id: str) -> dict[str, Any]:
         """
         data = self._requester.request("POST", f"/session/{session_id}/flush")
         return data  # type: ignore
+
+    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 = self._requester.request(
+            "GET", f"/session/{session_id}/get_learning_status"
+        )
+        return LearningStatus.model_validate(data)

src/client/acontext-py/src/acontext/types/__init__.py

@@ -13,6 +13,7 @@
     Asset,
     GetMessagesOutput,
     GetTasksOutput,
+    LearningStatus,
     ListSessionsOutput,
     Message,
     Part,
@@ -47,6 +48,7 @@
     "Asset",
     "GetMessagesOutput",
     "GetTasksOutput",
+    "LearningStatus",
     "ListSessionsOutput",
     "Message",
     "Part",

src/client/acontext-py/src/acontext/types/session.py

@@ -122,3 +122,14 @@ class GetTasksOutput(BaseModel):
     items: list[Task] = Field(..., description="List of tasks")
     next_cursor: str | None = Field(None, description="Cursor for pagination")
     has_more: bool = Field(..., description="Whether there are more items")
+
+
+class LearningStatus(BaseModel):
+    """Response model for learning status."""
+
+    space_digested_count: int = Field(
+        ..., description="Number of tasks that are space digested"
+    )
+    not_space_digested_count: int = Field(
+        ..., description="Number of tasks that are not space digested"
+    )

src/client/acontext-py/tests/test_async_client.py

@@ -438,6 +438,30 @@ async def test_async_sessions_get_tasks_with_filters(
     assert hasattr(result, "has_more")
 
 
+@patch("acontext.async_client.AcontextAsyncClient.request", new_callable=AsyncMock)
+@pytest.mark.asyncio
+async def test_async_sessions_get_learning_status(
+    mock_request, async_client: AcontextAsyncClient
+) -> None:
+    mock_request.return_value = {
+        "space_digested_count": 5,
+        "not_space_digested_count": 3,
+    }
+
+    result = await async_client.sessions.get_learning_status("session-id")
+
+    mock_request.assert_called_once()
+    args, kwargs = mock_request.call_args
+    method, path = args
+    assert method == "GET"
+    assert path == "/session/session-id/get_learning_status"
+    # Verify it returns a Pydantic model
+    assert hasattr(result, "space_digested_count")
+    assert hasattr(result, "not_space_digested_count")
+    assert result.space_digested_count == 5
+    assert result.not_space_digested_count == 3
+
+
 @patch("acontext.async_client.AcontextAsyncClient.request", new_callable=AsyncMock)
 @pytest.mark.asyncio
 async def test_async_blocks_list_without_filters(

src/client/acontext-py/tests/test_client.py

@@ -448,6 +448,27 @@ def test_sessions_get_tasks_with_filters(mock_request, client: AcontextClient) -
     assert hasattr(result, "has_more")
 
 
+@patch("acontext.client.AcontextClient.request")
+def test_sessions_get_learning_status(mock_request, client: AcontextClient) -> None:
+    mock_request.return_value = {
+        "space_digested_count": 5,
+        "not_space_digested_count": 3,
+    }
+
+    result = client.sessions.get_learning_status("session-id")
+
+    mock_request.assert_called_once()
+    args, kwargs = mock_request.call_args
+    method, path = args
+    assert method == "GET"
+    assert path == "/session/session-id/get_learning_status"
+    # Verify it returns a Pydantic model
+    assert hasattr(result, "space_digested_count")
+    assert hasattr(result, "not_space_digested_count")
+    assert result.space_digested_count == 5
+    assert result.not_space_digested_count == 3
+
+
 @patch("acontext.client.AcontextClient.request")
 def test_blocks_list_without_filters(mock_request, client: AcontextClient) -> None:
     mock_request.return_value = []