craft-ai-sdk 0.61.1rc1__py3-none-any.whl → 0.62.0__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.61.1rc1"
+__version__ = "0.62.0"

craft_ai_sdk/core/vector_database.py

@@ -1,6 +1,7 @@
 from craft_ai_sdk.shared.environments import get_environment_id
 
 from ..sdk import BaseCraftAiSdk
+from ..shared.logger import log_action, log_func_result
 
 
 def get_vector_database_credentials(sdk: BaseCraftAiSdk):
@@ -19,3 +20,48 @@ def get_vector_database_credentials(sdk: BaseCraftAiSdk):
     )
 
     return sdk._get(vector_database_url)
+
+
+@log_func_result("Connecting to Weaviate")
+def get_weaviate_client(sdk: BaseCraftAiSdk):
+    """Get the Weaviate client.
+
+    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://")
+
+    vector_database_url = (
+        credentials["vector_database_url"]
+        .replace("http://", "")
+        .replace("https://", "")
+    ).rstrip("/")
+
+    log_action(sdk, "Connecting to Weaviate")
+
+    weaviate_client = weaviate.connect_to_custom(
+        http_host=vector_database_url,
+        http_port=8080,
+        grpc_host=vector_database_url,
+        grpc_port=8082,
+        http_secure=is_secure,
+        grpc_secure=is_secure,
+        headers={
+            "craft-vector-database-token": credentials["vector_database_token"],
+            "craft-vector-database-url": credentials["vector_database_url"],
+        },
+        auth_credentials=None,
+    )
+
+    log_action(sdk, "Connected to Weaviate, using version", weaviate.__version__)
+
+    return weaviate_client

craft_ai_sdk/sdk.py

@@ -116,7 +116,11 @@ class CraftAiSdk(BaseCraftAiSdk):
         list_steps,
     )
     from .core.users import get_user
-    from .core.vector_database import get_vector_database_credentials
+
+    from .core.vector_database import (
+        get_vector_database_credentials,
+        get_weaviate_client,
+    )
 
     # Size (in bytes) from which datastore upload will switch to multipart
     # AWS: minimum part size is 5MiB
@@ -132,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.61.1rc1"  # Would be better to share it somewhere
+    _version = "0.62.0"  # Would be better to share it somewhere
 
     def __init__(
         self,

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: craft-ai-sdk
-Version: 0.61.1rc1
+Version: 0.62.0
 Summary: Craft AI MLOps platform SDK
 Home-page: https://www.craft.ai/
 License: Apache-2.0
@@ -16,6 +16,7 @@ Classifier: Programming Language :: Python :: 3.12
 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)
 Description-Content-Type: text/markdown
 
 # Craft AI Python SDK

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

@@ -1,4 +1,4 @@
-craft_ai_sdk/__init__.py,sha256=M372R1zDmPPZUxmt1d0eDpP51JoO4oCiaYvcD0ZwSGg,366
+craft_ai_sdk/__init__.py,sha256=nlUKe90vC3S66ZWnXees9yAs1ZtuNFFS5_zjSAiSNv4,363
 craft_ai_sdk/constants.py,sha256=rH4JrGlTpbjjjNRrKhk5oScbj5G5INrcVza6Bb6kIzY,980
 craft_ai_sdk/core/data_store.py,sha256=Q8eNYaiSmS6tWmEUTIYwS5hkMCfye0JNW0cU9QtfsoM,8560
 craft_ai_sdk/core/deployments.py,sha256=k3Eb3axVlZKJigOMy6PAozWxOzd0pZDy2d5nkGwSz34,34803
@@ -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=TONgdMe3v0rmKPQt8kAFHeo21XdTlW8HwrPlUK3UlK4,709
+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=pWsmjPvl6Oi2R0rLz2DdOpDMY79aEgbzlklQReTv9-s,10568
+craft_ai_sdk/sdk.py,sha256=va6TSKoFUASxt3ZvRY_lNgogXa32eAq626Dm0E99rwk,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,9 @@ 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=7-zDbWODdb_Fd7T_NVNkgpHtzCY5MRnUuYhhspn8m7E,338741
-craft_ai_sdk-0.61.1rc1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
-craft_ai_sdk-0.61.1rc1.dist-info/METADATA,sha256=AjctWE2eHnNIIOV3eBUi6O_16ageQsWBjopvTvjrFHc,1567
-craft_ai_sdk-0.61.1rc1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-craft_ai_sdk-0.61.1rc1.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
-craft_ai_sdk-0.61.1rc1.dist-info/RECORD,,
+documentation.pdf,sha256=gnwuYf86TbihQZBGoMtjuVauUscADjDD94_JAu2CN1s,338203
+craft_ai_sdk-0.62.0.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
+craft_ai_sdk-0.62.0.dist-info/METADATA,sha256=Dh9tB_76CWat9jhL6O8YO3ju_h_ZSXQnwYaFsPBf0i8,1613
+craft_ai_sdk-0.62.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+craft_ai_sdk-0.62.0.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
+craft_ai_sdk-0.62.0.dist-info/RECORD,,