frogml-core 0.0.86__tar.gz → 0.0.88__tar.gz

{frogml_core-0.0.86 → frogml_core-0.0.88}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-core
-Version: 0.0.86
+Version: 0.0.88
 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
@@ -31,7 +31,7 @@ Requires-Dist: protobuf (>=4.21.6) ; python_version >= "3.10"
 Requires-Dist: pyarrow (>=6.0.0) ; extra == "feature-store"
 Requires-Dist: pyathena (>=2.2.0,!=2.18.0) ; extra == "feature-store"
 Requires-Dist: pyspark (==3.4.2) ; extra == "feature-store"
-Requires-Dist: python-jose
+Requires-Dist: python-jose[cryptography] (>=3.4.0)
 Requires-Dist: python-json-logger (>=2.0.2)
 Requires-Dist: requests
 Requires-Dist: retrying (==1.3.4)

{frogml_core-0.0.86 → frogml_core-0.0.88}/frogml_core/__init__.py

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

frogml_core-0.0.88/frogml_core/clients/jfrog_gateway/client.py

@@ -0,0 +1,60 @@
+from typing import cast
+
+from dependency_injector.wiring import Provide, inject
+
+from frogml_core.inner.di_configuration import FrogmlContainer
+from frogml_core.inner.tool.grpc.grpc_try_wrapping import grpc_try_catch_wrapper
+from frogml_proto.qwak.ecosystem.jfrog.v0.jfrog_tenant_info_service_pb2 import (
+    GetJFrogTenantInfoResponse,
+    GetJFrogTenantInfoRequest,
+)
+from frogml_proto.qwak.ecosystem.jfrog.v0.jfrog_tenant_info_service_pb2_grpc import (
+    JFrogTenantInfoServiceStub,
+)
+from frogml_proto.qwak.jfrog.gateway.v0.repository_service_pb2 import (
+    GetRepositoryConfigurationResponse,
+    GetRepositoryConfigurationRequest,
+)
+from frogml_proto.qwak.jfrog.gateway.v0.repository_service_pb2_grpc import (
+    RepositoryServiceStub,
+)
+
+
+class JfrogGatewayClient:
+    """
+    Used for interacting with Feature Registry endpoints
+    """
+
+    @inject
+    def __init__(self, grpc_channel=Provide[FrogmlContainer.core_grpc_channel]):
+        self.__repository_service = RepositoryServiceStub(grpc_channel)
+        self.__jfrog_tenant_info_service = JFrogTenantInfoServiceStub(grpc_channel)
+
+    @grpc_try_catch_wrapper("Failed to get repository configuration")
+    def get_repository_configuration(
+        self, repository_key: str
+    ) -> GetRepositoryConfigurationResponse:
+        """
+        Get repository configuration
+        :param repository_key: Repository name
+        :return: Model
+        """
+        request = GetRepositoryConfigurationRequest(repository_key=repository_key)
+
+        return cast(
+            GetRepositoryConfigurationResponse,
+            self.__repository_service.GetRepositoryConfiguration(request),
+        )
+
+    @grpc_try_catch_wrapper("Failed to get JFrog tenant info")
+    def get_jfrog_tenant_info(self) -> GetJFrogTenantInfoResponse:
+        """
+        Get the customer's JFrog tenant info
+        :return: The JFrog tenant info response
+        """
+        return cast(
+            GetJFrogTenantInfoResponse,
+            self.__jfrog_tenant_info_service.GetJFrogTenantInfo(
+                GetJFrogTenantInfoRequest()
+            ),
+        )

frogml_core-0.0.88/frogml_core/inner/tool/grpc/grpc_try_wrapping.py

