samplehc 0.14.0__py3-none-any.whl → 0.15.0__py3-none-any.whl

samplehc/_base_client.py

@@ -1247,9 +1247,12 @@ class SyncAPIClient(BaseClient[httpx.Client, Stream[Any]]):
         *,
         cast_to: Type[ResponseT],
         body: Body | None = None,
+        files: RequestFiles | None = None,
         options: RequestOptions = {},
     ) -> ResponseT:
-        opts = FinalRequestOptions.construct(method="patch", url=path, json_data=body, **options)
+        opts = FinalRequestOptions.construct(
+            method="patch", url=path, json_data=body, files=to_httpx_files(files), **options
+        )
         return self.request(cast_to, opts)
 
     def put(
@@ -1767,9 +1770,12 @@ class AsyncAPIClient(BaseClient[httpx.AsyncClient, AsyncStream[Any]]):
         *,
         cast_to: Type[ResponseT],
         body: Body | None = None,
+        files: RequestFiles | None = None,
         options: RequestOptions = {},
     ) -> ResponseT:
-        opts = FinalRequestOptions.construct(method="patch", url=path, json_data=body, **options)
+        opts = FinalRequestOptions.construct(
+            method="patch", url=path, json_data=body, files=await async_to_httpx_files(files), **options
+        )
         return await self.request(cast_to, opts)
 
     async def put(

samplehc/_client.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 import os
-from typing import Any, Mapping
+from typing import TYPE_CHECKING, Any, Mapping
 from typing_extensions import Self, override
 
 import httpx
@@ -21,6 +21,7 @@ from ._types import (
     not_given,
 )
 from ._utils import is_given, get_async_library
+from ._compat import cached_property
 from ._version import __version__
 from ._streaming import Stream as Stream, AsyncStream as AsyncStream
 from ._exceptions import APIStatusError
@@ -29,8 +30,11 @@ from ._base_client import (
     SyncAPIClient,
     AsyncAPIClient,
 )
-from .resources.v1 import v1
-from .resources.v2 import v2
+
+if TYPE_CHECKING:
+    from .resources import v1, v2
+    from .resources.v1.v1 import V1Resource, AsyncV1Resource
+    from .resources.v2.v2 import V2Resource, AsyncV2Resource
 
 __all__ = [
     "Timeout",
@@ -45,11 +49,6 @@ __all__ = [
 
 
 class SampleHealthcare(SyncAPIClient):
-    v1: v1.V1Resource
-    v2: v2.V2Resource
-    with_raw_response: SampleHealthcareWithRawResponse
-    with_streaming_response: SampleHealthcareWithStreamedResponse
-
     # client options
     api_key: str | None
 
@@ -100,10 +99,25 @@ class SampleHealthcare(SyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
-        self.v1 = v1.V1Resource(self)
-        self.v2 = v2.V2Resource(self)
-        self.with_raw_response = SampleHealthcareWithRawResponse(self)
-        self.with_streaming_response = SampleHealthcareWithStreamedResponse(self)
+    @cached_property
+    def v1(self) -> V1Resource:
+        from .resources.v1 import V1Resource
+
+        return V1Resource(self)
+
+    @cached_property
+    def v2(self) -> V2Resource:
+        from .resources.v2 import V2Resource
+
+        return V2Resource(self)
+
+    @cached_property
+    def with_raw_response(self) -> SampleHealthcareWithRawResponse:
+        return SampleHealthcareWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> SampleHealthcareWithStreamedResponse:
+        return SampleHealthcareWithStreamedResponse(self)
 
     @property
     @override
@@ -224,11 +238,6 @@ class SampleHealthcare(SyncAPIClient):
 
 
 class AsyncSampleHealthcare(AsyncAPIClient):
-    v1: v1.AsyncV1Resource
-    v2: v2.AsyncV2Resource
-    with_raw_response: AsyncSampleHealthcareWithRawResponse
-    with_streaming_response: AsyncSampleHealthcareWithStreamedResponse
-
     # client options
     api_key: str | None
 
@@ -279,10 +288,25 @@ class AsyncSampleHealthcare(AsyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
-        self.v1 = v1.AsyncV1Resource(self)
-        self.v2 = v2.AsyncV2Resource(self)
-        self.with_raw_response = AsyncSampleHealthcareWithRawResponse(self)
-        self.with_streaming_response = AsyncSampleHealthcareWithStreamedResponse(self)
+    @cached_property
+    def v1(self) -> AsyncV1Resource:
+        from .resources.v1 import AsyncV1Resource
+
+        return AsyncV1Resource(self)
+
+    @cached_property
+    def v2(self) -> AsyncV2Resource:
+        from .resources.v2 import AsyncV2Resource
+
+        return AsyncV2Resource(self)
+
+    @cached_property
+    def with_raw_response(self) -> AsyncSampleHealthcareWithRawResponse:
+        return AsyncSampleHealthcareWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncSampleHealthcareWithStreamedResponse:
+        return AsyncSampleHealthcareWithStreamedResponse(self)
 
     @property
     @override
@@ -403,27 +427,79 @@ class AsyncSampleHealthcare(AsyncAPIClient):
 
 
 class SampleHealthcareWithRawResponse:
+    _client: SampleHealthcare
+
     def __init__(self, client: SampleHealthcare) -> None:
-        self.v1 = v1.V1ResourceWithRawResponse(client.v1)
-        self.v2 = v2.V2ResourceWithRawResponse(client.v2)
+        self._client = client
+
+    @cached_property
+    def v1(self) -> v1.V1ResourceWithRawResponse:
+        from .resources.v1 import V1ResourceWithRawResponse
+
+        return V1ResourceWithRawResponse(self._client.v1)
+
+    @cached_property
+    def v2(self) -> v2.V2ResourceWithRawResponse:
+        from .resources.v2 import V2ResourceWithRawResponse
+
+        return V2ResourceWithRawResponse(self._client.v2)
 
 
 class AsyncSampleHealthcareWithRawResponse:
+    _client: AsyncSampleHealthcare
+
     def __init__(self, client: AsyncSampleHealthcare) -> None:
-        self.v1 = v1.AsyncV1ResourceWithRawResponse(client.v1)
-        self.v2 = v2.AsyncV2ResourceWithRawResponse(client.v2)
+        self._client = client
+
+    @cached_property
+    def v1(self) -> v1.AsyncV1ResourceWithRawResponse:
+        from .resources.v1 import AsyncV1ResourceWithRawResponse
+
+        return AsyncV1ResourceWithRawResponse(self._client.v1)
+
+    @cached_property
+    def v2(self) -> v2.AsyncV2ResourceWithRawResponse:
+        from .resources.v2 import AsyncV2ResourceWithRawResponse
+
+        return AsyncV2ResourceWithRawResponse(self._client.v2)
 
 
 class SampleHealthcareWithStreamedResponse:
+    _client: SampleHealthcare
+
     def __init__(self, client: SampleHealthcare) -> None:
-        self.v1 = v1.V1ResourceWithStreamingResponse(client.v1)
-        self.v2 = v2.V2ResourceWithStreamingResponse(client.v2)
+        self._client = client
+
+    @cached_property
+    def v1(self) -> v1.V1ResourceWithStreamingResponse:
+        from .resources.v1 import V1ResourceWithStreamingResponse
+
+        return V1ResourceWithStreamingResponse(self._client.v1)
+
+    @cached_property
+    def v2(self) -> v2.V2ResourceWithStreamingResponse:
+        from .resources.v2 import V2ResourceWithStreamingResponse
+
+        return V2ResourceWithStreamingResponse(self._client.v2)
 
 
 class AsyncSampleHealthcareWithStreamedResponse:
+    _client: AsyncSampleHealthcare
+
     def __init__(self, client: AsyncSampleHealthcare) -> None:
-        self.v1 = v1.AsyncV1ResourceWithStreamingResponse(client.v1)
-        self.v2 = v2.AsyncV2ResourceWithStreamingResponse(client.v2)
+        self._client = client
+
+    @cached_property
+    def v1(self) -> v1.AsyncV1ResourceWithStreamingResponse:
+        from .resources.v1 import AsyncV1ResourceWithStreamingResponse
+
+        return AsyncV1ResourceWithStreamingResponse(self._client.v1)
+
+    @cached_property
+    def v2(self) -> v2.AsyncV2ResourceWithStreamingResponse:
+        from .resources.v2 import AsyncV2ResourceWithStreamingResponse
+
+        return AsyncV2ResourceWithStreamingResponse(self._client.v2)
 
 
 Client = SampleHealthcare

samplehc/_models.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import os
 import inspect
+import weakref
 from typing import TYPE_CHECKING, Any, Type, Union, Generic, TypeVar, Callable, Optional, cast
 from datetime import date, datetime
 from typing_extensions import (
@@ -256,15 +257,16 @@ class BaseModel(pydantic.BaseModel):
             mode: Literal["json", "python"] | str = "python",
             include: IncEx | None = None,
             exclude: IncEx | None = None,
+            context: Any | None = None,
             by_alias: bool | None = None,
             exclude_unset: bool = False,
             exclude_defaults: bool = False,
             exclude_none: bool = False,
+            exclude_computed_fields: bool = False,
             round_trip: bool = False,
             warnings: bool | Literal["none", "warn", "error"] = True,
-            context: dict[str, Any] | None = None,
-            serialize_as_any: bool = False,
             fallback: Callable[[Any], Any] | None = None,
+            serialize_as_any: bool = False,
         ) -> dict[str, Any]:
             """Usage docs: https://docs.pydantic.dev/2.4/concepts/serialization/#modelmodel_dump
 
@@ -272,16 +274,24 @@ class BaseModel(pydantic.BaseModel):
 
             Args:
                 mode: The mode in which `to_python` should run.
-                    If mode is 'json', the dictionary will only contain JSON serializable types.
-                    If mode is 'python', the dictionary may contain any Python objects.
-                include: A list of fields to include in the output.
-                exclude: A list of fields to exclude from the output.
+                    If mode is 'json', the output will only contain JSON serializable types.
+                    If mode is 'python', the output may contain non-JSON-serializable Python objects.
+                include: A set of fields to include in the output.
+                exclude: A set of fields to exclude from the output.
+                context: Additional context to pass to the serializer.
                 by_alias: Whether to use the field's alias in the dictionary key if defined.
-                exclude_unset: Whether to exclude fields that are unset or None from the output.
-                exclude_defaults: Whether to exclude fields that are set to their default value from the output.
-                exclude_none: Whether to exclude fields that have a value of `None` from the output.
-                round_trip: Whether to enable serialization and deserialization round-trip support.
-                warnings: Whether to log warnings when invalid fields are encountered.
+                exclude_unset: Whether to exclude fields that have not been explicitly set.
+                exclude_defaults: Whether to exclude fields that are set to their default value.
+                exclude_none: Whether to exclude fields that have a value of `None`.
+                exclude_computed_fields: Whether to exclude computed fields.
+                    While this can be useful for round-tripping, it is usually recommended to use the dedicated
+                    `round_trip` parameter instead.
+                round_trip: If True, dumped values should be valid as input for non-idempotent types such as Json[T].
+                warnings: How to handle serialization errors. False/"none" ignores them, True/"warn" logs errors,
+                    "error" raises a [`PydanticSerializationError`][pydantic_core.PydanticSerializationError].
+                fallback: A function to call when an unknown value is encountered. If not provided,
+                    a [`PydanticSerializationError`][pydantic_core.PydanticSerializationError] error is raised.
+                serialize_as_any: Whether to serialize fields with duck-typing serialization behavior.
 
             Returns:
                 A dictionary representation of the model.
@@ -298,6 +308,8 @@ class BaseModel(pydantic.BaseModel):
                 raise ValueError("serialize_as_any is only supported in Pydantic v2")
             if fallback is not None:
                 raise ValueError("fallback is only supported in Pydantic v2")
+            if exclude_computed_fields != False:
+                raise ValueError("exclude_computed_fields is only supported in Pydantic v2")
             dumped = super().dict(  # pyright: ignore[reportDeprecated]
                 include=include,
                 exclude=exclude,
@@ -314,15 +326,17 @@ class BaseModel(pydantic.BaseModel):
             self,
             *,
             indent: int | None = None,
+            ensure_ascii: bool = False,
             include: IncEx | None = None,
             exclude: IncEx | None = None,
+            context: Any | None = None,
             by_alias: bool | None = None,
             exclude_unset: bool = False,
             exclude_defaults: bool = False,
             exclude_none: bool = False,
+            exclude_computed_fields: bool = False,
             round_trip: bool = False,
             warnings: bool | Literal["none", "warn", "error"] = True,
-            context: dict[str, Any] | None = None,
             fallback: Callable[[Any], Any] | None = None,
             serialize_as_any: bool = False,
         ) -> str:
@@ -354,6 +368,10 @@ class BaseModel(pydantic.BaseModel):
                 raise ValueError("serialize_as_any is only supported in Pydantic v2")
             if fallback is not None:
                 raise ValueError("fallback is only supported in Pydantic v2")
+            if ensure_ascii != False:
+                raise ValueError("ensure_ascii is only supported in Pydantic v2")
+            if exclude_computed_fields != False:
+                raise ValueError("exclude_computed_fields is only supported in Pydantic v2")
             return super().json(  # type: ignore[reportDeprecated]
                 indent=indent,
                 include=include,
@@ -573,6 +591,9 @@ class CachedDiscriminatorType(Protocol):
     __discriminator__: DiscriminatorDetails
 
 
+DISCRIMINATOR_CACHE: weakref.WeakKeyDictionary[type, DiscriminatorDetails] = weakref.WeakKeyDictionary()
+
+
 class DiscriminatorDetails:
     field_name: str
     """The name of the discriminator field in the variant class, e.g.
@@ -615,8 +636,9 @@ class DiscriminatorDetails:
 
 
 def _build_discriminated_union_meta(*, union: type, meta_annotations: tuple[Any, ...]) -> DiscriminatorDetails | None:
-    if isinstance(union, CachedDiscriminatorType):
-        return union.__discriminator__
+    cached = DISCRIMINATOR_CACHE.get(union)
+    if cached is not None:
+        return cached
 
     discriminator_field_name: str | None = None
 
@@ -669,7 +691,7 @@ def _build_discriminated_union_meta(*, union: type, meta_annotations: tuple[Any,
         discriminator_field=discriminator_field_name,
         discriminator_alias=discriminator_alias,
     )
-    cast(CachedDiscriminatorType, union).__discriminator__ = details
+    DISCRIMINATOR_CACHE.setdefault(union, details)
     return details
 
 

samplehc/_streaming.py

@@ -54,12 +54,12 @@ class Stream(Generic[_T]):
         process_data = self._client._process_response_data
         iterator = self._iter_events()
 
-        for sse in iterator:
-            yield process_data(data=sse.json(), cast_to=cast_to, response=response)
-
-        # Ensure the entire stream is consumed
-        for _sse in iterator:
-            ...
+        try:
+            for sse in iterator:
+                yield process_data(data=sse.json(), cast_to=cast_to, response=response)
+        finally:
+            # Ensure the response is closed even if the consumer doesn't read all data
+            response.close()
 
     def __enter__(self) -> Self:
         return self
@@ -118,12 +118,12 @@ class AsyncStream(Generic[_T]):
         process_data = self._client._process_response_data
         iterator = self._iter_events()
 
-        async for sse in iterator:
-            yield process_data(data=sse.json(), cast_to=cast_to, response=response)
-
-        # Ensure the entire stream is consumed
-        async for _sse in iterator:
-            ...
+        try:
+            async for sse in iterator:
+                yield process_data(data=sse.json(), cast_to=cast_to, response=response)
+        finally:
+            # Ensure the response is closed even if the consumer doesn't read all data
+            await response.aclose()
 
     async def __aenter__(self) -> Self:
         return self

samplehc/_types.py

@@ -243,6 +243,9 @@ _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
+    #
+    # Note: index() and count() methods are intentionally omitted to allow pyright to properly
+    # infer TypedDict types when dict literals are used in lists assigned to SequenceNotStr.
     class SequenceNotStr(Protocol[_T_co]):
         @overload
         def __getitem__(self, index: SupportsIndex, /) -> _T_co: ...
@@ -251,8 +254,6 @@ if TYPE_CHECKING:
         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

samplehc/_utils/_sync.py

@@ -1,10 +1,8 @@
 from __future__ import annotations
 
-import sys
 import asyncio
 import functools
-import contextvars
-from typing import Any, TypeVar, Callable, Awaitable
+from typing import TypeVar, Callable, Awaitable
 from typing_extensions import ParamSpec
 
 import anyio
@@ -15,34 +13,11 @@ T_Retval = TypeVar("T_Retval")
 T_ParamSpec = ParamSpec("T_ParamSpec")
 
 
-if sys.version_info >= (3, 9):
-    _asyncio_to_thread = asyncio.to_thread
-else:
-    # backport of https://docs.python.org/3/library/asyncio-task.html#asyncio.to_thread
-    # for Python 3.8 support
-    async def _asyncio_to_thread(
-        func: Callable[T_ParamSpec, T_Retval], /, *args: T_ParamSpec.args, **kwargs: T_ParamSpec.kwargs
-    ) -> Any:
-        """Asynchronously run function *func* in a separate thread.
-
-        Any *args and **kwargs supplied for this function are directly passed
-        to *func*. Also, the current :class:`contextvars.Context` is propagated,
-        allowing context variables from the main thread to be accessed in the
-        separate thread.
-
-        Returns a coroutine that can be awaited to get the eventual result of *func*.
-        """
-        loop = asyncio.events.get_running_loop()
-        ctx = contextvars.copy_context()
-        func_call = functools.partial(ctx.run, func, *args, **kwargs)
-        return await loop.run_in_executor(None, func_call)
-
-
 async def to_thread(
     func: Callable[T_ParamSpec, T_Retval], /, *args: T_ParamSpec.args, **kwargs: T_ParamSpec.kwargs
 ) -> T_Retval:
     if sniffio.current_async_library() == "asyncio":
-        return await _asyncio_to_thread(func, *args, **kwargs)
+        return await asyncio.to_thread(func, *args, **kwargs)
 
     return await anyio.to_thread.run_sync(
         functools.partial(func, *args, **kwargs),
@@ -53,10 +28,7 @@ async def to_thread(
 def asyncify(function: Callable[T_ParamSpec, T_Retval]) -> Callable[T_ParamSpec, Awaitable[T_Retval]]:
     """
     Take a blocking function and create an async one that receives the same
-    positional and keyword arguments. For python version 3.9 and above, it uses
-    asyncio.to_thread to run the function in a separate thread. For python version
-    3.8, it uses locally defined copy of the asyncio.to_thread function which was
-    introduced in python 3.9.
+    positional and keyword arguments.
 
     Usage:
 

samplehc/_utils/_utils.py

@@ -133,7 +133,7 @@ def is_given(obj: _T | NotGiven | Omit) -> TypeGuard[_T]:
 # Type safe methods for narrowing types with TypeVars.
 # The default narrowing for isinstance(obj, dict) is dict[unknown, unknown],
 # however this cause Pyright to rightfully report errors. As we know we don't
-# care about the contained types we can safely use `object` in it's place.
+# care about the contained types we can safely use `object` in its place.
 #
 # There are two separate functions defined, `is_*` and `is_*_t` for different use cases.
 # `is_*` is for when you're dealing with an unknown input

samplehc/_version.py

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

samplehc/resources/v2/__init__.py

@@ -112,22 +112,6 @@ from .workflow_runs import (
     WorkflowRunsResourceWithStreamingResponse,
     AsyncWorkflowRunsResourceWithStreamingResponse,
 )
-from .browser_agents import (
-    BrowserAgentsResource,
-    AsyncBrowserAgentsResource,
-    BrowserAgentsResourceWithRawResponse,
-    AsyncBrowserAgentsResourceWithRawResponse,
-    BrowserAgentsResourceWithStreamingResponse,
-    AsyncBrowserAgentsResourceWithStreamingResponse,
-)
-from .browser_automation import (
-    BrowserAutomationResource,
-    AsyncBrowserAutomationResource,
-    BrowserAutomationResourceWithRawResponse,
-    AsyncBrowserAutomationResourceWithRawResponse,
-    BrowserAutomationResourceWithStreamingResponse,
-    AsyncBrowserAutomationResourceWithStreamingResponse,
-)
 
 __all__ = [
     "AsyncResultsResource",
@@ -190,12 +174,6 @@ __all__ = [
     "AsyncEventsResourceWithRawResponse",
     "EventsResourceWithStreamingResponse",
     "AsyncEventsResourceWithStreamingResponse",
-    "BrowserAgentsResource",
-    "AsyncBrowserAgentsResource",
-    "BrowserAgentsResourceWithRawResponse",
-    "AsyncBrowserAgentsResourceWithRawResponse",
-    "BrowserAgentsResourceWithStreamingResponse",
-    "AsyncBrowserAgentsResourceWithStreamingResponse",
     "PoliciesResource",
     "AsyncPoliciesResource",
     "PoliciesResourceWithRawResponse",
@@ -214,12 +192,6 @@ __all__ = [
     "AsyncDatabaseResourceWithRawResponse",
     "DatabaseResourceWithStreamingResponse",
     "AsyncDatabaseResourceWithStreamingResponse",
-    "BrowserAutomationResource",
-    "AsyncBrowserAutomationResource",
-    "BrowserAutomationResourceWithRawResponse",
-    "AsyncBrowserAutomationResourceWithRawResponse",
-    "BrowserAutomationResourceWithStreamingResponse",
-    "AsyncBrowserAutomationResourceWithStreamingResponse",
     "V2Resource",
     "AsyncV2Resource",
     "V2ResourceWithRawResponse",

samplehc/resources/v2/documents/documents.py

@@ -293,6 +293,7 @@ class DocumentsResource(SyncAPIResource):
         prompt: str,
         response_json_schema: Dict[str, object],
         model: Literal["reasoning-3-mini", "reasoning-3", "base-5", "base-5-mini", "base-5-nano"] | Omit = omit,
+        ocr_enhance: document_extract_params.OcrEnhance | Omit = omit,
         priority: Literal["interactive", "non-interactive"] | Omit = omit,
         reasoning_effort: Literal["low", "medium", "high"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -315,6 +316,8 @@ class DocumentsResource(SyncAPIResource):
 
           model: The model to use for extraction.
 
+          ocr_enhance: OCR enhancement configuration for figure and text analysis.
+
           priority: The priority of the extraction task. Non-interactive is lower priority.
 
           reasoning_effort: Optional control over the reasoning effort for extraction.
@@ -335,6 +338,7 @@ class DocumentsResource(SyncAPIResource):
                     "prompt": prompt,
                     "response_json_schema": response_json_schema,
                     "model": model,
+                    "ocr_enhance": ocr_enhance,
                     "priority": priority,
                     "reasoning_effort": reasoning_effort,
                 },
@@ -933,6 +937,7 @@ class AsyncDocumentsResource(AsyncAPIResource):
         prompt: str,
         response_json_schema: Dict[str, object],
         model: Literal["reasoning-3-mini", "reasoning-3", "base-5", "base-5-mini", "base-5-nano"] | Omit = omit,
+        ocr_enhance: document_extract_params.OcrEnhance | Omit = omit,
         priority: Literal["interactive", "non-interactive"] | Omit = omit,
         reasoning_effort: Literal["low", "medium", "high"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -955,6 +960,8 @@ class AsyncDocumentsResource(AsyncAPIResource):
 
           model: The model to use for extraction.
 
+          ocr_enhance: OCR enhancement configuration for figure and text analysis.
+
           priority: The priority of the extraction task. Non-interactive is lower priority.
 
           reasoning_effort: Optional control over the reasoning effort for extraction.
@@ -975,6 +982,7 @@ class AsyncDocumentsResource(AsyncAPIResource):
                     "prompt": prompt,
                     "response_json_schema": response_json_schema,
                     "model": model,
+                    "ocr_enhance": ocr_enhance,
                     "priority": priority,
                     "reasoning_effort": reasoning_effort,
                 },

samplehc/resources/v2/tasks/tasks.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 
-from typing import Any, Optional, cast
+from typing import Any, Union, Optional, cast
+from typing_extensions import Literal
 
 import httpx
 
@@ -241,7 +242,8 @@ class TasksResource(SyncAPIResource):
         task_id: str,
         *,
         key: str,
-        value: Optional[str],
+        value: Union[str, float, bool, None],
+        type: Literal["string", "number", "boolean", "date", "datetime"] | Omit = omit,
         # 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,
@@ -259,6 +261,8 @@ class TasksResource(SyncAPIResource):
 
           value: The value to set for the column.
 
+          type: The semantic type of the column. Defaults to string when omitted.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -275,6 +279,7 @@ class TasksResource(SyncAPIResource):
                 {
                     "key": key,
                     "value": value,
+                    "type": type,
                 },
                 task_update_column_params.TaskUpdateColumnParams,
             ),
@@ -533,7 +538,8 @@ class AsyncTasksResource(AsyncAPIResource):
         task_id: str,
         *,
         key: str,
-        value: Optional[str],
+        value: Union[str, float, bool, None],
+        type: Literal["string", "number", "boolean", "date", "datetime"] | Omit = omit,
         # 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,
@@ -551,6 +557,8 @@ class AsyncTasksResource(AsyncAPIResource):
 
           value: The value to set for the column.
 
+          type: The semantic type of the column. Defaults to string when omitted.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -567,6 +575,7 @@ class AsyncTasksResource(AsyncAPIResource):
                 {
                     "key": key,
                     "value": value,
+                    "type": type,
                 },
                 task_update_column_params.TaskUpdateColumnParams,
             ),