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

chunkr_ai/api/task.py

@@ -1,66 +0,0 @@
-from .config import Configuration
-from .misc import prepare_upload_data
-from .task_base import TaskBase
-import time
-
-
-class TaskResponse(TaskBase):
-    def _poll_request(self) -> dict:
-        while True:
-            try:
-                if not self.task_url:
-                    raise ValueError("Task URL not found in response")
-                if not self._client._session:
-                    raise ValueError("Client session not found")
-                r = self._client._session.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...")
-                time.sleep(0.5)
-            except Exception:
-                raise
-
-    def poll(self) -> "TaskResponse":
-        while True:
-            response = self._poll_request()
-            updated_task = TaskResponse(**response).with_client(self._client)
-            self.__dict__.update(updated_task.__dict__)
-            if result := self._check_status():
-                return result
-            time.sleep(0.5)
-
-    def update(self, config: Configuration) -> "TaskResponse":
-        if not self.task_url:
-            raise ValueError("Task URL not found")
-        if not self._client._session:
-            raise ValueError("Client session not found")
-        files = prepare_upload_data(None, config)
-        r = self._client._session.patch(
-            self.task_url, files=files, headers=self._client._headers()
-        )
-        r.raise_for_status()
-        updated = TaskResponse(**r.json()).with_client(self._client)
-        self.__dict__.update(updated.__dict__)
-        return self.poll()
-
-    def cancel(self):
-        if not self.task_url:
-            raise ValueError("Task URL not found")
-        if not self._client._session:
-            raise ValueError("Client session not found")
-        r = self._client._session.get(
-            f"{self.task_url}/cancel", headers=self._client._headers()
-        )
-        r.raise_for_status()
-        self.poll()
-
-    def delete(self):
-        if not self.task_url:
-            raise ValueError("Task URL not found")
-        if not self._client._session:
-            raise ValueError("Client session not found")
-        r = self._client._session.delete(self.task_url, headers=self._client._headers())
-        r.raise_for_status()

chunkr_ai/api/task_async.py

@@ -1,69 +0,0 @@
-from .config import Configuration
-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.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:
-            raise
-
-    async def poll(self) -> "TaskResponseAsync":
-        if not self.task_url:
-            raise ValueError("Task URL not found")
-        if not self._client._client:
-            raise ValueError("Client not found")
-        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 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.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")
-        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.raise_for_status()
-        return await self.poll()
-
-    async def delete(self):
-        if not self.task_url:
-            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.raise_for_status()

chunkr_ai/api/task_base.py

@@ -1,85 +0,0 @@
-from .config import Configuration, Status, OutputResponse
-from .protocol import ChunkrClientProtocol
-from abc import ABC, abstractmethod
-from typing import TypeVar, Optional, Generic
-from pydantic import BaseModel, PrivateAttr
-from datetime import datetime
-
-T = TypeVar("T", bound="TaskBase")
-
-class TaskBase(BaseModel, ABC, Generic[T]):
-    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[ChunkrClientProtocol] = PrivateAttr(default=None)
-
-    @abstractmethod
-    def _poll_request(self) -> dict:
-        """Helper method to make polling request with retry logic (synchronous)"""
-        pass
-
-    @abstractmethod
-    def poll(self) -> T:
-        """Poll the task for completion."""
-        pass
-
-    @abstractmethod
-    def update(self, config: Configuration) -> T:
-        """Update the task configuration."""
-        pass
-
-    @abstractmethod
-    def cancel(self) -> T:
-        """Cancel the task."""
-        pass
-
-    @abstractmethod
-    def delete(self) -> T:
-        """Delete the task."""
-        pass
-
-    def with_client(self, client: ChunkrClientProtocol) -> T:
-        self._client = client
-        return self
-
-    def _check_status(self) -> Optional[T]:
-        """Helper method to check task status and handle completion/failure"""
-        if self.status == "Failed":
-            raise ValueError(self.message)
-        if self.status not in ("Starting", "Processing"):
-            return self
-        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:
-        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)

chunkr_ai-0.0.16.dist-info/RECORD

@@ -1,21 +0,0 @@
-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,,