craft-ai-sdk 0.68.0rc2__tar.gz → 0.68.2rc1__tar.gz

{craft_ai_sdk-0.68.0rc2 → craft_ai_sdk-0.68.2rc1}/PKG-INFO

@@ -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 → craft_ai_sdk-0.68.2rc1}/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-0.68.0rc2 → craft_ai_sdk-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-0.68.0rc2 → craft_ai_sdk-0.68.2rc1}/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-0.68.0rc2 → craft_ai_sdk-0.68.2rc1}/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 → craft_ai_sdk-0.68.2rc1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "craft-ai-sdk"
-version = "0.68.0rc2"
+version = "0.68.2rc1"
 description = "Craft AI MLOps platform SDK"
 license = "Apache-2.0"
 authors = ["Craft AI <contact@craft.ai>"]