chunkr-ai 0.1.0a2__tar.gz → 0.1.0a4__tar.gz

{chunkr_ai-0.1.0a2 → chunkr_ai-0.1.0a4}/.gitignore

@@ -13,3 +13,5 @@ dist
 .envrc
 codegen.log
 Brewfile.lock.json
+
+.lh

chunkr_ai-0.1.0a4/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.1.0-alpha.4"
+}

{chunkr_ai-0.1.0a2 → chunkr_ai-0.1.0a4}/CHANGELOG.md

@@ -1,5 +1,38 @@
 # Changelog
 
+## 0.1.0-alpha.4 (2025-08-09)
+
+Full Changelog: [v0.1.0-alpha.3...v0.1.0-alpha.4](https://github.com/lumina-ai-inc/chunkr-python/compare/v0.1.0-alpha.3...v0.1.0-alpha.4)
+
+### Features
+
+* **api:** added poll() and apoll() methods to Task ([#16](https://github.com/lumina-ai-inc/chunkr-python/issues/16)) ([da329dd](https://github.com/lumina-ai-inc/chunkr-python/commit/da329dda8ac0818e2b770e156b6b69456a5cfc72))
+* **api:** manual updates ([a704c5d](https://github.com/lumina-ai-inc/chunkr-python/commit/a704c5deed4c91af239afb58fce99091f8784210))
+
+## 0.1.0-alpha.3 (2025-08-09)
+
+Full Changelog: [v0.1.0-alpha.2...v0.1.0-alpha.3](https://github.com/lumina-ai-inc/chunkr-python/compare/v0.1.0-alpha.2...v0.1.0-alpha.3)
+
+### Features
+
+* **api:** api update ([a00a95e](https://github.com/lumina-ai-inc/chunkr-python/commit/a00a95ec22d7141e9139dcdeac65f882e9be0d39))
+* **api:** api update ([145a438](https://github.com/lumina-ai-inc/chunkr-python/commit/145a43891e138f9b986815b105ce65ea2ebaaf17))
+* **api:** api update ([79170d4](https://github.com/lumina-ai-inc/chunkr-python/commit/79170d4b086300e3beb6cb536c574ba889164ee2))
+* **api:** api update ([29fe5ea](https://github.com/lumina-ai-inc/chunkr-python/commit/29fe5ea3cc4de3c097ea446927bc499848cd5a2d))
+* **api:** manual updates ([171e2f9](https://github.com/lumina-ai-inc/chunkr-python/commit/171e2f9046863086957002cb2eb77e726b2cac6e))
+* **api:** manual updates ([62d6d6e](https://github.com/lumina-ai-inc/chunkr-python/commit/62d6d6e39eb6b5a5dde6c2e2e9f5caa4a01c2828))
+* **api:** manual updates ([16190d7](https://github.com/lumina-ai-inc/chunkr-python/commit/16190d7c726aad97a749e3c606fa412f4f5fd191))
+* **api:** manual updates ([5d73fd1](https://github.com/lumina-ai-inc/chunkr-python/commit/5d73fd1fd0368c56777a69992c302d5ee088a8f1))
+* **api:** manual updates ([9269fd1](https://github.com/lumina-ai-inc/chunkr-python/commit/9269fd1f81567aa69a17a4d4b01adad8d2866e3b))
+* **api:** manual updates ([ac4cd88](https://github.com/lumina-ai-inc/chunkr-python/commit/ac4cd883378d3f4fb89df50fc4eac8d94655648d))
+* **api:** manual updates ([f29e11a](https://github.com/lumina-ai-inc/chunkr-python/commit/f29e11ad5014311faba9162e70a6aae40003c994))
+
+
+### Chores
+
+* **internal:** update comment in script ([f44c608](https://github.com/lumina-ai-inc/chunkr-python/commit/f44c60889e1b26b85c142aa318a6f7801ae87241))
+* update @stainless-api/prism-cli to v5.15.0 ([c0aab83](https://github.com/lumina-ai-inc/chunkr-python/commit/c0aab83a4ea61ea57f0424b05f19bbf46efb6ca0))
+
 ## 0.1.0-alpha.2 (2025-08-06)
 
 Full Changelog: [v0.1.0-alpha.1...v0.1.0-alpha.2](https://github.com/lumina-ai-inc/chunkr-python/compare/v0.1.0-alpha.1...v0.1.0-alpha.2)

{chunkr_ai-0.1.0a2 → chunkr_ai-0.1.0a4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: chunkr-ai
-Version: 0.1.0a2
+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
@@ -70,7 +70,7 @@ client = Chunkr(
     api_key=os.environ.get("CHUNKR_API_KEY"),  # This is the default and can be omitted
 )
 
-task = client.task.parse.create(
+task = client.tasks.parse.create(
     file="string",
 )
 print(task.task_id)
@@ -96,7 +96,7 @@ client = AsyncChunkr(
 
 
 async def main() -> None:
-    task = await client.task.parse.create(
+    task = await client.tasks.parse.create(
         file="string",
     )
     print(task.task_id)
@@ -131,7 +131,7 @@ async def main() -> None:
         api_key="My API Key",
         http_client=DefaultAioHttpClient(),
     ) as client:
-        task = await client.task.parse.create(
+        task = await client.tasks.parse.create(
             file="string",
         )
         print(task.task_id)
@@ -162,7 +162,7 @@ client = Chunkr()
 
 all_tasks = []
 # Automatically fetches more pages as needed.
-for task in client.task.list(
+for task in client.tasks.list(
     limit=100,
 ):
     # Do something with task here
@@ -182,7 +182,7 @@ client = AsyncChunkr()
 async def main() -> None:
     all_tasks = []
     # Iterate through items across all pages, issuing requests as needed.
-    async for task in client.task.list(
+    async for task in client.tasks.list(
         limit=100,
     ):
         all_tasks.append(task)
@@ -195,7 +195,7 @@ asyncio.run(main())
 Alternatively, you can use the `.has_next_page()`, `.next_page_info()`, or `.get_next_page()` methods for more granular control working with pages:
 
 ```python
-first_page = await client.task.list(
+first_page = await client.tasks.list(
     limit=100,
 )
 if first_page.has_next_page():
@@ -209,7 +209,7 @@ if first_page.has_next_page():
 Or just work directly with the returned data:
 
 ```python
-first_page = await client.task.list(
+first_page = await client.tasks.list(
     limit=100,
 )
 
@@ -229,13 +229,31 @@ from chunkr_ai import Chunkr
 
 client = Chunkr()
 
-task = client.task.parse.create(
+task = client.tasks.parse.create(
     file="file",
     chunk_processing={},
 )
 print(task.chunk_processing)
 ```
 
+## File uploads
+
+Request parameters that correspond to file uploads can be passed as `bytes`, or a [`PathLike`](https://docs.python.org/3/library/os.html#os.PathLike) instance or a tuple of `(filename, contents, media type)`.
+
+```python
+from pathlib import Path
+from chunkr_ai import Chunkr
+
+client = Chunkr()
+
+client.files.create(
+    file=Path("/path/to/file"),
+    file_metadata="file_metadata",
+)
+```
+
+The async client uses the exact same interface. If you pass a [`PathLike`](https://docs.python.org/3/library/os.html#os.PathLike) instance, the file contents will be read asynchronously automatically.
+
 ## Handling errors
 
 When the library is unable to connect to the API (for example, due to network connection problems or a timeout), a subclass of `chunkr_ai.APIConnectionError` is raised.
@@ -252,8 +270,8 @@ from chunkr_ai import Chunkr
 client = Chunkr()
 
 try:
-    client.task.parse.create(
-        file="base64 or url",
+    client.tasks.parse.create(
+        file="string",
     )
 except chunkr_ai.APIConnectionError as e:
     print("The server could not be reached")
@@ -297,8 +315,8 @@ client = Chunkr(
 )
 
 # Or, configure per-request:
-client.with_options(max_retries=5).task.parse.create(
-    file="base64 or url",
+client.with_options(max_retries=5).tasks.parse.create(
+    file="string",
 )
 ```
 
@@ -322,8 +340,8 @@ client = Chunkr(
 )
 
 # Override per-request:
-client.with_options(timeout=5.0).task.parse.create(
-    file="base64 or url",
+client.with_options(timeout=5.0).tasks.parse.create(
+    file="string",
 )
 ```
 
@@ -365,12 +383,12 @@ The "raw" Response object can be accessed by prefixing `.with_raw_response.` to
 from chunkr_ai import Chunkr
 
 client = Chunkr()
-response = client.task.parse.with_raw_response.create(
-    file="base64 or url",
+response = client.tasks.parse.with_raw_response.create(
+    file="string",
 )
 print(response.headers.get('X-My-Header'))
 
-parse = response.parse()  # get the object that `task.parse.create()` would have returned
+parse = response.parse()  # get the object that `tasks.parse.create()` would have returned
 print(parse.task_id)
 ```
 
@@ -385,8 +403,8 @@ The above interface eagerly reads the full response body when you make the reque
 To stream the response body, use `.with_streaming_response` instead, which requires a context manager and only reads the response body once you call `.read()`, `.text()`, `.json()`, `.iter_bytes()`, `.iter_text()`, `.iter_lines()` or `.parse()`. In the async client, these are async methods.
 
 ```python
-with client.task.parse.with_streaming_response.create(
-    file="base64 or url",
+with client.tasks.parse.with_streaming_response.create(
+    file="string",
 ) as response:
     print(response.headers.get("X-My-Header"))
 

{chunkr_ai-0.1.0a2 → chunkr_ai-0.1.0a4}/README.md

@@ -35,7 +35,7 @@ client = Chunkr(
     api_key=os.environ.get("CHUNKR_API_KEY"),  # This is the default and can be omitted
 )
 
-task = client.task.parse.create(
+task = client.tasks.parse.create(
     file="string",
 )
 print(task.task_id)
@@ -61,7 +61,7 @@ client = AsyncChunkr(
 
 
 async def main() -> None:
-    task = await client.task.parse.create(
+    task = await client.tasks.parse.create(
         file="string",
     )
     print(task.task_id)
@@ -96,7 +96,7 @@ async def main() -> None:
         api_key="My API Key",
         http_client=DefaultAioHttpClient(),
     ) as client:
-        task = await client.task.parse.create(
+        task = await client.tasks.parse.create(
             file="string",
         )
         print(task.task_id)
@@ -127,7 +127,7 @@ client = Chunkr()
 
 all_tasks = []
 # Automatically fetches more pages as needed.
-for task in client.task.list(
+for task in client.tasks.list(
     limit=100,
 ):
     # Do something with task here
@@ -147,7 +147,7 @@ client = AsyncChunkr()
 async def main() -> None:
     all_tasks = []
     # Iterate through items across all pages, issuing requests as needed.
-    async for task in client.task.list(
+    async for task in client.tasks.list(
         limit=100,
     ):
         all_tasks.append(task)
@@ -160,7 +160,7 @@ asyncio.run(main())
 Alternatively, you can use the `.has_next_page()`, `.next_page_info()`, or `.get_next_page()` methods for more granular control working with pages:
 
 ```python
-first_page = await client.task.list(
+first_page = await client.tasks.list(
     limit=100,
 )
 if first_page.has_next_page():
@@ -174,7 +174,7 @@ if first_page.has_next_page():
 Or just work directly with the returned data:
 
 ```python
-first_page = await client.task.list(
+first_page = await client.tasks.list(
     limit=100,
 )
 
@@ -194,13 +194,31 @@ from chunkr_ai import Chunkr
 
 client = Chunkr()
 
-task = client.task.parse.create(
+task = client.tasks.parse.create(
     file="file",
     chunk_processing={},
 )
 print(task.chunk_processing)
 ```
 
+## File uploads
+
+Request parameters that correspond to file uploads can be passed as `bytes`, or a [`PathLike`](https://docs.python.org/3/library/os.html#os.PathLike) instance or a tuple of `(filename, contents, media type)`.
+
+```python
+from pathlib import Path
+from chunkr_ai import Chunkr
+
+client = Chunkr()
+
+client.files.create(
+    file=Path("/path/to/file"),
+    file_metadata="file_metadata",
+)
+```
+
+The async client uses the exact same interface. If you pass a [`PathLike`](https://docs.python.org/3/library/os.html#os.PathLike) instance, the file contents will be read asynchronously automatically.
+
 ## Handling errors
 
 When the library is unable to connect to the API (for example, due to network connection problems or a timeout), a subclass of `chunkr_ai.APIConnectionError` is raised.
@@ -217,8 +235,8 @@ from chunkr_ai import Chunkr
 client = Chunkr()
 
 try:
-    client.task.parse.create(
-        file="base64 or url",
+    client.tasks.parse.create(
+        file="string",
     )
 except chunkr_ai.APIConnectionError as e:
     print("The server could not be reached")
@@ -262,8 +280,8 @@ client = Chunkr(
 )
 
 # Or, configure per-request:
-client.with_options(max_retries=5).task.parse.create(
-    file="base64 or url",
+client.with_options(max_retries=5).tasks.parse.create(
+    file="string",
 )
 ```
 
@@ -287,8 +305,8 @@ client = Chunkr(
 )
 
 # Override per-request:
-client.with_options(timeout=5.0).task.parse.create(
-    file="base64 or url",
+client.with_options(timeout=5.0).tasks.parse.create(
+    file="string",
 )
 ```
 
@@ -330,12 +348,12 @@ The "raw" Response object can be accessed by prefixing `.with_raw_response.` to
 from chunkr_ai import Chunkr
 
 client = Chunkr()
-response = client.task.parse.with_raw_response.create(
-    file="base64 or url",
+response = client.tasks.parse.with_raw_response.create(
+    file="string",
 )
 print(response.headers.get('X-My-Header'))
 
-parse = response.parse()  # get the object that `task.parse.create()` would have returned
+parse = response.parse()  # get the object that `tasks.parse.create()` would have returned
 print(parse.task_id)
 ```
 
@@ -350,8 +368,8 @@ The above interface eagerly reads the full response body when you make the reque
 To stream the response body, use `.with_streaming_response` instead, which requires a context manager and only reads the response body once you call `.read()`, `.text()`, `.json()`, `.iter_bytes()`, `.iter_text()`, `.iter_lines()` or `.parse()`. In the async client, these are async methods.
 
 ```python
-with client.task.parse.with_streaming_response.create(
-    file="base64 or url",
+with client.tasks.parse.with_streaming_response.create(
+    file="string",
 ) as response:
     print(response.headers.get("X-My-Header"))
 

chunkr_ai-0.1.0a4/api.md

@@ -0,0 +1,50 @@
+# Tasks
+
+Types:
+
+```python
+from chunkr_ai.types import Task
+```
+
+Methods:
+
+- <code title="get /tasks">client.tasks.<a href="./src/chunkr_ai/resources/tasks/tasks.py">list</a>(\*\*<a href="src/chunkr_ai/types/task_list_params.py">params</a>) -> <a href="./src/chunkr_ai/types/task.py">SyncTasksPage[Task]</a></code>
+- <code title="delete /tasks/{task_id}">client.tasks.<a href="./src/chunkr_ai/resources/tasks/tasks.py">delete</a>(task_id) -> None</code>
+- <code title="get /tasks/{task_id}/cancel">client.tasks.<a href="./src/chunkr_ai/resources/tasks/tasks.py">cancel</a>(task_id) -> None</code>
+- <code title="get /tasks/{task_id}">client.tasks.<a href="./src/chunkr_ai/resources/tasks/tasks.py">get</a>(task_id, \*\*<a href="src/chunkr_ai/types/task_get_params.py">params</a>) -> <a href="./src/chunkr_ai/types/task.py">Task</a></code>
+
+## Parse
+
+Methods:
+
+- <code title="post /tasks/parse">client.tasks.parse.<a href="./src/chunkr_ai/resources/tasks/parse.py">create</a>(\*\*<a href="src/chunkr_ai/types/tasks/parse_create_params.py">params</a>) -> <a href="./src/chunkr_ai/types/task.py">Task</a></code>
+- <code title="patch /tasks/parse/{task_id}">client.tasks.parse.<a href="./src/chunkr_ai/resources/tasks/parse.py">update</a>(task_id, \*\*<a href="src/chunkr_ai/types/tasks/parse_update_params.py">params</a>) -> <a href="./src/chunkr_ai/types/task.py">Task</a></code>
+
+# Files
+
+Types:
+
+```python
+from chunkr_ai.types import Delete, File, FilesListResponse, FileURL
+```
+
+Methods:
+
+- <code title="post /files">client.files.<a href="./src/chunkr_ai/resources/files.py">create</a>(\*\*<a href="src/chunkr_ai/types/file_create_params.py">params</a>) -> <a href="./src/chunkr_ai/types/file.py">File</a></code>
+- <code title="get /files">client.files.<a href="./src/chunkr_ai/resources/files.py">list</a>(\*\*<a href="src/chunkr_ai/types/file_list_params.py">params</a>) -> <a href="./src/chunkr_ai/types/file.py">SyncFilesPage[File]</a></code>
+- <code title="delete /files/{file_id}">client.files.<a href="./src/chunkr_ai/resources/files.py">delete</a>(file_id) -> <a href="./src/chunkr_ai/types/delete.py">Delete</a></code>
+- <code title="get /files/{file_id}/content">client.files.<a href="./src/chunkr_ai/resources/files.py">content</a>(file_id) -> None</code>
+- <code title="get /files/{file_id}">client.files.<a href="./src/chunkr_ai/resources/files.py">get</a>(file_id) -> <a href="./src/chunkr_ai/types/file.py">File</a></code>
+- <code title="get /files/{file_id}/url">client.files.<a href="./src/chunkr_ai/resources/files.py">url</a>(file_id, \*\*<a href="src/chunkr_ai/types/file_url_params.py">params</a>) -> <a href="./src/chunkr_ai/types/file_url.py">FileURL</a></code>
+
+# Health
+
+Types:
+
+```python
+from chunkr_ai.types import HealthCheckResponse
+```
+
+Methods:
+
+- <code title="get /health">client.health.<a href="./src/chunkr_ai/resources/health.py">check</a>() -> str</code>

{chunkr_ai-0.1.0a2 → chunkr_ai-0.1.0a4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "chunkr-ai"
-version = "0.1.0-alpha.2"
+version = "0.1.0-alpha.4"
 description = "The official Python library for the chunkr API"
 dynamic = ["readme"]
 license = "Apache-2.0"

{chunkr_ai-0.1.0a2 → chunkr_ai-0.1.0a4}/src/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-0.1.0a2 → chunkr_ai-0.1.0a4}/src/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 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,13 +29,14 @@ from ._base_client import (
     SyncAPIClient,
     AsyncAPIClient,
 )
-from .resources.task import task
+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
@@ -94,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)
@@ -205,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
@@ -264,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)
@@ -376,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-0.1.0a2 → chunkr_ai-0.1.0a4}/src/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-0.1.0a2 → chunkr_ai-0.1.0a4}/src/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.2"  # x-release-please-version
+__version__ = "0.1.0-alpha.4"  # x-release-please-version

chunkr_ai-0.1.0a4/src/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]
+
+