craft-ai-sdk 0.62.1rc1__tar.gz → 0.62.2rc1__tar.gz

{craft_ai_sdk-0.62.1rc1 → craft_ai_sdk-0.62.2rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: craft-ai-sdk
-Version: 0.62.1rc1
+Version: 0.62.2rc1
 Summary: Craft AI MLOps platform SDK
 License: Apache-2.0
 Author: Craft AI

{craft_ai_sdk-0.62.1rc1 → craft_ai_sdk-0.62.2rc1}/craft_ai_sdk/__init__.py

@@ -13,4 +13,4 @@ from .io import (  # noqa: F401
 )
 from .sdk import CraftAiSdk  # noqa: F401
 
-__version__ = "0.62.1rc1"
+__version__ = "0.62.2rc1"

{craft_ai_sdk-0.62.1rc1 → craft_ai_sdk-0.62.2rc1}/craft_ai_sdk/core/data_store.py

@@ -1,5 +1,4 @@
 import io
-import urllib.parse
 
 import requests
 
@@ -34,9 +33,13 @@ def get_data_store_object_information(sdk: BaseCraftAiSdk, object_path_in_datast
     return result
 
 
-def list_data_store_objects(sdk: BaseCraftAiSdk):
+def list_data_store_objects(sdk: BaseCraftAiSdk, folder_path=None):
     """Get the list of the objects stored in the data store.
 
+    Args:
+        folder_path (:obj:`str`, optional): Location of the requested folder in the data
+            store.
+
     Returns:
         :obj:`list` of :obj:`dict`: List of objects in the data store represented
         as :obj:`dict` with the following keys:
@@ -47,12 +50,12 @@ def list_data_store_objects(sdk: BaseCraftAiSdk):
             * ``"size"`` (:obj:`int`): The size of the object in bytes.
     """
     url = f"{sdk.base_environment_api_url}/data-store/list"
-
-    result = sdk._get(url)
+    params = {"folder_path": folder_path} if folder_path else {}
+    result = sdk._get(url, params=params)
     all_items = result["items"]
     while result.get("continuation_token", None):
-        query = {"continuation_token": result["continuation_token"]}
-        result = sdk._get(f"{url}?{urllib.parse.urlencode(query)}")
+        params = {"continuation_token": result["continuation_token"], **params}
+        result = sdk._get(url, params=params)
         all_items.extend(result["items"])
 
     for object in all_items:

{craft_ai_sdk-0.62.1rc1 → craft_ai_sdk-0.62.2rc1}/craft_ai_sdk/sdk.py

@@ -136,7 +136,7 @@ class CraftAiSdk(BaseCraftAiSdk):
         os.environ.get("CRAFT_AI__MULTIPART_PART_SIZE__B", str(38 * 256 * 1024))
     )
     _access_token_margin = timedelta(seconds=30)
-    _version = "0.62.1rc1"  # Would be better to share it somewhere
+    _version = "0.62.2rc1"  # Would be better to share it somewhere
 
     def __init__(
         self,

{craft_ai_sdk-0.62.1rc1 → craft_ai_sdk-0.62.2rc1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "craft-ai-sdk"
-version = "0.62.1rc1"
+version = "0.62.2rc1"
 description = "Craft AI MLOps platform SDK"
 license = "Apache-2.0"
 authors = ["Craft AI <contact@craft.ai>"]