mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-06-04 11:14:10 +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
|