diff --git a/mem0/configs/vector_stores/elasticsearch.py b/mem0/configs/vector_stores/elasticsearch.py index 0e9faac3f..58d0129d5 100644 --- a/mem0/configs/vector_stores/elasticsearch.py +++ b/mem0/configs/vector_stores/elasticsearch.py @@ -14,6 +14,7 @@ class ElasticsearchConfig(BaseModel): api_key: Optional[str] = Field(None, description="API key for authentication") embedding_model_dims: int = Field(1536, description="Dimension of the embedding vector") verify_certs: bool = Field(True, description="Verify SSL certificates") + ca_certs: Optional[str] = Field(None, description="Path to CA bundle for SSL certificate verification") use_ssl: bool = Field(True, description="Use SSL for connection") auto_create_index: bool = Field(True, description="Automatically create index during initialization") custom_search_query: Optional[Callable[[List[float], int, Optional[Dict]], Dict]] = Field( diff --git a/mem0/vector_stores/elasticsearch.py b/mem0/vector_stores/elasticsearch.py index d3abc2c26..687336ba2 100644 --- a/mem0/vector_stores/elasticsearch.py +++ b/mem0/vector_stores/elasticsearch.py @@ -31,6 +31,7 @@ class ElasticsearchDB(VectorStoreBase): cloud_id=config.cloud_id, api_key=config.api_key, verify_certs=config.verify_certs, + ca_certs=config.ca_certs, headers= config.headers or {}, ) else: @@ -38,6 +39,7 @@ class ElasticsearchDB(VectorStoreBase): hosts=[f"{config.host}" if config.port is None else f"{config.host}:{config.port}"], basic_auth=(config.user, config.password) if (config.user and config.password) else None, verify_certs=config.verify_certs, + ca_certs=config.ca_certs, headers= config.headers or {}, )