🚧 fix(wip): pass embeddings

Prismadic · Mar 28, 2024 · 3aa6cea · 3aa6cea
1 parent 3ddc0be
commit 3aa6cea
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/magnet/ize/memory.py b/magnet/ize/memory.py
@@ -103,14 +103,15 @@ def search(self, payload, limit: int = 100, cb: Optional[callable] = None, instr
             anns_field="embedding",
             param=self.config.index.options,
             limit=limit,
-            output_fields=['text', 'document']
+            output_fields=['text', 'document', 'embedding']
         )
         results = []
         for hits_i, hits in enumerate(_results):
             for hit in hits:
                 results.append({
                     'text': hit.entity.get('text'),
                     'document': hit.entity.get('document'),
+                    'embedding': hit.entity.get('embedding'),
                     'distance': hit.distance
                 })
         if cb: