craft-ai-sdk 0.68.2__tar.gz → 0.70.0__tar.gz

{craft_ai_sdk-0.68.2 → craft_ai_sdk-0.70.0}/PKG-INFO

@@ -1,23 +1,22 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: craft-ai-sdk
-Version: 0.68.2
+Version: 0.70.0
 Summary: Craft AI MLOps platform SDK
-License: Apache-2.0
+License-Expression: Apache-2.0
+License-File: LICENSE
 Author: Craft AI
 Author-email: contact@craft.ai
-Requires-Python: >=3.9,<4.0
-Classifier: License :: OSI Approved :: Apache Software License
+Requires-Python: >=3.10
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 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/
+Classifier: Programming Language :: Python :: 3.14
+Requires-Dist: StrEnum (>=0.4.9)
+Requires-Dist: pyjwt (>=2.4.0)
+Requires-Dist: requests (>=2.32.4)
+Requires-Dist: weaviate-client (>=4.15.3)
 Description-Content-Type: text/markdown
 
 # Craft AI Python SDK
@@ -25,7 +24,7 @@ Description-Content-Type: text/markdown
 This Python SDK lets you interact with Craft AI MLOps Platform.
 
 ## Installation
-This project relies on **Python 3.9+**. Once a supported version of Python is installed, you can install `craft-ai-sdk` from PyPI with:
+This project relies on **Python 3.10+**. Once a supported version of Python is installed, you can install `craft-ai-sdk` from PyPI with:
 
 ```console
 pip install craft-ai-sdk

{craft_ai_sdk-0.68.2 → craft_ai_sdk-0.70.0}/README.md

@@ -3,7 +3,7 @@
 This Python SDK lets you interact with Craft AI MLOps Platform.
 
 ## Installation
-This project relies on **Python 3.9+**. Once a supported version of Python is installed, you can install `craft-ai-sdk` from PyPI with:
+This project relies on **Python 3.10+**. Once a supported version of Python is installed, you can install `craft-ai-sdk` from PyPI with:
 
 ```console
 pip install craft-ai-sdk

{craft_ai_sdk-0.68.2 → craft_ai_sdk-0.70.0}/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.70.0"

{craft_ai_sdk-0.68.2 → craft_ai_sdk-0.70.0}/craft_ai_sdk/core/pipelines.py

