udl-sdk 0.1.0a11__py3-none-any.whl → 0.1.0a13__py3-none-any.whl

{udl_sdk-0.1.0a11.dist-info → udl_sdk-0.1.0a13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: udl-sdk
-Version: 0.1.0a11
+Version: 0.1.0a13
 Summary: The official Python library for the unifieddatalibrary API
 Project-URL: Homepage, https://github.com/Bluestaq/udl-python-sdk
 Project-URL: Repository, https://github.com/Bluestaq/udl-python-sdk
@@ -18,6 +18,7 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Typing :: Typed
 Requires-Python: >=3.8
@@ -29,12 +30,13 @@ Requires-Dist: sniffio
 Requires-Dist: typing-extensions<5,>=4.10
 Provides-Extra: aiohttp
 Requires-Dist: aiohttp; extra == 'aiohttp'
-Requires-Dist: httpx-aiohttp>=0.1.6; extra == 'aiohttp'
+Requires-Dist: httpx-aiohttp>=0.1.8; extra == 'aiohttp'
 Description-Content-Type: text/markdown
 
 # Unifieddatalibrary Python API library
 
-[![PyPI version](https://github.com/Bluestaq/udl-python-sdk/tree/main/<https://img.shields.io/pypi/v/udl-sdk.svg?label=pypi%20(stable)>)](https://pypi.org/project/udl-sdk/)
+<!-- prettier-ignore -->
+[![PyPI version](https://img.shields.io/pypi/v/udl-sdk.svg?label=pypi%20(stable))](https://pypi.org/project/udl-sdk/)
 
 The Unifieddatalibrary Python library provides convenient access to the Unifieddatalibrary REST API from any Python 3.8+
 application. The library includes type definitions for all request params and response fields,
@@ -114,7 +116,6 @@ pip install --pre udl-sdk[aiohttp]
 Then you can enable it by instantiating the client with `http_client=DefaultAioHttpClient()`:
 
 ```python
-import os
 import asyncio
 from unifieddatalibrary import DefaultAioHttpClient
 from unifieddatalibrary import AsyncUnifieddatalibrary
@@ -122,8 +123,8 @@ from unifieddatalibrary import AsyncUnifieddatalibrary
 
 async def main() -> None:
     async with AsyncUnifieddatalibrary(
-        username=os.environ.get("UDL_AUTH_USERNAME"),  # This is the default and can be omitted
-        password=os.environ.get("UDL_AUTH_PASSWORD"),  # This is the default and can be omitted
+        username="My Username",
+        password="My Password",
         http_client=DefaultAioHttpClient(),
     ) as client:
         page = await client.elsets.current.list()

{udl_sdk-0.1.0a11.dist-info → udl_sdk-0.1.0a13.dist-info}/RECORD

@@ -1,18 +1,18 @@
 unifieddatalibrary/__init__.py,sha256=562u3LZ5X6bc29nGRt5uXC91XbddnlSfk7tSqKA4yVk,2727
-unifieddatalibrary/_base_client.py,sha256=LZy7nMp9cjwI0TIUOO0mzkhuoHa4gjhj96WDINLReiM,66727
+unifieddatalibrary/_base_client.py,sha256=kv98LleyLFwecp-SyYKZVR-3Dt5OYNbmB_SQLIen6Yo,67579
 unifieddatalibrary/_client.py,sha256=mrBWoR1advnC0xyt7nMU-eiLqRf8ZBFqU8xFsrP4grU,136998
 unifieddatalibrary/_compat.py,sha256=VWemUKbj6DDkQ-O4baSpHVLJafotzeXmCQGJugfVTIw,6580
 unifieddatalibrary/_constants.py,sha256=S14PFzyN9-I31wiV7SmIlL5Ga0MLHxdvegInGdXH7tM,462
 unifieddatalibrary/_exceptions.py,sha256=rkk8r4oyqb4bxjMtx0OFWsh1m8gsMfrnIGnNPO6zRz0,3244
 unifieddatalibrary/_files.py,sha256=FZ264pl2ebvWlVztvuAE8hl1yitTQIT-2Bi6GMwMReA,3619
-unifieddatalibrary/_models.py,sha256=G1vczEodX0vUySeVKbF-mbzlaObNL1oVAYH4c65agRk,29131
+unifieddatalibrary/_models.py,sha256=viD5E6aDMhxslcFHDYvkHaKzE8YLcNmsPsMe8STixvs,29294
 unifieddatalibrary/_qs.py,sha256=AOkSz4rHtK4YI3ZU_kzea-zpwBUgEY8WniGmTPyEimc,4846
 unifieddatalibrary/_resource.py,sha256=Ik-pULzkvFIY2OgB9Ra7mIQKCle38FSP36dWWCH9vxk,1172
 unifieddatalibrary/_response.py,sha256=hdekMRMvxTvYdKfYIPvAxSpdiuRILRCYd5Dwcye-icg,28890
 unifieddatalibrary/_streaming.py,sha256=LwKrocz7ZRmYd47TA3q-PLXwgdTgjANE-TCIRZB958s,10148
 unifieddatalibrary/_types.py,sha256=mslWUKYM1Q3bMXxgq4Mr9fo3QvAwzQJrscV44EPbmGA,6209
-unifieddatalibrary/_version.py,sha256=E6BVaJ5lc_V2L0pH5HcTrZMelpC26PdpKUtkXqvw4wo,179
-unifieddatalibrary/pagination.py,sha256=Uuf4Q1bzJIgeFOadEoqFrzO8xbzmvTR6RJmO0cYM6zw,2415
+unifieddatalibrary/_version.py,sha256=4xoSZ-FMWPBL5-DynK0nN9F1JSHXcetMW2_4yc6quMk,179
+unifieddatalibrary/pagination.py,sha256=jRNsKPCwBHKIHIIX5V5yB7NUsvfy-k7UqnpPGrGyFNU,5758
 unifieddatalibrary/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unifieddatalibrary/_utils/__init__.py,sha256=PNZ_QJuzZEgyYXqkO1HVhGkj5IU9bglVUcw7H-Knjzw,2062
 unifieddatalibrary/_utils/_logs.py,sha256=mZd6C1F4ajWju5tJvYN_5ATHUi2zEzhcBQqJafS7Two,810
@@ -21,14 +21,14 @@ unifieddatalibrary/_utils/_reflection.py,sha256=ZmGkIgT_PuwedyNBrrKGbxoWtkpytJNU
 unifieddatalibrary/_utils/_resources_proxy.py,sha256=IEVfHgsswOT5HDg1cmdTtY8oOs3fo74UtR1aO6yMTZU,649
 unifieddatalibrary/_utils/_streams.py,sha256=SMC90diFFecpEg_zgDRVbdR3hSEIgVVij4taD-noMLM,289
 unifieddatalibrary/_utils/_sync.py,sha256=TpGLrrhRNWTJtODNE6Fup3_k7zrWm1j2RlirzBwre-0,2862
-unifieddatalibrary/_utils/_transform.py,sha256=sb9J55txMzvFJv2j7eEwiksN2T5VKvKaG80V_fpONlc,15827
+unifieddatalibrary/_utils/_transform.py,sha256=V2pcnNHKQvgqpShVjV1WjRtrWlCWWdIkZP_HduzypNg,15827
 unifieddatalibrary/_utils/_typing.py,sha256=D0DbbNu8GnYQTSICnTSHDGsYXj8TcAKyhejb0XcnjtY,4602
-unifieddatalibrary/_utils/_utils.py,sha256=PIEBlcoQMcd2YCTmSpJSTfYhH2g3FmjHCHCvtOfagm8,12334
+unifieddatalibrary/_utils/_utils.py,sha256=LLsABnJOF9OiH4aE5_nRwIya7lcn0b6BxBHY1Zgs_K8,12334
 unifieddatalibrary/lib/.keep,sha256=wuNrz-5SXo3jJaJOJgz4vFHM41YH_g20F5cRQo0vLes,224
-unifieddatalibrary/lib/common.py,sha256=x0DdmV6AIYIBkR4crwI3LAyL9wPL3lXFpZ9321A90TA,85
-unifieddatalibrary/lib/model_based_query.py,sha256=RZ1l-fBUj6HgUPb7V8V8ch-tLn4DbdrtHSAn8tX-hPE,4364
-unifieddatalibrary/lib/query_field_names.py,sha256=6YNz47bJGV5ICJXADECGCySjvnrREnfGSV-URAJkBew,5695
-unifieddatalibrary/lib/util.py,sha256=A2suZHIcXpW0ogkmluIhQvTRaSZWwpRGCZPkycRRspI,602
+unifieddatalibrary/lib/common.py,sha256=nKvnmTVUrTib17inEUTUcRKCk7xcVPfuRDJdjmzwHBU,83
+unifieddatalibrary/lib/model_based_query.py,sha256=xU9ixbT-sTejHNgLwJE8rcQ0KZA7V9iwV_7JxBtma9g,4373
+unifieddatalibrary/lib/query_field_names.py,sha256=KSLSGM6mKMqNRpEL67xq83m9jCOscsZlEzG1F1kzmx0,5699
+unifieddatalibrary/lib/util.py,sha256=9RfOZDHVPIAHh7Gp64zwZY0EFrHIldEgQ5rOhWFgGMI,604
 unifieddatalibrary/resources/__init__.py,sha256=lH1bSmd-k2lILj5bvgQyopgXics8PU-XX_7N5syfaEw,86715
 unifieddatalibrary/resources/air_events.py,sha256=mTbkac1Uttbqy6DWU0m_Xo4QQbjP6VkeMRA53XeC2Kc,83645
 unifieddatalibrary/resources/air_load_plans.py,sha256=-uLt1LheEw8MYv_DUs4MtjEvlsFy6NtekEqBOqYMkRg,61618
@@ -102,7 +102,7 @@ unifieddatalibrary/resources/rf_emitter_details.py,sha256=j_aQc0CrkkBYScJGyg2LPh
 unifieddatalibrary/resources/route_stats.py,sha256=tuKsGa6TVbKeS8xWFfKqTuXwi55CzZUZepXBaJIAhVU,71061
 unifieddatalibrary/resources/scientific.py,sha256=3z3Opivmv8PRULmRHrRBnr7J1yvcaZYuLdUcDbRxuUo,48200
 unifieddatalibrary/resources/scs_views.py,sha256=Eda3GPy4XyDVPzslQiyT3glccbM3n9xRSd03B2c7qKw,7308
-unifieddatalibrary/resources/secure_messaging.py,sha256=iVAJCU-ILKI4qV_5So91k0jVpBy86zW6y4xUSdHb6iI,18269
+unifieddatalibrary/resources/secure_messaging.py,sha256=L6Gn0JKk6mGea1AecP2B7CDPBHZYrLMMvlrO4VM0lZU,18646
 unifieddatalibrary/resources/sensor_observation_type.py,sha256=ZPc7MyrNehH8Zk__eXqf-NPYBUYLrS2jORYnrFMZveE,13935
 unifieddatalibrary/resources/sensor_type.py,sha256=6x9ASy-L4IBsceELFtbdZU2kuInXblLAAyYlz-8rTPY,13577
 unifieddatalibrary/resources/sera_data_comm_details.py,sha256=zEcbCknfEybPh-IqCHsMSfMPJRkZH5etYYYe_slusk4,65774
@@ -2610,7 +2610,7 @@ unifieddatalibrary/types/weather_report/history_count_params.py,sha256=KFXFsL7q4
 unifieddatalibrary/types/weather_report/history_count_response.py,sha256=ZAHTF5IoOvMAg9GyvtMvgeowh-ythtIJhcWtMbcM8wk,202
 unifieddatalibrary/types/weather_report/history_list_params.py,sha256=JUd53E70wjtn7HlQ7LOZ7lUJREDJkaDYHv2eYW60T3Q,1019
 unifieddatalibrary/types/weather_report/weather_report_full.py,sha256=iv7fHg9aQh6ztvl0ztve5Z3_8Ww-efPMBgrS0n5gXvM,20751
-udl_sdk-0.1.0a11.dist-info/METADATA,sha256=ZNG-6Rl3UDQks01rGPNRlVU7c0bYMm6E1w5-SEcQZOc,17131
-udl_sdk-0.1.0a11.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-udl_sdk-0.1.0a11.dist-info/licenses/LICENSE,sha256=YBMC8KbJHXtxIo1-d_G5WdWjtiwFyrcZ5jAsRwN4POI,11348
-udl_sdk-0.1.0a11.dist-info/RECORD,,
+udl_sdk-0.1.0a13.dist-info/METADATA,sha256=c8rfYagi-Yyvx5QmrHo8w20z1Nf1SwowVLNvbkTf_2Y,17014
+udl_sdk-0.1.0a13.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+udl_sdk-0.1.0a13.dist-info/licenses/LICENSE,sha256=YBMC8KbJHXtxIo1-d_G5WdWjtiwFyrcZ5jAsRwN4POI,11348
+udl_sdk-0.1.0a13.dist-info/RECORD,,

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,
         )
 

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()

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.11"  # x-release-please-version
+__version__ = "0.1.0-alpha.13"  # 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/secure_messaging.py

@@ -4,6 +4,8 @@ from __future__ import annotations
 
 import httpx
 
+from unifieddatalibrary.pagination import SyncKafkaOffsetPage, AsyncKafkaOffsetPage
+
 from ..types import (
     secure_messaging_get_messages_params,
     secure_messaging_describe_topic_params,
@@ -150,7 +152,7 @@ class SecureMessagingResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> SyncKafkaOffsetPage[object]:
         """Retrieve a set of messages from the given topic at the given offset.
 
         See Help >
@@ -168,8 +170,9 @@ class SecureMessagingResource(SyncAPIResource):
         if not topic:
             raise ValueError(f"Expected a non-empty value for `topic` but received {topic!r}")
         extra_headers = {"Accept": "*/*", **(extra_headers or {})}
-        return self._get(
+        return self._get_api_list(
             f"/sm/getMessages/{topic}/{offset}",
+            page=SyncKafkaOffsetPage.with_url_builder(lambda next_offset: f"/sm/getMessages/{topic}/{next_offset}"),
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -183,7 +186,7 @@ class SecureMessagingResource(SyncAPIResource):
                     secure_messaging_get_messages_params.SecureMessagingGetMessagesParams,
                 ),
             ),
-            cast_to=NoneType,
+            model=object,
         )
 
     def list_topics(
@@ -330,7 +333,7 @@ class AsyncSecureMessagingResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> AsyncKafkaOffsetPage[object]:
         """Retrieve a set of messages from the given topic at the given offset.
 
         See Help >
@@ -348,8 +351,9 @@ class AsyncSecureMessagingResource(AsyncAPIResource):
         if not topic:
             raise ValueError(f"Expected a non-empty value for `topic` but received {topic!r}")
         extra_headers = {"Accept": "*/*", **(extra_headers or {})}
-        return await self._get(
+        return await self._get_api_list(
             f"/sm/getMessages/{topic}/{offset}",
+            page=AsyncKafkaOffsetPage.with_url_builder(lambda next_offset: f"/sm/getMessages/{topic}/{next_offset}"),
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -363,7 +367,7 @@ class AsyncSecureMessagingResource(AsyncAPIResource):
                     secure_messaging_get_messages_params.SecureMessagingGetMessagesParams,
                 ),
             ),
-            cast_to=NoneType,
+            model=object,
         )
 
     async def list_topics(