feat: support setting database used in Milvus (#3003)

This commit is contained in:
Leo Q
2024-04-09 15:39:36 +08:00
committed by GitHub
parent a2c068d949
commit 9c01bcb3e5
4 changed files with 35 additions and 6 deletions

View File

@@ -67,6 +67,7 @@ DEFAULTS = {
'CODE_EXECUTION_ENDPOINT': '',
'CODE_EXECUTION_API_KEY': '',
'TOOL_ICON_CACHE_MAX_AGE': 3600,
'MILVUS_DATABASE': 'default',
'KEYWORD_DATA_SOURCE_TYPE': 'database',
}
@@ -212,6 +213,7 @@ class Config:
self.MILVUS_USER = get_env('MILVUS_USER')
self.MILVUS_PASSWORD = get_env('MILVUS_PASSWORD')
self.MILVUS_SECURE = get_env('MILVUS_SECURE')
self.MILVUS_DATABASE = get_env('MILVUS_DATABASE')
# weaviate settings
self.WEAVIATE_ENDPOINT = get_env('WEAVIATE_ENDPOINT')