craft-ai-sdk 0.68.2rc1__py3-none-any.whl → 0.68.3rc1__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.68.2rc1"
+__version__ = "0.68.3rc1"

craft_ai_sdk/core/vector_database.py

@@ -1,10 +1,16 @@
-from typing import TypedDict
+from typing import TYPE_CHECKING, TypedDict
 
 from craft_ai_sdk.shared.environments import get_environment_id
 
 from ..sdk import BaseCraftAiSdk
 from ..shared.logger import log_action, log_func_result
 
+if TYPE_CHECKING:
+    try:
+        import weaviate as weaviate_type
+    except ModuleNotFoundError:
+        pass
+
 
 class VectorDatabaseCredentials(TypedDict):
     vector_database_url: str
@@ -29,8 +35,32 @@ def get_vector_database_credentials(sdk: BaseCraftAiSdk) -> VectorDatabaseCreden
     return sdk._get(vector_database_url)
 
 
+class WeaviateParameters(VectorDatabaseCredentials):
+    is_secure: bool
+    vector_database_host: str
+
+
+def _get_weaviate_parameters(sdk: BaseCraftAiSdk):
+    credentials = get_vector_database_credentials(sdk)
+
+    is_secure = credentials["vector_database_url"].startswith("https://")
+
+    vector_database_host = (
+        credentials["vector_database_url"]
+        .replace("http://", "")
+        .replace("https://", "")
+    ).rstrip("/")
+
+    return WeaviateParameters(
+        vector_database_url=credentials["vector_database_url"],
+        vector_database_token=credentials["vector_database_token"],
+        is_secure=is_secure,
+        vector_database_host=vector_database_host,
+    )
+
+
 @log_func_result("Connecting to Weaviate")
-def get_weaviate_client(sdk: BaseCraftAiSdk):
+def get_weaviate_client(sdk: BaseCraftAiSdk) -> "weaviate_type.WeaviateClient":
     """Initializes and returns a Weaviate client for interacting with the vector
     database.
 
@@ -44,28 +74,65 @@ def get_weaviate_client(sdk: BaseCraftAiSdk):
             "The 'weaviate' package is required to use the vector database. "
             "You can install it with 'pip install weaviate-client'."
         ) from None
-    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("/")
+    weaviate_parameters = _get_weaviate_parameters(sdk)
 
     log_action(sdk, "Connecting to Weaviate")
 
     weaviate_client = weaviate.connect_to_custom(
-        http_host=vector_database_url,
+        http_host=weaviate_parameters["vector_database_host"],
+        http_port=8080,
+        grpc_host=weaviate_parameters["vector_database_host"],
+        grpc_port=8082,
+        http_secure=weaviate_parameters["is_secure"],
+        grpc_secure=weaviate_parameters["is_secure"],
+        headers={
+            "craft-vector-database-token": weaviate_parameters["vector_database_token"],
+            "craft-vector-database-url": weaviate_parameters["vector_database_url"],
+        },
+        auth_credentials=None,
+    )
+
+    log_action(
+        sdk,
+        f"Connected to Weaviate, using version {weaviate.__version__}",
+    )
+
+    return weaviate_client
+
+
+@log_func_result("Connecting to Weaviate")
+def get_async_weaviate_client(
+    sdk: BaseCraftAiSdk,
+) -> "weaviate_type.WeaviateAsyncClient":
+    """Initializes and returns an async Weaviate client for interacting with the vector
+    database.
+
+    Returns:
+        :obj:`weaviate.WeaviateAsyncClient`: 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'."
+        ) from None
+
+    weaviate_parameters = _get_weaviate_parameters(sdk)
+
+    log_action(sdk, "Connecting to Weaviate")
+
+    weaviate_client = weaviate.use_async_with_custom(
+        http_host=weaviate_parameters["vector_database_host"],
         http_port=8080,
-        grpc_host=vector_database_url,
+        grpc_host=weaviate_parameters["vector_database_host"],
         grpc_port=8082,
-        http_secure=is_secure,
-        grpc_secure=is_secure,
+        http_secure=weaviate_parameters["is_secure"],
+        grpc_secure=weaviate_parameters["is_secure"],
         headers={
-            "craft-vector-database-token": credentials["vector_database_token"],
-            "craft-vector-database-url": credentials["vector_database_url"],
+            "craft-vector-database-token": weaviate_parameters["vector_database_token"],
+            "craft-vector-database-url": weaviate_parameters["vector_database_url"],
         },
         auth_credentials=None,
     )

craft_ai_sdk/sdk.py

@@ -123,6 +123,7 @@ class CraftAiSdk(BaseCraftAiSdk):
     )
     from .core.users import get_user
     from .core.vector_database import (
+        get_async_weaviate_client,
         get_vector_database_credentials,
         get_weaviate_client,
     )
@@ -142,7 +143,7 @@ class CraftAiSdk(BaseCraftAiSdk):
     )
     # Default timeout of proxy network components
     _access_token_margin = timedelta(seconds=120)
