fix(grpc): fix grpc service

gnes-ai · Jul 22, 2019 · 08a9a4e · 08a9a4e
1 parent 6e6bbf8
commit 08a9a4e
Showing 1 changed file with 10 additions and 0 deletions.
diff --git 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):