fix SendNewToken

exo-explore · Dec 14, 2024 · a37b346 · a37b346
1 parent 45aeee9
commit a37b346
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/exo/networking/grpc/grpc_server.py b/exo/networking/grpc/grpc_server.py
@@ -118,12 +118,12 @@ async def CollectTopology(self, request, context):
     if DEBUG >= 5: print(f"CollectTopology {max_depth=} {visited=} {nodes=} {peer_graph=}")
     return node_service_pb2.Topology(nodes=nodes, peer_graph=peer_graph)
 
-  async def SendResult(self, request, context):
+  async def SendNewToken(self, request, context):
     request_id = request.request_id
-    result = request.result
+    token = request.token
     is_finished = request.is_finished
-    if DEBUG >= 5: print(f"Received SendResult request: {request_id=} {result=} {is_finished=}")
-    self.node.on_token.trigger_all(request_id, result, is_finished)
+    if DEBUG >= 5: print(f"Received SendNewToken request: {request_id=} {token=} {is_finished=}")
+    self.node.on_token.trigger_all(request_id, token, is_finished)
     return node_service_pb2.Empty()
 
   async def SendOpaqueStatus(self, request, context):