88
99from .... import _legacy_response
1010from ...._types import NOT_GIVEN , Body , Query , Headers , NotGiven
11- from ...._utils import maybe_transform
11+ from ...._utils import (
12+ maybe_transform ,
13+ async_maybe_transform ,
14+ )
1215from ...._compat import cached_property
1316from ...._resource import SyncAPIResource , AsyncAPIResource
1417from ...._response import to_streamed_response_wrapper , async_to_streamed_response_wrapper
2023 individual_retrieve_many_benefits_params ,
2124)
2225from ....types .hris .benefits .individual_benefit import IndividualBenefit
23- from ....types .hris .benefits .enrolled_individual import EnrolledIndividual
24- from ....types .hris .benefits .unenrolled_individual import UnenrolledIndividual
26+ from ....types .hris .benefits .individual_enroll_many_response import IndividualEnrollManyResponse
2527from ....types .hris .benefits .individual_enrolled_ids_response import IndividualEnrolledIDsResponse
2628
2729__all__ = ["Individuals" , "AsyncIndividuals" ]
@@ -58,7 +60,7 @@ def enroll_many(
5860 extra_query : Query | None = None ,
5961 extra_body : Body | None = None ,
6062 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
61- ) -> SyncSinglePage [ EnrolledIndividual ] :
63+ ) -> IndividualEnrollManyResponse :
6264 """Enroll an individual into a deduction or contribution.
6365
6466 This is an overwrite
@@ -79,15 +81,13 @@ def enroll_many(
7981 """
8082 if not benefit_id :
8183 raise ValueError (f"Expected a non-empty value for `benefit_id` but received { benefit_id !r} " )
82- return self ._get_api_list (
84+ return self ._post (
8385 f"/employer/benefits/{ benefit_id } /individuals" ,
84- page = SyncSinglePage [EnrolledIndividual ],
8586 body = maybe_transform (individuals , Iterable [individual_enroll_many_params .Individual ]),
8687 options = make_request_options (
8788 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
8889 ),
89- model = EnrolledIndividual ,
90- method = "post" ,
90+ cast_to = IndividualEnrollManyResponse ,
9191 )
9292
9393 def enrolled_ids (
@@ -179,7 +179,7 @@ def unenroll_many(
179179 extra_query : Query | None = None ,
180180 extra_body : Body | None = None ,
181181 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
182- ) -> SyncSinglePage [UnenrolledIndividual ]:
182+ ) -> SyncSinglePage [object ]:
183183 """
184184 Unenroll individuals from a deduction or contribution
185185
@@ -198,14 +198,14 @@ def unenroll_many(
198198 raise ValueError (f"Expected a non-empty value for `benefit_id` but received { benefit_id !r} " )
199199 return self ._get_api_list (
200200 f"/employer/benefits/{ benefit_id } /individuals" ,
201- page = SyncSinglePage [UnenrolledIndividual ],
201+ page = SyncSinglePage [object ],
202202 body = maybe_transform (
203203 {"individual_ids" : individual_ids }, individual_unenroll_many_params .IndividualUnenrollManyParams
204204 ),
205205 options = make_request_options (
206206 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
207207 ),
208- model = UnenrolledIndividual ,
208+ model = object ,
209209 method = "delete" ,
210210 )
211211
@@ -230,7 +230,7 @@ def with_streaming_response(self) -> AsyncIndividualsWithStreamingResponse:
230230 """
231231 return AsyncIndividualsWithStreamingResponse (self )
232232
233- def enroll_many (
233+ async def enroll_many (
234234 self ,
235235 benefit_id : str ,
236236 * ,
@@ -241,7 +241,7 @@ def enroll_many(
241241 extra_query : Query | None = None ,
242242 extra_body : Body | None = None ,
243243 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
244- ) -> AsyncPaginator [ EnrolledIndividual , AsyncSinglePage [ EnrolledIndividual ]] :
244+ ) -> IndividualEnrollManyResponse :
245245 """Enroll an individual into a deduction or contribution.
246246
247247 This is an overwrite
@@ -262,15 +262,13 @@ def enroll_many(
262262 """
263263 if not benefit_id :
264264 raise ValueError (f"Expected a non-empty value for `benefit_id` but received { benefit_id !r} " )
265- return self ._get_api_list (
265+ return await self ._post (
266266 f"/employer/benefits/{ benefit_id } /individuals" ,
267- page = AsyncSinglePage [EnrolledIndividual ],
268- body = maybe_transform (individuals , Iterable [individual_enroll_many_params .Individual ]),
267+ body = await async_maybe_transform (individuals , Iterable [individual_enroll_many_params .Individual ]),
269268 options = make_request_options (
270269 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
271270 ),
272- model = EnrolledIndividual ,
273- method = "post" ,
271+ cast_to = IndividualEnrollManyResponse ,
274272 )
275273
276274 async def enrolled_ids (
@@ -362,7 +360,7 @@ def unenroll_many(
362360 extra_query : Query | None = None ,
363361 extra_body : Body | None = None ,
364362 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
365- ) -> AsyncPaginator [UnenrolledIndividual , AsyncSinglePage [UnenrolledIndividual ]]:
363+ ) -> AsyncPaginator [object , AsyncSinglePage [object ]]:
366364 """
367365 Unenroll individuals from a deduction or contribution
368366
@@ -381,14 +379,14 @@ def unenroll_many(
381379 raise ValueError (f"Expected a non-empty value for `benefit_id` but received { benefit_id !r} " )
382380 return self ._get_api_list (
383381 f"/employer/benefits/{ benefit_id } /individuals" ,
384- page = AsyncSinglePage [UnenrolledIndividual ],
382+ page = AsyncSinglePage [object ],
385383 body = maybe_transform (
386384 {"individual_ids" : individual_ids }, individual_unenroll_many_params .IndividualUnenrollManyParams
387385 ),
388386 options = make_request_options (
389387 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
390388 ),
391- model = UnenrolledIndividual ,
389+ model = object ,
392390 method = "delete" ,
393391 )
394392
0 commit comments