chunkr-ai 0.0.7__tar.gz → 0.0.9__tar.gz

{chunkr_ai-0.0.7/src/chunkr_ai.egg-info → chunkr_ai-0.0.9}/PKG-INFO

@@ -1,20 +1,20 @@
 Metadata-Version: 2.2
 Name: chunkr-ai
-Version: 0.0.7
+Version: 0.0.9
 Summary: Python client for Chunkr: open source document intelligence
 Author-email: Ishaan Kapoor <ishaan@lumina.sh>
 Project-URL: Homepage, https://chunkr.ai
 Description-Content-Type: text/markdown
 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
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: pillow>=10.0.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: pytest-asyncio>=0.21.0
+Requires-Dist: python-dotenv>=0.19.0
+Requires-Dist: requests>=2.28.0
 Provides-Extra: test
-Requires-Dist: pytest>=8.3.4; extra == "test"
-Requires-Dist: pytest-xdist>=3.6.1; extra == "test"
+Requires-Dist: pytest>=7.0.0; extra == "test"
+Requires-Dist: pytest-xdist>=3.0.0; extra == "test"
 
 # Chunkr Python Client
 

{chunkr_ai-0.0.7 → chunkr_ai-0.0.9}/pyproject.toml

@@ -4,24 +4,24 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "chunkr-ai"
-version = "0.0.7"
+version = "0.0.9"
 authors = [{"name" = "Ishaan Kapoor", "email" = "ishaan@lumina.sh"}]
 description = "Python client for Chunkr: open source document intelligence"
 readme = "README.md"
 license = {"file" = "LICENSE"}
 urls = {Homepage = "https://chunkr.ai"} 
 dependencies = [
-    "httpx>=0.28.1",
-    "pillow>=11.1.0",
-    "pydantic>=2.10.4",
-    "pytest-asyncio>=0.25.2",
-    "python-dotenv>=1.0.1",
-    "requests>=2.32.3",
+    "httpx>=0.24.0",
+    "pillow>=10.0.0",
+    "pydantic>=2.0.0",
+    "pytest-asyncio>=0.21.0",
+    "python-dotenv>=0.19.0",
+    "requests>=2.28.0",
 ]
 
 [project.optional-dependencies]
 test = [
-    "pytest>=8.3.4",
-    "pytest-xdist>=3.6.1",
+    "pytest>=7.0.0",
+    "pytest-xdist>=3.0.0",
 ]
 

{chunkr_ai-0.0.7 → chunkr_ai-0.0.9}/src/chunkr_ai/api/config.py

@@ -86,9 +86,9 @@ class Segment(BaseModel):
     bbox: BoundingBox
     content: str
     page_height: float
-    html: Optional[str]
-    image: Optional[str]
-    markdown: Optional[str]
+    html: Optional[str] = None
+    image: Optional[str] = None
+    markdown: Optional[str] = None
     ocr: List[OCRResult]
     page_number: int
     page_width: float
@@ -104,8 +104,8 @@ class ExtractedJson(BaseModel):
     data: Dict
 
 class OutputResponse(BaseModel):
-    chunks: List[Chunk] = []
-    extracted_json: Optional[ExtractedJson]
+    chunks: List[Chunk]
+    extracted_json: Optional[ExtractedJson] = Field(default=None)
 
 class Model(str, Enum):
     FAST = "Fast"

{chunkr_ai-0.0.7 → chunkr_ai-0.0.9}/src/chunkr_ai/api/task.py

@@ -18,18 +18,18 @@ class Status(str, Enum):
 class TaskResponse(BaseModel):
     configuration: Configuration
     created_at: datetime
-    expires_at: Optional[datetime]
-    file_name: Optional[str]
-    finished_at: Optional[datetime]
-    input_file_url: Optional[str]
+    expires_at: Optional[datetime] = None
+    file_name: Optional[str] = None
+    finished_at: Optional[datetime] = None
+    input_file_url: Optional[str] = None
     message: str
