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

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
commit 80c94f02e9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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),