chunkr-ai 0.0.14__py3-none-any.whl → 0.0.16__py3-none-any.whl

chunkr_ai/api/task_async.py

@@ -3,21 +3,24 @@ from .misc import prepare_upload_data
 from .task_base import TaskBase
 import asyncio
 
+
 class TaskResponseAsync(TaskBase):
     async def _poll_request(self) -> dict:
         try:
             if not self._client._client:
                 raise ValueError("Client not found")
-            r = await self._client._client.get(self.task_url, headers=self._client._headers())
+            r = await self._client._client.get(
+                self.task_url, headers=self._client._headers()
+            )
             r.raise_for_status()
             return r.json()
         except (ConnectionError, TimeoutError) as _:
             print("Connection error while polling the task, retrying...")
             await asyncio.sleep(0.5)
-        except Exception as e:
+        except Exception:
             raise
 
-    async def poll(self) -> 'TaskResponseAsync':
+    async def poll(self) -> "TaskResponseAsync":
         if not self.task_url:
             raise ValueError("Task URL not found")
         if not self._client._client:
@@ -30,13 +33,15 @@ class TaskResponseAsync(TaskBase):
                 return res
             await asyncio.sleep(0.5)
 
-    async def update(self, config: Configuration) -> 'TaskResponseAsync':
+    async def update(self, config: Configuration) -> "TaskResponseAsync":
         if not self.task_url:
             raise ValueError("Task URL not found")
         if not self._client._client:
             raise ValueError("Client not found")
         f = prepare_upload_data(None, config)
-        r = await self._client._client.patch(self.task_url, files=f, headers=self._client._headers())
+        r = await self._client._client.patch(
+            self.task_url, files=f, headers=self._client._headers()
+        )
         r.raise_for_status()
         updated = TaskResponseAsync(**r.json()).with_client(self._client)
         self.__dict__.update(updated.__dict__)
@@ -47,7 +52,9 @@ class TaskResponseAsync(TaskBase):
             raise ValueError("Task URL not found")
         if not self._client._client:
             raise ValueError("Client not found")
-        r = await self._client._client.get(f"{self.task_url}/cancel", headers=self._client._headers())
+        r = await self._client._client.get(
+            f"{self.task_url}/cancel", headers=self._client._headers()
+        )
         r.raise_for_status()
         return await self.poll()
 
@@ -56,5 +63,7 @@ class TaskResponseAsync(TaskBase):
             raise ValueError("Task URL not found")
         if not self._client._client:
             raise ValueError("Client not found")
-        r = await self._client._client.delete(self.task_url, headers=self._client._headers())
+        r = await self._client._client.delete(
+            self.task_url, headers=self._client._headers()
+        )
         r.raise_for_status()

chunkr_ai/api/task_base.py

@@ -5,7 +5,7 @@ from typing import TypeVar, Optional, Generic
 from pydantic import BaseModel, PrivateAttr
 from datetime import datetime
 
-T = TypeVar('T', bound='TaskBase')
+T = TypeVar("T", bound="TaskBase")
 
 class TaskBase(BaseModel, ABC, Generic[T]):
     configuration: Configuration
@@ -62,12 +62,15 @@ class TaskBase(BaseModel, ABC, Generic[T]):
         return None
 
     def html(self) -> str:
+        """Get the full HTML of the task"""
         return self._get_content("html")
 
     def markdown(self) -> str:
+        """Get the full markdown of the task"""
         return self._get_content("markdown")
 
     def content(self) -> str:
+        """Get the full content of the task"""
         return self._get_content("content")
 
     def _get_content(self, t: str) -> str:

chunkr_ai/models.py

@@ -18,32 +18,33 @@ from .api.config import (
     SegmentType,
     SegmentationStrategy,
     Status,
+    PipelineType,
 )
 
 from .api.task import TaskResponse
 from .api.task_async import TaskResponseAsync
 
 __all__ = [
-    'BoundingBox',
-    'Chunk',
-    'ChunkProcessing',
-    'Configuration',
-    'CroppingStrategy',
-    'ExtractedJson',
-    'GenerationConfig',
-    'GenerationStrategy',
-    'JsonSchema',
-    'LlmConfig',
-    'Model',
-    'OCRResult',
-    'OcrStrategy',
-    'OutputResponse',
-    'Property',
-    'Segment',
-    'SegmentProcessing',
-    'SegmentType',
-    'SegmentationStrategy',
-    'Status',
-    'TaskResponse',
-    'TaskResponseAsync',
+    "BoundingBox",
+    "Chunk",
+    "ChunkProcessing",
+    "Configuration",
+    "CroppingStrategy",
+    "ExtractedJson",
+    "GenerationConfig",
+    "GenerationStrategy",
+    "JsonSchema",
+    "Model",
+    "OCRResult",
+    "OcrStrategy",
+    "OutputResponse",
+    "Property",
+    "Segment",
+    "SegmentProcessing",
+    "SegmentType",
+    "SegmentationStrategy",
+    "Status",
+    "TaskResponse",
+    "TaskResponseAsync",
+    "PipelineType",
 ]

