Skip to content

Commit cc295c9

Browse files
committed
Fix
1 parent 7457578 commit cc295c9

File tree

3 files changed

+6
-7
lines changed

3 files changed

+6
-7
lines changed

src/apify/storage_clients/_apify/_key_value_store_client.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from crawlee.storage_clients.models import KeyValueStoreRecord, KeyValueStoreRecordMetadata
1212

1313
from ._api_client_creation import create_storage_api_client
14-
from ._models import ApifyKeyValueStoreMetadata, KeyValueStoreListKeysPage
14+
from ._models import ApifyKeyValueStoreMetadata
1515

1616
if 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

src/apify/storage_clients/_apify/_request_queue_shared_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

src/apify/storage_clients/_apify/_request_queue_single_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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:

0 commit comments

Comments
 (0)