-    _version = "0.68.2rc1"  # Would be better to share it somewhere
+    _version = "0.68.3rc1"  # Would be better to share it somewhere
 
     def __init__(
         self,

{craft_ai_sdk-0.68.2rc1.dist-info → craft_ai_sdk-0.68.3rc1.dist-info}/METADATA

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

{craft_ai_sdk-0.68.2rc1.dist-info → craft_ai_sdk-0.68.3rc1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-craft_ai_sdk/__init__.py,sha256=eFFCciMenzXHcefjgWH5c0zm394Bnnv3KxLYwpt0VwQ,366
+craft_ai_sdk/__init__.py,sha256=H3m8hXsrWJjz_Qt2Z1bxkMPfM3jN0HuTf91pDMhlsOQ,366
 craft_ai_sdk/constants.py,sha256=rH4JrGlTpbjjjNRrKhk5oScbj5G5INrcVza6Bb6kIzY,980
 craft_ai_sdk/core/data_store.py,sha256=BPQalGcm6bujgkgu-MtNWWhKiJEg4MiVuCtByNbTbK8,9956
 craft_ai_sdk/core/deployments.py,sha256=Hig0Z6tZXbaLxms9ENtKI4OG3ki7G4Iz3wGk2V2Gmfg,37731
@@ -11,10 +11,10 @@ craft_ai_sdk/core/pipelines.py,sha256=FD6ai2_YhE1tz6QO57bj_lA-i93TXZzm0Y9PA0mpoc
 craft_ai_sdk/core/resource_metrics.py,sha256=ubCe6QOhud3f-EQr_TXKBsrFFaxZubqTS3fZqfV7kgg,3457
 craft_ai_sdk/core/steps.py,sha256=TwOWXAAVpLPuSZ_gvf-3E5uKEvfGIzR52Y58bqKn_Ds,23487
 craft_ai_sdk/core/users.py,sha256=q5et87q0SOMpRTBOiLX026oaEOTaY1aGqKdbUC51zbA,618
-craft_ai_sdk/core/vector_database.py,sha256=cidA9R-KL9nYR7d7VqBx8o8pIu1Rdp3XYLZn6pnwmok,2397
+craft_ai_sdk/core/vector_database.py,sha256=NzlYeEvq4eH75Y47Z0Gam7BWJIJ0E2IPXajmPipHUAU,4553
 craft_ai_sdk/exceptions.py,sha256=IC-JfZmmmaTsbMCgirOEByRmWnatQLjKe8BErRkuwM0,1075
 craft_ai_sdk/io.py,sha256=x1G7ga10W_SeTDLHe-FrhU26_WNwWW32-yprMmCZ5Ck,11836
-craft_ai_sdk/sdk.py,sha256=-BuLBOOlzg7CFHimo26cMnz5Ax_eZbfZOaxKWdjOT_0,11717
+craft_ai_sdk/sdk.py,sha256=CXQ0pV4J-EZenJz23LntmYxhGPybTWCJWtmltV7yLew,11752
 craft_ai_sdk/shared/authentication.py,sha256=mnsEhG0mWwG9ZY-GToHDNnYERoou_cjt4k8l_87QW6s,1024
 craft_ai_sdk/shared/environments.py,sha256=JOV3s0KIFVl_zskJb--bybzIXEGQJjrf_T4dL-szSlQ,500
 craft_ai_sdk/shared/execution_context.py,sha256=B2Ghq-wiUvq81q5mhsm79Oc59c8c00uQxMIpApFD03o,585
@@ -27,8 +27,8 @@ craft_ai_sdk/utils/__init__.py,sha256=A0sLCXSPD1Z3q2GP1uLDjvif4ivOr__Hzg9RQysEuq
 craft_ai_sdk/utils/datetime_utils.py,sha256=yYP5HVdI879WXxQCajPTnas1pWrwInOxMux-mxqQNQM,734
 craft_ai_sdk/utils/dict_utils.py,sha256=1HQ3A14SN48XPFDmQleujGAgksmkjIs3hyPLpwhwh24,748
 craft_ai_sdk/utils/file_utils.py,sha256=o10-CDt4qzgCJNPykvlNrL6WTouhVLY8C8BVpHJYt18,2795
-craft_ai_sdk-0.68.2rc1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
-craft_ai_sdk-0.68.2rc1.dist-info/METADATA,sha256=DbN2rnk6mZMqGMKju5vc0tPeAW6yQ51OX63LxQOleEg,1679
-craft_ai_sdk-0.68.2rc1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-craft_ai_sdk-0.68.2rc1.dist-info/entry_points.txt,sha256=QC96WcXvvUfLMRgFD-l_y7_TgC9SqZybLs9EQ8dsGiQ,417
-craft_ai_sdk-0.68.2rc1.dist-info/RECORD,,
+craft_ai_sdk-0.68.3rc1.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
+craft_ai_sdk-0.68.3rc1.dist-info/METADATA,sha256=jTCVA73g52xzvzwGOCjM9dg71A9nUHMJP_uUt130WUc,1679
+craft_ai_sdk-0.68.3rc1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+craft_ai_sdk-0.68.3rc1.dist-info/entry_points.txt,sha256=QC96WcXvvUfLMRgFD-l_y7_TgC9SqZybLs9EQ8dsGiQ,417
+craft_ai_sdk-0.68.3rc1.dist-info/RECORD,,