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

Feat/delete single dataset retrival (#6570)

Browse source
This commit is contained in:
Jyong 2024-07-24 12:50:11 +08:00 • committed by GitHub
commit e4bb943fe5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
22 changed files with 651 additions and 115 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/core/app/app_config/entities.py
View file

@ -159,7 +159,11 @@ class DatasetRetrieveConfigEntity(BaseModel):
retrieve_strategy: RetrieveStrategy
top_k: Optional[int] = None
score_threshold: Optional[float] = None
rerank_mode: Optional[str] = 'reranking_model'
reranking_model: Optional[dict] = None
weights: Optional[dict] = None
class DatasetEntity(BaseModel):