fixing Enum part in backend and making it same as front end (#22296)

9 months ago · 1b26f9a4c6
parent df886259bd
commit 1b26f9a4c6
1 changed files with 1 additions and 1 deletions
--- a/api/services/entities/knowledge_entities/knowledge_entities.py
+++ b/api/services/entities/knowledge_entities/knowledge_entities.py
@ -95,7 +95,7 @@ class WeightKeywordSetting(BaseModel):


 class WeightModel(BaseModel):
-    weight_type: Optional[str] = None
+    weight_type: Optional[Literal["semantic_first", "keyword_first", "customized"]] = None
    vector_setting: Optional[WeightVectorSetting] = None
    keyword_setting: Optional[WeightKeywordSetting] = None