klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: Deven Patel <deven298@yahoo.com>
This commit is contained in:
Deven Patel
2023-12-29 14:48:41 +05:30
committed by GitHub
parent d9d529987e
commit 19d80914df
15 changed files with 47 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/vectordb/test_elasticsearch_db.py
View File

@@ -66,8 +66,8 @@ class TestEsDB(unittest.TestCase):
results_with_citations = self.db.query(query, n_results=2, where={}, skip_embedding=False, citations=True)
expected_results_with_citations = [
("This is a document.", "url_1", "doc_id_1"),
("This is another document.", "url_2", "doc_id_2"),
("This is a document.", {"url": "url_1", "doc_id": "doc_id_1"}),
("This is another document.", {"url": "url_2", "doc_id": "doc_id_2"}),
]
self.assertEqual(results_with_citations, expected_results_with_citations)