chunkr-ai 0.1.0a1__py3-none-any.whl → 0.1.0a3__py3-none-any.whl

chunkr_ai/_client.py

@@ -21,7 +21,7 @@ from ._types import (
 )
 from ._utils import is_given, get_async_library
 from ._version import __version__
-from .resources import task, health
+from .resources import files, health
 from ._streaming import Stream as Stream, AsyncStream as AsyncStream
 from ._exceptions import ChunkrError, APIStatusError
 from ._base_client import (
@@ -29,12 +29,14 @@ from ._base_client import (
     SyncAPIClient,
     AsyncAPIClient,
 )
+from .resources.tasks import tasks
 
 __all__ = ["Timeout", "Transport", "ProxiesTypes", "RequestOptions", "Chunkr", "AsyncChunkr", "Client", "AsyncClient"]
 
 
 class Chunkr(SyncAPIClient):
-    task: task.TaskResource
+    tasks: tasks.TasksResource
+    files: files.FilesResource
     health: health.HealthResource
     with_raw_response: ChunkrWithRawResponse
     with_streaming_response: ChunkrWithStreamedResponse
@@ -93,7 +95,8 @@ class Chunkr(SyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
-        self.task = task.TaskResource(self)
+        self.tasks = tasks.TasksResource(self)
+        self.files = files.FilesResource(self)
         self.health = health.HealthResource(self)
         self.with_raw_response = ChunkrWithRawResponse(self)
         self.with_streaming_response = ChunkrWithStreamedResponse(self)
@@ -204,7 +207,8 @@ class Chunkr(SyncAPIClient):
 
 
 class AsyncChunkr(AsyncAPIClient):
-    task: task.AsyncTaskResource
+    tasks: tasks.AsyncTasksResource
+    files: files.AsyncFilesResource
     health: health.AsyncHealthResource
     with_raw_response: AsyncChunkrWithRawResponse
     with_streaming_response: AsyncChunkrWithStreamedResponse
@@ -263,7 +267,8 @@ class AsyncChunkr(AsyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
-        self.task = task.AsyncTaskResource(self)
+        self.tasks = tasks.AsyncTasksResource(self)
+        self.files = files.AsyncFilesResource(self)
         self.health = health.AsyncHealthResource(self)
         self.with_raw_response = AsyncChunkrWithRawResponse(self)
         self.with_streaming_response = AsyncChunkrWithStreamedResponse(self)
@@ -375,25 +380,29 @@ class AsyncChunkr(AsyncAPIClient):
 
 class ChunkrWithRawResponse:
     def __init__(self, client: Chunkr) -> None:
-        self.task = task.TaskResourceWithRawResponse(client.task)
+        self.tasks = tasks.TasksResourceWithRawResponse(client.tasks)
+        self.files = files.FilesResourceWithRawResponse(client.files)
         self.health = health.HealthResourceWithRawResponse(client.health)
 
 
 class AsyncChunkrWithRawResponse:
     def __init__(self, client: AsyncChunkr) -> None:
-        self.task = task.AsyncTaskResourceWithRawResponse(client.task)
+        self.tasks = tasks.AsyncTasksResourceWithRawResponse(client.tasks)
+        self.files = files.AsyncFilesResourceWithRawResponse(client.files)
         self.health = health.AsyncHealthResourceWithRawResponse(client.health)
 
 
 class ChunkrWithStreamedResponse:
     def __init__(self, client: Chunkr) -> None:
-        self.task = task.TaskResourceWithStreamingResponse(client.task)
+        self.tasks = tasks.TasksResourceWithStreamingResponse(client.tasks)
+        self.files = files.FilesResourceWithStreamingResponse(client.files)
         self.health = health.HealthResourceWithStreamingResponse(client.health)
 
 
 class AsyncChunkrWithStreamedResponse:
     def __init__(self, client: AsyncChunkr) -> None:
-        self.task = task.AsyncTaskResourceWithStreamingResponse(client.task)
+        self.tasks = tasks.AsyncTasksResourceWithStreamingResponse(client.tasks)
+        self.files = files.AsyncFilesResourceWithStreamingResponse(client.files)
         self.health = health.AsyncHealthResourceWithStreamingResponse(client.health)
 
 

chunkr_ai/_files.py

@@ -34,7 +34,7 @@ def assert_is_file_content(obj: object, *, key: str | None = None) -> None:
     if not is_file_content(obj):
         prefix = f"Expected entry at `{key}`" if key is not None else f"Expected file input `{obj!r}`"
         raise RuntimeError(
-            f"{prefix} to be bytes, an io.IOBase instance, PathLike or a tuple but received {type(obj)} instead."
+            f"{prefix} to be bytes, an io.IOBase instance, PathLike or a tuple but received {type(obj)} instead. See https://github.com/lumina-ai-inc/chunkr-python/tree/main#file-uploads"
         ) from None
 
 

chunkr_ai/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "chunkr_ai"
-__version__ = "0.1.0-alpha.1"  # x-release-please-version
+__version__ = "0.1.0-alpha.3"  # x-release-please-version

chunkr_ai/pagination.py

@@ -6,7 +6,7 @@ from typing_extensions import override
 
 from ._base_client import BasePage, PageInfo, BaseSyncPage, BaseAsyncPage
 
-__all__ = ["SyncTasksPage", "AsyncTasksPage"]
+__all__ = ["SyncTasksPage", "AsyncTasksPage", "SyncFilesPage", "AsyncFilesPage"]
 
 _T = TypeVar("_T")
 
@@ -69,3 +69,63 @@ class AsyncTasksPage(BaseAsyncPage[_T], BasePage[_T], Generic[_T]):
             return None
 
         return PageInfo(params={"cursor": next_cursor})
+
+
+class SyncFilesPage(BaseSyncPage[_T], BasePage[_T], Generic[_T]):
+    files: List[_T]
+    next_cursor: Optional[datetime] = None
+    has_more: Optional[bool] = None
+    """Whether there are more pages available"""
+
+    @override
+    def _get_page_items(self) -> List[_T]:
+        files = self.files
+        if not files:
+            return []
+        return files
+
+    @override
+    def has_next_page(self) -> bool:
+        has_more = self.has_more
+        if has_more is not None and has_more is False:
+            return False
+
+        return super().has_next_page()
+
+    @override
+    def next_page_info(self) -> Optional[PageInfo]:
+        next_cursor = self.next_cursor
+        if not next_cursor:
+            return None
+
+        return PageInfo(params={"cursor": next_cursor})
+
+
+class AsyncFilesPage(BaseAsyncPage[_T], BasePage[_T], Generic[_T]):
+    files: List[_T]
+    next_cursor: Optional[datetime] = None
+    has_more: Optional[bool] = None
+    """Whether there are more pages available"""
+
+    @override
+    def _get_page_items(self) -> List[_T]:
+        files = self.files
+        if not files:
+            return []
+        return files
+
+    @override
+    def has_next_page(self) -> bool:
+        has_more = self.has_more
+        if has_more is not None and has_more is False:
+            return False
+
+        return super().has_next_page()
+
+    @override
+    def next_page_info(self) -> Optional[PageInfo]:
+        next_cursor = self.next_cursor
+        if not next_cursor:
+            return None
+
+        return PageInfo(params={"cursor": next_cursor})

chunkr_ai/resources/__init__.py

@@ -1,12 +1,20 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from .task import (
-    TaskResource,
-    AsyncTaskResource,
-    TaskResourceWithRawResponse,
-    AsyncTaskResourceWithRawResponse,
-    TaskResourceWithStreamingResponse,
-    AsyncTaskResourceWithStreamingResponse,
+from .files import (
+    FilesResource,
+    AsyncFilesResource,
+    FilesResourceWithRawResponse,
+    AsyncFilesResourceWithRawResponse,
+    FilesResourceWithStreamingResponse,
+    AsyncFilesResourceWithStreamingResponse,
+)
+from .tasks import (
+    TasksResource,
+    AsyncTasksResource,
+    TasksResourceWithRawResponse,
+    AsyncTasksResourceWithRawResponse,
+    TasksResourceWithStreamingResponse,
+    AsyncTasksResourceWithStreamingResponse,
 )
 from .health import (
     HealthResource,
@@ -18,12 +26,18 @@ from .health import (
 )
 
 __all__ = [
-    "TaskResource",
-    "AsyncTaskResource",
-    "TaskResourceWithRawResponse",
-    "AsyncTaskResourceWithRawResponse",
-    "TaskResourceWithStreamingResponse",
-    "AsyncTaskResourceWithStreamingResponse",
+    "TasksResource",
+    "AsyncTasksResource",
+    "TasksResourceWithRawResponse",
+    "AsyncTasksResourceWithRawResponse",
+    "TasksResourceWithStreamingResponse",
+    "AsyncTasksResourceWithStreamingResponse",
+    "FilesResource",
+    "AsyncFilesResource",
+    "FilesResourceWithRawResponse",
+    "AsyncFilesResourceWithRawResponse",
+    "FilesResourceWithStreamingResponse",
+    "AsyncFilesResourceWithStreamingResponse",
     "HealthResource",
     "AsyncHealthResource",
     "HealthResourceWithRawResponse",