Feat/delete single dataset retrival (#6570)

2024-07-24 12:50:11 +08:00
parent 0fb741f269
commit e4bb943fe5
22 changed files with 651 additions and 115 deletions
--- a/api/core/app/app_config/entities.py
+++ b/api/core/app/app_config/entities.py
@@ -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):