-    output: Optional[OutputResponse]
-    page_count: Optional[int]
-    pdf_url: Optional[str]
-    started_at: Optional[datetime]
+    output: Optional[OutputResponse] = None
+    page_count: Optional[int] = None
+    pdf_url: Optional[str] = None
+    started_at: Optional[datetime] = None
     status: Status
     task_id: str
-    task_url: Optional[str]
+    task_url: Optional[str] = None
     _client: Optional[Union[ChunkrClientProtocol]] = PrivateAttr(default=None)
 
     def with_client(self, client: Union[ChunkrClientProtocol]) -> 'TaskResponse':

chunkr_ai-0.0.9/src/chunkr_ai/api/task_async.py

@@ -0,0 +1,111 @@
+import asyncio
+from pydantic import BaseModel, PrivateAttr
+from datetime import datetime
+from enum import Enum
+from typing import Optional, Union
+from .task_base import TaskBase
+from .protocol import ChunkrClientProtocol
+from .config import Configuration, OutputResponse
+from .misc import prepare_upload_data
+
+class Status(str, Enum):
+    STARTING = "Starting"
+    PROCESSING = "Processing"
+    SUCCEEDED = "Succeeded"
+    FAILED = "Failed"
+    CANCELLED = "Cancelled"
+
+class TaskResponseAsync(BaseModel, TaskBase):
+    configuration: Configuration
+    created_at: datetime
+    expires_at: Optional[datetime]
+    file_name: Optional[str]
+    finished_at: Optional[datetime]
+    input_file_url: Optional[str]
+    message: str
+    output: Optional[OutputResponse]
+    page_count: Optional[int]
+    pdf_url: Optional[str]
+    started_at: Optional[datetime]
+    status: Status
+    task_id: str
+    task_url: Optional[str]
+    _client: Optional[Union[ChunkrClientProtocol]] = PrivateAttr(default=None)
+
+    def with_client(self, client: Union[ChunkrClientProtocol]) -> 'TaskResponseAsync':
+        self._client = client
+        return self
+
+    async def poll(self) -> 'TaskResponseAsync':
+        while True:
+            j = await self._poll_request()
+            updated = TaskResponseAsync(**j).with_client(self._client)
+            self.__dict__.update(updated.__dict__)
+            if res := self._check_status():
+                return res
+            await asyncio.sleep(0.5)
+
+    async def _poll_request(self) -> dict:
+        if not self.task_url:
+            raise ValueError("Task URL not found")
+        while True:
+            try:
+                r = await self._client._client.get(self.task_url, headers=self._client._headers())
+                r.raise_for_status()
+                return r.json()
+            except Exception as e:
+                if self.status == Status.FAILED:
+                    raise ValueError(self.message) from e
+                await asyncio.sleep(0.5)
+
+    def _check_status(self) -> Optional['TaskResponseAsync']:
+        if self.status == Status.FAILED:
+            raise ValueError(f"Task failed: {self.message}")
+        if self.status == Status.CANCELLED:
+            return self
+        if self.status not in [Status.STARTING, Status.PROCESSING]:
+            return self
+        return None
+
+    async def update(self, config: Configuration) -> 'TaskResponseAsync':
+        if not self.task_url:
+            raise ValueError("Task URL not found")
+        f = prepare_upload_data(None, config)
+        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__)
+        return await self.poll()
+
+    async def cancel(self):
+        if not self.task_url:
+            raise ValueError("Task URL not found")
+        r = await self._client._client.get(f"{self.task_url}/cancel", headers=self._client._headers())
+        r.raise_for_status()
+        return await self.poll()
+
+    async def delete(self):
+        r = await self._client._client.delete(self.task_url, headers=self._client._headers())
+        r.raise_for_status()
+
+    def html(self) -> str:
+        return self._get_content("html")
+
+    def markdown(self) -> str:
+        return self._get_content("markdown")
+
+    def content(self) -> str:
+        return self._get_content("content")
+
+    def _get_content(self, t: str) -> str:
+        if not self.output:
+            return ""
+        parts = []
+        for c in self.output.chunks:
+            for s in c.segments:
+                v = getattr(s, t)
+                if v:
+                    parts.append(v)
+        return "\n".join(parts)
+
+    # Satisfying TaskBase abstract methods with stubs

