Change list[str] -> str for vectordbs (#1388)

This commit is contained in:
Anu
2024-06-07 10:15:40 -06:00
committed by GitHub
parent fd07513004
commit e0d0f6e94c
11 changed files with 27 additions and 27 deletions

View File

@@ -58,7 +58,7 @@ class TestEsDB(unittest.TestCase):
mock_client.return_value.search.return_value = search_response
# Query the database for the documents that are most similar to the query "This is a document".
query = ["This is a document"]
query = "This is a document"
results_without_citations = self.db.query(query, n_results=2, where={})
expected_results_without_citations = ["This is a document.", "This is another document."]
self.assertEqual(results_without_citations, expected_results_without_citations)