chunkr-ai 0.0.5__py3-none-any.whl → 0.0.6__py3-none-any.whl

chunkr_ai/api/task.py

@@ -24,6 +24,7 @@ class TaskResponse(BaseModel):
     output: Optional[OutputResponse]
     page_count: Optional[int]
     pdf_url: Optional[str]
+    started_at: Optional[datetime]
     status: Status
     task_id: str
     task_url: Optional[str]
@@ -57,8 +58,9 @@ class TaskResponse(BaseModel):
         while True:
             try:
                 r = await self._client._client.get(self.task_url, headers=self._client._headers())
-                await r.raise_for_status()
-                return await r.json()
+                r.raise_for_status()
+                response = r.json()
+                return response
             except (ConnectionError, TimeoutError) as _:
                 print("Connection error while polling the task, retrying...")
                 await asyncio.sleep(0.5)
@@ -117,15 +119,4 @@ class TaskResponse(BaseModel):
         
     def content(self) -> str:
         """Get full text for the task"""
-        return self._get_content("content")
-
-class TaskPayload(BaseModel):
-    current_configuration: Configuration
-    file_name: str
-    image_folder_location: str
-    input_location: str
-    output_location: str
-    pdf_location: str
-    previous_configuration: Optional[Configuration]
-    task_id: str
-    user_id: str
+        return self._get_content("content")

chunkr_ai/models.py

@@ -20,7 +20,7 @@ from .api.config import (
     SegmentationStrategy,
 )
 
-from .api.task import TaskResponse, TaskPayload, Status
+from .api.task import TaskResponse, Status
 
 __all__ = [
     'BoundingBox',
@@ -43,6 +43,5 @@ __all__ = [
     'SegmentType',
     'SegmentationStrategy',
     'Status',
-    'TaskPayload',
     'TaskResponse'
 ]

{chunkr_ai-0.0.5.dist-info → chunkr_ai-0.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: chunkr-ai
-Version: 0.0.5
+Version: 0.0.6
 Summary: Python client for Chunkr: open source document intelligence
 Author-email: Ishaan Kapoor <ishaan@lumina.sh>
 Project-URL: Homepage, https://chunkr.ai
@@ -9,6 +9,7 @@ License-File: LICENSE
 Requires-Dist: httpx>=0.28.1
 Requires-Dist: pillow>=11.1.0
 Requires-Dist: pydantic>=2.10.4
+Requires-Dist: pytest-asyncio>=0.25.2
 Requires-Dist: python-dotenv>=1.0.1
 Requires-Dist: requests>=2.32.3
 Provides-Extra: test

{chunkr_ai-0.0.5.dist-info → chunkr_ai-0.0.6.dist-info}/RECORD

@@ -1,6 +1,6 @@
 chunkr_ai/__init__.py,sha256=eXygrEhGxxIHXNYIlHF2eied8rGsx2RphgR8Wo4lRyo,110
 chunkr_ai/main.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-chunkr_ai/models.py,sha256=d-B4vfgZClJOoHdPaH3vagwUc4qxeQSmUxab77DKYtQ,874
+chunkr_ai/models.py,sha256=kNeYtBO4TFvQWKFCent7tLEQjyKlVUieKNiuTt3u564,842
 chunkr_ai/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chunkr_ai/api/api.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chunkr_ai/api/auth.py,sha256=iSd5Jek2BFaHGw9HY-RrqgwP56BHFU0xbSuJS4fU6AA,425
@@ -9,9 +9,9 @@ chunkr_ai/api/chunkr.py,sha256=LkBFzGB_T0y3fnBeIn_nwQW6Mb7eZO-iTlzWrmWBoko,3450
 chunkr_ai/api/chunkr_async.py,sha256=B9deRVoe4h3Csh_jEuQxuxQ-DKSuZPdwkanFTyfHmeM,3603
 chunkr_ai/api/config.py,sha256=K0s1giImciPksu-bO9gzRwUaK2Vo1nxNKQkXlRQ2cb8,3785
 chunkr_ai/api/protocol.py,sha256=XKS9RmtvBpJItYhPg18qlOCKpaSHdOuQTRSUxAdUz2g,479
-chunkr_ai/api/task.py,sha256=ALU-rYlObbitlM1MKEFeSz_IBUpzb9736Iqu9huWg7c,4392
-chunkr_ai-0.0.5.dist-info/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-chunkr_ai-0.0.5.dist-info/METADATA,sha256=Roj63O2Ms3D1vNfgEmnCYAJESFrOQ9nnsSlyXkvORU4,4806
-chunkr_ai-0.0.5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-chunkr_ai-0.0.5.dist-info/top_level.txt,sha256=0IZY7PZIiS8bw5r4NUQRUQ-ATi-L_3vLQVq3ZLouOW8,10
-chunkr_ai-0.0.5.dist-info/RECORD,,
+chunkr_ai/api/task.py,sha256=_WOGRirlLEow_wS9kJB_dNYb2RvYE9nlu7Spq16AhME,4172
+chunkr_ai-0.0.6.dist-info/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+chunkr_ai-0.0.6.dist-info/METADATA,sha256=TuBBU6n1g7kdLVky2vAx94TFWZVyu8PqQ_47vi6tN5E,4844
+chunkr_ai-0.0.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+chunkr_ai-0.0.6.dist-info/top_level.txt,sha256=0IZY7PZIiS8bw5r4NUQRUQ-ATi-L_3vLQVq3ZLouOW8,10
+chunkr_ai-0.0.6.dist-info/RECORD,,