fixing Enum part in backend and making it same as front end

10 months ago · 49be299955
parent 016ff0feae
commit 49be299955
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