JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(vector-stores): add agent_id and run_id to Elasticsearch/OpenSearch default mappings (#4906)

Browse Source 
Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
jessai2099
2026-04-21 01:40:59 +08:00
committed by GitHub
parent 8ba225cec8
commit 573e5212a4
4 changed files with 24 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mem0/vector_stores/elasticsearch.py
+8 -1
View File
@@ -66,7 +66,14 @@ class ElasticsearchDB(VectorStoreBase):
"index": True,
"similarity": "cosine",
},
"metadata": {"type": "object", "properties": {"user_id": {"type": "keyword"}}},
"metadata": {
"type": "object",
"properties": {
"user_id": {"type": "keyword"},
"agent_id": {"type": "keyword"},
"run_id": {"type": "keyword"},
},
},
}
},
}
mem0/vector_stores/opensearch.py
+8 -1
View File
@@ -55,7 +55,14 @@ class OpenSearchDB(VectorStoreBase):
"dimension": self.embedding_model_dims,
"method": {"engine": "nmslib", "name": "hnsw", "space_type": "cosinesimil"},
},
"metadata": {"type": "object", "properties": {"user_id": {"type": "keyword"}}},
"metadata": {
"type": "object",
"properties": {
"user_id": {"type": "keyword"},
"agent_id": {"type": "keyword"},
"run_id": {"type": "keyword"},
},
},
}
},
}
tests/vector_stores/test_elasticsearch.py
+4
View File
@@ -96,6 +96,10 @@ class TestElasticsearchDB(unittest.TestCase):
self.assertEqual(mappings["vector"]["index"], True)
self.assertEqual(mappings["vector"]["similarity"], "cosine")
self.assertEqual(mappings["metadata"]["type"], "object")
metadata_props = mappings["metadata"]["properties"]
self.assertEqual(metadata_props["user_id"]["type"], "keyword")
self.assertEqual(metadata_props["agent_id"]["type"], "keyword")
self.assertEqual(metadata_props["run_id"]["type"], "keyword")
# Reset mocks for next test
self.client_mock.reset_mock()
tests/vector_stores/test_opensearch.py
+4
View File
@@ -142,6 +142,10 @@ class TestOpenSearchDB(unittest.TestCase):
mappings = create_args["body"]["mappings"]["properties"]
self.assertEqual(mappings["vector_field"]["type"], "knn_vector")
self.assertEqual(mappings["vector_field"]["dimension"], 1536)
metadata_props = mappings["metadata"]["properties"]
self.assertEqual(metadata_props["user_id"]["type"], "keyword")
self.assertEqual(metadata_props["agent_id"]["type"], "keyword")
self.assertEqual(metadata_props["run_id"]["type"], "keyword")
self.client_mock.reset_mock()
self.client_mock.indices.exists.return_value = True
self.os_db.create_index()