diff --git a/gnes/service/indexer.py b/gnes/service/indexer.py index 7322ae44..928cf960 100644 --- a/gnes/service/indexer.py +++ b/gnes/service/indexer.py @@ -61,7 +61,7 @@ def _handler_chunk_search(self, msg: 'gnes_pb2.Message'): results = self._model.query(vecs, top_k=msg.request.search.top_k) q_weights = [qc.weight for qc in msg.request.search.query.chunks] for all_topks, qc_weight in zip(results, q_weights): - for _doc_id, _offset, _relevance, _weight in all_topks: + for _doc_id, _offset, _weight, _relevance in all_topks: r = msg.response.search.topk_results.add() r.chunk.doc_id = _doc_id r.chunk.offset_1d = _offset