From 08a9a4e3863afde2dbcb74633c1648e92c47abb5 Mon Sep 17 00:00:00 2001 From: Larry Yan Date: Mon, 22 Jul 2019 20:45:23 +0800 Subject: [PATCH] fix(grpc): fix grpc service --- gnes/service/grpc.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/gnes/service/grpc.py b/gnes/service/grpc.py index 1c881367..1f167317 100644 --- a/gnes/service/grpc.py +++ b/gnes/service/grpc.py @@ -129,6 +129,16 @@ def Index(self, request, context): def Search(self, request, context): return self._Call(request, context) + def TrainStream(self, request_iterator, context): + for request in request_iterator: + ret = self._Call(request, context) + return ret + + def IndexStream(self, request_iterator, context): + for request in request_iterator: + ret = self._Call(request, context) + return ret + class GRPCFrontend: def __init__(self, args):