hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-07 18:06:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

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
parent 793205afc5
commit 2ec6ffe478
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 (