@@ -0,0 +1,40 @@
+import logging
+from typing import Callable
+
+import grpc
+
+from frogml_core.exceptions import FrogmlException
+
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.INFO)
+
+
+def grpc_try_catch_wrapper(exception_message: str):
+    def decorator(function: Callable):
+        def _inner_wrapper(*args, **kwargs):
+            try:
+                logger.debug(
+                    "About to call gRPC function: %s where *args = %s and **kwargs = %s",
+                    function.__name__,
+                    args,
+                    kwargs,
+                )
+                return function(*args, **kwargs)
+            except Exception as e:
+                logger.exception(
+                    "An exception occurred in gRPC function %s. Exception: %s",
+                    function.__name__,
+                    e,
+                )
+                if isinstance(e, grpc.RpcError):
+                    # noinspection PyUnresolvedReferences
+                    raise FrogmlException(
+                        exception_message + f". Error is: {e.details()}."
+                    ) from e
+
+                raise e
+
+        return _inner_wrapper
+
+    return decorator

frogml_core-0.0.88/frogml_proto/qwak/ecosystem/jfrog/v0/jfrog_tenant_info_service_pb2.py

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml_proto.qwak.ecosystem/jfrog/v0/jfrog_tenant_info_service.proto
+"""Generated protocol buffer code."""
+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()
+
+
+from frogml_proto.qwak.ecosystem.jfrog.v0 import jfrog_tenant_pb2 as qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n7qwak/ecosystem/jfrog/v0/jfrog_tenant_info_service.proto\x12)qwak.ecosystem.jfrog.jfrog_tenant_info.v0\x1a*qwak/ecosystem/jfrog/v0/jfrog_tenant.proto\"\x1b\n\x19GetJFrogTenantInfoRequest\"m\n\x1aGetJFrogTenantInfoResponse\x12O\n\x0btenant_info\x18\x01 \x01(\x0b\x32:.qwak.ecosystem.jfrog.jfrog_tenant_info.v0.JFrogTenantInfo\"\x1a\n\x18ListJFrogProjectsRequest\"f\n\x19ListJFrogProjectsResponse\x12I\n\x08projects\x18\x01 \x03(\x0b\x32\x37.qwak.ecosystem.jfrog.jfrog_tenant_info.v0.JFrogProject2\xdd\x02\n\x16JFrogTenantInfoService\x12\xa1\x01\n\x12GetJFrogTenantInfo\x12\x44.qwak.ecosystem.jfrog.jfrog_tenant_info.v0.GetJFrogTenantInfoRequest\x1a\x45.qwak.ecosystem.jfrog.jfrog_tenant_info.v0.GetJFrogTenantInfoResponse\x12\x9e\x01\n\x11ListJFrogProjects\x12\x43.qwak.ecosystem.jfrog.jfrog_tenant_info.v0.ListJFrogProjectsRequest\x1a\x44.qwak.ecosystem.jfrog.jfrog_tenant_info.v0.ListJFrogProjectsResponseB\xbd\x01\n com.qwak.ai.jfrog.tenant.info.v0P\x01Z\x96\x01github.com/qwak-ai/qwak-platform/services/shared/java/libs/qwak-common/qwak-common-api/qwak-platform-ecosystem-api/pb/qwak/ecosystem/jfrog/v0;jfrog_v0b\x06proto3')
+
+
+
+_GETJFROGTENANTINFOREQUEST = DESCRIPTOR.message_types_by_name['GetJFrogTenantInfoRequest']
+_GETJFROGTENANTINFORESPONSE = DESCRIPTOR.message_types_by_name['GetJFrogTenantInfoResponse']
+_LISTJFROGPROJECTSREQUEST = DESCRIPTOR.message_types_by_name['ListJFrogProjectsRequest']
+_LISTJFROGPROJECTSRESPONSE = DESCRIPTOR.message_types_by_name['ListJFrogProjectsResponse']
+GetJFrogTenantInfoRequest = _reflection.GeneratedProtocolMessageType('GetJFrogTenantInfoRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETJFROGTENANTINFOREQUEST,
+  '__module__' : 'qwak.ecosystem.jfrog.v0.jfrog_tenant_info_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.ecosystem.jfrog.jfrog_tenant_info.v0.GetJFrogTenantInfoRequest)
+  })
+_sym_db.RegisterMessage(GetJFrogTenantInfoRequest)
+
+GetJFrogTenantInfoResponse = _reflection.GeneratedProtocolMessageType('GetJFrogTenantInfoResponse', (_message.Message,), {
+  'DESCRIPTOR' : _GETJFROGTENANTINFORESPONSE,
+  '__module__' : 'qwak.ecosystem.jfrog.v0.jfrog_tenant_info_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.ecosystem.jfrog.jfrog_tenant_info.v0.GetJFrogTenantInfoResponse)
+  })
+_sym_db.RegisterMessage(GetJFrogTenantInfoResponse)
+
+ListJFrogProjectsRequest = _reflection.GeneratedProtocolMessageType('ListJFrogProjectsRequest', (_message.Message,), {
+  'DESCRIPTOR' : _LISTJFROGPROJECTSREQUEST,
+  '__module__' : 'qwak.ecosystem.jfrog.v0.jfrog_tenant_info_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.ecosystem.jfrog.jfrog_tenant_info.v0.ListJFrogProjectsRequest)
+  })
+_sym_db.RegisterMessage(ListJFrogProjectsRequest)
+
+ListJFrogProjectsResponse = _reflection.GeneratedProtocolMessageType('ListJFrogProjectsResponse', (_message.Message,), {
+  'DESCRIPTOR' : _LISTJFROGPROJECTSRESPONSE,
+  '__module__' : 'qwak.ecosystem.jfrog.v0.jfrog_tenant_info_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.ecosystem.jfrog.jfrog_tenant_info.v0.ListJFrogProjectsResponse)
+  })
+_sym_db.RegisterMessage(ListJFrogProjectsResponse)
+
+_JFROGTENANTINFOSERVICE = DESCRIPTOR.services_by_name['JFrogTenantInfoService']
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n com.qwak.ai.jfrog.tenant.info.v0P\001Z\226\001github.com/qwak-ai/qwak-platform/services/shared/java/libs/qwak-common/qwak-common-api/qwak-platform-ecosystem-api/pb/qwak/ecosystem/jfrog/v0;jfrog_v0'
+  _GETJFROGTENANTINFOREQUEST._serialized_start=146
+  _GETJFROGTENANTINFOREQUEST._serialized_end=173
+  _GETJFROGTENANTINFORESPONSE._serialized_start=175
+  _GETJFROGTENANTINFORESPONSE._serialized_end=284
+  _LISTJFROGPROJECTSREQUEST._serialized_start=286
+  _LISTJFROGPROJECTSREQUEST._serialized_end=312
+  _LISTJFROGPROJECTSRESPONSE._serialized_start=314
+  _LISTJFROGPROJECTSRESPONSE._serialized_end=416
+  _JFROGTENANTINFOSERVICE._serialized_start=419
+  _JFROGTENANTINFOSERVICE._serialized_end=768
+# @@protoc_insertion_point(module_scope)

