udl-sdk 0.1.0a10__py3-none-any.whl → 0.1.0a12__py3-none-any.whl

unifieddatalibrary/__init__.py

@@ -36,7 +36,7 @@ from ._exceptions import (
     UnprocessableEntityError,
     APIResponseValidationError,
 )
-from ._base_client import DefaultHttpxClient, DefaultAsyncHttpxClient
+from ._base_client import DefaultHttpxClient, DefaultAioHttpClient, DefaultAsyncHttpxClient
 from ._utils._logs import setup_logging as _setup_logging
 
 __all__ = [
@@ -78,6 +78,7 @@ __all__ = [
     "DEFAULT_CONNECTION_LIMITS",
     "DefaultHttpxClient",
     "DefaultAsyncHttpxClient",
+    "DefaultAioHttpClient",
 ]
 
 if not _t.TYPE_CHECKING:

unifieddatalibrary/_base_client.py

@@ -89,6 +89,7 @@ log: logging.Logger = logging.getLogger(__name__)
 # TODO: make base page type vars covariant
 SyncPageT = TypeVar("SyncPageT", bound="BaseSyncPage[Any]")
 AsyncPageT = TypeVar("AsyncPageT", bound="BaseAsyncPage[Any]")
+_BasePageT = TypeVar("_BasePageT", bound="BasePage[Any]")
 
 
 _T = TypeVar("_T")
@@ -174,6 +175,7 @@ class BasePage(GenericModel, Generic[_T]):
         next_page_info(): Get the necessary information to make a request for the next page
     """
 
+    _response: httpx.Response = PrivateAttr()
     _options: FinalRequestOptions = PrivateAttr()
     _model: Type[_T] = PrivateAttr()
 
@@ -222,6 +224,23 @@ class BasePage(GenericModel, Generic[_T]):
 
         raise ValueError("Unexpected PageInfo state")
 
+    @classmethod
+    def build(cls: Type[_BasePageT], *, response: httpx.Response, data: object) -> _BasePageT:  # noqa: ARG003
+        return cls._with_response(
+            cls.construct(
+                None,
+                **{
+                    **(cast(Mapping[str, Any], data) if is_mapping(data) else {}),
+                },
+            ),
+            response,
+        )
+
+    @classmethod
+    def _with_response(cls, inst: _BasePageT, response: httpx.Response) -> _BasePageT:
+        inst._response = response
+        return inst
+
 
 class BaseSyncPage(BasePage[_T], Generic[_T]):
     _client: SyncAPIClient = pydantic.PrivateAttr()
@@ -529,6 +548,15 @@ class BaseClient(Generic[_HttpxClientT, _DefaultStreamT]):
             # work around https://github.com/encode/httpx/discussions/2880
             kwargs["extensions"] = {"sni_hostname": prepared_url.host.replace("_", "-")}
 
+        is_body_allowed = options.method.lower() != "get"
+
+        if is_body_allowed:
+            kwargs["json"] = json_data if is_given(json_data) else None
+            kwargs["files"] = files
+        else:
+            headers.pop("Content-Type", None)
+            kwargs.pop("data", None)
+
         # TODO: report this error to httpx
         return self._client.build_request(  # pyright: ignore[reportUnknownMemberType]
             headers=headers,
@@ -540,8 +568,6 @@ class BaseClient(Generic[_HttpxClientT, _DefaultStreamT]):
             # so that passing a `TypedDict` doesn't cause an error.
             # https://github.com/microsoft/pyright/issues/3526#event-6715453066
             params=self.qs.stringify(cast(Mapping[str, Any], params)) if params else None,
-            json=json_data if is_given(json_data) else None,
-            files=files,
             **kwargs,
         )
 
@@ -960,6 +986,9 @@ class SyncAPIClient(BaseClient[httpx.Client, Stream[Any]]):
             if self.custom_auth is not None:
                 kwargs["auth"] = self.custom_auth
 
+            if options.follow_redirects is not None:
+                kwargs["follow_redirects"] = options.follow_redirects
+
             log.debug("Sending HTTP Request: %s %s", request.method, request.url)
 
             response = None
@@ -1068,7 +1097,14 @@ class SyncAPIClient(BaseClient[httpx.Client, Stream[Any]]):
     ) -> ResponseT:
         origin = get_origin(cast_to) or cast_to
 
-        if inspect.isclass(origin) and issubclass(origin, BaseAPIResponse):
+        if (
+            inspect.isclass(origin)
+            and issubclass(origin, BaseAPIResponse)
+            # we only want to actually return the custom BaseAPIResponse class if we're
+            # returning the raw response, or if we're not streaming SSE, as if we're streaming
+            # SSE then `cast_to` doesn't actively reflect the type we need to parse into
+            and (not stream or bool(response.request.headers.get(RAW_RESPONSE_HEADER)))
+        ):
             if not issubclass(origin, APIResponse):
                 raise TypeError(f"API Response types must subclass {APIResponse}; Received {origin}")
 
@@ -1279,6 +1315,24 @@ class _DefaultAsyncHttpxClient(httpx.AsyncClient):
         super().__init__(**kwargs)
 
 
+try:
+    import httpx_aiohttp
+except ImportError:
+
+    class _DefaultAioHttpClient(httpx.AsyncClient):
+        def __init__(self, **_kwargs: Any) -> None:
+            raise RuntimeError("To use the aiohttp client you must have installed the package with the `aiohttp` extra")
+else:
+
+    class _DefaultAioHttpClient(httpx_aiohttp.HttpxAiohttpClient):  # type: ignore
+        def __init__(self, **kwargs: Any) -> None:
+            kwargs.setdefault("timeout", DEFAULT_TIMEOUT)
+            kwargs.setdefault("limits", DEFAULT_CONNECTION_LIMITS)
+            kwargs.setdefault("follow_redirects", True)
+
+            super().__init__(**kwargs)
+
+
 if TYPE_CHECKING:
     DefaultAsyncHttpxClient = httpx.AsyncClient
     """An alias to `httpx.AsyncClient` that provides the same defaults that this SDK
@@ -1287,8 +1341,12 @@ if TYPE_CHECKING:
     This is useful because overriding the `http_client` with your own instance of
     `httpx.AsyncClient` will result in httpx's defaults being used, not ours.
     """
+
+    DefaultAioHttpClient = httpx.AsyncClient
+    """An alias to `httpx.AsyncClient` that changes the default HTTP transport to `aiohttp`."""
 else:
     DefaultAsyncHttpxClient = _DefaultAsyncHttpxClient
+    DefaultAioHttpClient = _DefaultAioHttpClient
 
 
 class AsyncHttpxClientWrapper(DefaultAsyncHttpxClient):
@@ -1460,6 +1518,9 @@ class AsyncAPIClient(BaseClient[httpx.AsyncClient, AsyncStream[Any]]):
             if self.custom_auth is not None:
                 kwargs["auth"] = self.custom_auth
 
+            if options.follow_redirects is not None:
+                kwargs["follow_redirects"] = options.follow_redirects
+
             log.debug("Sending HTTP Request: %s %s", request.method, request.url)
 
             response = None
@@ -1568,7 +1629,14 @@ class AsyncAPIClient(BaseClient[httpx.AsyncClient, AsyncStream[Any]]):
     ) -> ResponseT:
         origin = get_origin(cast_to) or cast_to
 
-        if inspect.isclass(origin) and issubclass(origin, BaseAPIResponse):
+        if (
+            inspect.isclass(origin)
+            and issubclass(origin, BaseAPIResponse)
+            # we only want to actually return the custom BaseAPIResponse class if we're
+            # returning the raw response, or if we're not streaming SSE, as if we're streaming
+            # SSE then `cast_to` doesn't actively reflect the type we need to parse into
+            and (not stream or bool(response.request.headers.get(RAW_RESPONSE_HEADER)))
+        ):
             if not issubclass(origin, AsyncAPIResponse):
                 raise TypeError(f"API Response types must subclass {AsyncAPIResponse}; Received {origin}")
 

unifieddatalibrary/_client.py

@@ -668,7 +668,7 @@ class Unifieddatalibrary(SyncAPIClient):
             params = set_default_query
 
         http_client = http_client or self._client
-        return self.__class__(
+        client = self.__class__(
             password=password or self.password,
             username=username or self.username,
             base_url=base_url or self.base_url,
@@ -679,6 +679,8 @@ class Unifieddatalibrary(SyncAPIClient):
             default_query=params,
             **_extra_kwargs,
         )
+        client._base_url_overridden = self._base_url_overridden or base_url is not None
+        return client
 
     # Alias for `copy` for nicer inline usage, e.g.
     # client.with_options(timeout=10).foo.create(...)
@@ -1178,7 +1180,7 @@ class AsyncUnifieddatalibrary(AsyncAPIClient):
             params = set_default_query
 
         http_client = http_client or self._client
-        return self.__class__(
+        client = self.__class__(
             password=password or self.password,
             username=username or self.username,
             base_url=base_url or self.base_url,
@@ -1189,6 +1191,8 @@ class AsyncUnifieddatalibrary(AsyncAPIClient):
             default_query=params,
             **_extra_kwargs,
         )
+        client._base_url_overridden = self._base_url_overridden or base_url is not None
+        return client
 
     # Alias for `copy` for nicer inline usage, e.g.
     # client.with_options(timeout=10).foo.create(...)

unifieddatalibrary/_models.py

@@ -2,9 +2,10 @@ from __future__ import annotations
 
 import os
 import inspect
-from typing import TYPE_CHECKING, Any, Type, Union, Generic, TypeVar, Callable, cast
+from typing import TYPE_CHECKING, Any, Type, Union, Generic, TypeVar, Callable, Optional, cast
 from datetime import date, datetime
 from typing_extensions import (
+    List,
     Unpack,
     Literal,
     ClassVar,
@@ -366,7 +367,7 @@ def _construct_field(value: object, field: FieldInfo, key: str) -> object:
     if type_ is None:
         raise RuntimeError(f"Unexpected field type is None for {key}")
 
-    return construct_type(value=value, type_=type_)
+    return construct_type(value=value, type_=type_, metadata=getattr(field, "metadata", None))
 
 
 def is_basemodel(type_: type) -> bool:
@@ -420,7 +421,7 @@ def construct_type_unchecked(*, value: object, type_: type[_T]) -> _T:
     return cast(_T, construct_type(value=value, type_=type_))
 
 
-def construct_type(*, value: object, type_: object) -> object:
+def construct_type(*, value: object, type_: object, metadata: Optional[List[Any]] = None) -> object:
     """Loose coercion to the expected type with construction of nested values.
 
     If the given value does not match the expected type then it is returned as-is.
@@ -438,8 +439,10 @@ def construct_type(*, value: object, type_: object) -> object:
         type_ = type_.__value__  # type: ignore[unreachable]
 
     # unwrap `Annotated[T, ...]` -> `T`
-    if is_annotated_type(type_):
-        meta: tuple[Any, ...] = get_args(type_)[1:]
+    if metadata is not None:
+        meta: tuple[Any, ...] = tuple(metadata)
+    elif is_annotated_type(type_):
+        meta = get_args(type_)[1:]
         type_ = extract_type_arg(type_, 0)
     else:
         meta = tuple()
@@ -737,6 +740,7 @@ class FinalRequestOptionsInput(TypedDict, total=False):
     idempotency_key: str
     json_data: Body
     extra_json: AnyMapping
+    follow_redirects: bool
 
 
 @final
@@ -750,6 +754,7 @@ class FinalRequestOptions(pydantic.BaseModel):
     files: Union[HttpxRequestFiles, None] = None
     idempotency_key: Union[str, None] = None
     post_parser: Union[Callable[[Any], Any], NotGiven] = NotGiven()
+    follow_redirects: Union[bool, None] = None
 
     # It should be noted that we cannot use `json` here as that would override
     # a BaseModel method in an incompatible fashion.

unifieddatalibrary/_types.py

@@ -100,6 +100,7 @@ class RequestOptions(TypedDict, total=False):
     params: Query
     extra_json: AnyMapping
     idempotency_key: str
+    follow_redirects: bool
 
 
 # Sentinel class used until PEP 0661 is accepted
@@ -215,3 +216,4 @@ class _GenericAlias(Protocol):
 
 class HttpxSendArgs(TypedDict, total=False):
     auth: httpx.Auth
+    follow_redirects: bool

unifieddatalibrary/_utils/_transform.py

@@ -231,7 +231,7 @@ def _format_data(data: object, format_: PropertyFormat, format_template: str | N
     if isinstance(data, (date, datetime)):
         if format_ == "iso8601":
             if isinstance(data, datetime):
-                return data.strftime('%Y-%m-%dT%H:%M:%S.%fZ')
+                return data.strftime("%Y-%m-%dT%H:%M:%S.%fZ")
             return data.isoformat()
 
         if format_ == "custom" and format_template is not None:
@@ -395,7 +395,7 @@ async def _async_format_data(data: object, format_: PropertyFormat, format_templ
     if isinstance(data, (date, datetime)):
         if format_ == "iso8601":
             if isinstance(data, datetime):
-                return data.strftime('%Y-%m-%dT%H:%M:%S.%fZ')
+                return data.strftime("%Y-%m-%dT%H:%M:%S.%fZ")
             return data.isoformat()
 
         if format_ == "custom" and format_template is not None:

unifieddatalibrary/_utils/_utils.py

@@ -417,6 +417,6 @@ def json_safe(data: object) -> object:
         return [json_safe(item) for item in data]
 
     if isinstance(data, (datetime, date)):
-        return data.strftime('%Y-%m-%dT%H:%M:%S.%fZ')
+        return data.strftime("%Y-%m-%dT%H:%M:%S.%fZ")
 
     return data

unifieddatalibrary/_version.py

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

unifieddatalibrary/lib/common.py

@@ -1,3 +1,2 @@
 UDL_DATETIME_FORMAT = "%Y-%m-%dT%H:%M:%S.%fZ"
 KAFKA_HEADER = "KAFKA_NEXT_OFFSET"
-

unifieddatalibrary/lib/model_based_query.py

@@ -1,10 +1,12 @@
 # Proposed approach 2 for supporting query parameters with dynamic method generation
 
-from typing import Type, Dict, Any, Generic, TypeVar, Callable
+from typing import Any, Dict, Type, Generic, TypeVar, Callable
+
 from pydantic import BaseModel
 
 T = TypeVar("T", bound=BaseModel)
 
+
 class Query(Generic[T]):
     """
     A dynamic query builder that generates field-based filter methods at runtime.
@@ -48,7 +50,7 @@ class Query(Generic[T]):
         }
 
         # Dynamically create query methods for each field/operator combination
-        for field_name, field_info in model.model_fields.items():
+        for field_name, _field_info in model.model_fields.items():
             for suffix, operator in operators.items():
                 method_name = f"{field_name}{suffix}"
                 method = self._make_method(field_name, operator)
@@ -65,8 +67,10 @@ class Query(Generic[T]):
         Returns:
             A callable method that accepts a value and returns the updated Query object.
         """
+
         def method(self: "Query[T]", value: Any) -> "Query[T]":
             return self._add_filter(field_name, operator, value)
+
         return method
 
     def _add_filter(self, field_name: str, operator: str, value: Any) -> "Query[T]":

unifieddatalibrary/lib/query_field_names.py

@@ -1,10 +1,11 @@
 # Proposed approach 1 for supporting query parameters in a typed and composable way
 
-from typing import Type, Dict, Any, Generic, TypeVar, Protocol, Union, Tuple, cast
-from pydantic import BaseModel
+from typing import Any, Dict, Type, Tuple, Union, Generic, TypeVar, Protocol, cast
 from datetime import datetime
-from .util import sanitize_datetime
 
+from pydantic import BaseModel
+
+from .util import sanitize_datetime
 
 T = TypeVar("T", bound=BaseModel)
 
@@ -16,6 +17,7 @@ class QueryField(Protocol):
     Each method represents a filter operation that returns a modified Query object
     with the new filter applied.
     """
+
     def eq(self, value: Any) -> "Query[Any]": ...
     def gte(self, value: Any) -> "Query[Any]": ...
     def lte(self, value: Any) -> "Query[Any]": ...
@@ -91,7 +93,7 @@ class Query(Generic[T]):
                 if model_info is None:
                     raise KeyError(f"{self.field_name} does not exist in {self.query.model}")
 
-                key = getattr(model_info, 'alias', None) or self.field_name
+                key = getattr(model_info, "alias", None) or self.field_name
 
                 if operator == "between":
                     if not isinstance(value, tuple) or len(value) != 2:

unifieddatalibrary/lib/util.py

@@ -1,7 +1,8 @@
-
 from datetime import datetime, timezone
+
 from .common import UDL_DATETIME_FORMAT
 
+
 def sanitize_datetime(val: datetime) -> str:
     """Takes a datetime argument val and returns the same datetime converted to UTC.
     If tzinfo is not set, assumes local time in conversion."""

unifieddatalibrary/pagination.py

@@ -1,17 +1,23 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import Any, List, Type, Generic, Mapping, TypeVar, Optional, cast
+from typing import Any, List, Type, Generic, Mapping, TypeVar, Callable, Optional, cast
 from typing_extensions import override
 
-from httpx import Response
+from httpx import URL, Response
 
 from ._utils import is_mapping
 from ._models import BaseModel
 from ._base_client import BasePage, PageInfo, BaseSyncPage, BaseAsyncPage
 
-__all__ = ["SyncOffsetPage", "AsyncOffsetPage"]
+__all__ = [
+    "SyncOffsetPage",
+    "AsyncOffsetPage",
+    "SyncKafkaOffsetPage",
+    "AsyncKafkaOffsetPage",
+]
 
 _BaseModelT = TypeVar("_BaseModelT", bound=BaseModel)
+_BasePageT = TypeVar("_BasePageT", bound=BasePage[Any])
 
 _T = TypeVar("_T")
 
@@ -38,12 +44,16 @@ class SyncOffsetPage(BaseSyncPage[_T], BasePage[_T], Generic[_T]):
         return PageInfo(params={"firstResult": current_count})
 
     @classmethod
-    def build(cls: Type[_BaseModelT], *, response: Response, data: object) -> _BaseModelT:  # noqa: ARG003
-        return cls.construct(
-            None,
-            **{
-                **(cast(Mapping[str, Any], data) if is_mapping(data) else {"items": data}),
-            },
+    @override
+    def build(cls: Type[_BasePageT], *, response: Response, data: object) -> _BasePageT:  # noqa: ARG003
+        return cls._with_response(
+            cls.construct(
+                None,
+                **{
+                    **(cast(Mapping[str, Any], data) if is_mapping(data) else {"items": data}),
+                },
+            ),
+            response,
         )
 
 
@@ -69,10 +79,114 @@ class AsyncOffsetPage(BaseAsyncPage[_T], BasePage[_T], Generic[_T]):
         return PageInfo(params={"firstResult": current_count})
 
     @classmethod
-    def build(cls: Type[_BaseModelT], *, response: Response, data: object) -> _BaseModelT:  # noqa: ARG003
-        return cls.construct(
-            None,
-            **{
-                **(cast(Mapping[str, Any], data) if is_mapping(data) else {"items": data}),
-            },
+    @override
+    def build(cls: Type[_BasePageT], *, response: Response, data: object) -> _BasePageT:  # noqa: ARG003
+        return cls._with_response(
+            cls.construct(
+                None,
+                **{
+                    **(cast(Mapping[str, Any], data) if is_mapping(data) else {"items": data}),
+                },
+            ),
+            response,
+        )
+
+
+class SyncKafkaOffsetPage(BaseSyncPage[_T], BasePage[_T], Generic[_T]):
+    """Pagination for Kafka-style endpoints that return the next offset in a response header."""
+
+    items: List[_T]
+    url_builder: Callable[[int], str]
+
+    @staticmethod
+    def with_url_builder(fn: Callable[[int], str]) -> Type["SyncKafkaOffsetPage[object]"]:
+        """Create a page class with a URL builder for constructing next page URLs."""
+
+        class PageWithBuilder(SyncKafkaOffsetPage[object]):
+            url_builder = fn
+
+        return PageWithBuilder
+
+    @override
+    def _get_page_items(self) -> List[_T]:
+        items = self.items
+        if not items:
+            return []
+        return items
+
+    @override
+    def next_page_info(self) -> Optional[PageInfo]:
+        next_offset_str = self._response.headers.get("KAFKA_NEXT_OFFSET")
+        if not next_offset_str:
+            return None
+
+        try:
+            next_offset = int(next_offset_str)
+        except ValueError:
+            return None
+
+        new_url = self.url_builder(next_offset)
+        return PageInfo(url=URL(new_url))
+
+    @classmethod
+    @override
+    def build(cls: Type[_BasePageT], *, response: Response, data: object) -> _BasePageT:  # noqa: ARG003
+        return cls._with_response(
+            cls.construct(
+                None,
+                **{
+                    **(cast(Mapping[str, Any], data) if is_mapping(data) else {"items": data}),
+                },
+            ),
+            response,
+        )
+
+
+class AsyncKafkaOffsetPage(BaseAsyncPage[_T], BasePage[_T], Generic[_T]):
+    """Async pagination for Kafka-style endpoints that return the next offset in a response header."""
+
+    items: List[_T]
+    url_builder: Callable[[int], str]
+
+    @staticmethod
+    def with_url_builder(fn: Callable[[int], str]) -> Type["AsyncKafkaOffsetPage[object]"]:
+        """Create a page class with a URL builder for constructing next page URLs."""
+
+        class PageWithBuilder(AsyncKafkaOffsetPage[object]):
+            url_builder = fn
+
+        return PageWithBuilder
+
+    @override
+    def _get_page_items(self) -> List[_T]:
+        items = self.items
+        if not items:
+            return []
+        return items
+
+    @override
+    def next_page_info(self) -> Optional[PageInfo]:
+        next_offset_str = self._response.headers.get("KAFKA_NEXT_OFFSET")
+        if not next_offset_str:
+            return None
+
+        try:
+            next_offset = int(next_offset_str)
+        except ValueError:
+            return None
+
+        new_url = self.url_builder(next_offset)
+        return PageInfo(url=URL(new_url))
+
+    @classmethod
+    @override
+    def build(cls: Type[_BasePageT], *, response: Response, data: object) -> _BasePageT:  # noqa: ARG003
+        return cls._with_response(
+            cls.construct(
+                None,
+                **{
+                    **(cast(Mapping[str, Any], data) if is_mapping(data) else {"items": data}),
+                },
+            ),
+            response,
         )

unifieddatalibrary/resources/air_events.py

@@ -33,6 +33,7 @@ from .._base_client import AsyncPaginator, make_request_options
 from ..types.air_event_get_response import AirEventGetResponse
 from ..types.air_event_list_response import AirEventListResponse
 from ..types.air_event_tuple_response import AirEventTupleResponse
+from ..types.air_event_queryhelp_response import AirEventQueryhelpResponse
 
 __all__ = ["AirEventsResource", "AsyncAirEventsResource"]
 
@@ -765,18 +766,17 @@ class AirEventsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> AirEventQueryhelpResponse:
         """
         Service operation to provide detailed information on available dynamic query
         parameters for a particular data type.
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return self._get(
             "/udl/airevent/queryhelp",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=AirEventQueryhelpResponse,
         )
 
     def tuple(
@@ -1600,18 +1600,17 @@ class AsyncAirEventsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> AirEventQueryhelpResponse:
         """
         Service operation to provide detailed information on available dynamic query
         parameters for a particular data type.
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return await self._get(
             "/udl/airevent/queryhelp",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=AirEventQueryhelpResponse,
         )
 
     async def tuple(

unifieddatalibrary/resources/air_load_plans.py

@@ -30,6 +30,7 @@ from .._base_client import AsyncPaginator, make_request_options
 from ..types.airloadplan_full import AirloadplanFull
 from ..types.airloadplan_abridged import AirloadplanAbridged
 from ..types.air_load_plan_tuple_response import AirLoadPlanTupleResponse
+from ..types.air_load_plan_queryhelp_response import AirLoadPlanQueryhelpResponse
 
 __all__ = ["AirLoadPlansResource", "AsyncAirLoadPlansResource"]
 
@@ -564,18 +565,17 @@ class AirLoadPlansResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> AirLoadPlanQueryhelpResponse:
         """
         Service operation to provide detailed information on available dynamic query
         parameters for a particular data type.
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return self._get(
             "/udl/airloadplan/queryhelp",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=AirLoadPlanQueryhelpResponse,
         )
 
     def tuple(
@@ -1170,18 +1170,17 @@ class AsyncAirLoadPlansResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> AirLoadPlanQueryhelpResponse:
         """
         Service operation to provide detailed information on available dynamic query
         parameters for a particular data type.
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return await self._get(
             "/udl/airloadplan/queryhelp",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=AirLoadPlanQueryhelpResponse,
         )
 
     async def tuple(