zhuhai111's picture
Upload 43 files
7cc8bc0 verified
raw
history blame contribute delete
376 Bytes
from abc import ABC, abstractmethod
from typing import List, Dict
class BaseRetriever(ABC):
"""检索策略的基类"""
@abstractmethod
def retrieve(self, query: str, context: List[Dict]) -> List[Dict]:
"""执行检索"""
pass
@abstractmethod
def init_retriever(self, config: Dict):
"""初始化检索器"""
pass