craft-ai-sdk 0.61.1__py3-none-any.whl → 0.62.0rc1__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.1"
+__version__ = "0.62.0rc1"

craft_ai_sdk/core/vector_database.py

@@ -1,6 +1,9 @@
 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):
@@ -19,3 +22,41 @@ 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.
+    """
+    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.1"  # Would be better to share it somewhere
+    _version = "0.62.0rc1"  # Would be better to share it somewhere
 
     def __init__(
         self,

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: craft-ai-sdk
-Version: 0.61.1
+Version: 0.62.0rc1
 Summary: Craft AI MLOps platform SDK
 Home-page: https://www.craft.ai/
 License: Apache-2.0
@@ -13,6 +13,7 @@ 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
 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)

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

@@ -1,4 +1,4 @@
-craft_ai_sdk/__init__.py,sha256=bS3_oEVZ3Xrgg5faSoUrRzE-LQoDer6FFKd_KOxT9Xg,363
+craft_ai_sdk/__init__.py,sha256=MVFZ2jBU_33IVZbAgCNdllJe31oKKPVbwxxH9dpvBTs,366
 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=kUND4rpLSzCvn9O_7Xs6Ld-xhnB9nmDKAJntXP2HrYQ,1894
 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=9TjEdz939hhKKNomQWVvLmpDinARY7eXu6iJokwrr1k,10565
+craft_ai_sdk/sdk.py,sha256=m_6xnCrTnHjtNpsYMnFqwEiC8Ba34Bh7fPy9M0TvMiY,10615
 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=KWCoiiHWt3bORACTYL8V4WTVv6R3XU9f2SnB5wupw80,338186
-craft_ai_sdk-0.61.1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
-craft_ai_sdk-0.61.1.dist-info/METADATA,sha256=1seHIBIX0vqFP-ZUM5rWRaf2teen1-yoUx2xN5WEdLI,1564
-craft_ai_sdk-0.61.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-craft_ai_sdk-0.61.1.dist-info/entry_points.txt,sha256=eV9YD0zCAb88_wNMDV99sRxVKVC-WOQF3b1Pepaytcg,385
-craft_ai_sdk-0.61.1.dist-info/RECORD,,
+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,,