frogml-core 0.0.78__py3-none-any.whl → 0.0.79__py3-none-any.whl

frogml_core/__init__.py

@@ -1,7 +1,7 @@
 """Top-level package for frogml."""
 
 __author__ = "jfrog"
-__version__ = "0.0.78"
+__version__ = "0.0.79"
 
 from frogml_core.inner.di_configuration import wire_dependencies
 from frogml_core.model.model_version_tracking import (  # noqa: F401,E501

frogml_core/inner/tool/auth.py

@@ -5,40 +5,44 @@ from frogml_storage._utils import BearerAuth
 from frogml_storage.authentication._authentication_utils import get_credentials
 from frogml_storage.authentication.models._auth_config import AuthConfig
 from requests.auth import AuthBase
+from typing_extensions import Self
 
 from frogml_core.exceptions import FrogmlLoginException
 
 
 class FrogMLAuthClient:
-    def __init__(self, auth_config: Optional[AuthConfig] = None):
+    __MIN_TOKEN_LENGTH: int = 64
+
+    def __init__(self: Self, auth_config: Optional[AuthConfig] = None):
         self.auth_config: Optional[AuthConfig] = auth_config
         self._token: Optional[str] = None
         self._tenant_id: Optional[str] = None
 
-    def get_token(self) -> str:
+    def get_token(self: Self) -> str:
         if not self._token:
             self.login()
 
         return cast(str, self._token)
 
-    def get_tenant_id(self) -> str:
+    def get_tenant_id(self: Self) -> str:
         if not self._tenant_id:
             self.login()
 
         return cast(str, self._tenant_id)
 
-    def login(self):
+    def login(self: Self):
         artifactory_url, auth = get_credentials(self.auth_config)
 
         if isinstance(auth, BearerAuth):  # For BearerAuth
+            self.__validate_token(auth.token)
             self._token = auth.token
 
         self.__get_tenant_id(artifactory_url, auth)
 
-    def get_auth(self) -> Union[AuthBase]:
+    def get_auth(self: Self) -> Union[AuthBase]:
         return get_credentials(self.auth_config)[1]
 
-    def __get_tenant_id(self, artifactory_url: str, auth: AuthBase):
+    def __get_tenant_id(self: Self, artifactory_url: str, auth: AuthBase):
         login_exception = FrogmlLoginException(
             "Failed to authenticate with JFrog. Please check your credentials"
         )
@@ -63,3 +67,10 @@ class FrogMLAuthClient:
 
         except (requests.exceptions.RequestException, ValueError) as exc:
             raise login_exception from exc
+
+    def __validate_token(self: Self, token: Optional[str]):
+        if token is None or len(token) <= self.__MIN_TOKEN_LENGTH:
+            raise FrogmlLoginException(
+                "Authentication with JFrog failed: Only JWT Access Tokens are supported. "
+                "Please ensure you are using a valid JWT Access Token."
+            )

{frogml_core-0.0.78.dist-info → frogml_core-0.0.79.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-core
-Version: 0.0.78
+Version: 0.0.79
 Summary: frogml Core contains the necessary objects and communication tools for using the Jfrog ml Platform
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model

{frogml_core-0.0.78.dist-info → frogml_core-0.0.79.dist-info}/RECORD

@@ -1,4 +1,4 @@
-frogml_core/__init__.py,sha256=Uql7AUqVfJsciUbDjWMoJoILzqPYZkBjdkdyY8skAsM,777
+frogml_core/__init__.py,sha256=Obj8skKjosyVzjEKbjkibFypwfZ7R0PqFtH-YOurzqo,777
 frogml_core/automations/__init__.py,sha256=j2gD15MN-xVWhI5rAFsDwhL0CIyICLNT0scXsKvNBkU,1547
 frogml_core/automations/automation_executions.py,sha256=xpOb9Dq8gPPGNQDJTvBBZbNz4woZDRZY0HqnLSu7pwU,3230
 frogml_core/automations/automations.py,sha256=GKEQyQMi8sxX5oZn62PaxPi0zD8IaJRjBkhczRJxHNs,13070
@@ -281,7 +281,7 @@ frogml_core/inner/runtime_di/__init__.py,sha256=HXqtejgH3J9TXjnEBT6cNwnTDPq6v3sT
 frogml_core/inner/runtime_di/containers.py,sha256=LbapepWFjBjkELEOTK7m7AJEfj4KENVDlpD7lv7zw6o,577
 frogml_core/inner/singleton_meta.py,sha256=1cU99I0f9tjuMQLMJyLsK1oK3fZJMsO5-TbRHAMXqds,627
 frogml_core/inner/tool/__init__.py,sha256=rmOSE-ejnzDG_H7kbikPQxEO4TFIkhBWjOXhTIrldiU,35
-frogml_core/inner/tool/auth.py,sha256=XWBLkCz6xS5xCVH6ZFNC42RnWo1p27w3d2SMbNLMh30,2218
+frogml_core/inner/tool/auth.py,sha256=H0tQTc8JgEGLxv79IS_28jahi8cq8NNoP3lpxlQshPs,2714
 frogml_core/inner/tool/grpc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/inner/tool/grpc/grpc_auth.py,sha256=WPFWn7CfGp-pSqlnUTplHB6lfuJdzpS6jeidmpyondo,1421
 frogml_core/inner/tool/grpc/grpc_tools.py,sha256=8sXDWBd_kVRuwmUSdpQNobylT6u8H_83Q8WlZJYOi0c,7247
@@ -1096,6 +1096,6 @@ frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xT
 frogml_services_mock/services_mock.py,sha256=xfbYvl05gFUq2oT82cT-6K2CEJ6d9kFX_tk77QZHTFI,20787
 frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-frogml_core-0.0.78.dist-info/METADATA,sha256=6S_8ftAvq2OeIvAM9chAPCpSGRYiqX9G7-eEXRayQq0,2004
-frogml_core-0.0.78.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-frogml_core-0.0.78.dist-info/RECORD,,
+frogml_core-0.0.79.dist-info/METADATA,sha256=u1qoVO96CDfDjfgs8orTURNoTD-jG84jhksQOkQzZig,2004
+frogml_core-0.0.79.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+frogml_core-0.0.79.dist-info/RECORD,,