File tree Expand file tree Collapse file tree 3 files changed +6
-7
lines changed
src/apify/storage_clients/_apify Expand file tree Collapse file tree 3 files changed +6
-7
lines changed Original file line number Diff line number Diff line change 1111from crawlee .storage_clients .models import KeyValueStoreRecord , KeyValueStoreRecordMetadata
1212
1313from ._api_client_creation import create_storage_api_client
14- from ._models import ApifyKeyValueStoreMetadata , KeyValueStoreListKeysPage
14+ from ._models import ApifyKeyValueStoreMetadata
1515
1616if TYPE_CHECKING :
1717 from collections .abc import AsyncIterator
@@ -143,14 +143,13 @@ async def iterate_keys(
143143 count = 0
144144
145145 while True :
146- response = await self ._api_client .list_keys (exclusive_start_key = exclusive_start_key )
147- list_key_page = KeyValueStoreListKeysPage .model_validate (response )
146+ list_key_page = await self ._api_client .list_keys (exclusive_start_key = exclusive_start_key )
148147
149148 for item in list_key_page .items :
150149 # Convert KeyValueStoreKeyInfo to KeyValueStoreRecordMetadata
151150 record_metadata = KeyValueStoreRecordMetadata (
152151 key = item .key ,
153- size = item .size ,
152+ size = int ( item .size ) ,
154153 content_type = 'application/octet-stream' , # Content type not available from list_keys
155154 )
156155 yield record_metadata
Original file line number Diff line number Diff line change @@ -442,7 +442,7 @@ async def _list_head(
442442 self .metadata .had_multiple_clients = list_and_lost_data .had_multiple_clients
443443
444444 for request_data in list_and_lost_data .items :
445- request = Request .model_validate (request_data )
445+ request = Request .model_validate (request_data . model_dump ( by_alias = True ) )
446446 request_id = request_data .id
447447
448448 # Skip requests without ID or unique key
Original file line number Diff line number Diff line change @@ -297,7 +297,7 @@ async def _list_head(self) -> None:
297297
298298 # Update the cached data
299299 for request_data in response .items :
300- request = Request .model_validate (request_data )
300+ request = Request .model_validate (request_data . model_dump ( by_alias = True ) )
301301 request_id = request_data .id
302302
303303 if request_id in self ._requests_in_progress :
@@ -380,7 +380,7 @@ async def _init_caches(self) -> None:
380380 """
381381 response = await self ._api_client .list_requests (limit = 10_000 )
382382 for request_data in response .items :
383- request = Request .model_validate (request_data )
383+ request = Request .model_validate (request_data . model_dump ( by_alias = True ) )
384384 request_id = request_data .id
385385
386386 if request .was_already_handled :
You can’t perform that action at this time.
0 commit comments