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

add vector field for other vectordb (#7051)

Browse Source
This commit is contained in:
Jyong
2024-08-07 17:14:03 +08:00
committed by GitHub
parent aad02113c6
commit 80c94f02e9
10 changed files with 23 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/services/hit_testing_service.py
View File

@@ -44,7 +44,8 @@ class HitTestingService:
top_k=retrieval_model.get('top_k', 2),
score_threshold=retrieval_model.get('score_threshold', .0)
if retrieval_model['score_threshold_enabled'] else None,
reranking_model=retrieval_model.get('reranking_model', None),
reranking_model=retrieval_model.get('reranking_model', None)
if retrieval_model['reranking_enable'] else None,
reranking_mode=retrieval_model.get('reranking_mode')
if retrieval_model.get('reranking_mode') else 'reranking_model',
weights=retrieval_model.get('weights', None),