craft-ai-sdk 0.68.2__tar.gz → 0.68.3rc1__tar.gz

{craft_ai_sdk-0.68.2 → craft_ai_sdk-0.68.3rc1}/PKG-INFO

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

{craft_ai_sdk-0.68.2 → craft_ai_sdk-0.68.3rc1}/craft_ai_sdk/__init__.py

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

craft_ai_sdk-0.68.3rc1/craft_ai_sdk/core/vector_database.py

@@ -0,0 +1,145 @@
+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
+    vector_database_token: str
+
+
+def get_vector_database_credentials(sdk: BaseCraftAiSdk) -> VectorDatabaseCredentials:
+    """Get the credentials of the vector database.
+
+    Returns:
+        :obj:`dict`: The vector database credentials, with the following keys:
+            * ``"vector_database_url"`` (:obj:`str`): URL of the vector database.
+            * ``"vector_database_token"`` (:obj:`str`): Token to connect to the vector
+              database.
+    """
+    environment_id = get_environment_id(sdk)
+
+    vector_database_url = (
+        f"{sdk.base_control_api_url}/environments/{environment_id}/vector-database"
+    )
+
+    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) -> "weaviate_type.WeaviateClient":
+    """Initializes and returns a Weaviate client for interacting with the vector
+    database.
+
+    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'."
+        ) from None
+
+    weaviate_parameters = _get_weaviate_parameters(sdk)
+
+    log_action(sdk, "Connecting to Weaviate")
+
+    weaviate_client = weaviate.connect_to_custom(
+        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=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

{craft_ai_sdk-0.68.2 → craft_ai_sdk-0.68.3rc1}/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.2"  # 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.2 → craft_ai_sdk-0.68.3rc1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "craft-ai-sdk"
-version = "0.68.2"
+version = "0.68.3rc1"
 description = "Craft AI MLOps platform SDK"
 license = "Apache-2.0"
 authors = ["Craft AI <contact@craft.ai>"]
@@ -38,6 +38,7 @@ ipykernel = "^6.29.5"
 pylint = "^3.3.7"
 python-dotenv = "^1.0.1"
 pytest = "^8.3.4"
+pytest-asyncio = "^1.2.0"
 requests-mock = { extras = ["fixture"], version = "^1.9.3" }
 pytest-watch = "^4.2.0"
 pyarrow = "^19.0.1"

craft_ai_sdk-0.68.2/craft_ai_sdk/core/vector_database.py

@@ -1,78 +0,0 @@
-from typing import TypedDict
-
-from craft_ai_sdk.shared.environments import get_environment_id
-
-from ..sdk import BaseCraftAiSdk
-from ..shared.logger import log_action, log_func_result
-
-
-class VectorDatabaseCredentials(TypedDict):
-    vector_database_url: str
-    vector_database_token: str
-
-
-def get_vector_database_credentials(sdk: BaseCraftAiSdk) -> VectorDatabaseCredentials:
-    """Get the credentials of the vector database.
-
-    Returns:
-        :obj:`dict`: The vector database credentials, with the following keys:
-            * ``"vector_database_url"`` (:obj:`str`): URL of the vector database.
-            * ``"vector_database_token"`` (:obj:`str`): Token to connect to the vector
-              database.
-    """
-    environment_id = get_environment_id(sdk)
-
-    vector_database_url = (
-        f"{sdk.base_control_api_url}/environments/{environment_id}/vector-database"
-    )
-
-    return sdk._get(vector_database_url)
-
-
-@log_func_result("Connecting to Weaviate")
-def get_weaviate_client(sdk: BaseCraftAiSdk):
-    """Initializes and returns a Weaviate client for interacting with the vector
-    database.
-
-    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'."
-        ) 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("/")
-
-    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,
-        f"Connected to Weaviate, using version {weaviate.__version__}",
-    )
-
-    return weaviate_client