基础：BaseLlamaPack
用于运行代理搜索检索器的 AgentSearchRetrieverPack。
源代码位于 llama-index-packs/llama-index-packs-agent-search-retriever/llama_index/packs/agent_search_retriever/base.py
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94 | class AgentSearchRetrieverPack(BaseLlamaPack):
"""AgentSearchRetrieverPack for running an agent-search retriever."""
def __init__(
self,
similarity_top_k: int = 2,
search_provider: str = "agent-search",
api_key: Optional[str] = None,
api_base: Optional[str] = None,
) -> None:
self.retriever = AgentSearchRetriever(
search_provider=search_provider,
api_key=api_key,
api_base=api_base,
similarity_top_k=similarity_top_k,
)
super().__init__()
def get_modules(self) -> Dict[str, Any]:
"""Get modules."""
return {
"retriever": self.retriever,
}
def run(self, *args: Any, **kwargs: Any) -> Any:
"""Run the pipeline."""
return self._retriever.retrieve(*args, **kwargs)
|
get_modules
get_modules() -> Dict[str, Any]
获取模块。
源代码位于 llama-index-packs/llama-index-packs-agent-search-retriever/llama_index/packs/agent_search_retriever/base.py
| def get_modules(self) -> Dict[str, Any]:
"""Get modules."""
return {
"retriever": self.retriever,
}
|
run
run(*args: Any, **kwargs: Any) -> Any
运行流水线。
源代码位于 llama-index-packs/llama-index-packs-agent-search-retriever/llama_index/packs/agent_search_retriever/base.py
| def run(self, *args: Any, **kwargs: Any) -> Any:
"""Run the pipeline."""
return self._retriever.retrieve(*args, **kwargs)
|
