mirror of
https://github.com/langgenius/dify.git
synced 2026-03-06 08:06:37 +08:00
21 lines
297 B
Python
21 lines
297 B
Python
from pydantic import BaseModel
|
|
|
|
|
|
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]
|