mirror of
http://112.124.100.131/huang.ze/ebiz-dify-ai.git
synced 2025-12-10 11:26:52 +08:00
24 lines
375 B
Python
24 lines
375 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."""
|
|
|
|
weight_type: str
|
|
|
|
vector_setting: VectorSetting
|
|
|
|
keyword_setting: KeywordSetting
|