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

acontext/resources/async_sessions.py

@@ -15,6 +15,7 @@ from ..types.session import (
     ListSessionsOutput,
     Message,
     Session,
+    TokenCounts,
 )
 from ..uploads import FileUpload, normalize_file_upload
 from pydantic import BaseModel
@@ -321,3 +322,17 @@ class AsyncSessionsAPI:
             "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

@@ -15,6 +15,7 @@ from ..types.session import (
     ListSessionsOutput,
     Message,
     Session,
+    TokenCounts,
 )
 from ..uploads import FileUpload, normalize_file_upload
 from pydantic import BaseModel
@@ -321,3 +322,17 @@ class SessionsAPI:
             "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

@@ -20,6 +20,7 @@ from .session import (
     PublicURL,
     Session,
     Task,
+    TokenCounts,
 )
 from .block import Block
 from .space import (
@@ -55,6 +56,7 @@ __all__ = [
     "PublicURL",
     "Session",
     "Task",
+    "TokenCounts",
     # Space types
     "ListSpacesOutput",
     "SearchResultBlockItem",

acontext/types/session.py

@@ -133,3 +133,9 @@ class LearningStatus(BaseModel):
     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.4.dist-info → acontext-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: acontext
-Version: 0.0.4
+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.4.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=B7GgGV5M2wH1t9wyJdvdsCxrBfXxBKCAd6V6IGHAS2Y,11414
+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=0t2EREueidwSz0bXycmaysNmA4AT7EXKVGKr8C09X2w,11247
+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=vTSOk8ByG2qUJF9CRj73wYefQAxaFiqSKrMLClWb0hY,1244
+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=_suqVE_bgaQ6vGX0sgTuICREPZwWKG8cRy6aXoEee-w,5466
+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.4.dist-info/WHEEL,sha256=YUH1mBqsx8Dh2cQG2rlcuRYUhJddG9iClegy4IgnHik,79
-acontext-0.0.4.dist-info/METADATA,sha256=gNQPNnyKjtZFInTRTl4eIraKn-cHHq_T0UOGI3G6Lqg,11648
-acontext-0.0.4.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.4.dist-info → acontext-0.0.5.dist-info}/WHEEL

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