mirror of
https://github.com/langgenius/dify.git
synced 2025-11-23 00:13:22 +00:00
26 lines
661 B
Python
26 lines
661 B
Python
from abc import ABC, abstractmethod
|
|
|
|
from core.rag.models.document import Document
|
|
|
|
|
|
class BaseRerankRunner(ABC):
|
|
@abstractmethod
|
|
def run(
|
|
self,
|
|
query: str,
|
|
documents: list[Document],
|
|
score_threshold: float | None = None,
|
|
top_n: int | None = None,
|
|
user: str | None = None,
|
|
) -> list[Document]:
|
|
"""
|
|
Run rerank model
|
|
:param query: search query
|
|
:param documents: documents for reranking
|
|
:param score_threshold: score threshold
|
|
:param top_n: top n
|
|
:param user: unique user id if needed
|
|
:return:
|
|
"""
|
|
raise NotImplementedError
|