refactor(proto): assign doc id in request generator

gnes-ai · Aug 15, 2019 · 3a2b85b · 3a2b85b
1 parent da3d3a8
commit 3a2b85b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/gnes/router/reduce/doc_sum.py b/gnes/router/reduce/doc_sum.py
@@ -33,9 +33,9 @@ def apply(self, msg: 'gnes_pb2.Message', accum_msgs: List['gnes_pb2.Message'], *
         doc_score_explained = defaultdict(str)
 
         for d, s, ex in zip(all_docs, all_docs_socres, all_score_explained):
-            doc_id[d.meta_info] = d
-            doc_score[d.meta_info] += s
-            doc_score_explained[d.meta_info] += '%s\n' % ex
+            doc_id[d.doc_id] = d
+            doc_score[d.doc_id] += s
+            doc_score_explained[d.doc_id] += '%s\n' % ex
 
         msg.response.search.ClearField('topk_results')