class RetrievalToolkit(BaseToolkit):
def __init__(
self,
auto_retriever: Optional[AutoRetriever] = None,
timeout: Optional[float] = None
):
def information_retrieval(
self,
query: str,
contents: Union[str, List[str]],
top_k: int = Constants.DEFAULT_TOP_K_RESULTS,
similarity_threshold: float = Constants.DEFAULT_SIMILARITY_THRESHOLD
):
DEFAULT_TOP_K_RESULTS
.DEFAULT_SIMILARITY_THRESHOLD
.def get_tools(self):