estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat: support tidb vector (#4588)

Browse Source
This commit is contained in:
Weaxs
2024-06-05 18:19:53 +08:00
committed by GitHub
parent 602c4e51ec
commit 0797f9bc05
10 changed files with 332 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/controllers/console/datasets/datasets.py
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 in {"milvus", "relyt", "pgvector", "pgvecto_rs"}:
if vector_type in {"milvus", "relyt", "pgvector", "pgvecto_rs", 'tidb_vector'}:
return {
'retrieval_method': [
'semantic_search'
@@ -497,7 +497,7 @@ class DatasetRetrievalSettingMockApi(Resource):
@login_required
@account_initialization_required
def get(self, vector_type):
if vector_type in {'milvus', 'relyt', 'pgvector'}:
if vector_type in {'milvus', 'relyt', 'pgvector', 'tidb_vector'}:
return {
'retrieval_method': [
'semantic_search'