craft-ai-sdk 0.62.0rc1__py3-none-any.whl → 0.62.1__py3-none-any.whl

craft_ai_sdk/__init__.py

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

craft_ai_sdk/core/data_store.py

@@ -47,12 +47,11 @@ 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"
-    query = {"include_continuation_token": "t"}
 
-    result = sdk._get(f"{url}?{urllib.parse.urlencode(query)}")
+    result = sdk._get(url)
     all_items = result["items"]
     while result.get("continuation_token", None):
-        query["continuation_token"] = result["continuation_token"]
+        query = {"continuation_token": result["continuation_token"]}
         result = sdk._get(f"{url}?{urllib.parse.urlencode(query)}")
         all_items.extend(result["items"])
 

craft_ai_sdk/core/vector_database.py

@@ -3,8 +3,6 @@ from craft_ai_sdk.shared.environments import get_environment_id
 from ..sdk import BaseCraftAiSdk
 from ..shared.logger import log_action, log_func_result
 
-import weaviate
-
 
 def get_vector_database_credentials(sdk: BaseCraftAiSdk):
     """Get the credentials of the vector database.
@@ -31,6 +29,13 @@ def get_weaviate_client(sdk: BaseCraftAiSdk):
     Returns:
         :obj:`weaviate.WeaviateClient`: The Weaviate client.
     """