chunkr_ai-0.0.9/src/chunkr_ai/api/task_base.py

@@ -0,0 +1,31 @@
+from abc import ABC, abstractmethod
+from .config import Configuration
+
+class TaskBase(ABC):
+    @abstractmethod
+    def poll(self):
+        pass
+
+    @abstractmethod
+    def update(self, config: Configuration):
+        pass
+
+    @abstractmethod
+    def cancel(self):
+        pass
+
+    @abstractmethod
+    def delete(self):
+        pass
+
+    @abstractmethod
+    def html(self) -> str:
+        pass
+
+    @abstractmethod
+    def markdown(self) -> str:
+        pass
+
+    @abstractmethod
+    def content(self) -> str:
+        pass

{chunkr_ai-0.0.7 → chunkr_ai-0.0.9/src/chunkr_ai.egg-info}/PKG-INFO

@@ -1,20 +1,20 @@
 Metadata-Version: 2.2
 Name: chunkr-ai
-Version: 0.0.7
+Version: 0.0.9
 Summary: Python client for Chunkr: open source document intelligence
 Author-email: Ishaan Kapoor <ishaan@lumina.sh>
 Project-URL: Homepage, https://chunkr.ai
 Description-Content-Type: text/markdown
 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
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: pillow>=10.0.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: pytest-asyncio>=0.21.0
+Requires-Dist: python-dotenv>=0.19.0
+Requires-Dist: requests>=2.28.0
 Provides-Extra: test
-Requires-Dist: pytest>=8.3.4; extra == "test"
-Requires-Dist: pytest-xdist>=3.6.1; extra == "test"
+Requires-Dist: pytest>=7.0.0; extra == "test"
+Requires-Dist: pytest-xdist>=3.0.0; extra == "test"
 
 # Chunkr Python Client
 

{chunkr_ai-0.0.7 → chunkr_ai-0.0.9}/src/chunkr_ai.egg-info/SOURCES.txt

@@ -19,4 +19,6 @@ src/chunkr_ai/api/config.py
 src/chunkr_ai/api/misc.py
 src/chunkr_ai/api/protocol.py
 src/chunkr_ai/api/task.py
+src/chunkr_ai/api/task_async.py
+src/chunkr_ai/api/task_base.py
 tests/test_chunkr.py

chunkr_ai-0.0.9/src/chunkr_ai.egg-info/requires.txt

@@ -0,0 +1,10 @@
+httpx>=0.24.0
+pillow>=10.0.0
+pydantic>=2.0.0
+pytest-asyncio>=0.21.0
+python-dotenv>=0.19.0
+requests>=2.28.0
+
+[test]
+pytest>=7.0.0
+pytest-xdist>=3.0.0

{chunkr_ai-0.0.7 → chunkr_ai-0.0.9}/tests/test_chunkr.py

@@ -19,7 +19,7 @@ from chunkr_ai.models import (
 
 @pytest.fixture(params=[
     pytest.param(("sync", Chunkr()), id="sync"),
-    # pytest.param(("async", ChunkrAsync()), id="async")
+    pytest.param(("async", ChunkrAsync()), id="async")
 ])
 def chunkr_client(request):
     return request.param
@@ -206,9 +206,9 @@ async def test_json_schema(chunkr_client, sample_path):
     
     assert isinstance(response, TaskResponse)
     assert response.task_id is not None
-    assert response.status == "Succeeded"
+    if response.status != "Succeeded":
+        raise ValueError(f"Task failed with message: {response.message}")
     assert response.output is not None
-    
 @pytest.mark.asyncio
 async def test_delete_task(chunkr_client, sample_path):
     client_type, client = chunkr_client

chunkr_ai-0.0.7/src/chunkr_ai.egg-info/requires.txt

@@ -1,10 +0,0 @@
-httpx>=0.28.1
-pillow>=11.1.0
-pydantic>=2.10.4
-pytest-asyncio>=0.25.2
-python-dotenv>=1.0.1
-requests>=2.32.3
-
-[test]
-pytest>=8.3.4
-pytest-xdist>=3.6.1