From a37b346bebfd01326c4ed39eeb4567402b5167d6 Mon Sep 17 00:00:00 2001 From: Gary Date: Sat, 14 Dec 2024 21:11:00 +0000 Subject: [PATCH] fix SendNewToken --- exo/networking/grpc/grpc_server.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/exo/networking/grpc/grpc_server.py b/exo/networking/grpc/grpc_server.py index c6a56685b..a22f3e68b 100644 --- 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):