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

8 months ago · 9a18a98b58
parent 7b9e01aa07
commit 9a18a98b58
2 changed files with 11 additions and 11 deletions
--- a/api/services/entities/knowledge_entities/knowledge_entities.py
+++ b/api/services/entities/knowledge_entities/knowledge_entities.py
@ -101,7 +101,7 @@ class WeightModel(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_model: Optional[RerankingModel] = None
    reranking_mode: Optional[str] = None