@@ -474,6 +474,7 @@ def get_pipeline(
             the pipeline.
     """
     base_url = f"{sdk.base_environment_api_url}/pipelines/{pipeline_name}"
+    log_action(sdk, f'Get pipeline: "{pipeline_name}", "{wait_for_completion}".')
     if wait_for_completion:
         pipeline = wait_create_until_ready(
             sdk,
@@ -487,6 +488,7 @@ def get_pipeline(
         )
     else:
         pipeline = sdk._get(base_url)
+    log_action(sdk, "Get pipeline done.")
 
     latest_execution = sdk._get(
         f"{sdk.base_environment_api_url}/pipelines/{pipeline_name}/executions/latest"

craft_ai_sdk-0.70.0/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.70.0}/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.70.0"  # Would be better to share it somewhere
 
     def __init__(
         self,

{craft_ai_sdk-0.68.2 → craft_ai_sdk-0.70.0}/craft_ai_sdk/shared/helpers.py

@@ -1,15 +1,23 @@
 from craft_ai_sdk.exceptions import SdkException
+from craft_ai_sdk.shared.logger import log_action
 
 
 def wait_create_until_ready(sdk, name, get_func, timeout_s, start_time, get_log_func):
     elapsed_time = sdk._get_time() - start_time
     status = "creation_pending"
+    log_action(sdk, f'Start waiting for "{name}" to be created.')
     while status == "creation_pending" and (
         timeout_s is None or elapsed_time < timeout_s
     ):
         created_obj = get_func(sdk, name)
         status = created_obj.get("creation_info", {}).get("status", None)
         elapsed_time = sdk._get_time() - start_time
+        log_action(
+            sdk,
+            f'Still waiting for "{name}" to be created, \
+status="{status}", elapsed time={elapsed_time}s.',
+        )
+    log_action(sdk, f'End waiting for "{name}" to be created, status="{status}".')
 
     if status == "creation_failed":
         raise SdkException(

{craft_ai_sdk-0.68.2 → craft_ai_sdk-0.70.0}/pyproject.toml

@@ -1,15 +1,20 @@
-[tool.poetry]
+[project]
 name = "craft-ai-sdk"
-version = "0.68.2"
 description = "Craft AI MLOps platform SDK"
-license = "Apache-2.0"
-authors = ["Craft AI <contact@craft.ai>"]
+authors = [{name = "Craft AI", email = "contact@craft.ai"}]
 readme = "README.md"
 homepage = "https://www.craft.ai/"
-packages = [{ include = "craft_ai_sdk" }]
-include = ["documentation.pdf"]
+license = "Apache-2.0"
+version = "0.70.0"
+requires-python = ">=3.10"
+dependencies = [
+    "requests >= 2.32.4",
+    "pyjwt >= 2.4.0",
+    "StrEnum >= 0.4.9",
+    "weaviate-client >= 4.15.3",
+]
 
-[tool.poetry.scripts]
+[project.scripts]
 lint = "scripts.scripts:lint"
 lintfix = "scripts.scripts:lintfix"
 format = "scripts.scripts:format"
@@ -22,33 +27,38 @@ set-version = "scripts.scripts:set_version"
 pdfdoc = "scripts.scripts:generate_docs_in_pdf"
 rstdoc = "scripts.scripts:generate_docs_in_rst"
 
+[dependency-groups]
+dev = [
+    "ipython < 8.31", # >= 8.31.0 dropped support for python 3.10
+    "ipykernel >= 7.1.0",
+    "pylint >= 4.0.3",
+    "python-dotenv >= 1.2.1",
+    "pytest >= 9.0.1",
+    "pytest-asyncio >= 1.3.0",
+    "requests-mock[fixture] >= 1.12.1",
+    "pytest-watch >= 4.2.0",
+    "pyarrow >= 22.0.0",
+    "pandas >= 2.3.3",
+    "ruff >= 0.14.6",
+]
+docs = [
+    "doc8 >= 2.0.0",
+    "Sphinx < 8.2.0", # >= 8.2.0 dropped support for python 3.10
+    "sphinxcontrib-restbuilder >= 0.3",
+]
+
+[tool.poetry]
+version = "0.70.0"
+packages = [{ include = "craft_ai_sdk" }]
+include = ["documentation.pdf"]
+requires-poetry = ">=2.2.1"
+
+[tool.poetry.group.docs]
+optional = true
+
 [tool.pytest.ini_options]
 filterwarnings = ["ignore::UserWarning"]
 
-[tool.poetry.dependencies]
-python = "^3.9"
-requests = "^2.27.1"
-PyJWT = "^2.3.0"
-StrEnum = "^0.4.9"
-weaviate-client = "^4.10.4"
-
-[tool.poetry.group.dev.dependencies]
-ipython = "^8.18.1"
-ipykernel = "^6.29.5"
-pylint = "^3.3.7"
-python-dotenv = "^1.0.1"
-pytest = "^8.3.4"
-requests-mock = { extras = ["fixture"], version = "^1.9.3" }
-pytest-watch = "^4.2.0"
-pyarrow = "^19.0.1"
-pandas = "^2.2.3"
-ruff = "^0.12.2"
-
-[tool.poetry.group.docs.dependencies]
-doc8 = "^0.10.1"
-Sphinx = "^5.0.1"
-sphinxcontrib-restbuilder = "^0.3"
-
 [tool.ruff.lint]
 extend-select = ["B", "C", "E", "F", "I", "T", "W"]
 extend-ignore = ["C401", "C901"]

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