[Bugfix] IndexInfo API moved from Agent Client to Vald Client

Makefile

@@ -49,6 +49,7 @@ PROTOS = \
 	v1/vald/filter.proto \
 	v1/vald/insert.proto \
 	v1/vald/object.proto \
+	v1/vald/index.proto \
 	v1/vald/remove.proto \
 	v1/vald/search.proto \
 	v1/vald/update.proto \

VALD_SHA

@@ -1 +1 @@
-284fdb57b3afedecd330101968ba393381e58085
+702e85153e90cdfea37c857e6f3a5cfd56755afa

src/tests/test_e2e.py

@@ -10,6 +10,7 @@
 from vald.v1.vald import upsert_pb2_grpc
 from vald.v1.vald import remove_pb2_grpc
 from vald.v1.vald import object_pb2_grpc
+from vald.v1.vald import index_pb2_grpc
 from vald.v1.payload import payload_pb2
 
 
@@ -79,7 +80,7 @@ def test_save_index(self):
         self.assertIsInstance(result, payload_pb2.Empty)
 
     def test_index_info(self):
-        stub = agent_pb2_grpc.AgentStub(self.channel)
+        stub = index_pb2_grpc.IndexStub(self.channel)
         result = stub.IndexInfo(payload_pb2.Empty())
         self.assertIsInstance(result, payload_pb2.Info.Index.Count)
         self.assertEqual(result.stored, 99)

src/vald/v1/agent/core/agent_pb2_grpc.py

@@ -30,16 +30,6 @@ def __init__(self, channel):
                 request_serializer=vald_dot_v1_dot_payload_dot_payload__pb2.Control.CreateIndexRequest.SerializeToString,
                 response_deserializer=vald_dot_v1_dot_payload_dot_payload__pb2.Empty.FromString,
                 _registered_method=True)
-        self.IndexInfo = channel.unary_unary(
-                '/core.v1.Agent/IndexInfo',
-                request_serializer=vald_dot_v1_dot_payload_dot_payload__pb2.Empty.SerializeToString,
-                response_deserializer=vald_dot_v1_dot_payload_dot_payload__pb2.Info.Index.Count.FromString,
-                _registered_method=True)
-        self.GetTimestamp = channel.unary_unary(
-                '/core.v1.Agent/GetTimestamp',
-                request_serializer=vald_dot_v1_dot_payload_dot_payload__pb2.Object.GetTimestampRequest.SerializeToString,
-                response_deserializer=vald_dot_v1_dot_payload_dot_payload__pb2.Object.Timestamp.FromString,
-                _registered_method=True)
 
 
 class AgentServicer(object):
@@ -67,20 +57,6 @@ def CreateAndSaveIndex(self, request, context):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
-    def IndexInfo(self, request, context):
-        """Represent the RPC to get the agent index information.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def GetTimestamp(self, request, context):
-        """Represent the RPC to get the vector metadata. This RPC is mainly used for index correction process
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
 
 def add_AgentServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -99,16 +75,6 @@ def add_AgentServicer_to_server(servicer, server):
                     request_deserializer=vald_dot_v1_dot_payload_dot_payload__pb2.Control.CreateIndexRequest.FromString,
                     response_serializer=vald_dot_v1_dot_payload_dot_payload__pb2.Empty.SerializeToString,
             ),
-            'IndexInfo': grpc.unary_unary_rpc_method_handler(
-                    servicer.IndexInfo,
-                    request_deserializer=vald_dot_v1_dot_payload_dot_payload__pb2.Empty.FromString,
-                    response_serializer=vald_dot_v1_dot_payload_dot_payload__pb2.Info.Index.Count.SerializeToString,
-            ),
-            'GetTimestamp': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetTimestamp,
-                    request_deserializer=vald_dot_v1_dot_payload_dot_payload__pb2.Object.GetTimestampRequest.FromString,
-                    response_serializer=vald_dot_v1_dot_payload_dot_payload__pb2.Object.Timestamp.SerializeToString,
-            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'core.v1.Agent', rpc_method_handlers)
@@ -201,57 +167,3 @@ def CreateAndSaveIndex(request,
             timeout,
             metadata,
             _registered_method=True)
-
-    @staticmethod
-    def IndexInfo(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,
-            '/core.v1.Agent/IndexInfo',
-            vald_dot_v1_dot_payload_dot_payload__pb2.Empty.SerializeToString,
-            vald_dot_v1_dot_payload_dot_payload__pb2.Info.Index.Count.FromString,
-            options,
-            channel_credentials,
-            insecure,
-            call_credentials,
-            compression,
-            wait_for_ready,
-            timeout,
-            metadata,
-            _registered_method=True)
-
-    @staticmethod
-    def GetTimestamp(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,
-            '/core.v1.Agent/GetTimestamp',
-            vald_dot_v1_dot_payload_dot_payload__pb2.Object.GetTimestampRequest.SerializeToString,
-            vald_dot_v1_dot_payload_dot_payload__pb2.Object.Timestamp.FromString,
-            options,
-            channel_credentials,
-            insecure,
-            call_credentials,
-            compression,
-            wait_for_ready,
-            timeout,
-            metadata,
-            _registered_method=True)