qwak-core 0.4.271__py3-none-any.whl → 0.4.273__py3-none-any.whl

_qwak_proto/qwak/models/models_query_pb2.py

@@ -0,0 +1,77 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/models/models_query.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1eqwak/models/models_query.proto\x12\x16qwak.models.management\"\x99\x01\n\nSortModels\x12\x31\n\x07sort_by\x18\x01 \x01(\x0e\x32\x1e.qwak.models.management.SortByH\x00\x12\x37\n\nsort_order\x18\x02 \x01(\x0e\x32!.qwak.models.management.SortOrderH\x01\x42\r\n\x0bsort_by_optB\x10\n\x0esort_order_opt\"\xc8\x01\n\x06\x46ilter\x12\x1e\n\x14model_name_substring\x18\x01 \x01(\tH\x00\x12\x19\n\x11model_group_names\x18\x02 \x03(\t\x12V\n\x17\x64\x65ployment_model_status\x18\x03 \x01(\x0e\x32\x33.qwak.models.management.DeploymentModelStatusFilterH\x01\x42\x0c\n\nsearch_optB\x1d\n\x1b\x64\x65ployment_model_status_opt\"<\n\x12PaginationMetadata\x12\x11\n\tpage_size\x18\x01 \x01(\x05\x12\x13\n\x0bnext_cursor\x18\x02 \x01(\t*J\n\x06SortBy\x12\x13\n\x0fSORT_BY_INVALID\x10\x00\x12\x10\n\x0cSORT_BY_NAME\x10\x01\x12\x19\n\x15SORT_BY_CREATION_DATE\x10\x02*L\n\tSortOrder\x12\x16\n\x12SORT_ORDER_INVALID\x10\x00\x12\x12\n\x0eSORT_ORDER_ASC\x10\x01\x12\x13\n\x0fSORT_ORDER_DESC\x10\x02*\xa5\x01\n\x1b\x44\x65ploymentModelStatusFilter\x12*\n&DEPLOYMENT_MODEL_STATUS_FILTER_INVALID\x10\x00\x12+\n\'DEPLOYMENT_MODEL_STATUS_FILTER_DEPLOYED\x10\x01\x12-\n)DEPLOYMENT_MODEL_STATUS_FILTER_UNDEPLOYED\x10\x02\x42\x36\n!com.qwak.ai.management.models.apiB\x0fModelQueryProtoP\x01\x62\x06proto3')
+
+_SORTBY = DESCRIPTOR.enum_types_by_name['SortBy']
+SortBy = enum_type_wrapper.EnumTypeWrapper(_SORTBY)
+_SORTORDER = DESCRIPTOR.enum_types_by_name['SortOrder']
+SortOrder = enum_type_wrapper.EnumTypeWrapper(_SORTORDER)
+_DEPLOYMENTMODELSTATUSFILTER = DESCRIPTOR.enum_types_by_name['DeploymentModelStatusFilter']
+DeploymentModelStatusFilter = enum_type_wrapper.EnumTypeWrapper(_DEPLOYMENTMODELSTATUSFILTER)
+SORT_BY_INVALID = 0
+SORT_BY_NAME = 1
+SORT_BY_CREATION_DATE = 2
+SORT_ORDER_INVALID = 0
+SORT_ORDER_ASC = 1
+SORT_ORDER_DESC = 2
+DEPLOYMENT_MODEL_STATUS_FILTER_INVALID = 0
+DEPLOYMENT_MODEL_STATUS_FILTER_DEPLOYED = 1
+DEPLOYMENT_MODEL_STATUS_FILTER_UNDEPLOYED = 2
+
+
+_SORTMODELS = DESCRIPTOR.message_types_by_name['SortModels']
+_FILTER = DESCRIPTOR.message_types_by_name['Filter']
+_PAGINATIONMETADATA = DESCRIPTOR.message_types_by_name['PaginationMetadata']
+SortModels = _reflection.GeneratedProtocolMessageType('SortModels', (_message.Message,), {
+  'DESCRIPTOR' : _SORTMODELS,
+  '__module__' : 'qwak.models.models_query_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.models.management.SortModels)
+  })
+_sym_db.RegisterMessage(SortModels)
+
+Filter = _reflection.GeneratedProtocolMessageType('Filter', (_message.Message,), {
+  'DESCRIPTOR' : _FILTER,
+  '__module__' : 'qwak.models.models_query_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.models.management.Filter)
+  })
+_sym_db.RegisterMessage(Filter)
+
+PaginationMetadata = _reflection.GeneratedProtocolMessageType('PaginationMetadata', (_message.Message,), {
+  'DESCRIPTOR' : _PAGINATIONMETADATA,
+  '__module__' : 'qwak.models.models_query_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.models.management.PaginationMetadata)
+  })
+_sym_db.RegisterMessage(PaginationMetadata)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n!com.qwak.ai.management.models.apiB\017ModelQueryProtoP\001'
+  _SORTBY._serialized_start=479
+  _SORTBY._serialized_end=553
+  _SORTORDER._serialized_start=555
+  _SORTORDER._serialized_end=631
+  _DEPLOYMENTMODELSTATUSFILTER._serialized_start=634
+  _DEPLOYMENTMODELSTATUSFILTER._serialized_end=799
+  _SORTMODELS._serialized_start=59
+  _SORTMODELS._serialized_end=212
+  _FILTER._serialized_start=215
+  _FILTER._serialized_end=415
+  _PAGINATIONMETADATA._serialized_start=417
+  _PAGINATIONMETADATA._serialized_end=477
+# @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/models/models_query_pb2.pyi

