fix: Correct return type of storage get_info methods (#200)

Author: janbuchar

src/crawlee/storages/dataset.py

@@ -232,11 +232,7 @@ async def export_to(self, **kwargs: Unpack[ExportToKwargs]) -> None:
         raise ValueError(f'Unsupported content type: {content_type}')
 
     async def get_info(self) -> DatasetMetadata | None:
-        """Get an object containing general information about the dataset.
-
-        Returns:
-            Object returned by calling the GET dataset API endpoint.
-        """
+        """Get an object containing general information about the dataset."""
         metadata = await self._resource_client.get()
         if isinstance(metadata, DatasetMetadata):
             return metadata

src/crawlee/storages/key_value_store.py

@@ -4,7 +4,7 @@
 
 from typing_extensions import override
 
-from crawlee.models import KeyValueStoreKeyInfo
+from crawlee.models import KeyValueStoreKeyInfo, KeyValueStoreMetadata
 from crawlee.storages.base_storage import BaseStorage
 
 if TYPE_CHECKING:
@@ -59,6 +59,10 @@ def id(self) -> str:
     def name(self) -> str | None:
         return self._name
 
+    async def get_info(self) -> KeyValueStoreMetadata | None:
+        """Get an object containing general information about the key value store."""
+        return await self._resource_client.get()
+
     @override
     @classmethod
     async def open(

src/crawlee/storages/request_queue.py

@@ -18,6 +18,7 @@
     ProcessedRequest,
     Request,
     RequestQueueHeadState,
+    RequestQueueMetadata,
 )
 from crawlee.storages.base_storage import BaseStorage
 from crawlee.storages.request_provider import RequestProvider
@@ -27,7 +28,6 @@
 
     from crawlee.base_storage_client import BaseStorageClient
     from crawlee.configuration import Configuration
-    from crawlee.models import BaseStorageMetadata
 
 logger = getLogger(__name__)
 
@@ -458,12 +458,8 @@ async def is_finished(self) -> bool:
         is_head_consistent = await self.ensure_head_is_non_empty(ensure_consistency=True)
         return is_head_consistent and len(self._queue_head_dict) == 0 and self._in_progress_count() == 0
 
-    async def get_info(self) -> BaseStorageMetadata | None:
-        """Get an object containing general information about the request queue.
-
-        Returns:
-            Object returned by calling the GET request queue API endpoint.
-        """
+    async def get_info(self) -> RequestQueueMetadata | None:
+        """Get an object containing general information about the request queue."""
         return await self._resource_client.get()
 
     @override