mirror of
https://github.com/langgenius/dify.git
synced 2024-11-16 19:59:50 +08:00
22 lines
353 B
Python
22 lines
353 B
Python
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
|