
* Apply filter in eval only if no gold documents are given as input * change type annotation of input documents in eval * Update Documentation & Code Style * fix mypy * Update Documentation & Code Style Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
35 KiB
Module base
RootNode
class RootNode(BaseComponent)
RootNode feeds inputs together with corresponding params to a Pipeline.
BasePipeline
class BasePipeline()
Base class for pipelines, providing the most basic methods to load and save them in different ways.
See also the Pipeline
class for the actual pipeline logic.
get_config
def get_config(return_defaults: bool = False) -> dict
Returns a configuration for the Pipeline that can be used with BasePipeline.load_from_config()
.
Arguments:
return_defaults
: whether to output parameters that have the default values.
load_from_config
@classmethod
def load_from_config(cls, pipeline_config: Dict, pipeline_name: Optional[str] = None, overwrite_with_env_variables: bool = True)
Load Pipeline from a config dict defining the individual components and how they're tied together to form
a Pipeline. A single config can declare multiple Pipelines, in which case an explicit pipeline_name
must
be passed.
Here's a sample configuration:
```python
| {
| "version": "0.9",
| "components": [
| { # define all the building-blocks for Pipeline
| "name": "MyReader", # custom-name for the component; helpful for visualization & debugging
| "type": "FARMReader", # Haystack Class name for the component
| "params": {"no_ans_boost": -10, "model_name_or_path": "deepset/roberta-base-squad2"},
| },
| {
| "name": "MyESRetriever",
| "type": "ElasticsearchRetriever",
| "params": {
| "document_store": "MyDocumentStore", # params can reference other components defined in the YAML
| "custom_query": None,
| },
| },
| {"name": "MyDocumentStore", "type": "ElasticsearchDocumentStore", "params": {"index": "haystack_test"}},
| ],
| "pipelines": [
| { # multiple Pipelines can be defined using the components from above
| "name": "my_query_pipeline", # a simple extractive-qa Pipeline
| "nodes": [
| {"name": "MyESRetriever", "inputs": ["Query"]},
| {"name": "MyReader", "inputs": ["MyESRetriever"]},
| ],
| }
| ],
| }
```
Arguments:
pipeline_config
: the pipeline config as dictpipeline_name
: if the config contains multiple pipelines, the pipeline_name to load must be set.overwrite_with_env_variables
: Overwrite the configuration with environment variables. For example, to change index name param for an ElasticsearchDocumentStore, an env variable 'MYDOCSTORE_PARAMS_INDEX=documents-2021' can be set. Note that an_
sign must be used to specify nested hierarchical properties.
load_from_yaml
@classmethod
def load_from_yaml(cls, path: Path, pipeline_name: Optional[str] = None, overwrite_with_env_variables: bool = True)
Load Pipeline from a YAML file defining the individual components and how they're tied together to form
a Pipeline. A single YAML can declare multiple Pipelines, in which case an explicit pipeline_name
must
be passed.
Here's a sample configuration:
```yaml
| version: '0.9'
|
| components: # define all the building-blocks for Pipeline
| - name: MyReader # custom-name for the component; helpful for visualization & debugging
| type: FARMReader # Haystack Class name for the component
| params:
| no_ans_boost: -10
| model_name_or_path: deepset/roberta-base-squad2
| - name: MyESRetriever
| type: ElasticsearchRetriever
| params:
| document_store: MyDocumentStore # params can reference other components defined in the YAML
| custom_query: null
| - name: MyDocumentStore
| type: ElasticsearchDocumentStore
| params:
| index: haystack_test
|
| pipelines: # multiple Pipelines can be defined using the components from above
| - name: my_query_pipeline # a simple extractive-qa Pipeline
| nodes:
| - name: MyESRetriever
| inputs: [Query]
| - name: MyReader
| inputs: [MyESRetriever]
```
Arguments:
path
: path of the YAML file.pipeline_name
: if the YAML contains multiple pipelines, the pipeline_name to load must be set.overwrite_with_env_variables
: Overwrite the YAML configuration with environment variables. For example, to change index name param for an ElasticsearchDocumentStore, an env variable 'MYDOCSTORE_PARAMS_INDEX=documents-2021' can be set. Note that an_
sign must be used to specify nested hierarchical properties.
load_from_deepset_cloud
@classmethod
def load_from_deepset_cloud(cls, pipeline_config_name: str, pipeline_name: str = "query", workspace: str = "default", api_key: Optional[str] = None, api_endpoint: Optional[str] = None, overwrite_with_env_variables: bool = False)
Load Pipeline from Deepset Cloud defining the individual components and how they're tied together to form
a Pipeline. A single config can declare multiple Pipelines, in which case an explicit pipeline_name
must
be passed.
In order to get a list of all available pipeline_config_names, call list_pipelines_on_deepset_cloud()
.
Use the returned name
as pipeline_config_name
.
Arguments:
pipeline_config_name
: name of the config file inside the Deepset Cloud workspace. To get a list of all available pipeline_config_names, calllist_pipelines_on_deepset_cloud()
.pipeline_name
: specifies which pipeline to load from config. Deepset Cloud typically provides a 'query' and a 'index' pipeline per config.workspace
: workspace in Deepset Cloudapi_key
: Secret value of the API key. If not specified, will be read from DEEPSET_CLOUD_API_KEY environment variable.api_endpoint
: The URL of the Deepset Cloud API. If not specified, will be read from DEEPSET_CLOUD_API_ENDPOINT environment variable.overwrite_with_env_variables
: Overwrite the config with environment variables. For example, to change return_no_answer param for a FARMReader, an env variable 'READER_PARAMS_RETURN_NO_ANSWER=False' can be set. Note that an_
sign must be used to specify nested hierarchical properties.
list_pipelines_on_deepset_cloud
@classmethod
def list_pipelines_on_deepset_cloud(cls, workspace: str = "default", api_key: Optional[str] = None, api_endpoint: Optional[str] = None) -> List[dict]
Lists all pipeline configs available on Deepset Cloud.
Arguments:
workspace
: workspace in Deepset Cloudapi_key
: Secret value of the API key. If not specified, will be read from DEEPSET_CLOUD_API_KEY environment variable.api_endpoint
: The URL of the Deepset Cloud API. If not specified, will be read from DEEPSET_CLOUD_API_ENDPOINT environment variable.
Returns:
list of dictionaries: List[dict]
each dictionary: {
"name": str -> pipeline_config_name
to be used in load_from_deepset_cloud()
,
"..." -> additional pipeline meta information
}
example:
[{'name': 'my_super_nice_pipeline_config',
'pipeline_id': '2184e0c1-c6ec-40a1-9b28-5d2768e5efa2',
'status': 'DEPLOYED',
'created_at': '2022-02-01T09:57:03.803991+00:00',
'deleted': False,
'is_default': False,
'indexing': {'status': 'IN_PROGRESS',
'pending_file_count': 3,
'total_file_count': 31}}]
save_to_deepset_could
@classmethod
def save_to_deepset_could(cls, query_pipeline: BasePipeline, index_pipeline: BasePipeline, pipeline_config_name: str, workspace: str = "default", api_key: Optional[str] = None, api_endpoint: Optional[str] = None, overwrite: bool = False)
Saves a Pipeline config to Deepset Cloud defining the individual components and how they're tied together to form
a Pipeline. A single config must declare a query pipeline and a index pipeline.
Arguments:
query_pipeline
: the query pipeline to save.index_pipeline
: the index pipeline to save.pipeline_config_name
: name of the config file inside the Deepset Cloud workspace.workspace
: workspace in Deepset Cloudapi_key
: Secret value of the API key. If not specified, will be read from DEEPSET_CLOUD_API_KEY environment variable.api_endpoint
: The URL of the Deepset Cloud API. If not specified, will be read from DEEPSET_CLOUD_API_ENDPOINT environment variable.overwrite
: Whether to overwrite the config if it already exists. Otherwise an error is being raised.
Pipeline
class Pipeline(BasePipeline)
Pipeline brings together building blocks to build a complex search pipeline with Haystack & user-defined components.
Under-the-hood, a pipeline is represented as a directed acyclic graph of component nodes. It enables custom query flows with options to branch queries(eg, extractive qa vs keyword match query), merge candidate documents for a Reader from multiple Retrievers, or re-ranking of candidate documents.
add_node
def add_node(component, name: str, inputs: List[str])
Add a new node to the pipeline.
Arguments:
component
: The object to be called when the data is passed to the node. It can be a Haystack component (like Retriever, Reader, or Generator) or a user-defined object that implements a run() method to process incoming data from predecessor node.name
: The name for the node. It must not contain any dots.inputs
: A list of inputs to the node. If the predecessor node has a single outgoing edge, just the name of node is sufficient. For instance, a 'ElasticsearchRetriever' node would always output a single edge with a list of documents. It can be represented as ["ElasticsearchRetriever"].
In cases when the predecessor node has multiple outputs, e.g., a "QueryClassifier", the output must be specified explicitly as "QueryClassifier.output_2".
get_node
def get_node(name: str) -> Optional[BaseComponent]
Get a node from the Pipeline.
Arguments:
name
: The name of the node.
set_node
def set_node(name: str, component)
Set the component for a node in the Pipeline.
Arguments:
name
: The name of the node.component
: The component object to be set at the node.
run
def run(query: Optional[str] = None, file_paths: Optional[List[str]] = None, labels: Optional[MultiLabel] = None, documents: Optional[List[Document]] = None, meta: Optional[dict] = None, params: Optional[dict] = None, debug: Optional[bool] = None)
Runs the pipeline, one node at a time.
Arguments:
query
: The search query (for query pipelines only)file_paths
: The files to index (for indexing pipelines only)labels
:documents
:meta
:params
: Dictionary of parameters to be dispatched to the nodes. If you want to pass a param to all nodes, you can just use: {"top_k":10} If you want to pass it to targeted nodes, you can do: {"Retriever": {"top_k": 10}, "Reader": {"top_k": 3, "debug": True}}debug
: Whether the pipeline should instruct nodes to collect debug information about their execution. By default these include the input parameters they received and the output they generated. All debug information can then be found in the dict returned by this method under the key "_debug"
eval
def eval(labels: List[MultiLabel], documents: Optional[List[List[Document]]] = None, params: Optional[dict] = None, sas_model_name_or_path: str = None, add_isolated_node_eval: bool = False) -> EvaluationResult
Evaluates the pipeline by running the pipeline once per query in debug mode
and putting together all data that is needed for evaluation, e.g. calculating metrics.
Arguments:
labels
: The labels to evaluate ondocuments
: List of List of Document that the first node in the pipeline should get as input per multilabel. Can be used to evaluate a pipeline that consists of a reader without a retriever.params
: Dictionary of parameters to be dispatched to the nodes. If you want to pass a param to all nodes, you can just use: {"top_k":10} If you want to pass it to targeted nodes, you can do: {"Retriever": {"top_k": 10}, "Reader": {"top_k": 3, "debug": True}}sas_model_name_or_path
: Name or path of "Semantic Answer Similarity (SAS) model". When set, the model will be used to calculate similarity between predictions and labels and generate the SAS metric. The SAS metric correlates better with human judgement of correct answers as it does not rely on string overlaps. Example: Prediction = "30%", Label = "thirty percent", EM and F1 would be overly pessimistic with both being 0, while SAS paints a more realistic picture. More info in the paper: https://arxiv.org/abs/2108.06130 Models:- You can use Bi Encoders (sentence transformers) or cross encoders trained on Semantic Textual Similarity (STS) data. Not all cross encoders can be used because of different return types. If you use custom cross encoders please make sure they work with sentence_transformers.CrossEncoder class
- Good default for multiple languages: "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
- Large, powerful, but slow model for English only: "cross-encoder/stsb-roberta-large"
- Large model for German only: "deepset/gbert-large-sts"
add_isolated_node_eval
: If set to True, in addition to the integrated evaluation of the pipeline, each node is evaluated in isolated evaluation mode. This mode helps to understand the bottlenecks of a pipeline in terms of output quality of each individual node. If a node performs much better in the isolated evaluation than in the integrated evaluation, the previous node needs to be optimized to improve the pipeline's performance. If a node's performance is similar in both modes, this node itself needs to be optimized to improve the pipeline's performance. The isolated evaluation calculates the upper bound of each node's evaluation metrics under the assumption that it received perfect inputs from the previous node. To this end, labels are used as input to the node instead of the output of the previous node in the pipeline. The generated dataframes in the EvaluationResult then contain additional rows, which can be distinguished from the integrated evaluation results based on the values "integrated" or "isolated" in the column "eval_mode" and the evaluation report then additionally lists the upper bound of each node's evaluation metrics.
get_nodes_by_class
def get_nodes_by_class(class_type) -> List[Any]
Gets all nodes in the pipeline that are an instance of a certain class (incl. subclasses).
This is for example helpful if you loaded a pipeline and then want to interact directly with the document store. Example: | from haystack.document_stores.base import BaseDocumentStore | INDEXING_PIPELINE = Pipeline.load_from_yaml(Path(PIPELINE_YAML_PATH), pipeline_name=INDEXING_PIPELINE_NAME) | res = INDEXING_PIPELINE.get_nodes_by_class(class_type=BaseDocumentStore)
Returns:
List of components that are an instance the requested class
get_document_store
def get_document_store() -> Optional[BaseDocumentStore]
Return the document store object used in the current pipeline.
Returns:
Instance of DocumentStore or None
draw
def draw(path: Path = Path("pipeline.png"))
Create a Graphviz visualization of the pipeline.
Arguments:
path
: the path to save the image.
load_from_config
@classmethod
def load_from_config(cls, pipeline_config: Dict, pipeline_name: Optional[str] = None, overwrite_with_env_variables: bool = True)
Load Pipeline from a config dict defining the individual components and how they're tied together to form
a Pipeline. A single config can declare multiple Pipelines, in which case an explicit pipeline_name
must
be passed.
Here's a sample configuration:
```python
| {
| "version": "0.9",
| "components": [
| { # define all the building-blocks for Pipeline
| "name": "MyReader", # custom-name for the component; helpful for visualization & debugging
| "type": "FARMReader", # Haystack Class name for the component
| "params": {"no_ans_boost": -10, "model_name_or_path": "deepset/roberta-base-squad2"},
| },
| {
| "name": "MyESRetriever",
| "type": "ElasticsearchRetriever",
| "params": {
| "document_store": "MyDocumentStore", # params can reference other components defined in the YAML
| "custom_query": None,
| },
| },
| {"name": "MyDocumentStore", "type": "ElasticsearchDocumentStore", "params": {"index": "haystack_test"}},
| ],
| "pipelines": [
| { # multiple Pipelines can be defined using the components from above
| "name": "my_query_pipeline", # a simple extractive-qa Pipeline
| "nodes": [
| {"name": "MyESRetriever", "inputs": ["Query"]},
| {"name": "MyReader", "inputs": ["MyESRetriever"]},
| ],
| }
| ],
| }
```
Arguments:
pipeline_config
: the pipeline config as dictpipeline_name
: if the config contains multiple pipelines, the pipeline_name to load must be set.overwrite_with_env_variables
: Overwrite the configuration with environment variables. For example, to change index name param for an ElasticsearchDocumentStore, an env variable 'MYDOCSTORE_PARAMS_INDEX=documents-2021' can be set. Note that an_
sign must be used to specify nested hierarchical properties.
save_to_yaml
def save_to_yaml(path: Path, return_defaults: bool = False)
Save a YAML configuration for the Pipeline that can be used with Pipeline.load_from_yaml()
.
Arguments:
path
: path of the output YAML file.return_defaults
: whether to output parameters that have the default values.
get_config
def get_config(return_defaults: bool = False) -> dict
Returns a configuration for the Pipeline that can be used with Pipeline.load_from_config()
.
Arguments:
return_defaults
: whether to output parameters that have the default values.
print_eval_report
def print_eval_report(eval_result: EvaluationResult, n_wrong_examples: int = 3, metrics_filter: Optional[Dict[str, List[str]]] = None)
Prints evaluation report containing a metrics funnel and worst queries for further analysis.
Arguments:
eval_result
: The evaluation result, can be obtained by running eval().n_wrong_examples
: The number of worst queries to show.metrics_filter
: The metrics to show per node. If None all metrics will be shown.
RayPipeline
class RayPipeline(Pipeline)
Ray (https://ray.io) is a framework for distributed computing.
Ray allows distributing a Pipeline's components across a cluster of machines. The individual components of a Pipeline can be independently scaled. For instance, an extractive QA Pipeline deployment can have three replicas of the Reader and a single replica for the Retriever. It enables efficient resource utilization by horizontally scaling Components.
To set the number of replicas, add replicas
in the YAML config for the node in a pipeline:
```yaml
| components:
| ...
|
| pipelines:
| - name: ray_query_pipeline
| type: RayPipeline
| nodes:
| - name: ESRetriever
| replicas: 2 # number of replicas to create on the Ray cluster
| inputs: [ Query ]
```
A RayPipeline can only be created with a YAML Pipeline config.
from haystack.pipeline import RayPipeline pipeline = RayPipeline.load_from_yaml(path="my_pipelines.yaml", pipeline_name="my_query_pipeline") pipeline.run(query="What is the capital of Germany?")
By default, RayPipelines creates an instance of RayServe locally. To connect to an existing Ray instance,
set the address
parameter when creating the RayPipeline instance.
load_from_yaml
@classmethod
def load_from_yaml(cls, path: Path, pipeline_name: Optional[str] = None, overwrite_with_env_variables: bool = True, address: Optional[str] = None, **kwargs, ,)
Load Pipeline from a YAML file defining the individual components and how they're tied together to form
a Pipeline. A single YAML can declare multiple Pipelines, in which case an explicit pipeline_name
must
be passed.
Here's a sample configuration:
```yaml
| version: '0.9'
|
| components: # define all the building-blocks for Pipeline
| - name: MyReader # custom-name for the component; helpful for visualization & debugging
| type: FARMReader # Haystack Class name for the component
| params:
| no_ans_boost: -10
| model_name_or_path: deepset/roberta-base-squad2
| - name: MyESRetriever
| type: ElasticsearchRetriever
| params:
| document_store: MyDocumentStore # params can reference other components defined in the YAML
| custom_query: null
| - name: MyDocumentStore
| type: ElasticsearchDocumentStore
| params:
| index: haystack_test
|
| pipelines: # multiple Pipelines can be defined using the components from above
| - name: my_query_pipeline # a simple extractive-qa Pipeline
| type: RayPipeline
| nodes:
| - name: MyESRetriever
| inputs: [Query]
| replicas: 2 # number of replicas to create on the Ray cluster
| - name: MyReader
| inputs: [MyESRetriever]
```
Arguments:
path
: path of the YAML file.pipeline_name
: if the YAML contains multiple pipelines, the pipeline_name to load must be set.overwrite_with_env_variables
: Overwrite the YAML configuration with environment variables. For example, to change index name param for an ElasticsearchDocumentStore, an env variable 'MYDOCSTORE_PARAMS_INDEX=documents-2021' can be set. Note that an_
sign must be used to specify nested hierarchical properties.address
: The IP address for the Ray cluster. If set to None, a local Ray instance is started.
Module standard_pipelines
BaseStandardPipeline
class BaseStandardPipeline(ABC)
Base class for pre-made standard Haystack pipelines. This class does not inherit from Pipeline.
add_node
def add_node(component, name: str, inputs: List[str])
Add a new node to the pipeline.
Arguments:
component
: The object to be called when the data is passed to the node. It can be a Haystack component (like Retriever, Reader, or Generator) or a user-defined object that implements a run() method to process incoming data from predecessor node.name
: The name for the node. It must not contain any dots.inputs
: A list of inputs to the node. If the predecessor node has a single outgoing edge, just the name of node is sufficient. For instance, a 'ElasticsearchRetriever' node would always output a single edge with a list of documents. It can be represented as ["ElasticsearchRetriever"].
In cases when the predecessor node has multiple outputs, e.g., a "QueryClassifier", the output must be specified explicitly as "QueryClassifier.output_2".
get_node
def get_node(name: str)
Get a node from the Pipeline.
Arguments:
name
: The name of the node.
set_node
def set_node(name: str, component)
Set the component for a node in the Pipeline.
Arguments:
name
: The name of the node.component
: The component object to be set at the node.
draw
def draw(path: Path = Path("pipeline.png"))
Create a Graphviz visualization of the pipeline.
Arguments:
path
: the path to save the image.
save_to_yaml
def save_to_yaml(path: Path, return_defaults: bool = False)
Save a YAML configuration for the Pipeline that can be used with Pipeline.load_from_yaml()
.
Arguments:
path
: path of the output YAML file.return_defaults
: whether to output parameters that have the default values.
load_from_yaml
@classmethod
def load_from_yaml(cls, path: Path, pipeline_name: Optional[str] = None, overwrite_with_env_variables: bool = True)
Load Pipeline from a YAML file defining the individual components and how they're tied together to form
a Pipeline. A single YAML can declare multiple Pipelines, in which case an explicit pipeline_name
must
be passed.
Here's a sample configuration:
```yaml
| version: '0.8'
|
| components: # define all the building-blocks for Pipeline
| - name: MyReader # custom-name for the component; helpful for visualization & debugging
| type: FARMReader # Haystack Class name for the component
| params:
| no_ans_boost: -10
| model_name_or_path: deepset/roberta-base-squad2
| - name: MyESRetriever
| type: ElasticsearchRetriever
| params:
| document_store: MyDocumentStore # params can reference other components defined in the YAML
| custom_query: null
| - name: MyDocumentStore
| type: ElasticsearchDocumentStore
| params:
| index: haystack_test
|
| pipelines: # multiple Pipelines can be defined using the components from above
| - name: my_query_pipeline # a simple extractive-qa Pipeline
| nodes:
| - name: MyESRetriever
| inputs: [Query]
| - name: MyReader
| inputs: [MyESRetriever]
```
Arguments:
path
: path of the YAML file.pipeline_name
: if the YAML contains multiple pipelines, the pipeline_name to load must be set.overwrite_with_env_variables
: Overwrite the YAML configuration with environment variables. For example, to change index name param for an ElasticsearchDocumentStore, an env variable 'MYDOCSTORE_PARAMS_INDEX=documents-2021' can be set. Note that an_
sign must be used to specify nested hierarchical properties.
get_nodes_by_class
def get_nodes_by_class(class_type) -> List[Any]
Gets all nodes in the pipeline that are an instance of a certain class (incl. subclasses).
This is for example helpful if you loaded a pipeline and then want to interact directly with the document store. Example:
| from haystack.document_stores.base import BaseDocumentStore
| INDEXING_PIPELINE = Pipeline.load_from_yaml(Path(PIPELINE_YAML_PATH), pipeline_name=INDEXING_PIPELINE_NAME)
| res = INDEXING_PIPELINE.get_nodes_by_class(class_type=BaseDocumentStore)
Returns:
List of components that are an instance of the requested class
get_document_store
def get_document_store() -> Optional[BaseDocumentStore]
Return the document store object used in the current pipeline.
Returns:
Instance of DocumentStore or None
eval
def eval(labels: List[MultiLabel], params: Optional[dict] = None, sas_model_name_or_path: Optional[str] = None, add_isolated_node_eval: bool = False) -> EvaluationResult
Evaluates the pipeline by running the pipeline once per query in debug mode
and putting together all data that is needed for evaluation, e.g. calculating metrics.
Arguments:
labels
: The labels to evaluate onparams
: Params for theretriever
andreader
. For instance, params={"Retriever": {"top_k": 10}, "Reader": {"top_k": 5}}sas_model_name_or_path
: SentenceTransformers semantic textual similarity model to be used for sas value calculation, should be path or string pointing to downloadable models.add_isolated_node_eval
: Whether to additionally evaluate the reader based on labels as input instead of output of previous node in pipeline
ExtractiveQAPipeline
class ExtractiveQAPipeline(BaseStandardPipeline)
Pipeline for Extractive Question Answering.
run
def run(query: str, params: Optional[dict] = None, debug: Optional[bool] = None)
Arguments:
query
: The search query string.params
: Params for theretriever
andreader
. For instance, params={"Retriever": {"top_k": 10}, "Reader": {"top_k": 5}}debug
: Whether the pipeline should instruct nodes to collect debug information about their execution. By default these include the input parameters they received and the output they generated. All debug information can then be found in the dict returned by this method under the key "_debug"
DocumentSearchPipeline
class DocumentSearchPipeline(BaseStandardPipeline)
Pipeline for semantic document search.
run
def run(query: str, params: Optional[dict] = None, debug: Optional[bool] = None)
Arguments:
query
: the query string.params
: params for theretriever
andreader
. For instance, params={"Retriever": {"top_k": 10}}debug
: Whether the pipeline should instruct nodes to collect debug information about their execution. By default these include the input parameters they received and the output they generated. All debug information can then be found in the dict returned by this method under the key "_debug"
GenerativeQAPipeline
class GenerativeQAPipeline(BaseStandardPipeline)
Pipeline for Generative Question Answering.
run
def run(query: str, params: Optional[dict] = None, debug: Optional[bool] = None)
Arguments:
query
: the query string.params
: params for theretriever
andgenerator
. For instance, params={"Retriever": {"top_k": 10}, "Generator": {"top_k": 5}}debug
: Whether the pipeline should instruct nodes to collect debug information about their execution. By default these include the input parameters they received and the output they generated. All debug information can then be found in the dict returned by this method under the key "_debug"
SearchSummarizationPipeline
class SearchSummarizationPipeline(BaseStandardPipeline)
Pipeline that retrieves documents for a query and then summarizes those documents.
run
def run(query: str, params: Optional[dict] = None, debug: Optional[bool] = None)
Arguments:
query
: the query string.params
: params for theretriever
andsummarizer
. For instance, params={"Retriever": {"top_k": 10}, "Summarizer": {"generate_single_summary": True}}debug
: Whether the pipeline should instruct nodes to collect debug information about their execution. By default these include the input parameters they received and the output they generated. All debug information can then be found in the dict returned by this method under the key "_debug"
FAQPipeline
class FAQPipeline(BaseStandardPipeline)
Pipeline for finding similar FAQs using semantic document search.
run
def run(query: str, params: Optional[dict] = None, debug: Optional[bool] = None)
Arguments:
query
: the query string.params
: params for theretriever
. For instance, params={"Retriever": {"top_k": 10}}debug
: Whether the pipeline should instruct nodes to collect debug information about their execution. By default these include the input parameters they received and the output they generated. All debug information can then be found in the dict returned by this method under the key "_debug"
TranslationWrapperPipeline
class TranslationWrapperPipeline(BaseStandardPipeline)
Takes an existing search pipeline and adds one "input translation node" after the Query and one "output translation" node just before returning the results
QuestionGenerationPipeline
class QuestionGenerationPipeline(BaseStandardPipeline)
A simple pipeline that takes documents as input and generates questions that it thinks can be answered by the documents.
RetrieverQuestionGenerationPipeline
class RetrieverQuestionGenerationPipeline(BaseStandardPipeline)
A simple pipeline that takes a query as input, performs retrieval, and then generates questions that it thinks can be answered by the retrieved documents.
QuestionAnswerGenerationPipeline
class QuestionAnswerGenerationPipeline(BaseStandardPipeline)
This is a pipeline which takes a document as input, generates questions that the model thinks can be answered by this document, and then performs question answering of this questions using that single document.
MostSimilarDocumentsPipeline
class MostSimilarDocumentsPipeline(BaseStandardPipeline)
run
def run(document_ids: List[str], top_k: int = 5)
Arguments:
document_ids
: document idstop_k
: How many documents id to return against single document