frogml_core-0.0.88/frogml_proto/qwak/ecosystem/jfrog/v0/jfrog_tenant_info_service_pb2.pyi

@@ -0,0 +1,69 @@
+"""
+@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.message
+import frogml_proto.qwak.ecosystem.jfrog.v0.jfrog_tenant_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class GetJFrogTenantInfoRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___GetJFrogTenantInfoRequest = GetJFrogTenantInfoRequest
+
+class GetJFrogTenantInfoResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TENANT_INFO_FIELD_NUMBER: builtins.int
+    @property
+    def tenant_info(self) -> frogml_proto.qwak.ecosystem.jfrog.v0.jfrog_tenant_pb2.JFrogTenantInfo:
+        """ Information about the tenant"""
+    def __init__(
+        self,
+        *,
+        tenant_info: frogml_proto.qwak.ecosystem.jfrog.v0.jfrog_tenant_pb2.JFrogTenantInfo | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["tenant_info", b"tenant_info"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["tenant_info", b"tenant_info"]) -> None: ...
+
+global___GetJFrogTenantInfoResponse = GetJFrogTenantInfoResponse
+
+class ListJFrogProjectsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ListJFrogProjectsRequest = ListJFrogProjectsRequest
+
+class ListJFrogProjectsResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PROJECTS_FIELD_NUMBER: builtins.int
+    @property
+    def projects(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.ecosystem.jfrog.v0.jfrog_tenant_pb2.JFrogProject]:
+        """A list of projects for which the requesting user is an admin"""
+    def __init__(
+        self,
+        *,
+        projects: collections.abc.Iterable[qwak.ecosystem.jfrog.v0.jfrog_tenant_pb2.JFrogProject] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["projects", b"projects"]) -> None: ...
+
+global___ListJFrogProjectsResponse = ListJFrogProjectsResponse

frogml_core-0.0.88/frogml_proto/qwak/ecosystem/jfrog/v0/jfrog_tenant_info_service_pb2_grpc.py

@@ -0,0 +1,102 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from frogml_proto.qwak.ecosystem.jfrog.v0 import jfrog_tenant_info_service_pb2 as qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__info__service__pb2
+
+
+class JFrogTenantInfoServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetJFrogTenantInfo = channel.unary_unary(
+                '/qwak.ecosystem.jfrog.jfrog_tenant_info.v0.JFrogTenantInfoService/GetJFrogTenantInfo',
+                request_serializer=qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__info__service__pb2.GetJFrogTenantInfoRequest.SerializeToString,
+                response_deserializer=qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__info__service__pb2.GetJFrogTenantInfoResponse.FromString,
+                )
+        self.ListJFrogProjects = channel.unary_unary(
+                '/qwak.ecosystem.jfrog.jfrog_tenant_info.v0.JFrogTenantInfoService/ListJFrogProjects',
+                request_serializer=qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__info__service__pb2.ListJFrogProjectsRequest.SerializeToString,
+                response_deserializer=qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__info__service__pb2.ListJFrogProjectsResponse.FromString,
+                )
+
+
+class JFrogTenantInfoServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetJFrogTenantInfo(self, request, context):
+        """Get Tenant Info
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ListJFrogProjects(self, request, context):
+        """List Tenant Projects
+        Returns all projects for which the requesting user is an admin
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_JFrogTenantInfoServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'GetJFrogTenantInfo': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetJFrogTenantInfo,
+                    request_deserializer=qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__info__service__pb2.GetJFrogTenantInfoRequest.FromString,
+                    response_serializer=qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__info__service__pb2.GetJFrogTenantInfoResponse.SerializeToString,
+            ),
+            'ListJFrogProjects': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListJFrogProjects,
+                    request_deserializer=qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__info__service__pb2.ListJFrogProjectsRequest.FromString,
+                    response_serializer=qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__info__service__pb2.ListJFrogProjectsResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.ecosystem.jfrog.jfrog_tenant_info.v0.JFrogTenantInfoService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class JFrogTenantInfoService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetJFrogTenantInfo(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.ecosystem.jfrog.jfrog_tenant_info.v0.JFrogTenantInfoService/GetJFrogTenantInfo',
+            qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__info__service__pb2.GetJFrogTenantInfoRequest.SerializeToString,
+            qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__info__service__pb2.GetJFrogTenantInfoResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def ListJFrogProjects(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.ecosystem.jfrog.jfrog_tenant_info.v0.JFrogTenantInfoService/ListJFrogProjects',
+            qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__info__service__pb2.ListJFrogProjectsRequest.SerializeToString,
+            qwak_dot_ecosystem_dot_jfrog_dot_v0_dot_jfrog__tenant__info__service__pb2.ListJFrogProjectsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

frogml_core-0.0.88/frogml_proto/qwak/ecosystem/jfrog/v0/jfrog_tenant_pb2.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml_proto.qwak.ecosystem/jfrog/v0/jfrog_tenant.proto
+"""Generated protocol buffer code."""
+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*qwak/ecosystem/jfrog/v0/jfrog_tenant.proto\x12)qwak.ecosystem.jfrog.jfrog_tenant_info.v0\"\'\n\x0fJFrogTenantInfo\x12\x14\n\x0cplatform_url\x18\x01 \x01(\t\"N\n\x0cJFrogProject\x12\x13\n\x0bproject_key\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\tB\xbd\x01\n com.qwak.ai.jfrog.tenant.info.v0P\x01Z\x96\x01github.com/qwak-ai/qwak-platform/services/shared/java/libs/qwak-common/qwak-common-api/qwak-platform-ecosystem-api/pb/qwak/ecosystem/jfrog/v0;jfrog_v0b\x06proto3')
+
+
+
+_JFROGTENANTINFO = DESCRIPTOR.message_types_by_name['JFrogTenantInfo']
+_JFROGPROJECT = DESCRIPTOR.message_types_by_name['JFrogProject']
+JFrogTenantInfo = _reflection.GeneratedProtocolMessageType('JFrogTenantInfo', (_message.Message,), {
+  'DESCRIPTOR' : _JFROGTENANTINFO,
+  '__module__' : 'qwak.ecosystem.jfrog.v0.jfrog_tenant_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.ecosystem.jfrog.jfrog_tenant_info.v0.JFrogTenantInfo)
+  })
+_sym_db.RegisterMessage(JFrogTenantInfo)
+
+JFrogProject = _reflection.GeneratedProtocolMessageType('JFrogProject', (_message.Message,), {
+  'DESCRIPTOR' : _JFROGPROJECT,
+  '__module__' : 'qwak.ecosystem.jfrog.v0.jfrog_tenant_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.ecosystem.jfrog.jfrog_tenant_info.v0.JFrogProject)
+  })
+_sym_db.RegisterMessage(JFrogProject)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n com.qwak.ai.jfrog.tenant.info.v0P\001Z\226\001github.com/qwak-ai/qwak-platform/services/shared/java/libs/qwak-common/qwak-common-api/qwak-platform-ecosystem-api/pb/qwak/ecosystem/jfrog/v0;jfrog_v0'
+  _JFROGTENANTINFO._serialized_start=89
+  _JFROGTENANTINFO._serialized_end=128
+  _JFROGPROJECT._serialized_start=130
+  _JFROGPROJECT._serialized_end=208
+# @@protoc_insertion_point(module_scope)

frogml_core-0.0.88/frogml_proto/qwak/ecosystem/jfrog/v0/jfrog_tenant_pb2.pyi

@@ -0,0 +1,53 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class JFrogTenantInfo(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PLATFORM_URL_FIELD_NUMBER: builtins.int
+    platform_url: builtins.str
+    """The JFrog platform base url"""
+    def __init__(
+        self,
+        *,
+        platform_url: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["platform_url", b"platform_url"]) -> None: ...
+
+global___JFrogTenantInfo = JFrogTenantInfo
+
+class JFrogProject(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PROJECT_KEY_FIELD_NUMBER: builtins.int
+    DISPLAY_NAME_FIELD_NUMBER: builtins.int
+    DESCRIPTION_FIELD_NUMBER: builtins.int
+    project_key: builtins.str
+    """The project's key"""
+    display_name: builtins.str
+    """The project's display name"""
+    description: builtins.str
+    """The project's description"""
+    def __init__(
+        self,
+        *,
+        project_key: builtins.str = ...,
+        display_name: builtins.str = ...,
+        description: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["description", b"description", "display_name", b"display_name", "project_key", b"project_key"]) -> None: ...
+
+global___JFrogProject = JFrogProject

