tmp

Author: vdusek

src/apify/storage_clients/_apify/_dataset_client.py

@@ -126,6 +126,8 @@ async def open(
         # Normalize 'default' alias to None
         alias = None if alias == 'default' else alias
 
+        logger.info(f'Opening dataset with id={id}, name={name}, and alias={alias}')
+
         # Handle alias resolution
         if alias:
             # Try to resolve alias to existing storage ID

src/apify/storage_clients/_apify/_utils.py

@@ -37,9 +37,13 @@ async def resolve_alias_to_id(
 
     try:
         record = await default_kvs_client.get_record(_ALIAS_MAPPING_KEY)
+        logger.info(f'resolving record: {record}')
 
-        if isinstance(record, dict) and alias_key in record:
-            return str(record[alias_key])
+        # Extract the actual data from the KVS record
+        alias_data = record.get('value') if isinstance(record, dict) else record
+
+        if isinstance(alias_data, dict) and alias_key in alias_data:
+            return str(alias_data[alias_key])
 
     except Exception as exc:
         # If there's any error accessing the record, treat it as not found
@@ -69,15 +73,19 @@ async def store_alias_mapping(
 
     try:
         record = await default_kvs_client.get_record(_ALIAS_MAPPING_KEY)
+        logger.info(f'storing record: {record}')
+
+        # Extract the actual data from the KVS record
+        alias_data = record.get('value') if isinstance(record, dict) else None
 
         # Update or create the record with the new alias mapping
-        if isinstance(record, dict) and alias_key in record:
-            record[alias_key] = storage_id
+        if isinstance(alias_data, dict):
+            alias_data[alias_key] = storage_id
         else:
-            record = {alias_key: storage_id}
+            alias_data = {alias_key: storage_id}
 
         # Store the mapping back in the KVS.
-        await default_kvs_client.set_record(_ALIAS_MAPPING_KEY, record)
+        await default_kvs_client.set_record(_ALIAS_MAPPING_KEY, alias_data)
     except Exception as exc:
         logger.warning(f'Error accessing alias mapping for {alias}: {exc}')