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

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/vectordb/test_zilliz_db.py
View File

@@ -139,7 +139,7 @@ class TestZillizDBCollection:
]
]
query_result = zilliz_db.query(input_query=["query_text"], n_results=1, where={})
query_result = zilliz_db.query(input_query="query_text", n_results=1, where={})
# Assert that MilvusClient.search was called with the correct parameters
mock_search.assert_called_with(
@@ -154,7 +154,7 @@ class TestZillizDBCollection:
assert query_result == ["result_doc"]
query_result_with_citations = zilliz_db.query(
input_query=["query_text"], n_results=1, where={}, citations=True
input_query="query_text", n_results=1, where={}, citations=True
)
mock_search.assert_called_with(