fix(indexer): fix bug in indexer service

gnes-ai · Jul 26, 2019 · 5743e25 · 5743e25
1 parent 11dde2b
commit 5743e25
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/gnes/service/indexer.py b/gnes/service/indexer.py
@@ -39,11 +39,11 @@ def _handler_index(self, msg: 'gnes_pb2.Message'):
             if d.chunks:
                 all_vecs.append(blob2array(d.chunk_embeddings))
                 doc_ids += [d.doc_id] * len(d.chunks)
-                if msg.request.index.docs.doc_type == 'TEXT':
+                if d.doc_type == 'TEXT':
                     offsets += [c.offset_1d for c in d.chunks]
-                elif msg.request.index.docs.doc_type == 'IMAGE':
+                elif d.doc_type == 'IMAGE':
                     offsets += [c.offset_nd for c in d.chunks]
-                elif msg.request.index.docs.doc_type == 'VIDEO':
+                elif d.doc_type == 'VIDEO':
                     offsets += [c.offset_1d for c in d.chunks]
                 weights += [c.weight for c in d.chunks]