关键词表检索器

Query for GPTKeywordTableIndex.

class llama_index.indices.keyword_table.retrievers.BaseKeywordTableRetriever(index: BaseGPTKeywordTableIndex, keyword_extract_template: Optional[Prompt] = None, query_keyword_extract_template: Optional[Prompt] = None, max_keywords_per_query: int = 10, num_chunks_per_query: int = 10, **kwargs: Any)

Base GPT Keyword Table Index Query.

Arguments are shared among subclasses.

参数

  • keyword_extract_template (Optional[KeywordExtractPrompt]) -- A Keyword Extraction Prompt (see Prompt-Templates).

  • query_keyword_extract_template (Optional[QueryKeywordExtractPrompt]) -- A Query Keyword Extraction Prompt (see Prompt-Templates).

  • refine_template (Optional[RefinePrompt]) -- A Refinement Prompt (see Prompt-Templates).

  • text_qa_template (Optional[QuestionAnswerPrompt]) -- A Question Answering Prompt (see Prompt-Templates).

  • max_keywords_per_query (int) -- Maximum number of keywords to extract from query.

  • num_chunks_per_query (int) -- Maximum number of text chunks to query.

retrieve(str_or_query_bundle: Union[str, QueryBundle]) → List[NodeWithScore]

Retrieve nodes given query.

参数

str_or_query_bundle (QueryType) -- Either a query string or a QueryBundle object.

class llama_index.indices.keyword_table.retrievers.KeywordTableGPTRetriever(index: BaseGPTKeywordTableIndex, keyword_extract_template: Optional[Prompt] = None, query_keyword_extract_template: Optional[Prompt] = None, max_keywords_per_query: int = 10, num_chunks_per_query: int = 10, **kwargs: Any)

Keyword Table Index GPT Retriever.

Extracts keywords using GPT. Set when using retriever_mode="default".

See BaseGPTKeywordTableQuery for arguments.

retrieve(str_or_query_bundle: Union[str, QueryBundle]) → List[NodeWithScore]

Retrieve nodes given query.

参数

str_or_query_bundle (QueryType) -- Either a query string or a QueryBundle object.

class llama_index.indices.keyword_table.retrievers.KeywordTableRAKERetriever(index: BaseGPTKeywordTableIndex, keyword_extract_template: Optional[Prompt] = None, query_keyword_extract_template: Optional[Prompt] = None, max_keywords_per_query: int = 10, num_chunks_per_query: int = 10, **kwargs: Any)

Keyword Table Index RAKE Retriever.

Extracts keywords using RAKE keyword extractor. Set when retriever_mode="rake".

See BaseGPTKeywordTableQuery for arguments.

retrieve(str_or_query_bundle: Union[str, QueryBundle]) → List[NodeWithScore]

Retrieve nodes given query.

参数

str_or_query_bundle (QueryType) -- Either a query string or a QueryBundle object.

class llama_index.indices.keyword_table.retrievers.KeywordTableSimpleRetriever(index: BaseGPTKeywordTableIndex, keyword_extract_template: Optional[Prompt] = None, query_keyword_extract_template: Optional[Prompt] = None, max_keywords_per_query: int = 10, num_chunks_per_query: int = 10, **kwargs: Any)

Keyword Table Index Simple Retriever.

Extracts keywords using simple regex-based keyword extractor. Set when retriever_mode="simple".

See BaseGPTKeywordTableQuery for arguments.

retrieve(str_or_query_bundle: Union[str, QueryBundle]) → List[NodeWithScore]

Retrieve nodes given query.

参数

str_or_query_bundle (QueryType) -- Either a query string or a QueryBundle object.