File size: 539 Bytes
5374a2d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import logging
from enum import Enum
from typing import List
from abc import ABC, abstractmethod
from evoagentx.rag.schema import Query, RagResult
class RerankerType(str, Enum):
SIMPLE = "simple"
BGE = "bge"
class BasePostprocessor(ABC):
"""Base interface for post-processors."""
def __init__(self):
self.logger = logging.getLogger(__name__)
@abstractmethod
def postprocess(self, query: Query, results: List[RagResult]) -> RagResult:
"""Post-process retrieval results."""
pass |