Raptor

RaptorPack

Bases: BaseLlamaPack

猛禽包。

Source code in llama_index/packs/raptor/base.py

class RaptorPack(BaseLlamaPack):
    """猛禽包。"""

    def __init__(
        self,
        documents: List[BaseNode],
        llm: Optional[LLM] = None,
        embed_model: Optional[BaseEmbedding] = None,
        vector_store: Optional[BasePydanticVectorStore] = None,
        similarity_top_k: int = 2,
        mode: QueryModes = "collapsed",
        verbose: bool = True,
        **kwargs: Any,
    ) -> None:
        """初始化参数。"""
        self.retriever = RaptorRetriever(
            documents,
            embed_model=embed_model,
            llm=llm,
            similarity_top_k=similarity_top_k,
            vector_store=vector_store,
            mode=mode,
            verbose=verbose,
            **kwargs,
        )

    def get_modules(self) -> Dict[str, Any]:
        """获取模块。"""
        return {
            "retriever": self.retriever,
        }

    def run(
        self,
        query: str,
        mode: Optional[QueryModes] = None,
    ) -> Any:
        """运行流水线。"""
        return self.retriever.retrieve(query, mode=mode)

get_modules

get_modules() -> Dict[str, Any]

获取模块。

Source code in llama_index/packs/raptor/base.py

def get_modules(self) -> Dict[str, Any]:
    """获取模块。"""
    return {
        "retriever": self.retriever,
    }

run

run(query: str, mode: Optional[QueryModes] = None) -> Any

运行流水线。

Source code in llama_index/packs/raptor/base.py

def run(
    self,
    query: str,
    mode: Optional[QueryModes] = None,
) -> Any:
    """运行流水线。"""
    return self.retriever.retrieve(query, mode=mode)