基类: BaseRetriever
SummaryIndex 的简单检索器，返回所有节点。
参数
源代码位于 llama-index-core/llama_index/core/indices/list/retrievers.py
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62 | class SummaryIndexRetriever(BaseRetriever):
"""
Simple retriever for SummaryIndex that returns all nodes.
Args:
index (SummaryIndex): The index to retrieve from.
"""
def __init__(
self,
index: SummaryIndex,
callback_manager: Optional[CallbackManager] = None,
object_map: Optional[dict] = None,
verbose: bool = False,
**kwargs: Any,
) -> None:
self._index = index
super().__init__(
callback_manager=callback_manager, object_map=object_map, verbose=verbose
)
def _retrieve(
self,
query_bundle: QueryBundle,
) -> List[NodeWithScore]:
"""Retrieve nodes."""
del query_bundle
node_ids = self._index.index_struct.nodes
nodes = self._index.docstore.get_nodes(node_ids)
return [NodeWithScore(node=node) for node in nodes]
|
