Remove dep. on application_flows & misc updates.

CF_Code.py

@@ -1,4 +1,4 @@
-from flows.application_flows import OpenAIChatAtomicFlow
+from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
 
 
 class CF_Code(OpenAIChatAtomicFlow):

CF_Code.yaml

@@ -21,7 +21,7 @@ model_name: "gpt-4"
 
 generation_parameters:
   n: 1
-  max_tokens: 2000
+  max_tokens: 3000
   temperature: 0.3
 
   model_kwargs:

CF_CodeCollab.py

@@ -1,6 +1,6 @@
-from flows.application_flows import GeneratorCriticFlow
+from flows.base_flows import CircularFlow
 
 
-class CF_CodeCollab(GeneratorCriticFlow):
+class CF_CodeCollab(CircularFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

CF_CodeCollab.yaml

@@ -2,8 +2,8 @@ name: "CodeCollab_Flow"
 description: |2-
   Given a problem description, alternate between a step in which code is generated, and a step in which the produced code is evaluated and useful feedback is provided.
 
-
 max_rounds: 4
+early_exit_key: "end_of_interaction"
 
 input_interface:
   - "problem_description"
@@ -89,5 +89,3 @@ topology:
         api_output: "code_feedback"
 
     reset: true
-
-early_exit_key: "end_of_interaction"

CF_CodeCritic.py

@@ -1,4 +1,4 @@
-from flows.application_flows import OpenAIChatAtomicFlow
+from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
 
 
 class CF_CodeCritic(OpenAIChatAtomicFlow):

CF_CodeCritic.yaml

@@ -83,5 +83,3 @@ init_human_message_prompt_template:
     - "io_examples_and_explanation"
     - "code"
   template_format: jinja2
-
-

CF_CodeCriticWrongAttempt.py

@@ -1,4 +1,4 @@
-from flows.application_flows import OpenAIChatAtomicFlow
+from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
 
 
 class CF_CodeCriticWrongAttempt(OpenAIChatAtomicFlow):

CF_CodeDebug.py

@@ -1,7 +1,7 @@
-from flows.application_flows import GeneratorCriticFlow
+from flows.base_flows import CircularFlow
 
 
-class CF_CodeDebug(GeneratorCriticFlow):
+class CF_CodeDebug(CircularFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
 

CF_CodeDebugCollab.py

@@ -1,7 +1,7 @@
-from flows.application_flows import GeneratorCriticFlow
+from flows.base_flows import CircularFlow
 
 
-class CF_CodeDebugCollab(GeneratorCriticFlow):
+class CF_CodeDebugCollab(CircularFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
 

CF_CodeDebugCritic.yaml

@@ -1,5 +1,6 @@
 name: "CodeDebugCritic_Flow"
-description: "Given a problem description and a candidate solution, test the code and provide useful feedback concerning the correctness of the solution and the potential mistakes."
+description: |-2
+  Given a problem description and a candidate solution, test the code and provide useful feedback concerning the correctness of the solution and the potential mistakes.
 
 # ~~~ Input interface specification ~~~
 input_interface:
@@ -17,8 +18,6 @@ output_interface:
   - "code_feedback"
 
 # ~~~ Flow specification ~~~
-public_tests_key: "public_tests_individual_io"
-
 subflows_config:
   CodeTestingCritic:
     _target_: .CF_CodeTesting.instantiate_from_default_config

CF_CodeReflect.py

@@ -1,6 +1,6 @@
-from flows.application_flows import GeneratorCriticFlow
+from flows.base_flows import CircularFlow
 
 
-class CF_CodeReflect(GeneratorCriticFlow):
+class CF_CodeReflect(CircularFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

CF_CodeReflect.yaml

@@ -1,5 +1,6 @@
 name: "CodeReflect_Flow"
-description: "Given a problem description, generate code, reflecting on it and improving it until a message suggesting that the code seems correct or a maximum number of rounds is reached."
+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:
@@ -14,6 +15,7 @@ output_interface:
 
 # ~~~ Flow specification ~~~
 max_rounds: 4
+early_exit_key: "end_of_interaction"
 
 ### Subflows specification
 subflows_config:
@@ -81,5 +83,3 @@ topology:
 
     ### Reset flag
     reset: true
-
-early_exit_key: "end_of_interaction"

CF_CodeTesting.py

@@ -1,7 +1,7 @@
 from typing import Any, Dict
 
 from flows import logging
-from flows.utils.general_helpers import validate_parameters
+from flows.utils.general_helpers import validate_flow_config
 from .src.evaluation import testing_utils_codeforces
 from .CodeTesting import CodeTesting
 
@@ -16,10 +16,10 @@ class CF_CodeTesting(CodeTesting):
         super().__init__(**kwargs)
 
     @classmethod
-    def _validate_parameters(cls, kwargs):
-        validate_parameters(cls, kwargs)
+    def _validate_flow_config(cls, kwargs):
+        validate_flow_config(cls, kwargs)
 
-        if "public_tests_key" not in kwargs["flow_config"] and "hidden_tests_key" not in kwargs["flow_config"]:
+        if "public_tests_key" not in kwargs and "hidden_tests_key" not in kwargs:
             raise ValueError("At least one of 'public_tests_key' "
                              "and 'hidden_tests_key' must be specified in the config.")
 

FixedReply_CodeReflect.py

@@ -1,4 +1,4 @@
-from flows.application_flows import FixedReplyFlow
+from flow_modules.aiflows.FixedReplyFlowModule import FixedReplyFlow
 
 
 class FixedReply_CodeReflect(FixedReplyFlow):

FixedReply_CodeReflect.yaml

@@ -1,5 +1,6 @@
 name: "CodeReflectCritic"
-description: "A flow that prompts the caller to reflect on the generated code and provide a corrected version if necessary."
+description: |-2
+  A flow that prompts the caller to reflect on the generated code and provide a corrected version if necessary.
 
 input_interface: []
 output_interface: