Fix/vdb index issue (#1776)

Co-authored-by: jyong <jyong@dify.ai>
2023-12-18 21:33:54 +08:00 · 2023-12-18 21:33:54 +08:00 · 2fd56cb01c
commit 2fd56cb01c
parent 4f0e272549
7 changed files with 18 additions and 10 deletions
--- a/api/core/completion.py
+++ b/api/core/completion.py
@ -371,7 +371,8 @@ class Completion:
                vector_index = VectorIndex(
                    dataset=dataset,
                    config=current_app.config,
-                    embeddings=embeddings
+                    embeddings=embeddings,
+                    attributes=['doc_id', 'annotation_id', 'app_id']
                )

                documents = vector_index.search(