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

Init embedding_model_dims in all vectordbs (#2572)

Browse Source
This commit is contained in:
Dev Khant
2025-04-19 10:53:01 +05:30
committed by GitHub
parent 78912928bc
commit 3ee4768c14
7 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mem0/vector_stores/elasticsearch.py
View File

@@ -40,7 +40,7 @@ class ElasticsearchDB(VectorStoreBase):
)
self.collection_name = config.collection_name
self.vector_dim = config.embedding_model_dims
self.embedding_model_dims = config.embedding_model_dims
# Create index only if auto_create_index is True
if config.auto_create_index:
@@ -58,7 +58,7 @@ class ElasticsearchDB(VectorStoreBase):
"mappings": {
"properties": {
"text": {"type": "text"},
"vector": {"type": "dense_vector", "dims": self.vector_dim, "index": True, "similarity": "cosine"},
"vector": {"type": "dense_vector", "dims": self.embedding_model_dims, "index": True, "similarity": "cosine"},
"metadata": {"type": "object", "properties": {"user_id": {"type": "keyword"}}},
}
},