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

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

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: craft-ai-sdk
-Version: 0.62.0rc1
+Version: 0.62.1rc1
 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 → craft_ai_sdk-0.62.1rc1}/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.1rc1"

{craft_ai_sdk-0.62.0rc1 → craft_ai_sdk-0.62.1rc1}/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-0.62.0rc1 → craft_ai_sdk-0.62.1rc1}/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-0.62.0rc1 → craft_ai_sdk-0.62.1rc1}/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.1rc1"  # Would be better to share it somewhere
 
     def __init__(
         self,

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "craft-ai-sdk"
-version = "0.62.0rc1"
+version = "0.62.1rc1"
 description = "Craft AI MLOps platform SDK"
 license = "Apache-2.0"
 authors = ["Craft AI <contact@craft.ai>"]
@@ -29,9 +29,7 @@ python = "^3.9"
 requests = "^2.27.1"
 PyJWT = "^2.3.0"
 StrEnum = "^0.4.9"
-
-[tool.poetry.group.weaviate.dependencies]
-weaviate-client = { version = "^4.10.4" }
+weaviate-client = "^4.10.4"
 
 [tool.poetry.group.dev.dependencies]
 ipython = "^8.18.1"
@@ -51,9 +49,6 @@ doc8 = "^0.10.1"
 Sphinx = "^5.0.1"
 sphinxcontrib-restbuilder = "^0.3"
 
-[tool.poetry.extras]
-full = ["weaviate-client"]
-
 [build-system]
 requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"