[Improvement] return all the metadata when citations flag is True (#1059)

Co-authored-by: Deven Patel <deven298@yahoo.com>
2023-12-29 14:48:41 +05:30
parent d9d529987e
commit 19d80914df
15 changed files with 47 additions and 56 deletions
--- a/embedchain/vectordb/chroma.py
+++ b/embedchain/vectordb/chroma.py
@@ -250,9 +250,7 @@ class ChromaDB(BaseVectorDB):
            context = result[0].page_content
            if citations:
                metadata = result[0].metadata
-                source = metadata["url"]
-                doc_id = metadata["doc_id"]
-                contexts.append((context, source, doc_id))
+                contexts.append((context, metadata))
            else:
                contexts.append(context)
        return contexts