Update base.py

run-llama · Jun 27, 2024 · bc122b2 · bc122b2
1 parent d51d323
commit bc122b2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/...s/llama-index-vector-stores-elasticsearch/llama_index/vector_stores/elasticsearch/base.py b/...s/llama-index-vector-stores-elasticsearch/llama_index/vector_stores/elasticsearch/base.py
@@ -215,6 +215,7 @@ def __init__(
         batch_size: int = 200,
         distance_strategy: Optional[DISTANCE_STRATEGIES] = "COSINE",
         retrieval_strategy: Optional[AsyncRetrievalStrategy] = None,
+        **kwargs
     ) -> None:
         nest_asyncio.apply()
 
@@ -232,12 +233,16 @@ def __init__(
                 distance=DistanceMetric[distance_strategy]
             )
 
-        metadata_mappings = {
+        base_metadata_mappings = {
             "document_id": {"type": "keyword"},
             "doc_id": {"type": "keyword"},
             "ref_doc_id": {"type": "keyword"},
         }
 
+        metadata_mappings = kwargs.pop('metadata_mappings', None)
+            if metadata_mappings:
+                base_metadata_mappings.update(metadata_mappings)
+
         self._store = AsyncVectorStore(
             user_agent=get_user_agent(),
             client=es_client,