dify

dify/api/core/rag/rerank/entity/weight.py

@@ -0,0 +1,21 @@
+from pydantic import BaseModel
+
+
+class VectorSetting(BaseModel):
+    vector_weight: float
+
+    embedding_provider_name: str
+
+    embedding_model_name: str
+
+
+class KeywordSetting(BaseModel):
+    keyword_weight: float
+
+
+class Weights(BaseModel):
+    """Model for weighted rerank."""
+
+    vector_setting: VectorSetting
+
+    keyword_setting: KeywordSetting