Browse Source

fix keyword search top-k not initial (#21202)

tags/1.5.0
Jyong 4 months ago
parent
commit
9a18a98b58
No account linked to committer's email address

+ 10
- 10
api/services/dataset_service.py View File

knowledge_config.embedding_model, # type: ignore knowledge_config.embedding_model, # type: ignore
) )
dataset_collection_binding_id = dataset_collection_binding.id dataset_collection_binding_id = dataset_collection_binding.id
if knowledge_config.retrieval_model:
retrieval_model = knowledge_config.retrieval_model
else:
retrieval_model = RetrievalModel(
search_method=RetrievalMethod.SEMANTIC_SEARCH.value,
reranking_enable=False,
reranking_model=RerankingModel(reranking_provider_name="", reranking_model_name=""),
top_k=2,
score_threshold_enabled=False,
)
if knowledge_config.retrieval_model:
retrieval_model = knowledge_config.retrieval_model
else:
retrieval_model = RetrievalModel(
search_method=RetrievalMethod.SEMANTIC_SEARCH.value,
reranking_enable=False,
reranking_model=RerankingModel(reranking_provider_name="", reranking_model_name=""),
top_k=2,
score_threshold_enabled=False,
)
# save dataset # save dataset
dataset = Dataset( dataset = Dataset(
tenant_id=tenant_id, tenant_id=tenant_id,

+ 1
- 1
api/services/entities/knowledge_entities/knowledge_entities.py View File





class RetrievalModel(BaseModel): class RetrievalModel(BaseModel):
search_method: Literal["hybrid_search", "semantic_search", "full_text_search"]
search_method: Literal["hybrid_search", "semantic_search", "full_text_search", "keyword_search"]
reranking_enable: bool reranking_enable: bool
reranking_model: Optional[RerankingModel] = None reranking_model: Optional[RerankingModel] = None
reranking_mode: Optional[str] = None reranking_mode: Optional[str] = None

Loading…
Cancel
Save