qwak-core 0.4.246__py3-none-any.whl → 0.4.248__py3-none-any.whl

_qwak_proto/qwak/service_discovery/service_discovery_location_pb2.py

@@ -0,0 +1,65 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/service_discovery/service_discovery_location.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'\n7qwak/service_discovery/service_discovery_location.proto\x12\"qwak.service_discovery.location.v0\"w\n\x19ServiceLocationDescriptor\x12\x13\n\x0bservice_url\x18\x01 \x01(\t\x12\x45\n\x08location\x18\x02 \x01(\x0b\x32\x33.qwak.service_discovery.location.v0.ServiceLocation\"\x99\x01\n\x0fServiceLocation\x12\x38\n\x04\x63ore\x18\x01 \x01(\x0b\x32(.qwak.service_discovery.location.v0.CoreH\x00\x12\x38\n\x04\x65\x64ge\x18\x02 \x01(\x0b\x32(.qwak.service_discovery.location.v0.EdgeH\x00\x42\x12\n\x10service_location\"\x06\n\x04\x43ore\"\x06\n\x04\x45\x64geB-\n)com.qwak.ai.service_discovery.location.v0P\x01\x62\x06proto3')
+
+
+
+_SERVICELOCATIONDESCRIPTOR = DESCRIPTOR.message_types_by_name['ServiceLocationDescriptor']
+_SERVICELOCATION = DESCRIPTOR.message_types_by_name['ServiceLocation']
+_CORE = DESCRIPTOR.message_types_by_name['Core']
+_EDGE = DESCRIPTOR.message_types_by_name['Edge']
+ServiceLocationDescriptor = _reflection.GeneratedProtocolMessageType('ServiceLocationDescriptor', (_message.Message,), {
+  'DESCRIPTOR' : _SERVICELOCATIONDESCRIPTOR,
+  '__module__' : 'qwak.service_discovery.service_discovery_location_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.service_discovery.location.v0.ServiceLocationDescriptor)
+  })
+_sym_db.RegisterMessage(ServiceLocationDescriptor)
+
+ServiceLocation = _reflection.GeneratedProtocolMessageType('ServiceLocation', (_message.Message,), {
+  'DESCRIPTOR' : _SERVICELOCATION,
+  '__module__' : 'qwak.service_discovery.service_discovery_location_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.service_discovery.location.v0.ServiceLocation)
+  })
+_sym_db.RegisterMessage(ServiceLocation)
+
+Core = _reflection.GeneratedProtocolMessageType('Core', (_message.Message,), {
+  'DESCRIPTOR' : _CORE,
+  '__module__' : 'qwak.service_discovery.service_discovery_location_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.service_discovery.location.v0.Core)
+  })
+_sym_db.RegisterMessage(Core)
+
+Edge = _reflection.GeneratedProtocolMessageType('Edge', (_message.Message,), {
+  'DESCRIPTOR' : _EDGE,
+  '__module__' : 'qwak.service_discovery.service_discovery_location_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.service_discovery.location.v0.Edge)
+  })
+_sym_db.RegisterMessage(Edge)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n)com.qwak.ai.service_discovery.location.v0P\001'
+  _SERVICELOCATIONDESCRIPTOR._serialized_start=95
+  _SERVICELOCATIONDESCRIPTOR._serialized_end=214
+  _SERVICELOCATION._serialized_start=217
+  _SERVICELOCATION._serialized_end=370
+  _CORE._serialized_start=372
+  _CORE._serialized_end=378
+  _EDGE._serialized_start=380
+  _EDGE._serialized_end=386
+# @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/service_discovery/service_discovery_location_pb2.pyi

