acontext 0.0.3__py3-none-any.whl → 0.0.5__py3-none-any.whl

acontext/resources/async_sessions.py

@@ -11,9 +11,11 @@ from ..messages import AcontextMessage
 from ..types.session import (
     GetMessagesOutput,
     GetTasksOutput,
+    LearningStatus,
     ListSessionsOutput,
     Message,
     Session,
+    TokenCounts,
 )
 from ..uploads import FileUpload, normalize_file_upload
 from pydantic import BaseModel
@@ -303,3 +305,34 @@ 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.
+
+        Args:
+            session_id: The UUID of the session.
+
+        Returns:
+            TokenCounts object containing total_tokens.
+        """
+        data = await self._requester.request(
+            "GET", f"/session/{session_id}/token_counts"
+        )
+        return TokenCounts.model_validate(data)

acontext/resources/sessions.py

@@ -11,19 +11,21 @@ from ..messages import AcontextMessage
 from ..types.session import (
     GetMessagesOutput,
     GetTasksOutput,
+    LearningStatus,
     ListSessionsOutput,
     Message,
     Session,
+    TokenCounts,
 )
 from ..uploads import FileUpload, normalize_file_upload
 from pydantic import BaseModel
 from openai.types.chat import ChatCompletionMessageParam
-from anthropic.types import Message as AnthropicMessage
+from anthropic.types import MessageParam
 
 UploadPayload = (
     FileUpload | tuple[str, BinaryIO | bytes] | tuple[str, BinaryIO | bytes, str | None]
 )
-MessageBlob = AcontextMessage | ChatCompletionMessageParam | AnthropicMessage
+MessageBlob = AcontextMessage | ChatCompletionMessageParam | MessageParam
 
 
 class SessionsAPI:
@@ -303,3 +305,34 @@ class SessionsAPI:
         """
         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)
+
+    def get_token_counts(self, session_id: str) -> TokenCounts:
+        """Get total token counts for all text and tool-call parts in a session.
+
+        Args:
+            session_id: The UUID of the session.
+
+        Returns:
+            TokenCounts object containing total_tokens.
+        """
+        data = self._requester.request(
+            "GET", f"/session/{session_id}/token_counts"
+        )
+        return TokenCounts.model_validate(data)

acontext/types/__init__.py

@@ -13,12 +13,14 @@ from .session import (
     Asset,
     GetMessagesOutput,
     GetTasksOutput,
+    LearningStatus,
     ListSessionsOutput,
     Message,
     Part,
     PublicURL,
     Session,
     Task,
+    TokenCounts,
 )
 from .block import Block
 from .space import (
@@ -47,12 +49,14 @@ __all__ = [
     "Asset",
     "GetMessagesOutput",
     "GetTasksOutput",
+    "LearningStatus",
     "ListSessionsOutput",
     "Message",
     "Part",
     "PublicURL",
     "Session",
     "Task",
+    "TokenCounts",
     # Space types
     "ListSpacesOutput",
     "SearchResultBlockItem",

acontext/types/session.py

@@ -122,3 +122,20 @@ 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"
+    )
+
+
+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")

{acontext-0.0.3.dist-info → acontext-0.0.5.dist-info}/METADATA

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

{acontext-0.0.3.dist-info → acontext-0.0.5.dist-info}/RECORD

@@ -10,21 +10,21 @@ 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=ls_s9r4St-XPE9pB5nUOTYnqIWXN4MOdM0skvY9i-NI,10760
+acontext/resources/async_sessions.py,sha256=b5oha4np1-RmNf2eRsnuJ2UJjdPyztExYQF-7_1pIAc,11894
 acontext/resources/async_spaces.py,sha256=MnHJ6c8bqT2RgSsDeyYS8fgrvnMxZ3yOkd5d9hSHjPE,6414
 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=gWrwQeBUEmI8yOK9-ByGGFff5SmEApH8JqZd9VHZlfg,10624
+acontext/resources/sessions.py,sha256=9iVd_2ZCxwuUFIfdsp1lRSTHb9JoFTHFkhzT7065kwY,11715
 acontext/resources/spaces.py,sha256=ugmma1jY3k4i87MdgBSoCTP_Slxg895qeWtiKNM5SqQ,6273
 acontext/resources/tools.py,sha256=II_185B0HYKSP43hizE6C1zs7kjkkPLKihuEG8s-DRY,1046
-acontext/types/__init__.py,sha256=tlxMx78rn7Bybcxv5KIhzAmWidUsriz0wF4TNEqaY5A,1202
+acontext/types/__init__.py,sha256=B5ydKtwbiyoX6ZxtHzeLXF0FrYRuotM9U1ByTUlHzRI,1280
 acontext/types/block.py,sha256=CzKByunk642rWXNNnh8cx67UzKLKDAxODmC_whwjP90,1078
 acontext/types/disk.py,sha256=g9u3rgCh05rK-gay19NUi-WyzI5Vt1GQwcyKv-5TSC4,2361
-acontext/types/session.py,sha256=o4IEvRumgvQVnx4hpjXU16qzSXV30vOYrDwnKEP54XY,5152
+acontext/types/session.py,sha256=OprkkCsEbDHrQ_Kpd6yv8lcyoCfV54IcTmNft-gAyPI,5656
 acontext/types/space.py,sha256=lhik5PMG0CthQAqgfi9MSlsP1N_sD7ubBbgAldFtO7s,1589
 acontext/types/tool.py,sha256=-mVn-vgk2SENK0Ubt-ZgWFZxKa-ddABqcAgXQ69YY-E,805
 acontext/uploads.py,sha256=6twnqQOY_eerNuEjeSKsE_3S0IfJUiczXtAy4aXqDl8,1379
-acontext-0.0.3.dist-info/WHEEL,sha256=w4ZtLaDgMAZW2MMZZwtH8zENekoQYBCeullI-zsXJQk,78
-acontext-0.0.3.dist-info/METADATA,sha256=gvgo15nko-Yqxb4qfpcP4allp8xX6EA-bX48lE8R_90,11648
-acontext-0.0.3.dist-info/RECORD,,
+acontext-0.0.5.dist-info/WHEEL,sha256=AaqHSNJgTyoT6I9ETCXrbV_7cVSjA_q07lkDGeNjGdQ,79
+acontext-0.0.5.dist-info/METADATA,sha256=GkLxYWgnBJXR9uPOccdQIAlWlUfuIC_OP-9frA0i9D4,11648
+acontext-0.0.5.dist-info/RECORD,,

{acontext-0.0.3.dist-info → acontext-0.0.5.dist-info}/WHEEL

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