diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-23 22:08:03 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-23 22:08:03 +0200 |
| commit | 75be0914f6bd2cdeda1539f83b38fcbc854d5cfa (patch) | |
| tree | 8d518f43926a98ee2cbae262d152662d0c07e9f6 /rag/generator/abstract.py | |
| parent | 694a4ad0e5a9e4c7eb6d11fff5ae414292ef8169 (diff) | |
Add reranking as a separate step
Diffstat (limited to 'rag/generator/abstract.py')
| -rw-r--r-- | rag/generator/abstract.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/rag/generator/abstract.py b/rag/generator/abstract.py index 1beacfb..439c1b5 100644 --- a/rag/generator/abstract.py +++ b/rag/generator/abstract.py @@ -16,3 +16,7 @@ class AbstractGenerator(type): @abstractmethod def generate(self, prompt: Prompt) -> Generator[Any, Any, Any]: pass + + @abstractmethod + def rerank(self, prompt: Prompt) -> Prompt: + return prompt |