@@ -0,0 +1,73 @@
+"""
+@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 ServiceLocationDescriptor(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SERVICE_URL_FIELD_NUMBER: builtins.int
+    LOCATION_FIELD_NUMBER: builtins.int
+    service_url: builtins.str
+    @property
+    def location(self) -> global___ServiceLocation: ...
+    def __init__(
+        self,
+        *,
+        service_url: builtins.str = ...,
+        location: global___ServiceLocation | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["location", b"location"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["location", b"location", "service_url", b"service_url"]) -> None: ...
+
+global___ServiceLocationDescriptor = ServiceLocationDescriptor
+
+class ServiceLocation(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CORE_FIELD_NUMBER: builtins.int
+    EDGE_FIELD_NUMBER: builtins.int
+    @property
+    def core(self) -> global___Core: ...
+    @property
+    def edge(self) -> global___Edge: ...
+    def __init__(
+        self,
+        *,
+        core: global___Core | None = ...,
+        edge: global___Edge | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["core", b"core", "edge", b"edge", "service_location", b"service_location"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["core", b"core", "edge", b"edge", "service_location", b"service_location"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["service_location", b"service_location"]) -> typing_extensions.Literal["core", "edge"] | None: ...
+
+global___ServiceLocation = ServiceLocation
+
+class Core(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___Core = Core
+
+class Edge(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___Edge = Edge

_qwak_proto/qwak/service_discovery/service_discovery_location_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
+

_qwak_proto/qwak/service_discovery/service_discovery_location_service_pb2.py

@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/service_discovery/service_discovery_location_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 _qwak_proto.qwak.service_discovery import service_discovery_location_pb2 as qwak_dot_service__discovery_dot_service__discovery__location__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n?qwak/service_discovery/service_discovery_location_service.proto\x12\"qwak.service_discovery.location.v0\x1a\x37qwak/service_discovery/service_discovery_location.proto\"\x16\n\x14GetServingUrlRequest\"o\n\x1cGetServingUrlRequestResponse\x12O\n\x08location\x18\x01 \x01(\x0b\x32=.qwak.service_discovery.location.v0.ServiceLocationDescriptor2\xa0\x07\n\x18LocationDiscoveryService\x12\x92\x01\n\x14GetOfflineServingUrl\x12\x38.qwak.service_discovery.location.v0.GetServingUrlRequest\x1a@.qwak.service_discovery.location.v0.GetServingUrlRequestResponse\x12\x97\x01\n\x19GetDistributionManagerUrl\x12\x38.qwak.service_discovery.location.v0.GetServingUrlRequest\x1a@.qwak.service_discovery.location.v0.GetServingUrlRequestResponse\x12\x93\x01\n\x15GetAnalyticsEngineUrl\x12\x38.qwak.service_discovery.location.v0.GetServingUrlRequest\x1a@.qwak.service_discovery.location.v0.GetServingUrlRequestResponse\x12\x92\x01\n\x14GetMetricsGatewayUrl\x12\x38.qwak.service_discovery.location.v0.GetServingUrlRequest\x1a@.qwak.service_discovery.location.v0.GetServingUrlRequestResponse\x12\x94\x01\n\x16GetFeaturesOperatorUrl\x12\x38.qwak.service_discovery.location.v0.GetServingUrlRequest\x1a@.qwak.service_discovery.location.v0.GetServingUrlRequestResponse\x12\x92\x01\n\x14GetHostingGatewayUrl\x12\x38.qwak.service_discovery.location.v0.GetServingUrlRequest\x1a@.qwak.service_discovery.location.v0.GetServingUrlRequestResponseB-\n)com.qwak.ai.service_discovery.location.v0P\x01\x62\x06proto3')
+
+
+
+_GETSERVINGURLREQUEST = DESCRIPTOR.message_types_by_name['GetServingUrlRequest']
+_GETSERVINGURLREQUESTRESPONSE = DESCRIPTOR.message_types_by_name['GetServingUrlRequestResponse']
+GetServingUrlRequest = _reflection.GeneratedProtocolMessageType('GetServingUrlRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETSERVINGURLREQUEST,
+  '__module__' : 'qwak.service_discovery.service_discovery_location_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.service_discovery.location.v0.GetServingUrlRequest)
+  })
+_sym_db.RegisterMessage(GetServingUrlRequest)
+
+GetServingUrlRequestResponse = _reflection.GeneratedProtocolMessageType('GetServingUrlRequestResponse', (_message.Message,), {
+  'DESCRIPTOR' : _GETSERVINGURLREQUESTRESPONSE,
+  '__module__' : 'qwak.service_discovery.service_discovery_location_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.service_discovery.location.v0.GetServingUrlRequestResponse)
+  })
+_sym_db.RegisterMessage(GetServingUrlRequestResponse)
+
+_LOCATIONDISCOVERYSERVICE = DESCRIPTOR.services_by_name['LocationDiscoveryService']
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n)com.qwak.ai.service_discovery.location.v0P\001'
+  _GETSERVINGURLREQUEST._serialized_start=160
+  _GETSERVINGURLREQUEST._serialized_end=182
+  _GETSERVINGURLREQUESTRESPONSE._serialized_start=184
+  _GETSERVINGURLREQUESTRESPONSE._serialized_end=295
+  _LOCATIONDISCOVERYSERVICE._serialized_start=298
+  _LOCATIONDISCOVERYSERVICE._serialized_end=1226
+# @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/service_discovery/service_discovery_location_service_pb2.pyi

@@ -0,0 +1,41 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import qwak.service_discovery.service_discovery_location_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class GetServingUrlRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___GetServingUrlRequest = GetServingUrlRequest
+
+class GetServingUrlRequestResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    LOCATION_FIELD_NUMBER: builtins.int
+    @property
+    def location(self) -> qwak.service_discovery.service_discovery_location_pb2.ServiceLocationDescriptor: ...
+    def __init__(
+        self,
+        *,
+        location: qwak.service_discovery.service_discovery_location_pb2.ServiceLocationDescriptor | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["location", b"location"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["location", b"location"]) -> None: ...
+
+global___GetServingUrlRequestResponse = GetServingUrlRequestResponse

_qwak_proto/qwak/service_discovery/service_discovery_location_service_pb2_grpc.py

@@ -0,0 +1,231 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from _qwak_proto.qwak.service_discovery import service_discovery_location_service_pb2 as qwak_dot_service__discovery_dot_service__discovery__location__service__pb2
+
+
+class LocationDiscoveryServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetOfflineServingUrl = channel.unary_unary(
+                '/qwak.service_discovery.location.v0.LocationDiscoveryService/GetOfflineServingUrl',
+                request_serializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.SerializeToString,
+                response_deserializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.FromString,
+                )
+        self.GetDistributionManagerUrl = channel.unary_unary(
+                '/qwak.service_discovery.location.v0.LocationDiscoveryService/GetDistributionManagerUrl',
+                request_serializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.SerializeToString,
+                response_deserializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.FromString,
+                )
+        self.GetAnalyticsEngineUrl = channel.unary_unary(
+                '/qwak.service_discovery.location.v0.LocationDiscoveryService/GetAnalyticsEngineUrl',
+                request_serializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.SerializeToString,
+                response_deserializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.FromString,
+                )
+        self.GetMetricsGatewayUrl = channel.unary_unary(
+                '/qwak.service_discovery.location.v0.LocationDiscoveryService/GetMetricsGatewayUrl',
+                request_serializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.SerializeToString,
+                response_deserializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.FromString,
+                )
+        self.GetFeaturesOperatorUrl = channel.unary_unary(
+                '/qwak.service_discovery.location.v0.LocationDiscoveryService/GetFeaturesOperatorUrl',
+                request_serializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.SerializeToString,
+                response_deserializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.FromString,
+                )
+        self.GetHostingGatewayUrl = channel.unary_unary(
+                '/qwak.service_discovery.location.v0.LocationDiscoveryService/GetHostingGatewayUrl',
+                request_serializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.SerializeToString,
+                response_deserializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.FromString,
+                )
+
+
+class LocationDiscoveryServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetOfflineServingUrl(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetDistributionManagerUrl(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetAnalyticsEngineUrl(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetMetricsGatewayUrl(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetFeaturesOperatorUrl(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetHostingGatewayUrl(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_LocationDiscoveryServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'GetOfflineServingUrl': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetOfflineServingUrl,
+                    request_deserializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.FromString,
+                    response_serializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.SerializeToString,
+            ),
+            'GetDistributionManagerUrl': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetDistributionManagerUrl,
+                    request_deserializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.FromString,
+                    response_serializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.SerializeToString,
+            ),
+            'GetAnalyticsEngineUrl': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetAnalyticsEngineUrl,
+                    request_deserializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.FromString,
+                    response_serializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.SerializeToString,
+            ),
+            'GetMetricsGatewayUrl': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetMetricsGatewayUrl,
+                    request_deserializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.FromString,
+                    response_serializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.SerializeToString,
+            ),
+            'GetFeaturesOperatorUrl': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetFeaturesOperatorUrl,
+                    request_deserializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.FromString,
+                    response_serializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.SerializeToString,
+            ),
+            'GetHostingGatewayUrl': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetHostingGatewayUrl,
+                    request_deserializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.FromString,
+                    response_serializer=qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.service_discovery.location.v0.LocationDiscoveryService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class LocationDiscoveryService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetOfflineServingUrl(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.service_discovery.location.v0.LocationDiscoveryService/GetOfflineServingUrl',
+            qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.SerializeToString,
+            qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetDistributionManagerUrl(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.service_discovery.location.v0.LocationDiscoveryService/GetDistributionManagerUrl',
+            qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.SerializeToString,
+            qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetAnalyticsEngineUrl(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.service_discovery.location.v0.LocationDiscoveryService/GetAnalyticsEngineUrl',
+            qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.SerializeToString,
+            qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetMetricsGatewayUrl(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.service_discovery.location.v0.LocationDiscoveryService/GetMetricsGatewayUrl',
+            qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.SerializeToString,
+            qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetFeaturesOperatorUrl(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.service_discovery.location.v0.LocationDiscoveryService/GetFeaturesOperatorUrl',
+            qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.SerializeToString,
+            qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetHostingGatewayUrl(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.service_discovery.location.v0.LocationDiscoveryService/GetHostingGatewayUrl',
+            qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequest.SerializeToString,
+            qwak_dot_service__discovery_dot_service__discovery__location__service__pb2.GetServingUrlRequestResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

qwak/__init__.py

@@ -1,7 +1,7 @@
 """Top-level package for qwak-core."""
 
 __author__ = "Qwak.ai"
-__version__ = "0.4.246"
+__version__ = "0.4.248"
 
 from qwak.inner.di_configuration import wire_dependencies
 from qwak.model.experiment_tracking import log_metric, log_param

qwak/clients/feature_store/offline_serving_client.py

@@ -1,6 +1,7 @@
 import logging
 from typing import List, Optional
 
+import grpc
 from _qwak_proto.qwak.offline.serving.v1.feature_values_pb2 import (
     FeaturesetFeatures as ProtoFeaturesetFeatures,
 )
@@ -27,9 +28,9 @@ from _qwak_proto.qwak.offline.serving.v1.population_pb2 import (
     PopulationFileUploadUrlType as ProtoPopulationFileUploadUrlType,
     TimedPopulation as ProtoTimedPopulation,
 )
-from dependency_injector.wiring import Provide
 from google.protobuf.timestamp_pb2 import Timestamp as ProtoTimestamp
-from qwak.inner.di_configuration import QwakContainer
+from qwak.clients.location_discovery import LocationDiscoveryClient
+from qwak.inner.tool.grpc.grpc_tools import create_grpc_channel
 from qwak.inner.tool.retry_utils import retry
 
 
@@ -46,8 +47,32 @@ class FsOfflineServingClient:
     Querying offline features store
     """
 