frogml_core-0.0.88/frogml_proto/qwak/workspace/workspace_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_core-0.0.86 → frogml_core-0.0.88}/frogml_services_mock/mocks/frogml_mocks.py

@@ -58,6 +58,7 @@ from frogml_services_mock.mocks.integration_management_service import (
 from frogml_services_mock.mocks.internal_build_orchestrator_service import (
     InternalBuildOrchestratorServiceMock,
 )
+from frogml_services_mock.mocks.jfrog_tenant_info_service_mock import JFrogTenantInfoServiceMock
 from frogml_services_mock.mocks.job_registry_service_api import (
     JobRegistryServiceApiMock,
 )
@@ -135,3 +136,4 @@ class FrogmlMocks:
     model_version_manager_service: ModelVersionManagerServiceMock
     repository_service: RepositoryServiceMock
     location_discovery_service: LocationDiscoveryServiceApiMock
+    jfrog_tenant_info_service: JFrogTenantInfoServiceMock

frogml_core-0.0.88/frogml_services_mock/mocks/jfrog_tenant_info_service_mock.py

@@ -0,0 +1,42 @@
+from datetime import datetime
+from typing import Dict
+
+from google.protobuf.timestamp_pb2 import Timestamp
+
+from frogml_core.exceptions import FrogmlNotFoundException
+from frogml_proto.qwak.ecosystem.jfrog.v0.jfrog_tenant_info_service_pb2 import (
+    GetJFrogTenantInfoRequest,
+    GetJFrogTenantInfoResponse,
+)
+from frogml_proto.qwak.ecosystem.jfrog.v0.jfrog_tenant_info_service_pb2_grpc import (
+    JFrogTenantInfoServiceServicer,
+)
+from frogml_proto.qwak.ecosystem.jfrog.v0.jfrog_tenant_pb2 import JFrogTenantInfo
+from frogml_proto.qwak.jfrog.gateway.v0.repository_pb2 import RepositorySpec
+from frogml_services_mock.mocks.utils.exception_handlers import (
+    raise_not_found_grpc_error,
+)
+
+timestamp = Timestamp()
+timestamp.FromDatetime(datetime.now())
+
+
+class JFrogTenantInfoServiceMock(JFrogTenantInfoServiceServicer):
+    def __init__(self):
+        super(JFrogTenantInfoServiceMock, self).__init__()
+        self.repositories: Dict[str, RepositorySpec] = {}
+        self.should_raise_exception: bool = False
+        self.platform_url: str = "mock.jfrog.io"
+
+    def GetJFrogTenantInfo(
+        self, request: GetJFrogTenantInfoRequest, context
+    ) -> GetJFrogTenantInfoResponse:
+        if self.should_raise_exception:
+            raise_not_found_grpc_error(
+                context,
+                FrogmlNotFoundException(f"JFrog tenant info not found"),
+            )
+
+        return GetJFrogTenantInfoResponse(
+            tenant_info=JFrogTenantInfo(platform_url=self.platform_url)
+        )

{frogml_core-0.0.86 → frogml_core-0.0.88}/frogml_services_mock/services_mock.py

@@ -3,6 +3,7 @@ from typing import Any, Generator, List, Tuple
 
 import grpc
 import pytest
+
 from frogml_core.inner.di_configuration import FrogmlContainer
 from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc import (
     add_ModelVersionManagerServiceServicer_to_server,
@@ -53,6 +54,8 @@ from frogml_proto.qwak.deployment.alert_service_pb2_grpc import (
 from frogml_proto.qwak.deployment.deployment_service_pb2_grpc import (
     add_DeploymentManagementServiceServicer_to_server,
 )
+from frogml_proto.qwak.ecosystem.jfrog.v0.jfrog_tenant_info_service_pb2_grpc import \
+    add_JFrogTenantInfoServiceServicer_to_server
 from frogml_proto.qwak.ecosystem.v0.ecosystem_runtime_service_pb2_grpc import (
     add_QwakEcosystemRuntimeServicer_to_server,
 )
@@ -183,6 +186,7 @@ from frogml_services_mock.mocks.integration_management_service import (
 from frogml_services_mock.mocks.internal_build_orchestrator_service import (
     InternalBuildOrchestratorServiceMock,
 )
+from frogml_services_mock.mocks.jfrog_tenant_info_service_mock import JFrogTenantInfoServiceMock
 from frogml_services_mock.mocks.job_registry_service_api import (
     JobRegistryServiceApiMock,
 )
@@ -539,6 +543,11 @@ def attach_servicers(free_port, server):
                 LocationDiscoveryServiceApiMock,
                 add_LocationDiscoveryServiceServicer_to_server,
             ),
+            (
+                "jfrog_tenant_info_service",
+                JFrogTenantInfoServiceMock,
+                add_JFrogTenantInfoServiceServicer_to_server,
+            ),
             ("port", free_port, None),
         ],
     )

{frogml_core-0.0.86 → frogml_core-0.0.88}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "frogml-core"
-version = "0.0.86"
+version = "0.0.88"
 description = "frogml Core contains the necessary objects and communication tools for using the Jfrog ml Platform"
 authors = ["Jfrog"]
 readme = "README.md"
@@ -38,7 +38,7 @@ protobuf = [
 ]
 dependency-injector = ">=4.0"
 requests = "*"
-python-jose = "*"
+python-jose = {version = ">=3.4.0", extras = ["cryptography"]}
 PyYAML = "*"
 filelock = "*"
 marshmallow-dataclass = "^8.5.8"

frogml_core-0.0.86/frogml_core/clients/jfrog_gateway/client.py

@@ -1,41 +0,0 @@
-from typing import cast
-
-from dependency_injector.wiring import Provide, inject
-
-from frogml_core.exceptions import FrogmlException
-from frogml_core.inner.di_configuration import FrogmlContainer
-from frogml_proto.qwak.jfrog.gateway.v0.repository_service_pb2 import (
-    GetRepositoryConfigurationResponse,
-    GetRepositoryConfigurationRequest,
-)
-from frogml_proto.qwak.jfrog.gateway.v0.repository_service_pb2_grpc import (
-    RepositoryServiceStub,
-)
-
-
-class JfrogGatewayClient:
-    """
-    Used for interacting with Feature Registry endpoints
-    """
-
-    @inject
-    def __init__(self, grpc_channel=Provide[FrogmlContainer.core_grpc_channel]):
-        self.__repository_service = RepositoryServiceStub(grpc_channel)
-
-    def get_repository_configuration(
-        self, repository_key: str
-    ) -> GetRepositoryConfigurationResponse:
-        """
-        Get repository configuration
-        :param repository_key: Repository name
-        :return: Model
-        """
-        request = GetRepositoryConfigurationRequest(repository_key=repository_key)
-
-        try:
-            return cast(
-                GetRepositoryConfigurationResponse,
-                self.__repository_service.GetRepositoryConfiguration(request),
-            )
-        except Exception as e:
-            raise FrogmlException(f"Failed to get repository configuration: {e}") from e

frogml_core-0.0.86/frogml_core/inner/tool/grpc/grpc_try_wrapping.py

@@ -1,16 +0,0 @@
-import grpc
-
-from frogml_core.exceptions import FrogmlException
-
-
-def grpc_try_catch_wrapper(exception_message):
-    def decorator(function):
-        def _inner_wrapper(*args, **kwargs):
-            try:
-                return function(*args, **kwargs)
-            except grpc.RpcError as e:
-                raise FrogmlException(exception_message + f". Error is: {e.details()}.")
-
-        return _inner_wrapper
-
-    return decorator