jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat:support baidu vector db (#9185)

Browse source
This commit is contained in:
Shili Cao 2024-10-12 23:24:17 +08:00 • committed by GitHub
commit 2ec6ffe478
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
15 changed files with 582 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/controllers/console/datasets/datasets.py
View file

@ -617,6 +617,7 @@ class DatasetRetrievalSettingApi(Resource):
| VectorType.CHROMA
| VectorType.TENCENT
| VectorType.PGVECTO_RS
| VectorType.BAIDU
):
return {"retrieval_method": [RetrievalMethod.SEMANTIC_SEARCH.value]}
case (
@ -653,6 +654,7 @@ class DatasetRetrievalSettingMockApi(Resource):
| VectorType.CHROMA
| VectorType.TENCENT
| VectorType.PGVECTO_RS
| VectorType.BAIDU
):
return {"retrieval_method": [RetrievalMethod.SEMANTIC_SEARCH.value]}
case (