Fixed up test_vectorstore. (#16113)

run-llama · Sep 19, 2024 · a18b946 · a18b946
1 parent 52fb788
commit a18b946
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-mongodb/pyproject.toml b/llama-index-integrations/vector_stores/llama-index-vector-stores-mongodb/pyproject.toml
@@ -29,7 +29,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-vector-stores-mongodb"
 readme = "README.md"
-version = "0.3.0"
+version = "0.3.1"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"

diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-mongodb/tests/conftest.py b/llama-index-integrations/vector_stores/llama-index-vector-stores-mongodb/tests/conftest.py
@@ -23,7 +23,7 @@ def documents() -> List[Document]:
     text = Document.example().text
     metadata = Document.example().metadata
     texts = text.split("\n")
-    return [Document(text=text, metadata=metadata) for text in texts]
+    return [Document(text=text, metadata={"text": text}) for text in texts]
 
 
 @pytest.fixture(scope="session")

diff --git a/...ex-integrations/vector_stores/llama-index-vector-stores-mongodb/tests/test_vectorstore.py b/...ex-integrations/vector_stores/llama-index-vector-stores-mongodb/tests/test_vectorstore.py
@@ -107,7 +107,7 @@ def test_vectorstore(
             "How do we best augment LLMs with our own private data?",
             "easily used with LLMs.",
         ]
-        filters_compoound = MetadataFilters(
+        filters_compound = MetadataFilters(
             condition=FilterCondition.AND,
             filters=[
                 MetadataFilter(
@@ -122,7 +122,7 @@ def test_vectorstore(
             query_str=query_str,
             query_embedding=query_embedding,
             similarity_top_k=n_similar,
-            filters=filters_compoound,
+            filters=filters_compound,
         )
         responses_with_filter_compound = vector_store.query(query=query)
         assert len(responses_with_filter_compound.ids) == n_similar
@@ -158,7 +158,7 @@ def test_vectorstore(
         query = VectorStoreQuery(
             query_str="llamaindex",
             query_embedding=query_embedding,  # "What are LLMs useful for?"
-            hybrid_top_k=n_similar,
+            similarity_top_k=n_similar,
             mode=VectorStoreQueryMode.HYBRID,
             alpha=0.5,
         )