mirror of
https://github.com/langgenius/dify.git
synced 2026-04-11 03:56:55 +08:00
28 lines
408 B
Python
28 lines
408 B
Python
from typing import TypedDict
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class MultimodalRerankInput(TypedDict):
|
|
content: str
|
|
content_type: str
|
|
|
|
|
|
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]
|