-    def __init__(self, grpc_channel=Provide[QwakContainer.core_grpc_channel]):
-        self._client = FeatureStoreOfflineServingAsyncServiceStub(grpc_channel)
+    def __init__(self, grpc_channel: Optional[grpc.Channel] = None):
+        self._client = self._create_client(grpc_channel)
+
+    @staticmethod
+    def _get_offline_serving_url() -> str:
+        """
+        Fetches the offline serving service URL from the LocationDiscoveryService.
+        """
+        return LocationDiscoveryClient().get_offline_serving().service_url
+
+    def _create_client(
+        self, grpc_channel: Optional[grpc.Channel] = None
+    ) -> FeatureStoreOfflineServingAsyncServiceStub:
+        if grpc_channel:
+            return FeatureStoreOfflineServingAsyncServiceStub(grpc_channel)
+
+        channel = create_grpc_channel(
+            url=self._get_offline_serving_url(),
+            enable_ssl=True,
+            status_for_retry=(
+                grpc.StatusCode.UNAVAILABLE,
+                grpc.StatusCode.CANCELLED,
+                grpc.StatusCode.DEADLINE_EXCEEDED,
+            ),
+        )
+        return FeatureStoreOfflineServingAsyncServiceStub(channel)
 
     def get_population_file_upload_url(
         self,

qwak/clients/location_discovery/__init__.py

@@ -0,0 +1 @@
+from .client import LocationDiscoveryClient

qwak/clients/location_discovery/client.py

@@ -0,0 +1,73 @@
+from typing import Callable
+
+import grpc
+from _qwak_proto.qwak.service_discovery.service_discovery_location_pb2 import (
+    ServiceLocationDescriptor,
+)
+from _qwak_proto.qwak.service_discovery.service_discovery_location_service_pb2 import (
+    GetServingUrlRequest as ProtoGetServingUrlRequest,
+    GetServingUrlRequestResponse as ProtoGetServingUrlRequestResponse,
+)
+from _qwak_proto.qwak.service_discovery.service_discovery_location_service_pb2_grpc import (
+    LocationDiscoveryServiceStub,
+)
+from dependency_injector.wiring import Provide
+from qwak.exceptions import QwakException
+from qwak.inner.di_configuration import QwakContainer
+
+
+class LocationDiscoveryClient:
+    """
+    Client for querying service locations from the LocationDiscoveryService.
+    """
+
+    def __init__(self, grpc_channel=Provide[QwakContainer.core_grpc_channel]):
+        self._service = LocationDiscoveryServiceStub(grpc_channel)
+
+    @staticmethod
+    def _get_location(
+        method: Callable[
+            [ProtoGetServingUrlRequest], ProtoGetServingUrlRequestResponse
+        ],
+    ) -> ServiceLocationDescriptor:
+        """
+        Calls a specific RPC and extracts the service location descriptor.
+
+        Args:
+            method: The gRPC method to call.
+
+        Returns:
+            ServiceLocationDescriptor: Contains service_url and location metadata.
+        """
+        try:
+            request = ProtoGetServingUrlRequest()
+            response = method(request)
+            return response.location
+        except grpc.RpcError as e:
+            raise QwakException(
+                f"Failed to retrieve service location. Error is: {e.details()}"
+            ) from e
+
+    def get_offline_serving(self) -> ServiceLocationDescriptor:
+        """Fetches the offline serving service location."""
+        return self._get_location(self._service.GetOfflineServingUrl)
+
+    def get_distribution_manager(self) -> ServiceLocationDescriptor:
+        """Fetches the distribution manager service location."""
+        return self._get_location(self._service.GetDistributionManagerUrl)
+
+    def get_analytics_engine(self) -> ServiceLocationDescriptor:
+        """Fetches the analytics engine service location."""
+        return self._get_location(self._service.GetAnalyticsEngineUrl)
+
+    def get_metrics_gateway(self) -> ServiceLocationDescriptor:
+        """Fetches the metrics gateway service location."""
+        return self._get_location(self._service.GetMetricsGatewayUrl)
+
+    def get_features_operator(self) -> ServiceLocationDescriptor:
+        """Fetches the features operator service location."""
+        return self._get_location(self._service.GetFeaturesOperatorUrl)
+
+    def get_hosting_gateway(self) -> ServiceLocationDescriptor:
+        """Fetches the hosting gateway service location."""
+        return self._get_location(self._service.GetHostingGatewayUrl)

qwak/feature_store/_common/functions.py

@@ -16,22 +16,3 @@ def _generate_new_col(col_name: str, duplicate_cols_dict: dict):
         return col_name
     elif len(feature_full_name) == 2 and duplicate_cols_dict[feature_full_name[1]] == 1:
         return feature_full_name[1]
-
-
-def normalize_cols(cols) -> List[str]:
-    """
-    Normalize cols - try to remove leading feature set name from features
-    Args:
-        cols: list of column
-    Return normalized columns
-    """
-    duplicate_columns_dict = {}
-    for col in cols:
-        feature_full_name: List[str] = col.split(".")
-        if len(feature_full_name) != 2:
-            duplicate_columns_dict[col] = 1
-        else:
-            fs_name = feature_full_name[1]
-            duplicate_columns_dict[fs_name] = duplicate_columns_dict.get(fs_name, 0) + 1
-    cols = [_generate_new_col(col_name, duplicate_columns_dict) for col_name in cols]
-    return cols

qwak/feature_store/offline/__init__.py

@@ -1,4 +1,3 @@
-from qwak.feature_store.offline.client import OfflineClient
 from qwak.feature_store.offline.client_v2 import OfflineClientV2
 
-__all__ = ["OfflineClient", "OfflineClientV2"]
+__all__ = ["OfflineClientV2"]