fix: upgrade MongoDB vector store from deprecated knnVector to GA vectorSearch (#3995)

Co-authored-by: kartik-mem0 <kartik.labhshetwar@mem0.ai>
Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Varun Chawla
2026-03-22 23:50:17 -07:00
committed by GitHub
parent 5dabf24809
commit 7cebaba0a2
2 changed files with 21 additions and 23 deletions
+10 -11
View File
@@ -48,17 +48,16 @@ def test_initalize_create_col(mongo_vector_fixture):
search_index_model = args[0].document
assert search_index_model == {
"name": "test_collection_vector_index",
"type": "vectorSearch",
"definition": {
"mappings": {
"dynamic": False,
"fields": {
"embedding": {
"type": "knnVector",
"dimensions": 1536,
"similarity": "cosine",
}
},
}
"fields": [
{
"type": "vector",
"path": "embedding",
"numDimensions": 1536,
"similarity": "cosine",
}
]
},
}
assert mongo_vector.collection == mock_collection
@@ -95,7 +94,7 @@ def test_search(mongo_vector_fixture):
"$vectorSearch": {
"index": "test_collection_vector_index",
"limit": 2,
"numCandidates": 2,
"numCandidates": 40,
"queryVector": query_vector,
"path": "embedding",
},