{chunkr_ai-0.0.14.dist-info → chunkr_ai-0.0.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: chunkr-ai
-Version: 0.0.14
+Version: 0.0.16
 Summary: Python client for Chunkr: open source document intelligence
 Author-email: Ishaan Kapoor <ishaan@lumina.sh>
 Project-URL: Homepage, https://chunkr.ai

chunkr_ai-0.0.16.dist-info/RECORD

@@ -0,0 +1,21 @@
+chunkr_ai/__init__.py,sha256=q5YosvCNXPNGjV10pZY1gcvdosqUh38nVQTQA9g8EuM,110
+chunkr_ai/models.py,sha256=iy8bqswbiNf-rCwmDnAF2jkZT7apOpxIxUMX3fWiE7k,924
+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=hlv0GiUmlsbFO1wLL9sslqOnsBSoBqkL_6Mk2SDvxgE,413
+chunkr_ai/api/base.py,sha256=QvHl8FInKHYKPLWDeEPpCchB1uktzOwTW7iPnyXccUc,6449
+chunkr_ai/api/chunkr.py,sha256=0extAWVeZtI7B-g14smTfFZD_csdJNCcVNXx2_L69OQ,2617
+chunkr_ai/api/chunkr_async.py,sha256=aa0s_tnYoujHBsfe8uLiPpVEnb2l9A3CXwPP34w9Mk8,4127
+chunkr_ai/api/chunkr_base.py,sha256=k34Dyt1f21NBWZvZJ3w6Svvpg4SKnzr2ldGQ4ib96Wc,4951
+chunkr_ai/api/config.py,sha256=TWl0Az6acKQCS1LIpKD4qr_lQ_63wqQ5M6calpLOlDM,5040
+chunkr_ai/api/misc.py,sha256=bQpURc7soT5GL2ZpY7EiYyvPYWEzDM9qaX-UHa-oFeI,4909
+chunkr_ai/api/protocol.py,sha256=lxIR_qoCA2a1OXjpq3LrWMdS0jRHct1bEmBlUzV8gvE,526
+chunkr_ai/api/schema.py,sha256=yYesvueGgtmRa7Fi_Tpdv8A2bzHlx-B-5DxRAPlaDHo,4926
+chunkr_ai/api/task.py,sha256=28J4dR8BDjvtkh3CQjW_YUEkgPXhCHBGu0wH6AQKKuE,2474
+chunkr_ai/api/task_async.py,sha256=K5hTEOnmD42snPZg_JtJsVWg6QBUFZ1aBz1Abwv58-A,2529
+chunkr_ai/api/task_base.py,sha256=KLiMhvvbCgcilguQKrtEPMlNs8oaatfQUtn8pYt9t6g,2467
+chunkr_ai-0.0.16.dist-info/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+chunkr_ai-0.0.16.dist-info/METADATA,sha256=ZZnNe88CL56fIw5-3UouYX0I10N9D7NXGzoBQmtd-oE,4839
+chunkr_ai-0.0.16.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+chunkr_ai-0.0.16.dist-info/top_level.txt,sha256=0IZY7PZIiS8bw5r4NUQRUQ-ATi-L_3vLQVq3ZLouOW8,10
+chunkr_ai-0.0.16.dist-info/RECORD,,

chunkr_ai-0.0.14.dist-info/RECORD

@@ -1,21 +0,0 @@
-chunkr_ai/__init__.py,sha256=eXygrEhGxxIHXNYIlHF2eied8rGsx2RphgR8Wo4lRyo,110
-chunkr_ai/models.py,sha256=-dbwtTHTcGhH3LXUdVUPkobbPoeFNXRizeAW8BCGSkE,903
-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
-chunkr_ai/api/base.py,sha256=WDHx8tU0fl9_-yvYTKL-U0uaxHv-8_bRfiw9Xkl-mWM,6499
-chunkr_ai/api/chunkr.py,sha256=A1KEjA4zRJkGZhYwhWde4CFncrljEMky4SO4LPzYvws,2652
-chunkr_ai/api/chunkr_async.py,sha256=OvXd9Ma3rsp6q8nJsWzNgTKAGk-py93DqAENE8fMQfc,4153
-chunkr_ai/api/chunkr_base.py,sha256=run4UJVKa7Gx8I_ME0Mol-c_b-NIcexNkgxHu_hbX5M,4996
-chunkr_ai/api/config.py,sha256=joTn7jiOlJXTwwza-jHauLV-39CMzaxZVGB9JBm8Cok,4862
-chunkr_ai/api/misc.py,sha256=9vnfrbJ7sFlZqwEIQ4NTMb5rhPOmETT7e1jR-b42PXM,4977
-chunkr_ai/api/protocol.py,sha256=li-zy7Z-ChR9kZqJlixQv1kUYrmesPHxwUtnE5p16tQ,529
-chunkr_ai/api/schema.py,sha256=OeLOhBRXeRBgEImg0Q6O9Z10ojT6aSEVvwnDR8UeENo,4971
-chunkr_ai/api/task.py,sha256=j-Odecnbj3NjAGDyUNmZHgqsRDYhHs3xBWgsT2rrHjs,2517
-chunkr_ai/api/task_async.py,sha256=HpuVW928s-V3jjPa8L5i86lvKcpKbMhkCuV1nKSyXVA,2437
-chunkr_ai/api/task_base.py,sha256=9S8UCsrEOAH48PmOpLlyAifKDrtaUtrBMUmMZq4Dceg,2328
-chunkr_ai-0.0.14.dist-info/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-chunkr_ai-0.0.14.dist-info/METADATA,sha256=1reEDprWgXg6bKCGKv6LXJxZZR-dQeJn9DeXGbp4Iwk,4839
-chunkr_ai-0.0.14.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-chunkr_ai-0.0.14.dist-info/top_level.txt,sha256=0IZY7PZIiS8bw5r4NUQRUQ-ATi-L_3vLQVq3ZLouOW8,10
-chunkr_ai-0.0.14.dist-info/RECORD,,