|
--- |
|
license: mit |
|
--- |
|
|
|
# Table of Contents |
|
|
|
* [\_\_init\_\_](#__init__) |
|
* [DemonstrationsAtomicFlow](#DemonstrationsAtomicFlow) |
|
* [DemonstrationsAtomicFlow](#DemonstrationsAtomicFlow.DemonstrationsAtomicFlow) |
|
* [instantiate\_from\_config](#DemonstrationsAtomicFlow.DemonstrationsAtomicFlow.instantiate_from_config) |
|
* [run](#DemonstrationsAtomicFlow.DemonstrationsAtomicFlow.run) |
|
* [ChatWithDemonstrationsFlow](#ChatWithDemonstrationsFlow) |
|
* [ChatWithDemonstrationsFlow](#ChatWithDemonstrationsFlow.ChatWithDemonstrationsFlow) |
|
|
|
<a id="__init__"></a> |
|
|
|
# \_\_init\_\_ |
|
|
|
<a id="DemonstrationsAtomicFlow"></a> |
|
|
|
# DemonstrationsAtomicFlow |
|
|
|
<a id="DemonstrationsAtomicFlow.DemonstrationsAtomicFlow"></a> |
|
|
|
## DemonstrationsAtomicFlow Objects |
|
|
|
```python |
|
class DemonstrationsAtomicFlow(AtomicFlow) |
|
``` |
|
|
|
This class implements a Demonstrations Atomic Flow. It is a flow which is usually used to pass demonstrations (of user assistant interactions) |
|
|
|
to the ChatAtomicFlow. |
|
|
|
*Configuration Parameters*: |
|
|
|
- `name` (str): The name of the flow. Default: "DemonstrationsAtomicFlow" |
|
- `description` (str): A description of the flow. This description is used to generate the help message of the flow. |
|
Default: "A flow that passes demonstrations to the ChatFlow" |
|
- `data` (List[Dict[str, Any]]): The data of the demonstrations. |
|
If data is None, the data is loaded from the file specified in the params["data_dir"]. |
|
Default: No default value this field must be set. |
|
- `params` (Dict[str, Any]): The parameters specific to the dataset of the demonstrations. Its default parameters are: |
|
- `data_dir` (str): The directory where the demonstrations are stored. If the data is not directly passed to the flow through `data` then |
|
the data is loaded from this directory. Default: No default value this field must be set. |
|
- `demonstrations_id` (str): The id of the demonstrations (name of the data file). If the data is not directly passed to the flow through `data` then |
|
the data is loaded from this file. Default: No default value this field must be set. |
|
- `demonstrations_k` (int): The number of demonstrations to pass to the ChatFlow. |
|
If None, all the demonstrations are passed to the ChatFlow. Default: None |
|
- `query_prompt_template` (Dict[str, Any]): The prompt template used to generate the query of the demonstrations. |
|
By default its of type aiflows.prompt_template.JinjaPrompt. None of the parameters of the prompt are defined by default and therefore need to be defined if one |
|
wants to use the query_prompt_template. Default parameters are defined in aiflows.prompt_template.jinja2_prompts.JinjaPrompt. |
|
- `response_prompt_template` (Dict[str, Any]): The prompt template used to generate the response of the demonstrations. By default its of type aiflows.prompt_template.JinjaPrompt. |
|
None of the parameters of the prompt are defined by default and therefore need to be defined if one |
|
wants to use the response_prompt_template. Default parameters are defined in aiflows.prompt_template.jinja2_prompts.JinjaPrompt. |
|
|
|
*Input Interface*: |
|
|
|
- The input interface expected by its successor flow (e.g. typically ChatAtomicFlow so the input interface is the one expected by ChatAtomicFlow) |
|
|
|
*Output Interface*: |
|
|
|
- The input interface expected by its successor flow (e.g. typically ChatAtomicFlow so the input interface expected by ChatAtomicFlow)) |
|
- `demonstrations` (List[Dict[str, Any]]): A list of demonstrations. Each demonstration is a dictionary with the following keys: |
|
- idx (int): The index of the demonstration |
|
- query (str): The query of the demonstration |
|
- response (str): The response of the demonstration |
|
|
|
**Arguments**: |
|
|
|
- `params` (`Dict[str, Any]`): The parameters specific to the dataset of the demonstrations. It must sould contain the following keys: |
|
- 'data_dir' (str): The directory where the demonstrations are stored. This field is used if the data is not directly passed to the flow through the 'data' field. |
|
- 'demonstrations_id' (str): The id of the demonstrations (name of the data file). This field is used if the data is not directly passed to the flow through the 'data' field. |
|
- 'demonstrations_k' (int): The number of demonstrations to pass to the ChatFlow. If None, all the demonstrations are passed to the ChatFlow. |
|
- 'ids_to_keep' (Optional[Union[str, List[str]]]): The ids of the demonstrations to keep. If None, all the demonstrations are kept. |
|
- `query_prompt_template` (`JinjaPrompt`): The prompt template used to generate the query of the demonstrations. |
|
- `response_prompt_template` (`JinjaPrompt`): The prompt template used to generate the response of the demonstrations. |
|
- `data` (`Optional[List[Dict[str, Any]]]`): The data of the demonstrations. If None, the data is loaded from the file specified in the params. |
|
|
|
<a id="DemonstrationsAtomicFlow.DemonstrationsAtomicFlow.instantiate_from_config"></a> |
|
|
|
#### instantiate\_from\_config |
|
|
|
```python |
|
@classmethod |
|
def instantiate_from_config(cls, config) |
|
``` |
|
|
|
This method instantiates the flow from a config file. |
|
|
|
**Arguments**: |
|
|
|
- `config` (`Dict[str, Any]`): The configuration of the flow. |
|
|
|
**Returns**: |
|
|
|
`Flow`: The instantiated flow. |
|
|
|
<a id="DemonstrationsAtomicFlow.DemonstrationsAtomicFlow.run"></a> |
|
|
|
#### run |
|
|
|
```python |
|
def run(input_data: Dict[str, Any]) -> Dict[str, Any] |
|
``` |
|
|
|
This method runs the flow. It returns the input data of the flow with the demonstrations added to it. |
|
|
|
**Arguments**: |
|
|
|
- `input_data` (`Dict[str, Any]`): The input data of the flow. |
|
|
|
**Returns**: |
|
|
|
`Dict[str, Any]`: The input data of the flow with the demonstrations added to it. |
|
|
|
<a id="ChatWithDemonstrationsFlow"></a> |
|
|
|
# ChatWithDemonstrationsFlow |
|
|
|
<a id="ChatWithDemonstrationsFlow.ChatWithDemonstrationsFlow"></a> |
|
|
|
## ChatWithDemonstrationsFlow Objects |
|
|
|
```python |
|
class ChatWithDemonstrationsFlow(SequentialFlow) |
|
``` |
|
|
|
A Chat with Demonstrations Flow. It is a flow that consists of multiple sub-flows that are executed sequentially. |
|
|
|
It's parent class is SequentialFlow. |
|
|
|
It Contains the following subflows: |
|
- A Demonstration Flow: It is a flow that passes demonstrations to the ChatFlow |
|
- A Chat Flow: It is a flow that uses the demonstrations to answer queries asked by the user/human. |
|
|
|
An illustration of the flow is as follows: |
|
|
|
-------> Demonstration Flow -------> Chat Flow -------> |
|
|
|
*Configuration Parameters*: |
|
|
|
- `name` (str): The name of the flow. Default: "ChatAtomic_Flow_with_Demonstrations" |
|
- `description` (str): A description of the flow. This description is used to generate the help message of the flow. |
|
Default: "A sequential flow that answers questions with demonstrations" |
|
- `subflows_config` (Dict[str,Any]): A dictionary of subflows configurations of the sequential Flow. Default: |
|
- `Demonstration Flow`: The configuration of the Demonstration Flow. By default, it a DemonstrationsAtomicFlow. |
|
Its default parmaters are defined in DemonstrationsAtomicFlow). |
|
- `Chat Flow`: The configuration of the Chat Flow. By default, its a ChatAtomicFlow. |
|
Its default parmaters are defined in ChatAtomicFlowModule (see Flowcard, i.e. README.md, of ChatAtomicFlowModule). |
|
- `topology` (str): The topology of the flow which is "sequential". By default, the topology is the one shown in the |
|
illustration above (the topology is also described in ChatWithDemonstrationsFlow.yaml). |
|
|
|
*Input Interface*: |
|
|
|
- `query` (str): A query asked to the flow (e.g. "What is the capital of France?") |
|
|
|
Output Interface: |
|
|
|
- `answer` (str): The answer of the flow to the query |
|
|
|
**Arguments**: |
|
|
|
- `\**kwargs`: Arguments to be passed to the parent class SequentialFlow constructor. |
|
|
|
|