File size: 2,544 Bytes
3c7fd6a
91b62cb
 
3c7fd6a
 
 
 
 
 
 
 
 
 
 
 
 
 
91b62cb
3c7fd6a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
a44c20f
3c7fd6a
a44c20f
3c7fd6a
 
 
 
 
 
 
 
a44c20f
3c7fd6a
a44c20f
3c7fd6a
 
 
 
 
 
a44c20f
3c7fd6a
 
 
a44c20f
3c7fd6a
 
 
 
 
 
 
 
 
 
 
a44c20f
3c7fd6a
a44c20f
3c7fd6a
 
 
 
 
 
 
a44c20f
3c7fd6a
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
name: "CodeReflect_Flow"
description: |-2
  Given a problem description, generate code, reflect on it and improve it until a message suggesting that the code seems correct or a maximum number of rounds is reached.

# ~~~ Input interface specification ~~~
input_interface:
  - "problem_description"
  - "input_description"
  - "output_description"
  - "io_examples_and_explanation"

# ~~~ Output interface specification ~~~
output_interface:
  - "code"

# ~~~ Flow specification ~~~
max_rounds: 4
early_exit_key: "end_of_interaction"

### Subflows specification
subflows_config:
  CodeGenerator:
    _target_: .CF_Code.instantiate_from_default_config
  CodeReflectCritic:
    _target_: .FixedReply_CodeReflect.instantiate_from_default_config

### Topology specification (specifies how the sequence of messages will flow from one of the subflows to another)
topology:
  # ~~~ Code Generator ~~~
  - goal: "Generate/refine a solution."

    ### Input Interface
    input_interface:
      _target_: aiflows.interfaces.KeyInterface
      additional_transformations:
        - _target_: aiflows.data_transformations.KeyMatchInput
      keys_to_rename:
        code_reflect_message: "query"

    ### Flow Specification
    flow: CodeGenerator

    ### Output Interface
    output_interface:
      _target_: aiflows.interfaces.KeyInterface
      additional_transformations:
        - _target_: aiflows.data_transformations.RegexFirstOccurrenceExtractor
          regex: '(?<=```python)([\s\S]*?)(?=```)'
          regex_fallback: '(?<=```)([\s\S]*?)(?=```)'
          input_key: "api_output"
          output_key: "code"
          strip: True
          assert_unique: True
        - _target_: aiflows.data_transformations.EndOfInteraction
          end_of_interaction_string: "Final answer"
          input_key: "api_output"
          output_key: "end_of_interaction"
        - _target_: aiflows.data_transformations.PrintPreviousMessages
      keys_to_select:
        - "code"
        - "end_of_interaction"

    ### Reset flag
    reset: false

  - goal: "Generate a message that encourages reflection."

    ### Input Interface
    input_interface:
      _target_: aiflows.interfaces.KeyInterface
      additional_transformations:
        - _target_: aiflows.data_transformations.KeyMatchInput


    ### Flow Specification
    flow: CodeReflectCritic

    ### Output Interface
    output_interface:
      _target_: aiflows.interfaces.KeyInterface
      keys_to_rename:
        fixed_reply: "code_reflect_message"

    ### Reset flag
    reset: true