chunkr-ai 0.1.0a6__py3-none-any.whl → 0.1.0a7__py3-none-any.whl

chunkr_ai/__init__.py

@@ -24,6 +24,7 @@ from ._exceptions import (
     InternalServerError,
     PermissionDeniedError,
     UnprocessableEntityError,
+    APIWebhookValidationError,
     APIResponseValidationError,
 )
 from ._base_client import DefaultHttpxClient, DefaultAioHttpClient, DefaultAsyncHttpxClient
@@ -45,6 +46,7 @@ __all__ = [
     "APITimeoutError",
     "APIConnectionError",
     "APIResponseValidationError",
+    "APIWebhookValidationError",
     "BadRequestError",
     "AuthenticationError",
     "PermissionDeniedError",

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 files, health
+from .resources import files, health, webhooks
 from ._streaming import Stream as Stream, AsyncStream as AsyncStream
 from ._exceptions import ChunkrError, APIStatusError
 from ._base_client import (
@@ -38,16 +38,19 @@ class Chunkr(SyncAPIClient):
     tasks: tasks.TasksResource
     files: files.FilesResource
     health: health.HealthResource
+    webhooks: webhooks.WebhooksResource
     with_raw_response: ChunkrWithRawResponse
     with_streaming_response: ChunkrWithStreamedResponse
 
     # client options
     api_key: str
+    webhook_key: str | None
 
     def __init__(
         self,
         *,
         api_key: str | None = None,
+        webhook_key: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN,
         max_retries: int = DEFAULT_MAX_RETRIES,
@@ -69,7 +72,9 @@ class Chunkr(SyncAPIClient):
     ) -> None:
         """Construct a new synchronous Chunkr client instance.
 
-        This automatically infers the `api_key` argument from the `CHUNKR_API_KEY` environment variable if it is not provided.
+        This automatically infers the following arguments from their corresponding environment variables if they are not provided:
+        - `api_key` from `CHUNKR_API_KEY`
+        - `webhook_key` from `CHUNKR_WEBHOOK_KEY`
         """
         if api_key is None:
             api_key = os.environ.get("CHUNKR_API_KEY")
@@ -79,6 +84,10 @@ class Chunkr(SyncAPIClient):
             )
         self.api_key = api_key
 
+        if webhook_key is None:
+            webhook_key = os.environ.get("CHUNKR_WEBHOOK_KEY")
+        self.webhook_key = webhook_key
+
         if base_url is None:
             base_url = os.environ.get("CHUNKR_BASE_URL")
         if base_url is None:
@@ -100,6 +109,7 @@ class Chunkr(SyncAPIClient):
         self.tasks = tasks.TasksResource(self)
         self.files = files.FilesResource(self)
         self.health = health.HealthResource(self)
+        self.webhooks = webhooks.WebhooksResource(self)
         self.with_raw_response = ChunkrWithRawResponse(self)
         self.with_streaming_response = ChunkrWithStreamedResponse(self)
 
@@ -127,6 +137,7 @@ class Chunkr(SyncAPIClient):
         self,
         *,
         api_key: str | None = None,
+        webhook_key: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = NOT_GIVEN,
         http_client: httpx.Client | None = None,
@@ -161,6 +172,7 @@ class Chunkr(SyncAPIClient):
         http_client = http_client or self._client
         return self.__class__(
             api_key=api_key or self.api_key,
+            webhook_key=webhook_key or self.webhook_key,
             base_url=base_url or self.base_url,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
             http_client=http_client,
@@ -212,16 +224,19 @@ class AsyncChunkr(AsyncAPIClient):
     tasks: tasks.AsyncTasksResource
     files: files.AsyncFilesResource
     health: health.AsyncHealthResource
+    webhooks: webhooks.AsyncWebhooksResource
     with_raw_response: AsyncChunkrWithRawResponse
     with_streaming_response: AsyncChunkrWithStreamedResponse
 
     # client options
     api_key: str
+    webhook_key: str | None
 
     def __init__(
         self,
         *,
         api_key: str | None = None,
+        webhook_key: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN,
         max_retries: int = DEFAULT_MAX_RETRIES,
@@ -243,7 +258,9 @@ class AsyncChunkr(AsyncAPIClient):
     ) -> None:
         """Construct a new async AsyncChunkr client instance.
 
-        This automatically infers the `api_key` argument from the `CHUNKR_API_KEY` environment variable if it is not provided.
+        This automatically infers the following arguments from their corresponding environment variables if they are not provided:
+        - `api_key` from `CHUNKR_API_KEY`
+        - `webhook_key` from `CHUNKR_WEBHOOK_KEY`
         """
         if api_key is None:
             api_key = os.environ.get("CHUNKR_API_KEY")
@@ -253,6 +270,10 @@ class AsyncChunkr(AsyncAPIClient):
             )
         self.api_key = api_key
 
+        if webhook_key is None:
+            webhook_key = os.environ.get("CHUNKR_WEBHOOK_KEY")
+        self.webhook_key = webhook_key
+
         if base_url is None:
             base_url = os.environ.get("CHUNKR_BASE_URL")
         if base_url is None:
@@ -274,6 +295,7 @@ class AsyncChunkr(AsyncAPIClient):
         self.tasks = tasks.AsyncTasksResource(self)
         self.files = files.AsyncFilesResource(self)
         self.health = health.AsyncHealthResource(self)
+        self.webhooks = webhooks.AsyncWebhooksResource(self)
         self.with_raw_response = AsyncChunkrWithRawResponse(self)
         self.with_streaming_response = AsyncChunkrWithStreamedResponse(self)
 
@@ -301,6 +323,7 @@ class AsyncChunkr(AsyncAPIClient):
         self,
         *,
         api_key: str | None = None,
+        webhook_key: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = NOT_GIVEN,
         http_client: httpx.AsyncClient | None = None,
@@ -335,6 +358,7 @@ class AsyncChunkr(AsyncAPIClient):
         http_client = http_client or self._client
         return self.__class__(
             api_key=api_key or self.api_key,
+            webhook_key=webhook_key or self.webhook_key,
             base_url=base_url or self.base_url,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
             http_client=http_client,
@@ -387,6 +411,7 @@ class ChunkrWithRawResponse:
         self.tasks = tasks.TasksResourceWithRawResponse(client.tasks)
         self.files = files.FilesResourceWithRawResponse(client.files)
         self.health = health.HealthResourceWithRawResponse(client.health)
+        self.webhooks = webhooks.WebhooksResourceWithRawResponse(client.webhooks)
 
 
 class AsyncChunkrWithRawResponse:
@@ -394,6 +419,7 @@ class AsyncChunkrWithRawResponse:
         self.tasks = tasks.AsyncTasksResourceWithRawResponse(client.tasks)
         self.files = files.AsyncFilesResourceWithRawResponse(client.files)
         self.health = health.AsyncHealthResourceWithRawResponse(client.health)
+        self.webhooks = webhooks.AsyncWebhooksResourceWithRawResponse(client.webhooks)
 
 
 class ChunkrWithStreamedResponse:
@@ -401,6 +427,7 @@ class ChunkrWithStreamedResponse:
         self.tasks = tasks.TasksResourceWithStreamingResponse(client.tasks)
         self.files = files.FilesResourceWithStreamingResponse(client.files)
         self.health = health.HealthResourceWithStreamingResponse(client.health)
+        self.webhooks = webhooks.WebhooksResourceWithStreamingResponse(client.webhooks)
 
 
 class AsyncChunkrWithStreamedResponse:
@@ -408,6 +435,7 @@ class AsyncChunkrWithStreamedResponse:
         self.tasks = tasks.AsyncTasksResourceWithStreamingResponse(client.tasks)
         self.files = files.AsyncFilesResourceWithStreamingResponse(client.files)
         self.health = health.AsyncHealthResourceWithStreamingResponse(client.health)
+        self.webhooks = webhooks.AsyncWebhooksResourceWithStreamingResponse(client.webhooks)
 
 
 Client = Chunkr

chunkr_ai/_constants.py

@@ -5,10 +5,10 @@ import httpx
 RAW_RESPONSE_HEADER = "X-Stainless-Raw-Response"
 OVERRIDE_CAST_TO_HEADER = "____stainless_override_cast_to"
 
-# default timeout is 30 seconds
-DEFAULT_TIMEOUT = httpx.Timeout(timeout=30, connect=5.0)
-DEFAULT_MAX_RETRIES = 50
+# default timeout is 1 minute
+DEFAULT_TIMEOUT = httpx.Timeout(timeout=60, connect=5.0)
+DEFAULT_MAX_RETRIES = 2
 DEFAULT_CONNECTION_LIMITS = httpx.Limits(max_connections=100, max_keepalive_connections=20)
 
-INITIAL_RETRY_DELAY = 1.0
-MAX_RETRY_DELAY = 10.0
+INITIAL_RETRY_DELAY = 0.5
+MAX_RETRY_DELAY = 8.0

chunkr_ai/_exceptions.py

@@ -54,6 +54,10 @@ class APIResponseValidationError(APIError):
         self.status_code = response.status_code
 
 
+class APIWebhookValidationError(APIError):
+    pass
+
+
 class APIStatusError(APIError):
     """Raised when an API response has a status code of 4xx or 5xx."""
 

chunkr_ai/_models.py

@@ -304,7 +304,7 @@ class BaseModel(pydantic.BaseModel):
                 exclude_none=exclude_none,
             )
 
-            return cast(dict[str, Any], json_safe(dumped)) if mode == "json" else dumped
+            return cast("dict[str, Any]", json_safe(dumped)) if mode == "json" else dumped
 
         @override
         def model_dump_json(

chunkr_ai/_types.py

@@ -13,10 +13,21 @@ from typing import (
     Mapping,
     TypeVar,
     Callable,
+    Iterator,
     Optional,
     Sequence,
 )
-from typing_extensions import Set, Literal, Protocol, TypeAlias, TypedDict, override, runtime_checkable
+from typing_extensions import (
+    Set,
+    Literal,
+    Protocol,
+    TypeAlias,
+    TypedDict,
+    SupportsIndex,
+    overload,
+    override,
+    runtime_checkable,
+)
 
 import httpx
 import pydantic
@@ -217,3 +228,26 @@ class _GenericAlias(Protocol):
 class HttpxSendArgs(TypedDict, total=False):
     auth: httpx.Auth
     follow_redirects: bool
+
+
+_T_co = TypeVar("_T_co", covariant=True)
+
+
+if TYPE_CHECKING:
+    # This works because str.__contains__ does not accept object (either in typeshed or at runtime)
+    # https://github.com/hauntsaninja/useful_types/blob/5e9710f3875107d068e7679fd7fec9cfab0eff3b/useful_types/__init__.py#L285
+    class SequenceNotStr(Protocol[_T_co]):
+        @overload
+        def __getitem__(self, index: SupportsIndex, /) -> _T_co: ...
+        @overload
+        def __getitem__(self, index: slice, /) -> Sequence[_T_co]: ...
+        def __contains__(self, value: object, /) -> bool: ...
+        def __len__(self) -> int: ...
+        def __iter__(self) -> Iterator[_T_co]: ...
+        def index(self, value: Any, start: int = 0, stop: int = ..., /) -> int: ...
+        def count(self, value: Any, /) -> int: ...
+        def __reversed__(self) -> Iterator[_T_co]: ...
+else:
+    # just point this to a normal `Sequence` at runtime to avoid having to special case
+    # deserializing our custom sequence type
+    SequenceNotStr = Sequence

chunkr_ai/_utils/__init__.py

@@ -38,6 +38,7 @@ from ._typing import (
     extract_type_arg as extract_type_arg,
     is_iterable_type as is_iterable_type,
     is_required_type as is_required_type,
+    is_sequence_type as is_sequence_type,
     is_annotated_type as is_annotated_type,
     is_type_alias_type as is_type_alias_type,
     strip_annotated_type as strip_annotated_type,

chunkr_ai/_utils/_typing.py

@@ -26,6 +26,11 @@ def is_list_type(typ: type) -> bool:
     return (get_origin(typ) or typ) == list
 
 
+def is_sequence_type(typ: type) -> bool:
+    origin = get_origin(typ) or typ
+    return origin == typing_extensions.Sequence or origin == typing.Sequence or origin == _c_abc.Sequence
+
+
 def is_iterable_type(typ: type) -> bool:
     """If the given type is `typing.Iterable[T]`"""
     origin = get_origin(typ) or typ

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

chunkr_ai/resources/__init__.py

@@ -24,6 +24,14 @@ from .health import (
     HealthResourceWithStreamingResponse,
     AsyncHealthResourceWithStreamingResponse,
 )
+from .webhooks import (
+    WebhooksResource,
+    AsyncWebhooksResource,
+    WebhooksResourceWithRawResponse,
+    AsyncWebhooksResourceWithRawResponse,
+    WebhooksResourceWithStreamingResponse,
+    AsyncWebhooksResourceWithStreamingResponse,
+)
 
 __all__ = [
     "TasksResource",
@@ -44,4 +52,10 @@ __all__ = [
     "AsyncHealthResourceWithRawResponse",
     "HealthResourceWithStreamingResponse",
     "AsyncHealthResourceWithStreamingResponse",
+    "WebhooksResource",
+    "AsyncWebhooksResource",
+    "WebhooksResourceWithRawResponse",
+    "AsyncWebhooksResourceWithRawResponse",
+    "WebhooksResourceWithStreamingResponse",
+    "AsyncWebhooksResourceWithStreamingResponse",
 ]

chunkr_ai/resources/files.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Union, Mapping, cast
+from typing import Union, Mapping, Optional, cast
 from datetime import datetime
 from typing_extensions import Literal
 
@@ -52,7 +52,7 @@ class FilesResource(SyncAPIResource):
         self,
         *,
         file: FileTypes,
-        file_metadata: str,
+        file_metadata: Optional[str] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -360,7 +360,7 @@ class AsyncFilesResource(AsyncAPIResource):
         self,
         *,
         file: FileTypes,
-        file_metadata: str,
+        file_metadata: Optional[str] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,

chunkr_ai/resources/tasks/__init__.py

@@ -16,8 +16,22 @@ from .tasks import (
     TasksResourceWithStreamingResponse,
     AsyncTasksResourceWithStreamingResponse,
 )
+from .extract import (
+    ExtractResource,
+    AsyncExtractResource,
+    ExtractResourceWithRawResponse,
+    AsyncExtractResourceWithRawResponse,
+    ExtractResourceWithStreamingResponse,
+    AsyncExtractResourceWithStreamingResponse,
+)
 
 __all__ = [
+    "ExtractResource",
+    "AsyncExtractResource",
+    "ExtractResourceWithRawResponse",
+    "AsyncExtractResourceWithRawResponse",
+    "ExtractResourceWithStreamingResponse",
+    "AsyncExtractResourceWithStreamingResponse",
     "ParseResource",
     "AsyncParseResource",
     "ParseResourceWithRawResponse",