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

chunkr_ai/__init__.py

@@ -72,6 +72,9 @@ __all__ = [
 ]
 
 if not _t.TYPE_CHECKING:
+    # Load custom helpers that monkey-patch generated types.
+    # This keeps custom code separate from generated files, per Stainless guidance.
+    from .lib import tasks_poll as _tasks_poll  # noqa: F401
     from ._utils._resources_proxy import resources as resources
 
 _setup_logging()

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.3"  # x-release-please-version
+__version__ = "0.1.0-alpha.4"  # x-release-please-version

chunkr_ai/lib/tasks_poll.py

@@ -0,0 +1,122 @@
+from __future__ import annotations
+
+"""
+Custom helpers for task polling.
+
+This module adds `Task.poll()` and `Task.apoll()` methods at runtime to the
+generated `Task` model, without modifying generated code directly.
+
+Usage:
+    task = client.tasks.get(task_id)
+    task = task.poll(client)  # blocks until terminal state
+
+    # async
+    task = await async_client.tasks.get(task_id)
+    task = await task.apoll(async_client)
+"""
+
+import time
+import asyncio
+from typing import Protocol, cast
+
+from .._types import NOT_GIVEN, NotGiven
+from .._client import Chunkr, AsyncChunkr
+from ..types.task import Task as _Task
+from .._exceptions import ChunkrError
+
+TERMINAL_STATUSES = {"Succeeded", "Failed", "Cancelled"}
+
+
+def _task_poll(
+    self: _Task,
+    client: Chunkr,
+    *,
+    interval: float = 0.5,
+    timeout: float = 600.0,
+    include_chunks: bool | NotGiven = NOT_GIVEN,
+    base64_urls: bool | NotGiven = NOT_GIVEN,
+) -> _Task:
+    """Poll the task until it reaches a terminal status.
+
+    Args:
+        client: Synchronous Chunkr client instance.
+        interval: Seconds to sleep between polls.
+        timeout: Maximum total seconds to wait before raising an error.
+        include_chunks: Whether to include chunks in the output response for each poll.
+        base64_urls: Whether to return base64 encoded URLs.
+    """
+    start_time = time.monotonic()
+    current: _Task = self
+
+    class _TasksGetProtocol(Protocol):
+        def get(
+            self,
+            task_id: str,
+            *,
+            base64_urls: bool | NotGiven = NOT_GIVEN,
+            include_chunks: bool | NotGiven = NOT_GIVEN,
+        ) -> _Task: ...
+
+    resource = cast(_TasksGetProtocol, client.tasks)
+
+    while current.status not in TERMINAL_STATUSES:
+        if time.monotonic() - start_time > timeout:
+            raise ChunkrError("Task polling timed out.")
+
+        if interval > 0:
+            time.sleep(interval)
+
+        current = resource.get(
+            current.task_id,
+            include_chunks=include_chunks,
+            base64_urls=base64_urls,
+        )
+
+    return current
+
+
+async def _task_apoll(
+    self: _Task,
+    client: AsyncChunkr,
+    *,
+    interval: float = 0.5,
+    timeout: float = 600.0,
+    include_chunks: bool | NotGiven = NOT_GIVEN,
+    base64_urls: bool | NotGiven = NOT_GIVEN,
+) -> _Task:
+    """Async poll the task until it reaches a terminal status."""
+    start_time = time.monotonic()
+    current: _Task = self
+
+    class _AsyncTasksGetProtocol(Protocol):
+        async def get(
+            self,
+            task_id: str,
+            *,
+            base64_urls: bool | NotGiven = NOT_GIVEN,
+            include_chunks: bool | NotGiven = NOT_GIVEN,
+        ) -> _Task: ...
+
+    aresource = cast(_AsyncTasksGetProtocol, client.tasks)
+
+    while current.status not in TERMINAL_STATUSES:
+        if time.monotonic() - start_time > timeout:
+            raise ChunkrError("Task polling timed out.")
+
+        if interval > 0:
+            await asyncio.sleep(interval)
+
+        current = await aresource.get(
+            current.task_id,
+            include_chunks=include_chunks,
+            base64_urls=base64_urls,
+        )
+
+    return current
+
+
+# Attach methods to the generated Task model
+_Task.poll = _task_poll  # type: ignore[attr-defined]
+_Task.apoll = _task_apoll  # type: ignore[attr-defined]
+
+

chunkr_ai/types/__init__.py

@@ -11,5 +11,5 @@ from .task_get_params import TaskGetParams as TaskGetParams
 from .file_list_params import FileListParams as FileListParams
 from .task_list_params import TaskListParams as TaskListParams
 from .file_create_params import FileCreateParams as FileCreateParams