@@ -0,0 +1,157 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _SortBy:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _SortByEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_SortBy.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    SORT_BY_INVALID: _SortBy.ValueType  # 0
+    SORT_BY_NAME: _SortBy.ValueType  # 1
+    """Sort models by name (default)."""
+    SORT_BY_CREATION_DATE: _SortBy.ValueType  # 2
+    """Sort models by their creation date."""
+
+class SortBy(_SortBy, metaclass=_SortByEnumTypeWrapper): ...
+
+SORT_BY_INVALID: SortBy.ValueType  # 0
+SORT_BY_NAME: SortBy.ValueType  # 1
+"""Sort models by name (default)."""
+SORT_BY_CREATION_DATE: SortBy.ValueType  # 2
+"""Sort models by their creation date."""
+global___SortBy = SortBy
+
+class _SortOrder:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _SortOrderEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_SortOrder.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    SORT_ORDER_INVALID: _SortOrder.ValueType  # 0
+    SORT_ORDER_ASC: _SortOrder.ValueType  # 1
+    """Sort in ascending order (default)."""
+    SORT_ORDER_DESC: _SortOrder.ValueType  # 2
+    """Sort in descending order."""
+
+class SortOrder(_SortOrder, metaclass=_SortOrderEnumTypeWrapper): ...
+
+SORT_ORDER_INVALID: SortOrder.ValueType  # 0
+SORT_ORDER_ASC: SortOrder.ValueType  # 1
+"""Sort in ascending order (default)."""
+SORT_ORDER_DESC: SortOrder.ValueType  # 2
+"""Sort in descending order."""
+global___SortOrder = SortOrder
+
+class _DeploymentModelStatusFilter:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _DeploymentModelStatusFilterEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DeploymentModelStatusFilter.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    DEPLOYMENT_MODEL_STATUS_FILTER_INVALID: _DeploymentModelStatusFilter.ValueType  # 0
+    DEPLOYMENT_MODEL_STATUS_FILTER_DEPLOYED: _DeploymentModelStatusFilter.ValueType  # 1
+    DEPLOYMENT_MODEL_STATUS_FILTER_UNDEPLOYED: _DeploymentModelStatusFilter.ValueType  # 2
+
+class DeploymentModelStatusFilter(_DeploymentModelStatusFilter, metaclass=_DeploymentModelStatusFilterEnumTypeWrapper): ...
+
+DEPLOYMENT_MODEL_STATUS_FILTER_INVALID: DeploymentModelStatusFilter.ValueType  # 0
+DEPLOYMENT_MODEL_STATUS_FILTER_DEPLOYED: DeploymentModelStatusFilter.ValueType  # 1
+DEPLOYMENT_MODEL_STATUS_FILTER_UNDEPLOYED: DeploymentModelStatusFilter.ValueType  # 2
+global___DeploymentModelStatusFilter = DeploymentModelStatusFilter
+
+class SortModels(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SORT_BY_FIELD_NUMBER: builtins.int
+    SORT_ORDER_FIELD_NUMBER: builtins.int
+    sort_by: global___SortBy.ValueType
+    sort_order: global___SortOrder.ValueType
+    def __init__(
+        self,
+        *,
+        sort_by: global___SortBy.ValueType = ...,
+        sort_order: global___SortOrder.ValueType = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["sort_by", b"sort_by", "sort_by_opt", b"sort_by_opt", "sort_order", b"sort_order", "sort_order_opt", b"sort_order_opt"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["sort_by", b"sort_by", "sort_by_opt", b"sort_by_opt", "sort_order", b"sort_order", "sort_order_opt", b"sort_order_opt"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["sort_by_opt", b"sort_by_opt"]) -> typing_extensions.Literal["sort_by"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["sort_order_opt", b"sort_order_opt"]) -> typing_extensions.Literal["sort_order"] | None: ...
+
+global___SortModels = SortModels
+
+class Filter(google.protobuf.message.Message):
+    """Filter models by specific properties using a filter expression.
+    The following fields are used to filter the models returned by the query. An AND operation is applied between each of these fields.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_NAME_SUBSTRING_FIELD_NUMBER: builtins.int
+    MODEL_GROUP_NAMES_FIELD_NUMBER: builtins.int
+    DEPLOYMENT_MODEL_STATUS_FIELD_NUMBER: builtins.int
+    model_name_substring: builtins.str
+    @property
+    def model_group_names(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """Filter models by one or more model group names.
+        Return models that belong to one of the specified model groups.
+        Optional field. If not specified, models from all groups are included.
+        """
+    deployment_model_status: global___DeploymentModelStatusFilter.ValueType
+    def __init__(
+        self,
+        *,
+        model_name_substring: builtins.str = ...,
+        model_group_names: collections.abc.Iterable[builtins.str] | None = ...,
+        deployment_model_status: global___DeploymentModelStatusFilter.ValueType = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["deployment_model_status", b"deployment_model_status", "deployment_model_status_opt", b"deployment_model_status_opt", "model_name_substring", b"model_name_substring", "search_opt", b"search_opt"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["deployment_model_status", b"deployment_model_status", "deployment_model_status_opt", b"deployment_model_status_opt", "model_group_names", b"model_group_names", "model_name_substring", b"model_name_substring", "search_opt", b"search_opt"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["deployment_model_status_opt", b"deployment_model_status_opt"]) -> typing_extensions.Literal["deployment_model_status"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["search_opt", b"search_opt"]) -> typing_extensions.Literal["model_name_substring"] | None: ...
+
+global___Filter = Filter
+
+class PaginationMetadata(google.protobuf.message.Message):
+    """Metadata for pagination."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PAGE_SIZE_FIELD_NUMBER: builtins.int
+    NEXT_CURSOR_FIELD_NUMBER: builtins.int
+    page_size: builtins.int
+    """Number of items per page."""
+    next_cursor: builtins.str
+    """Cursor for the next page.
+    If there are no more pages, this field will be empty.
+    """
+    def __init__(
+        self,
+        *,
+        page_size: builtins.int = ...,
+        next_cursor: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["next_cursor", b"next_cursor", "page_size", b"page_size"]) -> None: ...
+
+global___PaginationMetadata = PaginationMetadata

_qwak_proto/qwak/models/models_query_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

frogml_storage/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.4.273"

frogml_storage/artifactory/__init__.py

@@ -0,0 +1 @@
+from ._artifactory_api import ArtifactoryApi, StartTransactionResponse

frogml_storage/artifactory/_artifactory_api.py

@@ -0,0 +1,315 @@
+import json
+import os
+from typing import Optional
+
+from requests import Response
+from tqdm.auto import tqdm
+from tqdm.utils import CallbackIOWrapper
+from urllib3 import Retry
+
+from frogml_storage.logging import logger
+from frogml_storage.utils import join_url
+from frogml_storage.constants import CHECKSUM_SHA2_HEADER
+from frogml_storage.models.entity_manifest import Checksums, EntityManifest
+from frogml_storage.models.frogml_entity_type_info import FrogMLEntityTypeInfo
+from frogml_storage.http import HTTPClient
+from frogml_storage.models import DownloadContext
+
+
+class StartTransactionResponse:
+    files_upload_path: str
+    lead_upload_path: str
+    dependencies_upload_path: str
+    code_upload_path: str
+    transaction_id: str
+
+    def __init__(
+        self,
+        files_upload_path,
+        lead_upload_path,
+        dependencies_upload_path,
+        code_upload_path,
+        transaction_id,
+    ):
+        self.files_upload_path = files_upload_path
+        self.lead_upload_path = lead_upload_path
+        self.dependencies_upload_path = dependencies_upload_path
+        self.code_upload_path = code_upload_path
+        self.transaction_id = transaction_id
+
+
+class ArtifactoryApi:
+    def __init__(self, uri, auth=None, http_client=None):
+        self.uri = uri
+        if http_client is not None:
+            self.http_client = http_client
+        else:
+            self.auth = auth
+            self.http_client = HTTPClient(auth=auth)
+
+    def start_transaction(
+        self,
+        entity_type_info: FrogMLEntityTypeInfo,
+        repository: str,
+        entity_name: str,
+        version: Optional[str],
+    ) -> StartTransactionResponse:
+        """
+        Initializes an upload. Returns transaction ID and upload path
+        """
+        if version is None:
+            start_transaction_url = (
+                f"{self.uri}/api/machinelearning/{repository}/"
+                f"{entity_type_info.entity_type}/{entity_name}/start-transaction"
+            )
+        else:
+            start_transaction_url = (
+                f"{self.uri}/api/machinelearning/{repository}/{entity_type_info.entity_type}"
+                f"/{entity_name}/start-transaction/{version}"
+            )
+        try:
+            response = self.http_client.post(start_transaction_url)
+            response.raise_for_status()
+            files_upload_path = response.json()["filesUploadPath"]
+            lead_upload_path = response.json()["leadUploadPath"]
+            dependencies_upload_path = response.json()["dependenciesUploadPath"]
+            code_upload_path = response.json()["codeUploadPath"]
+            transaction_id = response.json()["transactionId"]
+        except Exception as exception:
+            err = (
+                f"Error occurred while trying to start an upload transaction for "
+                f"{entity_type_info.entity_type}: '{entity_name}'"
+                f" Error: '{exception}'"
+            )
+            logger.error(err, exc_info=False)
+            raise exception
+        return StartTransactionResponse(
+            files_upload_path=files_upload_path,
+            lead_upload_path=lead_upload_path,
+            dependencies_upload_path=dependencies_upload_path,
+            code_upload_path=code_upload_path,
+            transaction_id=transaction_id,
+        )
+
+    def end_transaction(
+        self,
+        entity_type: FrogMLEntityTypeInfo,
+        repository: str,
+        entity_name: str,
+        entity_manifest: EntityManifest,
+        transaction_id: str,
+        version: str,
+        properties: Optional[dict[str, str]],
+    ) -> None:
+        """
+        Upload model-manifest.json | dataset-manifest.json file, makes the model | dataset available in the repository
+        """
+        filename = entity_type.metadata_file_name
+
+        url = join_url(
+            self.uri,
+            "api",
+            "machinelearning",
+            repository,
+            entity_type.entity_type,
+            "entity-manifest",
+            entity_name,
+            version,
+            transaction_id,
+        )
+
+        json_entity_manifest = entity_manifest.to_json()
+        self.upload_entity_manifest(
+            entity_type=entity_type,
+            filename=filename,
+            payload=json_entity_manifest,
+            url=url,
+            properties=properties,
+        )
+
+    def download_file(self, args: DownloadContext) -> None:
+        filename = os.path.basename(args.target_path)
+        try:
+            url = f"{self.uri}/{args.repo_key}/{args.source_url}"
+            with self.http_client.get(url=url, stream=True) as response:
+                response.raise_for_status()
+                total_size = int(response.headers.get("content-length", 0))
+                with open(args.target_path, "wb") as file:
+                    with self.__initialize_progress_bar(total_size, filename) as pbar:
+                        for chunk in response.iter_content(chunk_size=8192):
+                            if chunk:
+                                file.write(chunk)
+                                pbar.update(len(chunk))
+
+        except Exception as exception:
+            self.__handle_download_exception(exception, args.target_path, filename)
+
+    def get_entity_manifest(
+        self,
+        entity_type_info: FrogMLEntityTypeInfo,
+        repository: str,
+        entity_name: str,
+        namespace: Optional[str],
+        version: Optional[str],
+    ) -> dict:
+        url = join_url(
+            self.uri,
+            "api",
+            "machinelearning",
+            repository,
+            entity_type_info.entity_type,
+            "entity-manifest",
+            namespace,
+            entity_name,
+            version,
+        )
+        try:
+            with self.http_client.get(url=url) as r:
+                r.raise_for_status()
+                return r.json()
+        except Exception as exception:
+            err = f"Error occurred while trying to get {entity_type_info.entity_type} info file. Error: '{exception}'"
+            logger.error(err, exc_info=False)
+            raise exception
+
+    @staticmethod
+    def __handle_download_exception(
+        exception: Exception, target_path: str, filename: str
+    ) -> None:
+        if os.path.exists(target_path):
+            os.remove(target_path)
+        err = f"Error occurred while trying to download file: '{filename}' Error: '{exception}'"
+        logger.error(err, exc_info=False)
+        raise exception
+
+    def get_artifact_checksum(self, download_context: DownloadContext) -> str:
+        url = f"{self.uri}/{download_context.repo_key}/{download_context.source_url}"
+        try:
+            with self.http_client.head(url=url, stream=True) as response:
+                response.raise_for_status()
+                return response.headers.get(CHECKSUM_SHA2_HEADER)
+
+        except Exception as exception:
+            logger.error(exception.__cause__, exc_info=False)
+            raise exception
+
+    def upload_entity_manifest(
+        self,
+        entity_type: FrogMLEntityTypeInfo,
+        filename: str,
+        payload: str,
+        url: str,
+        properties: Optional[dict[str, str]],
+        stream: bool = False,
+    ) -> None:
+        body_part_name = f"{entity_type.body_part_stream}"
+
+        try:
+            files = {
+                f"{body_part_name}": (
+                    f"{body_part_name}",
+                    payload,
+                    "application/octet-stream",
+                ),  # Include the InputStream
+                "additionalData": (
+                    "additionalData",
+                    json.dumps(properties),
+                    "application/octet-stream",
+                ),  # Include the object
+            }
+            with self.http_client.put(url=url, files=files, stream=stream) as response:
+                response.raise_for_status()
+        except Exception as exception:
+            err = f"Error occurred while trying to upload file: '{filename}' Error: '{exception}'"
+            logger.error(err, exc_info=False)
+            raise exception
+
+    def upload_file(self, file_path: str, url: str) -> None:
+        wrapped_file = None
+        try:
+            file_size = os.stat(file_path).st_size
+            with (
+                self.__initialize_progress_bar(file_size, file_path) as pbar,
+                open(file_path, "rb") as file,
+            ):
+                wrapped_file = CallbackIOWrapper(pbar.update, file, "read")
+                with self.http_client.put(url=url, payload=wrapped_file) as response:
+                    response.raise_for_status()
+        except Exception as exception:
+            err = f"Error occurred while trying to upload file: '{file_path}' Error: '{exception}'"
+            logger.error(err, exc_info=False)
+            raise type(exception)(f"{err} File: {file_path}") from exception
+        finally:
+            if wrapped_file is not None:
+                wrapped_file.close()
+
+    def checksum_deployment(
+        self,
+        checksum: Checksums,
+        url: str,
+        full_path: str,
+        stream: bool = False,
+    ) -> bool:
+        response = self.http_client.put(
+            url=url,
+            headers={"X-Checksum-Sha256": checksum.sha2, "X-Checksum-Deploy": "true"},
+            stream=stream,
+        )
+        if response.status_code != 200 and response.status_code != 201:
+            return False
+        else:
+            file_size = os.path.getsize(full_path)
+            pbar = self.__initialize_progress_bar(file_size, full_path)
+            pbar.update(file_size)
+            pbar.close()
+            return True
+
+    @staticmethod
+    def __initialize_progress_bar(total_size: int, filename: str) -> tqdm:
+        return tqdm(
+            total=total_size, unit="B", unit_scale=True, desc=filename, initial=0
+        )
+
+    def encrypt_password(self) -> Response:
+        """
+        returns encrypted password as text
+        """
+        return self.http_client.get(
+            url=join_url(self.uri, "/api/security/encryptedPassword")
+        )
+
+    def ping(self) -> Response:
+        """
+        Sends a ping to Artifactory to validate login status
+        """
+        url = join_url(self.uri, "api/system/ping")
+        return self.http_client.get(url=url)
+
+    def get_artifactory_version(self) -> Response:
+        return self.http_client.get(url=join_url(self.uri, "/api/system/version"))
+
+    def create_machinelearning_local_repo(self, repo_name: str) -> Response:
+        data = {
+            "rclass": "local",
+            "packageType": "machinelearning",
+        }
+        return self.http_client.put(
+            url=join_url(self.uri, "/api/repositories/" + repo_name), json=data
+        )
+
+    def delete_frogml_local_repo(self, repo_name: str) -> Response:
+        return self.http_client.delete(
+            url=join_url(self.uri, "/api/repositories/" + repo_name)
+        )
+
+
+class RetryWithLog(Retry):
+    """
+    Adding extra logs before making a retry request
+    """
+
+    def __init__(self, *args, **kwargs):
+        history = kwargs.get("history")
+        if history is not None:
+            logger.debug(f"Error: ${history[-1].error}\nretrying...")
+        super().__init__(*args, **kwargs)

frogml_storage/authentication/login/__init__.py

@@ -0,0 +1 @@
+from ._login_cli import login as frogml_login