Просмотр исходного кода

fixed score threshold is none (#8342)

tags/0.8.2
Jyong 1 год назад
Родитель
Сommit
49cee773c5
Аккаунт пользователя с таким Email не найден

+ 1
- 1
api/core/rag/retrieval/dataset_retrieval.py Просмотреть файл

top_k=top_k, top_k=top_k,
score_threshold=retrieval_model.get("score_threshold", 0.0) score_threshold=retrieval_model.get("score_threshold", 0.0)
if retrieval_model["score_threshold_enabled"] if retrieval_model["score_threshold_enabled"]
else None,
else 0.0,
reranking_model=retrieval_model.get("reranking_model", None) reranking_model=retrieval_model.get("reranking_model", None)
if retrieval_model["reranking_enable"] if retrieval_model["reranking_enable"]
else None, else None,

+ 1
- 1
api/core/tools/tool/dataset_retriever/dataset_multi_retriever_tool.py Просмотреть файл

top_k=self.top_k, top_k=self.top_k,
score_threshold=retrieval_model.get("score_threshold", 0.0) score_threshold=retrieval_model.get("score_threshold", 0.0)
if retrieval_model["score_threshold_enabled"] if retrieval_model["score_threshold_enabled"]
else None,
else 0.0,
reranking_model=retrieval_model.get("reranking_model", None) reranking_model=retrieval_model.get("reranking_model", None)
if retrieval_model["reranking_enable"] if retrieval_model["reranking_enable"]
else None, else None,

+ 1
- 1
api/core/tools/tool/dataset_retriever/dataset_retriever_tool.py Просмотреть файл

top_k=self.top_k, top_k=self.top_k,
score_threshold=retrieval_model.get("score_threshold", 0.0) score_threshold=retrieval_model.get("score_threshold", 0.0)
if retrieval_model["score_threshold_enabled"] if retrieval_model["score_threshold_enabled"]
else None,
else 0.0,
reranking_model=retrieval_model.get("reranking_model", None) reranking_model=retrieval_model.get("reranking_model", None)
if retrieval_model["reranking_enable"] if retrieval_model["reranking_enable"]
else None, else None,

+ 1
- 1
api/services/hit_testing_service.py Просмотреть файл

top_k=retrieval_model.get("top_k", 2), top_k=retrieval_model.get("top_k", 2),
score_threshold=retrieval_model.get("score_threshold", 0.0) score_threshold=retrieval_model.get("score_threshold", 0.0)
if retrieval_model["score_threshold_enabled"] if retrieval_model["score_threshold_enabled"]
else None,
else 0.0,
reranking_model=retrieval_model.get("reranking_model", None) reranking_model=retrieval_model.get("reranking_model", None)
if retrieval_model["reranking_enable"] if retrieval_model["reranking_enable"]
else None, else None,

Загрузка…
Отмена
Сохранить