mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-08-26 18:56:41 -04:00
Yeuoly
32 lines
609 B
Python
32 lines
609 B
Python
from pydantic import BaseModel, ConfigDict
|
|
|
|
from ...entities.model import BaseModelConfig, ModelType
|
|
|
|
|
|
class RerankDocument(BaseModel):
|
|
"""
|
|
Model class for rerank document.
|
|
"""
|
|
index: int
|
|
text: str
|
|
score: float
|
|
|
|
|
|
class RerankResult(BaseModel):
|
|
"""
|
|
Model class for rerank result.
|
|
"""
|
|
model: str
|
|
docs: list[RerankDocument]
|
|
|
|
|
|
class RerankModelConfig(BaseModelConfig):
|
|
"""
|
|
Model class for rerank model config.
|
|
"""
|
|
|
|
model_type: ModelType = ModelType.RERANK
|
|
score_threshold: float
|
|
top_n: int
|
|
|
|
model_config = ConfigDict(protected_namespaces=()) |