10
10
from ...._compat import cached_property
11
11
from ...._resource import SyncAPIResource , AsyncAPIResource
12
12
from ...._response import (
13
- to_raw_response_wrapper ,
14
- to_streamed_response_wrapper ,
15
- async_to_raw_response_wrapper ,
16
- async_to_streamed_response_wrapper ,
13
+ BinaryAPIResponse ,
14
+ AsyncBinaryAPIResponse ,
15
+ StreamedBinaryAPIResponse ,
16
+ AsyncStreamedBinaryAPIResponse ,
17
+ to_custom_raw_response_wrapper ,
18
+ to_custom_streamed_response_wrapper ,
19
+ async_to_custom_raw_response_wrapper ,
20
+ async_to_custom_streamed_response_wrapper ,
17
21
)
18
22
from ...._base_client import (
19
23
make_request_options ,
@@ -42,7 +46,7 @@ def get(
42
46
extra_query : Query | None = None ,
43
47
extra_body : Body | None = None ,
44
48
timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
45
- ) -> object :
49
+ ) -> BinaryAPIResponse :
46
50
"""
47
51
Download specified LOA document under the account.
48
52
@@ -63,12 +67,13 @@ def get(
63
67
raise ValueError (f"Expected a non-empty value for `account_id` but received { account_id !r} " )
64
68
if not loa_document_id :
65
69
raise ValueError (f"Expected a non-empty value for `loa_document_id` but received { loa_document_id !r} " )
70
+ extra_headers = {"Accept" : "application/pdf" , ** (extra_headers or {})}
66
71
return self ._get (
67
72
f"/accounts/{ account_id } /addressing/loa_documents/{ loa_document_id } /download" ,
68
73
options = make_request_options (
69
74
extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
70
75
),
71
- cast_to = object ,
76
+ cast_to = BinaryAPIResponse ,
72
77
)
73
78
74
79
@@ -92,7 +97,7 @@ async def get(
92
97
extra_query : Query | None = None ,
93
98
extra_body : Body | None = None ,
94
99
timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
95
- ) -> object :
100
+ ) -> AsyncBinaryAPIResponse :
96
101
"""
97
102
Download specified LOA document under the account.
98
103
@@ -113,46 +118,51 @@ async def get(
113
118
raise ValueError (f"Expected a non-empty value for `account_id` but received { account_id !r} " )
114
119
if not loa_document_id :
115
120
raise ValueError (f"Expected a non-empty value for `loa_document_id` but received { loa_document_id !r} " )
121
+ extra_headers = {"Accept" : "application/pdf" , ** (extra_headers or {})}
116
122
return await self ._get (
117
123
f"/accounts/{ account_id } /addressing/loa_documents/{ loa_document_id } /download" ,
118
124
options = make_request_options (
119
125
extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
120
126
),
121
- cast_to = object ,
127
+ cast_to = AsyncBinaryAPIResponse ,
122
128
)
123
129
124
130
125
131
class DownloadsResourceWithRawResponse :
126
132
def __init__ (self , downloads : DownloadsResource ) -> None :
127
133
self ._downloads = downloads
128
134
129
- self .get = to_raw_response_wrapper (
135
+ self .get = to_custom_raw_response_wrapper (
130
136
downloads .get ,
137
+ BinaryAPIResponse ,
131
138
)
132
139
133
140
134
141
class AsyncDownloadsResourceWithRawResponse :
135
142
def __init__ (self , downloads : AsyncDownloadsResource ) -> None :
136
143
self ._downloads = downloads
137
144
138
- self .get = async_to_raw_response_wrapper (
145
+ self .get = async_to_custom_raw_response_wrapper (
139
146
downloads .get ,
147
+ AsyncBinaryAPIResponse ,
140
148
)
141
149
142
150
143
151
class DownloadsResourceWithStreamingResponse :
144
152
def __init__ (self , downloads : DownloadsResource ) -> None :
145
153
self ._downloads = downloads
146
154
147
- self .get = to_streamed_response_wrapper (
155
+ self .get = to_custom_streamed_response_wrapper (
148
156
downloads .get ,
157
+ StreamedBinaryAPIResponse ,
149
158
)
150
159
151
160
152
161
class AsyncDownloadsResourceWithStreamingResponse :
153
162
def __init__ (self , downloads : AsyncDownloadsResource ) -> None :
154
163
self ._downloads = downloads
155
164
156
- self .get = async_to_streamed_response_wrapper (
165
+ self .get = async_to_custom_streamed_response_wrapper (
157
166
downloads .get ,
167
+ AsyncStreamedBinaryAPIResponse ,
158
168
)
0 commit comments