-from .files_page_response import FilesPageResponse as FilesPageResponse
+from .files_list_response import FilesListResponse as FilesListResponse
 from .health_check_response import HealthCheckResponse as HealthCheckResponse

chunkr_ai/types/{files_page_response.py → files_list_response.py}

@@ -6,10 +6,10 @@ from datetime import datetime
 from .file import File
 from .._models import BaseModel
 
-__all__ = ["FilesPageResponse"]
+__all__ = ["FilesListResponse"]
 
 
-class FilesPageResponse(BaseModel):
+class FilesListResponse(BaseModel):
     files: List[File]
     """List of files"""
 

{chunkr_ai-0.1.0a3.dist-info → chunkr_ai-0.1.0a4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: chunkr-ai
-Version: 0.1.0a3
+Version: 0.1.0a4
 Summary: The official Python library for the chunkr API
 Project-URL: Homepage, https://github.com/lumina-ai-inc/chunkr-python
 Project-URL: Repository, https://github.com/lumina-ai-inc/chunkr-python

{chunkr_ai-0.1.0a3.dist-info → chunkr_ai-0.1.0a4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-chunkr_ai/__init__.py,sha256=scS30uHiCpLbaalKTAJSCFSTqnu_b9R5JCkTu2hmbzU,2587
+chunkr_ai/__init__.py,sha256=RqteAJ-1Ma7DnoNz_AJJIjmynGeoGJ3F4ZKrSnjp9zs,2793
 chunkr_ai/_base_client.py,sha256=Nv5b_rmVdmmPbF42mlOfymbSC6lxcYsrsvBhKSBDXWQ,67038
 chunkr_ai/_client.py,sha256=fseZHGtnXGw3uSa1Le8SxH2oSBeHczn6mOsLeLGj4rY,15867
 chunkr_ai/_compat.py,sha256=VWemUKbj6DDkQ-O4baSpHVLJafotzeXmCQGJugfVTIw,6580
@@ -11,7 +11,7 @@ chunkr_ai/_resource.py,sha256=f5tiwjxcKdbeMor8idoHtMFTUhqD9yc2xXtq5rqeLLk,1100
 chunkr_ai/_response.py,sha256=xXNpF53hiYARmAW7npKuxQ5UHAEjgAzm7ME_L3eIstY,28800
 chunkr_ai/_streaming.py,sha256=ZmyrVWk7-AWkLAATR55WgNxnyFzYmaqJt2LthA_PTqQ,10100
 chunkr_ai/_types.py,sha256=dnzU2Q2tLcuk29QFEcnPC1wp0-4XB4Cpef_3AnRhV5Y,6200
-chunkr_ai/_version.py,sha256=ZlaSb0jnVJ7N4kkTfvr8UrgWy--cDr9CwSzUZLsiWbo,169
+chunkr_ai/_version.py,sha256=hJYiv4ePWLGN-Ur1VkK5zJERczdAZjPDNh7APrmHgBE,169
 chunkr_ai/pagination.py,sha256=bT-ErcJ80YlKBV6tWq2s9uqg-wv7o66SKe_AgUAGrKc,3533
 chunkr_ai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chunkr_ai/_utils/__init__.py,sha256=PNZ_QJuzZEgyYXqkO1HVhGkj5IU9bglVUcw7H-Knjzw,2062
@@ -25,20 +25,21 @@ chunkr_ai/_utils/_transform.py,sha256=n7kskEWz6o__aoNvhFoGVyDoalNe6mJwp-g7BWkdj8
 chunkr_ai/_utils/_typing.py,sha256=D0DbbNu8GnYQTSICnTSHDGsYXj8TcAKyhejb0XcnjtY,4602
 chunkr_ai/_utils/_utils.py,sha256=ts4CiiuNpFiGB6YMdkQRh2SZvYvsl7mAF-JWHCcLDf4,12312
 chunkr_ai/lib/.keep,sha256=wuNrz-5SXo3jJaJOJgz4vFHM41YH_g20F5cRQo0vLes,224
+chunkr_ai/lib/tasks_poll.py,sha256=3yosl_hH5j6NVNH9mANqneAW0FJSbIV9dMoTcF-OdJU,3341
 chunkr_ai/resources/__init__.py,sha256=K-axuAEg2pJQl45N5ao1tm8AnRwpQVVNp_b6qSMgB6A,1426
 chunkr_ai/resources/files.py,sha256=Dez080pD_xUr1jOW3y6QSg92sSSZhEYObPze2RWktoY,26304
 chunkr_ai/resources/health.py,sha256=XTvUtRs5hEK-uccb_40mcIex85eEUo1a171nQUjpSOs,4965
 chunkr_ai/resources/tasks/__init__.py,sha256=W-sclAx_Kfm7OBGlSs694QzNCMkewtz9LU9KRcb8Ud0,976
 chunkr_ai/resources/tasks/parse.py,sha256=um0sw2ZU7bY6AK7LKAS0GxAHUyuSzav_NlhxXPjNjxY,28491
 chunkr_ai/resources/tasks/tasks.py,sha256=UC15zZNpY7u85X_JJudDuNrnpaeULiISaBde4BRHGSw,21653
-chunkr_ai/types/__init__.py,sha256=70x8dXFkNKZJZMQyHDswdY2c-gppLNAda9aqGksZwjM,728
+chunkr_ai/types/__init__.py,sha256=DSRAMgXVRTZM2t8s2yrFU-FHt3FTs_wpZfVILH1zjJ0,728
 chunkr_ai/types/delete.py,sha256=EU78fjXpc8-fqvgcFTuJ0ejs5u_UjbhOz5frkeUHvxY,225
 chunkr_ai/types/file.py,sha256=kOxR0g-3A-qOxz2cjuTcq0wFMqPoph9uQuLYQ56zb-c,718
 chunkr_ai/types/file_create_params.py,sha256=eR5tUPv1ZxGk94y6Ps5PDF5AoHsTfPRM9jNxnrEEnCQ,440
 chunkr_ai/types/file_list_params.py,sha256=oJGTf88aAxBhNfmQDbxGT63b95HdSbMXUubKjXM22_U,822
 chunkr_ai/types/file_url.py,sha256=L434WnOXkNmt59dJiaAgT1_3pN3BIsxm2q14zHQK6xY,365
 chunkr_ai/types/file_url_params.py,sha256=ZHfKiy_6B25StdDemulavGcsPggNNMKLWf6KN7xfPTY,413
-chunkr_ai/types/files_page_response.py,sha256=N0Wwr8NeDxF-AX4Zs1HYDMeFoxJ8jxkFoCADgt6bZik,506
+chunkr_ai/types/files_list_response.py,sha256=ggSRWhTzZWjcDXxStyCzrYICXXB5TqnL2j-SN9mHH_g,506
 chunkr_ai/types/health_check_response.py,sha256=6Zn5YYHCQf2RgMjDlf39mtiTPqfaBfC9Vv599U_rKCI,200
 chunkr_ai/types/task.py,sha256=aew6aT0ngKtwgfUCSCvMTJOBQL1Xp0F0otB_wxumIGQ,46703
 chunkr_ai/types/task_get_params.py,sha256=Nx2luhebcoaiuRln4KP4FarWvBPd1OYi__efi56zHPM,460
@@ -46,7 +47,7 @@ chunkr_ai/types/task_list_params.py,sha256=fCku42QW6QUsLmZgKJBaxisGvUcmcQ5fa6LgH
 chunkr_ai/types/tasks/__init__.py,sha256=VdLEmQvgPoiykSEYaRhkMYVaIueGDkR4P_MjCq9SbQY,267
 chunkr_ai/types/tasks/parse_create_params.py,sha256=PBg2VR_OnBdB8K4NihuefGJXgUXBn7v5317LZG7PDks,34340
 chunkr_ai/types/tasks/parse_update_params.py,sha256=B1cKfdX_cNDh0m2zDoH0FiZP_Qc-a5GFy-5iXHDHuy8,34113
-chunkr_ai-0.1.0a3.dist-info/METADATA,sha256=1KrH5ko1te-nf_LOutXz2FeuCG8BErWw2WdXFDrXMKc,16441
-chunkr_ai-0.1.0a3.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-chunkr_ai-0.1.0a3.dist-info/licenses/LICENSE,sha256=3FDRL-L-DFkrFy8yJpb1Nxhuztm0PB2kawcCgK5utFg,11336
-chunkr_ai-0.1.0a3.dist-info/RECORD,,
+chunkr_ai-0.1.0a4.dist-info/METADATA,sha256=241RRJb1pTZBg9IG3oGx3_WASrJDN8a2myyJhQ9TUNE,16441
+chunkr_ai-0.1.0a4.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+chunkr_ai-0.1.0a4.dist-info/licenses/LICENSE,sha256=3FDRL-L-DFkrFy8yJpb1Nxhuztm0PB2kawcCgK5utFg,11336
+chunkr_ai-0.1.0a4.dist-info/RECORD,,