github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-21 17:54:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

os environment variables should take precedences over the defaults, f… (#2100)

Browse Source
This commit is contained in:
Alexis Yushin 2024-10-28 00:24:28 -07:00 committed by GitHub
parent 6c682befa8
commit 21c282e659
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
dbgpt/storage/vector_store/elastic_store.py
View File

@ -130,18 +130,18 @@ class ElasticStore(VectorStoreBase):
connect_kwargs = {}
elasticsearch_vector_config = vector_store_config.dict()
self.uri = elasticsearch_vector_config.get("uri") or os.getenv(
self.uri = os.getenv(
"ELASTICSEARCH_URL", "localhost"
)
self.port = elasticsearch_vector_config.get("post") or os.getenv(
) or elasticsearch_vector_config.get("uri")
self.port = os.getenv(
"ELASTICSEARCH_PORT", "9200"
)
self.username = elasticsearch_vector_config.get("username") or os.getenv(
) or elasticsearch_vector_config.get("post")
self.username = os.getenv(
"ELASTICSEARCH_USERNAME"
)
self.password = elasticsearch_vector_config.get("password") or os.getenv(
) or elasticsearch_vector_config.get("username")
self.password = os.getenv(
"ELASTICSEARCH_PASSWORD"
)
) or elasticsearch_vector_config.get("password")
self.collection_name = (
elasticsearch_vector_config.get("name") or vector_store_config.name