autorag.nodes.queryexpansion package¶

Submodules¶

class autorag.nodes.queryexpansion.base.BaseQueryExpansion(project_dir: str | Path, *args, **kwargs)[source]¶

cast_to_run(previous_result: DataFrame, *args, **kwargs)[source]¶: This function is for cast function (a.k.a decorator) only for pure function in the whole node.

autorag.nodes.queryexpansion.base.check_expanded_query(query: str, expanded_query_list: List[str])[source]¶

class autorag.nodes.queryexpansion.hyde.HyDE(project_dir: str | Path, *args, **kwargs)[source]¶

class autorag.nodes.queryexpansion.multi_query_expansion.MultiQueryExpansion(project_dir: str | Path, *args, **kwargs)[source]¶

autorag.nodes.queryexpansion.multi_query_expansion.get_multi_query_expansion(query: str, answer: str) → List[str][source]¶

class autorag.nodes.queryexpansion.pass_query_expansion.PassQueryExpansion(project_dir: str | Path, *args, **kwargs)[source]¶

pure(previous_result: DataFrame, *args, **kwargs)[source]¶: Do not perform query expansion. Return with the same queries. The dimension will be 2-d list, and the column name will be ‘queries’.

class autorag.nodes.queryexpansion.query_decompose.QueryDecompose(project_dir: str | Path, *args, **kwargs)[source]¶

autorag.nodes.queryexpansion.query_decompose.get_query_decompose(query: str, answer: str) → List[str][source]¶: decompose query to little piece of questions. :param query: str, query to decompose. :param answer: str, answer from query_decompose function. :return: List[str], list of a decomposed query. Return input query if query is not decomposable.