frogml-core 0.0.86__py3-none-any.whl → 0.0.88__py3-none-any.whl

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/clients/jfrog_gateway/client.py

@@ -2,8 +2,15 @@ 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_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,
@@ -21,7 +28,9 @@ class JfrogGatewayClient:
     @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:
@@ -32,10 +41,20 @@ class JfrogGatewayClient:
         """
         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
+        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/inner/tool/grpc/grpc_try_wrapping.py

@@ -1,15 +1,39 @@
+import logging
+from typing import Callable
+
 import grpc
 
 from frogml_core.exceptions import FrogmlException
 
 
-def grpc_try_catch_wrapper(exception_message):
-    def decorator(function):
+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 grpc.RpcError as e:
-                raise FrogmlException(exception_message + f". Error is: {e.details()}.")
+            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
 

{frogml_core-0.0.86.dist-info → frogml_core-0.0.88.dist-info}/METADATA

@@ -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.dist-info → frogml_core-0.0.88.dist-info}/RECORD

@@ -1,4 +1,4 @@
-frogml_core/__init__.py,sha256=Ed-B3bIfukVJ22Ht4IpZ_A6-BAXGXgCI-XF79-6UH1M,777
+frogml_core/__init__.py,sha256=twgBbCi-3DxE4i01CUscvYdGwQFPLUrynAGv-9irXi0,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
@@ -65,7 +65,7 @@ frogml_core/clients/integration_management/integration_utils.py,sha256=j5gomMtYi
 frogml_core/clients/integration_management/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/clients/integration_management/openai/openai_system_secret.py,sha256=eP3blUextKicLnjpDcyElpShKCwsSnq7_OIt5pf3kLc,2145
 frogml_core/clients/jfrog_gateway/__init__.py,sha256=E_BrYKBESU3wGNiR_RQncbAhLWyvJ-Ub2Akt_6FpfgM,39
-frogml_core/clients/jfrog_gateway/client.py,sha256=H4hF3MNi0QBYwGgD9c9A-4-VQlBB7zlQgH3ImpX8aBQ,1357
+frogml_core/clients/jfrog_gateway/client.py,sha256=-7eN9nq7BUVDCC_rBBHlOd7Ngu6sL1Lwn4G-pLKA51g,2101
 frogml_core/clients/kube_deployment_captain/__init__.py,sha256=rJUEEy3zNH0aTFyuO_UBexzaUKdjvwU9P2vV1MDj684,41
 frogml_core/clients/kube_deployment_captain/client.py,sha256=oz7VF37TSO0S07MqXOYu2Xmx_rl9IVrfHOz_8MWnBZ8,9340
 frogml_core/clients/location_discovery/__init__.py,sha256=sqGQ75YHFE6nvOcir38fykUUmAa6cFEIze8PJYgYWRc,44
@@ -285,7 +285,7 @@ frogml_core/inner/tool/auth.py,sha256=H0tQTc8JgEGLxv79IS_28jahi8cq8NNoP3lpxlQshP
 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
-frogml_core/inner/tool/grpc/grpc_try_wrapping.py,sha256=zSk7netTKMLoBf8WW3uf5Bw5bqlhGVDtx83Q3wQ8JkU,434
+frogml_core/inner/tool/grpc/grpc_try_wrapping.py,sha256=nMNzmHQcgROBfd_QOTTUCZy-bIcpxPGPiUwsseqIrLg,1172
 frogml_core/inner/tool/protobuf_factory.py,sha256=QBk7ySxHRkxvrC8ICZR7sYizDHIZKQHOcGfo6qXnrDA,1699
 frogml_core/inner/tool/retry_utils.py,sha256=KcSFJuj02RKF-H9INpCmdiTNXlywEMJ2ClBa00N9aNM,435
 frogml_core/inner/tool/run_config/__init__.py,sha256=krOWmfbiUyMxa4Z7FHZk3gGZBbMiJINxLxD7XwyUefE,277
@@ -666,6 +666,12 @@ frogml_proto/qwak/deployment/deployment_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8
 frogml_proto/qwak/deployment/deployment_service_pb2.py,sha256=nElkI-aRZBXTRbWTn6yZ9-m_sSMG7S_KKSEn2qPH_jg,46324
 frogml_proto/qwak/deployment/deployment_service_pb2.pyi,sha256=U43wEm09zS8pyjl_bHqym9b778pgodxPl-LrsP8gTrc,40993
 frogml_proto/qwak/deployment/deployment_service_pb2_grpc.py,sha256=JTfQWEXtDmSa7E_wlCPFGuWe4mWF3BhbU5kphAJmEUk,27757
+frogml_proto/qwak/ecosystem/jfrog/v0/jfrog_tenant_info_service_pb2.py,sha256=IYnu38-bjQHONTmtb57FrcqR6MOIIfxOvIXIb_wpN2w,4798
+frogml_proto/qwak/ecosystem/jfrog/v0/jfrog_tenant_info_service_pb2.pyi,sha256=hnVo03rTflR9nBvNIYqHf-RB5PfrxJhn3hnKgnTQ5ac,2448
+frogml_proto/qwak/ecosystem/jfrog/v0/jfrog_tenant_info_service_pb2_grpc.py,sha256=yrYrxN01Ty295xL0HDECbKVudyw98UKxAP_qsYx1gTQ,5294
+frogml_proto/qwak/ecosystem/jfrog/v0/jfrog_tenant_pb2.py,sha256=GLErGwAdocN1Cn_06ye-NFdCGEzBfYkAutYf6Eoh4FU,2459
+frogml_proto/qwak/ecosystem/jfrog/v0/jfrog_tenant_pb2.pyi,sha256=685WP-kcl5wFSPW9H8wPRES9XSrRdFnIAOknpfMsrbc,1639
+frogml_proto/qwak/ecosystem/jfrog/v0/jfrog_tenant_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 frogml_proto/qwak/ecosystem/jfrog/v0/token_pb2.py,sha256=8hJ7R8WfDA3c8ey_kgKFcPsi4_V-Vw0tNwsN9aEhxDQ,13287
 frogml_proto/qwak/ecosystem/jfrog/v0/token_pb2.pyi,sha256=eZhdgcdWYETdyH8uKi8qHfqx97_i_JTzlI2SY_Pppoc,17104
 frogml_proto/qwak/ecosystem/jfrog/v0/token_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -1070,11 +1076,12 @@ frogml_services_mock/mocks/features_operator_v3_service.py,sha256=_RcUeL9LRFbxL7
 frogml_services_mock/mocks/features_set_state_service_api.py,sha256=jLtDYRBQUXP9x2DaywRPNFTtAGewP1JF1wv28ZTRbL4,2290
 frogml_services_mock/mocks/feedback_service.py,sha256=NH8IskHnwbHGfDJCunSWMWQd9UfVBL7rPzVrFWrCZC4,1140
 frogml_services_mock/mocks/file_versioning_service.py,sha256=MtxGcWoB_hkJUMBRSso9-G_6_WBbHkrgzG6Rf_37Ysk,2606
-frogml_services_mock/mocks/frogml_mocks.py,sha256=bR8pW3pdz5ogHBiHTA_RrQNf85x1CDSnUnwMvOpt4p0,6546
+frogml_services_mock/mocks/frogml_mocks.py,sha256=zJx_jIOkXtO0ngdZ1iHNykku1KcaHsmdRSdxGdyotGc,6701
 frogml_services_mock/mocks/fs_offline_serving_service.py,sha256=O4hd4kQ-sXm9zMPVJYHXO4ARPuc3UN0E9rcOtDkSJRk,2093
 frogml_services_mock/mocks/instance_template_management_service.py,sha256=8J8NlD667kWfjhSXsyH31jjr7qKIaF77K1Fc7FgxtHY,4762
 frogml_services_mock/mocks/integration_management_service.py,sha256=XvWyif8pGuqJsrjTs6m29cneVuYdjVptPpRndwIdqq4,2771
 frogml_services_mock/mocks/internal_build_orchestrator_service.py,sha256=saWQOWbJC5uoAcr053rmd0Jj2TI4TH3Kyr2D5lsL87w,1059
+frogml_services_mock/mocks/jfrog_tenant_info_service_mock.py,sha256=wvRJS5Jn1y_7nRR5PXmoJwBrA3qDB8fu8tOXXlHxkfA,1510
 frogml_services_mock/mocks/job_registry_service_api.py,sha256=Zd5lVM6h4jFfKHxnQAux1FiBEw2tXFA284OJ33a_IH4,2711
 frogml_services_mock/mocks/kube_captain_service_api.py,sha256=WVCaoOHY-kFdS73bd7kuOssr1RAK1F6MUlJ-NO0eLfY,1596
 frogml_services_mock/mocks/location_discovery_service_api.py,sha256=TUdku1zdmIZXZYbkGurK-OhScPfMeGUzRVa7FgxZnrQ,3682
@@ -1093,9 +1100,9 @@ frogml_services_mock/mocks/utils/exception_handlers.py,sha256=k_8mez3cwjNjKE9yGQ
 frogml_services_mock/mocks/vector_serving_api.py,sha256=ZljLOw9_ee-nlvEMU0HNzmK2tcsmBY6VzuVzLmInYj4,5838
 frogml_services_mock/mocks/vectors_management_api.py,sha256=-GtKow3JmBj6LRZw625WdD8pt9VKtGZUs2VXTbtEPg0,3602
 frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xTicwJO7jdY-gN_5hF_s9GOU-ZO5P_2_M,7745
-frogml_services_mock/services_mock.py,sha256=xfbYvl05gFUq2oT82cT-6K2CEJ6d9kFX_tk77QZHTFI,20787
+frogml_services_mock/services_mock.py,sha256=sgKgwhu2W0YOHtzil8x7f1znK_sZr_i27XSeiF4xqVE,21200
 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.86.dist-info/METADATA,sha256=zvWjQmk9m28_k_0lcWSrXF1_XPWb8hYbD9S_21jWo58,2004
-frogml_core-0.0.86.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-frogml_core-0.0.86.dist-info/RECORD,,
+frogml_core-0.0.88.dist-info/METADATA,sha256=d-_5vh5OvKAIllVEA8nyZrJWJ1zdLFT4gBDIRKxeEvc,2028
+frogml_core-0.0.88.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+frogml_core-0.0.88.dist-info/RECORD,,

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_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_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_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_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_proto/qwak/ecosystem/jfrog/v0/jfrog_tenant_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_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_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_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),
         ],
     )