+    try:
+        import weaviate
+    except ModuleNotFoundError:
+        raise ModuleNotFoundError(
+            "The 'weaviate' package is required to use the vector database. "
+            "You can install it with 'pip install weaviate-client'."
+        )
     credentials = get_vector_database_credentials(sdk)
 
     is_secure = credentials["vector_database_url"].startswith("https://")

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.0rc1"  # Would be better to share it somewhere
+    _version = "0.62.1"  # Would be better to share it somewhere
 
     def __init__(
         self,

{craft_ai_sdk-0.62.0rc1.dist-info → craft_ai_sdk-0.62.1.dist-info}/METADATA

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: craft-ai-sdk
-Version: 0.62.0rc1
+Version: 0.62.1
 Summary: Craft AI MLOps platform SDK
-Home-page: https://www.craft.ai/
 License: Apache-2.0
 Author: Craft AI
 Author-email: contact@craft.ai
@@ -13,10 +12,12 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Provides-Extra: full
+Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: PyJWT (>=2.3.0,<3.0.0)
 Requires-Dist: StrEnum (>=0.4.9,<0.5.0)
 Requires-Dist: requests (>=2.27.1,<3.0.0)
+Requires-Dist: weaviate-client (>=4.10.4,<5.0.0)
+Project-URL: Homepage, https://www.craft.ai/
 Description-Content-Type: text/markdown
 
 # Craft AI Python SDK

{craft_ai_sdk-0.62.0rc1.dist-info → craft_ai_sdk-0.62.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
-craft_ai_sdk/__init__.py,sha256=MVFZ2jBU_33IVZbAgCNdllJe31oKKPVbwxxH9dpvBTs,366
+craft_ai_sdk/__init__.py,sha256=HmsS0beJdUKSM3p1WkX_ZTazY-wWR_YjvZIBJtvrpdo,363
 craft_ai_sdk/constants.py,sha256=rH4JrGlTpbjjjNRrKhk5oScbj5G5INrcVza6Bb6kIzY,980
-craft_ai_sdk/core/data_store.py,sha256=Q8eNYaiSmS6tWmEUTIYwS5hkMCfye0JNW0cU9QtfsoM,8560
+craft_ai_sdk/core/data_store.py,sha256=e2dfemMGZEQvVpL-tMOVa4IT-5lJaDVwAhWFBjdAxSM,8477
 craft_ai_sdk/core/deployments.py,sha256=k3Eb3axVlZKJigOMy6PAozWxOzd0pZDy2d5nkGwSz34,34803
 craft_ai_sdk/core/endpoints.py,sha256=eZMBGj7RCIdCr2IBXPyR2hsargmkzcZGoKL6sgs5xlc,5194
 craft_ai_sdk/core/environment_variables.py,sha256=H1fo5RKQXNSQ8JAgtxZ9AdrdSQGrUtt7QPHbfkr0zTI,2053
@@ -10,10 +10,10 @@ craft_ai_sdk/core/pipelines.py,sha256=-lGP6ynV_rGJf4fi9dzKNNl5JfMgsylOjBeE2Y0iuY
 craft_ai_sdk/core/resource_metrics.py,sha256=DsvlHNFabXobYnt52w7FS_V95KRGqZu9HWFpkpYsb6Y,2333
 craft_ai_sdk/core/steps.py,sha256=qOyULKzYfiyLwOxRU0Z6Zz--PYLJxK4RkTLC3RrP8ZU,20953
 craft_ai_sdk/core/users.py,sha256=Qi3xg7Vzk_0MepccNIZbFzjI0bBqouqkFStTuejUY6s,509
-craft_ai_sdk/core/vector_database.py,sha256=kUND4rpLSzCvn9O_7Xs6Ld-xhnB9nmDKAJntXP2HrYQ,1894
+craft_ai_sdk/core/vector_database.py,sha256=9zgu9WFHYgSoBm_CYuI02-1mAE_y03IXNEk1D4ZIKks,2134
 craft_ai_sdk/exceptions.py,sha256=IC-JfZmmmaTsbMCgirOEByRmWnatQLjKe8BErRkuwM0,1075
 craft_ai_sdk/io.py,sha256=dbXAa6T1dR3V93PA_ctLY-UWzxfeI6pz2nDooLO58b0,12656
-craft_ai_sdk/sdk.py,sha256=m_6xnCrTnHjtNpsYMnFqwEiC8Ba34Bh7fPy9M0TvMiY,10615
+craft_ai_sdk/sdk.py,sha256=XLhdnUI6HN1I2gfblmGvLTyeQ_Mc6vSYjM82cSvxb6M,10612
 craft_ai_sdk/shared/authentication.py,sha256=z0wVkl1P-P8Hod06T4vYLq0P2o0ZCxStpX2dEaHANWs,774
 craft_ai_sdk/shared/environments.py,sha256=IjLZYcL3eyOn_H0j3c05KZBCWcZJVlHf4-N3BupOOTc,494
 craft_ai_sdk/shared/execution_context.py,sha256=B2Ghq-wiUvq81q5mhsm79Oc59c8c00uQxMIpApFD03o,585
@@ -25,9 +25,8 @@ craft_ai_sdk/utils/__init__.py,sha256=A0sLCXSPD1Z3q2GP1uLDjvif4ivOr__Hzg9RQysEuq
 craft_ai_sdk/utils/datetime_utils.py,sha256=QpI2YgHDKbp7yHqvImDuZeSbtQWZDlGDGKR1AehA6x0,650
 craft_ai_sdk/utils/dict_utils.py,sha256=geLcoBg5CPNqrPRm4jSf7MFym5r882QFyNz-4xKzYz8,742
 craft_ai_sdk/utils/file_utils.py,sha256=rspzONrXhiaKYvkY54m0_yKDAYnaun09jvglzsZt0ro,2117
-documentation.pdf,sha256=0sxjjEhfuJTsgMRFTyDVUR831r9HT3D26jwx07UUKLA,338835
-craft_ai_sdk-0.62.0rc1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
-craft_ai_sdk-0.62.0rc1.dist-info/METADATA,sha256=cGteO29GQAvrkNV6EfBkbcfi6miXG78LEYXeObutLkg,1588
-craft_ai_sdk-0.62.0rc1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-craft_ai_sdk-0.62.0rc1.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
-craft_ai_sdk-0.62.0rc1.dist-info/RECORD,,
+craft_ai_sdk-0.62.1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
+craft_ai_sdk-0.62.1.dist-info/METADATA,sha256=ww0N-0PARXB-86JxYuY_Wpd6ahIBQdB2twbD3w44hT4,1676
+craft_ai_sdk-0.62.1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+craft_ai_sdk-0.62.1.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
+craft_ai_sdk-0.62.1.dist-info/RECORD,,

{craft_ai_sdk-0.62.0rc1.dist-info → craft_ai_sdk-0.62.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: poetry-core 2.1.1
 Root-Is-Purelib: true
 Tag: py3-none-any