fix: using api can not execute relyt vector database (#3766)

Co-authored-by: jingsi <jingsi@leadincloud.com>
This commit is contained in:
Jingpan Xiong
2024-04-25 19:46:20 +08:00
committed by GitHub
parent bf9fc8fef4
commit 1be222af2e
4 changed files with 186 additions and 50 deletions

View File

@@ -476,7 +476,7 @@ class DatasetRetrievalSettingApi(Resource):
@account_initialization_required
def get(self):
vector_type = current_app.config['VECTOR_STORE']
if vector_type == 'milvus':
if vector_type == 'milvus' or vector_type == 'relyt':
return {
'retrieval_method': [
'semantic_search'
@@ -498,7 +498,7 @@ class DatasetRetrievalSettingMockApi(Resource):
@account_initialization_required
def get(self, vector_type):
if vector_type == 'milvus':
if vector_type == 'milvus' or vector_type == 'relyt':
return {
'retrieval_method': [
'semantic_search'