craft-ai-sdk 0.68.0rc2__py3-none-any.whl → 0.68.2rc1__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.0rc2"
+__version__ = "0.68.2rc1"

craft_ai_sdk/core/vector_database.py

@@ -31,7 +31,8 @@ def get_vector_database_credentials(sdk: BaseCraftAiSdk) -> VectorDatabaseCreden
 
 @log_func_result("Connecting to Weaviate")
 def get_weaviate_client(sdk: BaseCraftAiSdk):
-    """Get the Weaviate client.
+    """Initializes and returns a Weaviate client for interacting with the vector
+    database.
 
     Returns:
         :obj:`weaviate.WeaviateClient`: The Weaviate client.

craft_ai_sdk/sdk.py

@@ -3,7 +3,7 @@ import sys
 import time
 import warnings
 from abc import ABC, abstractmethod
-from datetime import timedelta
+from datetime import datetime, timedelta
 from typing import Any
 
 import jwt
@@ -140,8 +140,9 @@ class CraftAiSdk(BaseCraftAiSdk):
     _MULTIPART_PART_SIZE = int(
         os.environ.get("CRAFT_AI__MULTIPART_PART_SIZE__B", str(38 * 256 * 1024))
     )
-    _access_token_margin = timedelta(seconds=30)
-    _version = "0.68.0rc2"  # Would be better to share it somewhere
+    # Default timeout of proxy network components
+    _access_token_margin = timedelta(seconds=120)
+    _version = "0.68.2rc1"  # Would be better to share it somewhere
 
     def __init__(
         self,
@@ -285,11 +286,21 @@ class CraftAiSdk(BaseCraftAiSdk):
         return self._session.post(url, json=data)
 
     def _refresh_access_token(self):
+        # While a bit weird, those time calculations support any time configuration
+        # on the client & control. Incorrect local datetime should be supported.
+        token_refreshed_at = datetime.now()
         response = self._query_refresh_access_token()
         self._access_token = response["access_token"]
         self._access_token_data = jwt.decode(
             self._access_token, options={"verify_signature": False}
         )
+        # `exp` & `iat` are both relative to control datetime.
+        # Diff between them is an absolute duration
+        token_validity_duration = timedelta(
+            seconds=self._access_token_data["exp"] - self._access_token_data["iat"]
+        )
+        # Adding that duration to local time gives a local expiration time.
+        self._access_token_valid_until = token_refreshed_at + token_validity_duration
 
     def _clear_access_token(self):
         self._access_token = None

craft_ai_sdk/shared/authentication.py

@@ -19,10 +19,8 @@ def use_authentication(
     @functools.wraps(action_func)
     def wrapper(sdk, *args, headers=None, **kwargs):
         actual_headers = None
-        if (
-            sdk._access_token_data is None
-            or sdk._access_token_data["exp"]
-            < (datetime.now() + sdk._access_token_margin).timestamp()
+        if sdk._access_token_data is None or (
+            datetime.now() > sdk._access_token_valid_until - sdk._access_token_margin
         ):
             sdk._refresh_access_token()
         actual_headers = {"Authorization": f"Bearer {sdk._access_token}"}

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

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

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

@@ -1,4 +1,4 @@
-craft_ai_sdk/__init__.py,sha256=tQ-JG-V2iIWjxexEj0I_3oCwgBx3uzOfVMQgxpFtC4E,366
+craft_ai_sdk/__init__.py,sha256=eFFCciMenzXHcefjgWH5c0zm394Bnnv3KxLYwpt0VwQ,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,11 +11,11 @@ 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=S3h68Ej1FnHYJdc5LdLlNjobYYdmaCloSq2UaylPWko,2334
+craft_ai_sdk/core/vector_database.py,sha256=cidA9R-KL9nYR7d7VqBx8o8pIu1Rdp3XYLZn6pnwmok,2397
 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=55KwLEN2oa57_LpLb2QTM3CXfFHY4FPwMetBAvwG9K4,11031
-craft_ai_sdk/shared/authentication.py,sha256=5lJS_aNphkgI0-wvHFK4aW316ZRfOq9oywNpv-Oyn2M,1062
+craft_ai_sdk/sdk.py,sha256=-BuLBOOlzg7CFHimo26cMnz5Ax_eZbfZOaxKWdjOT_0,11717
+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
 craft_ai_sdk/shared/helpers.py,sha256=5ZBiHlS-k1OgKejZgMlDJiYwIPYk7yg-lzeJI67poCc,1070
@@ -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.0rc2.dist-info/LICENSE,sha256=_2oYRJic9lZK05LceuJ9aZZw5mPHYc1WQhJiVS-oGFU,10754
-craft_ai_sdk-0.68.0rc2.dist-info/METADATA,sha256=Ss_ejMAL2d4PYxqq65vkwDBGWO9Vs-yPLhMSVY4BpY8,1679
-craft_ai_sdk-0.68.0rc2.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-craft_ai_sdk-0.68.0rc2.dist-info/entry_points.txt,sha256=QC96WcXvvUfLMRgFD-l_y7_TgC9SqZybLs9EQ8dsGiQ,417
-craft_ai_sdk-0.68.0rc2.dist-info/RECORD,,
+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,,