aiflows
/

CCFlows

Model card Files Files and versions Community

nbaldwin commited on Dec 8, 2023

Commit

a44c20f

•

1 Parent(s): 8d8c91e

renamed flows to aiflows

Browse files

This view is limited to 50 files because it contains too many changes. See raw diff

Files changed (50) hide show

CF_Code.py +2 -2
CF_Code.yaml +4 -4
CF_CodeCollab.py +1 -1
CF_CodeCollab.yaml +10 -10
CF_CodeCritic.py +2 -2
CF_CodeCritic.yaml +4 -4
CF_CodeCriticWrongAttempt.py +2 -2
CF_CodeCriticWrongAttempt.yaml +4 -4
CF_CodeCriticWrongAttemptWithPlan.py +2 -2
CF_CodeCriticWrongAttemptWithPlan.yaml +4 -4
CF_CodeDebug.py +1 -1
CF_CodeDebug.yaml +9 -9
CF_CodeDebugCollab.py +1 -1
CF_CodeDebugCollab.yaml +8 -8
CF_CodeDebugCollabWithPlan.py +1 -1
CF_CodeDebugCollabWithPlan.yaml +8 -8
CF_CodeDebugCritic.py +1 -1
CF_CodeDebugCritic.yaml +7 -7
CF_CodeDebugCriticWithPlan.py +1 -1
CF_CodeDebugCriticWithPlan.yaml +7 -7
CF_CodeReflect.py +1 -1
CF_CodeReflect.yaml +9 -9
CF_CodeTesting.py +2 -2
CF_CodeWithPlan.py +2 -2
CF_CodeWithPlan.yaml +4 -4
CF_Plan.py +2 -2
CF_Plan.yaml +3 -3
CF_PlanCollab.py +1 -1
CF_PlanCollab.yaml +10 -10
CF_PlanCollab_Code.py +1 -1
CF_PlanCollab_Code.yaml +8 -8
CF_PlanCritic.py +2 -2
CF_PlanCritic.yaml +4 -4
CF_PlanReflect.py +1 -1
CF_PlanReflect.yaml +9 -9
CF_PlanReflect_Code.py +1 -1
CF_PlanReflect_Code.yaml +8 -8
CF_Plan_Code.py +1 -1
CF_Plan_Code.yaml +10 -10
CodeTesting.py +2 -2
LC_Code.py +2 -2
LC_Code.yaml +4 -4
LC_CodeCollab.py +1 -1
LC_CodeCollab.yaml +10 -10
LC_CodeCritic.py +2 -2
LC_CodeCritic.yaml +4 -4
LC_CodeCriticWrongAttempt.py +2 -2
LC_CodeCriticWrongAttempt.yaml +4 -4
LC_CodeDebug.py +1 -1
LC_CodeDebug.yaml +9 -9

CF_Code.py CHANGED Viewed

@@ -1,6 +1,6 @@
-from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
-class CF_Code(OpenAIChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

+from flow_modules.aiflows.ChatFlowModule import ChatAtomicFlow
+class CF_Code(ChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

CF_Code.yaml CHANGED Viewed

@@ -18,7 +18,7 @@ output_interface:
 # ~~~ Flow specification ~~~
 backend:
-  _target_: flows.backends.llm_lite.LiteLLMBackend
   wait_time_per_key: 6
   model_name:
     openai: "gpt-4"
@@ -31,7 +31,7 @@ backend:
   presence_penalty: 0
 system_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to provide executable Python code that solves a competitive programming problem. The code should correctly handle all corner cases in order to pass the hidden test cases, which are used to evaluate the correctness of the solution.
@@ -47,14 +47,14 @@ system_message_prompt_template:
 human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

 # ~~~ Flow specification ~~~
 backend:
+  _target_: aiflows.backends.llm_lite.LiteLLMBackend
   wait_time_per_key: 6
   model_name:
     openai: "gpt-4"
   presence_penalty: 0
 system_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to provide executable Python code that solves a competitive programming problem. The code should correctly handle all corner cases in order to pass the hidden test cases, which are used to evaluate the correctness of the solution.
 human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

CF_CodeCollab.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import CircularFlow
 class CF_CodeCollab(CircularFlow):


1	+ from aiflows.base_flows import CircularFlow
2
3
4	class CF_CodeCollab(CircularFlow):

CF_CodeCollab.yaml CHANGED Viewed

@@ -18,7 +18,7 @@ subflows_config:
     _target_: .CF_Code.instantiate_from_default_config
     name: "CodeGenerator"
     human_message_prompt_template:
-      _target_: flows.prompt_template.JinjaPrompt
       template: |2-
         # Feedback on the last proposed solution
         {{code_feedback}}
@@ -45,29 +45,29 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeGenerator
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=```python)([\s\S]*?)(?=```)'
           regex_fallback: '(?<=```)([\s\S]*?)(?=```)'
           input_key: "api_output"
           output_key: "code"
           strip: True
           assert_unique: True
-        - _target_: flows.data_transformations.EndOfInteraction
           end_of_interaction_string: "Final answer"
           input_key: "api_output"
           output_key: "end_of_interaction"
-        - _target_: flows.data_transformations.PrintPreviousMessages
     reset: false
   # ~~~ Code Critic ~~~
@@ -75,16 +75,16 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeCritic
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       keys_to_rename:
         api_output: "code_feedback"

     _target_: .CF_Code.instantiate_from_default_config
     name: "CodeGenerator"
     human_message_prompt_template:
+      _target_: aiflows.prompt_template.JinjaPrompt
       template: |2-
         # Feedback on the last proposed solution
         {{code_feedback}}
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### 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
     reset: false
   # ~~~ Code Critic ~~~
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeCritic
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       keys_to_rename:
         api_output: "code_feedback"

CF_CodeCritic.py CHANGED Viewed

@@ -1,6 +1,6 @@
-from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
-class CF_CodeCritic(OpenAIChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

+from flow_modules.aiflows.ChatFlowModule import ChatAtomicFlow
+class CF_CodeCritic(ChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

CF_CodeCritic.yaml CHANGED Viewed

@@ -19,7 +19,7 @@ output_interface:
 # ~~~ Flow specification ~~~
 backend:
-  _target_: flows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
       openai: "gpt-4"
@@ -33,7 +33,7 @@ backend:
   presence_penalty: 0
 system_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to identify potential issues with a competitive programming solution attempt.
@@ -51,14 +51,14 @@ system_message_prompt_template:
 human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

 # ~~~ Flow specification ~~~
 backend:
+  _target_: aiflows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
       openai: "gpt-4"
   presence_penalty: 0
 system_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to identify potential issues with a competitive programming solution attempt.
 human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

CF_CodeCriticWrongAttempt.py CHANGED Viewed

@@ -1,6 +1,6 @@
-from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
-class CF_CodeCriticWrongAttempt(OpenAIChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

+from flow_modules.aiflows.ChatFlowModule import ChatAtomicFlow
+class CF_CodeCriticWrongAttempt(ChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

CF_CodeCriticWrongAttempt.yaml CHANGED Viewed

@@ -20,7 +20,7 @@ output_interface:
 # ~~~ Flow specification ~~~
 backend:
-  _target_: flows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
     openai: "gpt-4"
@@ -35,7 +35,7 @@ backend:
   presence_penalty: 0
 system_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to identify the issues with an incorrect competitive programming solution attempt.
@@ -54,14 +54,14 @@ system_message_prompt_template:
 human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

 # ~~~ Flow specification ~~~
 backend:
+  _target_: aiflows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
     openai: "gpt-4"
   presence_penalty: 0
 system_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to identify the issues with an incorrect competitive programming solution attempt.
 human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

CF_CodeCriticWrongAttemptWithPlan.py CHANGED Viewed

@@ -1,6 +1,6 @@
-from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
-class CF_CodeCriticWrongAttemptWithPlan(OpenAIChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

+from flow_modules.aiflows.ChatFlowModule import ChatAtomicFlow
+class CF_CodeCriticWrongAttemptWithPlan(ChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

CF_CodeCriticWrongAttemptWithPlan.yaml CHANGED Viewed

@@ -21,7 +21,7 @@ output_interface:
 # ~~~ Flow specification ~~~
 backend:
-  _target_: flows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
       openai: "gpt-4"
@@ -36,7 +36,7 @@ backend:
   presence_penalty: 0
 system_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to identify the issues with an incorrect competitive programming solution attempt.
@@ -57,14 +57,14 @@ system_message_prompt_template:
 human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

 # ~~~ Flow specification ~~~
 backend:
+  _target_: aiflows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
       openai: "gpt-4"
   presence_penalty: 0
 system_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to identify the issues with an incorrect competitive programming solution attempt.
 human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

CF_CodeDebug.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import CircularFlow
 class CF_CodeDebug(CircularFlow):


1	+ from aiflows.base_flows import CircularFlow
2
3
4	class CF_CodeDebug(CircularFlow):

CF_CodeDebug.yaml CHANGED Viewed

@@ -23,7 +23,7 @@ subflows_config:
     _target_: .CF_Code.instantiate_from_default_config
     name: "CodeGenerator"
     backend:
-      _target_: flows.backends.llm_lite.LiteLLMBackend
       model_name:
         openai: "gpt-4"
         azure: "azure/gpt-4"
@@ -52,25 +52,25 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeGenerator
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=```python)([\s\S]*?)(?=```)'
           regex_fallback: '(?<=```)([\s\S]*?)(?=```)'
           input_key: "api_output"
           output_key: "code"
           strip: True
           assert_unique: True
-        - _target_: flows.data_transformations.PrintPreviousMessages
       keys_to_select:
         - "code"
@@ -82,16 +82,16 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeTestingCritic
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
         - _target_: .src.data_transformations.CorrectnessFlag
           input_key: "public_tests_results"

     _target_: .CF_Code.instantiate_from_default_config
     name: "CodeGenerator"
     backend:
+      _target_: aiflows.backends.llm_lite.LiteLLMBackend
       model_name:
         openai: "gpt-4"
         azure: "azure/gpt-4"
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### 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.PrintPreviousMessages
       keys_to_select:
         - "code"
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeTestingCritic
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
         - _target_: .src.data_transformations.CorrectnessFlag
           input_key: "public_tests_results"

CF_CodeDebugCollab.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import CircularFlow
 class CF_CodeDebugCollab(CircularFlow):


1	+ from aiflows.base_flows import CircularFlow
2
3
4	class CF_CodeDebugCollab(CircularFlow):

CF_CodeDebugCollab.yaml CHANGED Viewed

@@ -22,12 +22,12 @@ subflows_config:
     _target_: .CF_Code.instantiate_from_default_config
     name: "CodeGenerator"
     backend:
-      _target_: flows.backends.llm_lite.LiteLLMBackend
       model_name:
         openai: "gpt-4"
         azure: "azure/gpt-4"
     human_message_prompt_template:
-      _target_: flows.prompt_template.JinjaPrompt
       template: |2-
         {{testing_results_summary}}
@@ -55,18 +55,18 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeGenerator
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=```python)([\s\S]*?)(?=```)'
           regex_fallback: '(?<=```)([\s\S]*?)(?=```)'
           input_key: "api_output"
@@ -83,9 +83,9 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeDebugCritic

     _target_: .CF_Code.instantiate_from_default_config
     name: "CodeGenerator"
     backend:
+      _target_: aiflows.backends.llm_lite.LiteLLMBackend
       model_name:
         openai: "gpt-4"
         azure: "azure/gpt-4"
     human_message_prompt_template:
+      _target_: aiflows.prompt_template.JinjaPrompt
       template: |2-
         {{testing_results_summary}}
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### 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"
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeDebugCritic

CF_CodeDebugCollabWithPlan.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import CircularFlow
 class CF_CodeDebugCollabWithPlan(CircularFlow):


1	+ from aiflows.base_flows import CircularFlow
2
3
4	class CF_CodeDebugCollabWithPlan(CircularFlow):

CF_CodeDebugCollabWithPlan.yaml CHANGED Viewed

@@ -23,13 +23,13 @@ subflows_config:
     _target_: .CF_CodeWithPlan.instantiate_from_default_config
     name: "CodeGenerator"
     backend:
-      _target_: flows.backends.llm_lite.LiteLLMBackend
       api_infos: ???
       model_name:
         openai: "gpt-4"
         azure: "azure/gpt-4"
     human_message_prompt_template:
-      _target_: flows.prompt_template.JinjaPrompt
       template: |2-
         {{testing_results_summary}}
@@ -57,18 +57,18 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeGenerator
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=```python)([\s\S]*?)(?=```)'
           regex_fallback: '(?<=```)([\s\S]*?)(?=```)'
           input_key: "api_output"
@@ -85,9 +85,9 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeDebugCritic

     _target_: .CF_CodeWithPlan.instantiate_from_default_config
     name: "CodeGenerator"
     backend:
+      _target_: aiflows.backends.llm_lite.LiteLLMBackend
       api_infos: ???
       model_name:
         openai: "gpt-4"
         azure: "azure/gpt-4"
     human_message_prompt_template:
+      _target_: aiflows.prompt_template.JinjaPrompt
       template: |2-
         {{testing_results_summary}}
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### 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"
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeDebugCritic

CF_CodeDebugCritic.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import SequentialFlow
 class CF_CodeDebugCritic(SequentialFlow):


1	+ from aiflows.base_flows import SequentialFlow
2
3
4	class CF_CodeDebugCritic(SequentialFlow):

CF_CodeDebugCritic.yaml CHANGED Viewed

@@ -30,16 +30,16 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeTestingCritic
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
         - _target_: .src.data_transformations.CorrectnessFlag
           input_key: "public_tests_results"
@@ -111,18 +111,18 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeCriticWrongAttempt
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyRename
           old_key2new_key:
             api_output: "code_feedback"

     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeTestingCritic
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
         - _target_: .src.data_transformations.CorrectnessFlag
           input_key: "public_tests_results"
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeCriticWrongAttempt
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyRename
           old_key2new_key:
             api_output: "code_feedback"

CF_CodeDebugCriticWithPlan.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import SequentialFlow
 class CF_CodeDebugCriticWithPlan(SequentialFlow):


1	+ from aiflows.base_flows import SequentialFlow
2
3
4	class CF_CodeDebugCriticWithPlan(SequentialFlow):

CF_CodeDebugCriticWithPlan.yaml CHANGED Viewed

@@ -31,16 +31,16 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeTestingCritic
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
         - _target_: .src.data_transformations.CorrectnessFlag
           input_key: "public_tests_results"
@@ -112,18 +112,18 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeCriticWrongAttempt
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyRename
           old_key2new_key:
             api_output: "code_feedback"

     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeTestingCritic
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
         - _target_: .src.data_transformations.CorrectnessFlag
           input_key: "public_tests_results"
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeCriticWrongAttempt
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyRename
           old_key2new_key:
             api_output: "code_feedback"

CF_CodeReflect.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import CircularFlow
 class CF_CodeReflect(CircularFlow):


1	+ from aiflows.base_flows import CircularFlow
2
3
4	class CF_CodeReflect(CircularFlow):

CF_CodeReflect.yaml CHANGED Viewed

@@ -31,9 +31,9 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
       keys_to_rename:
         code_reflect_message: "query"
@@ -42,20 +42,20 @@ topology:
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=```python)([\s\S]*?)(?=```)'
           regex_fallback: '(?<=```)([\s\S]*?)(?=```)'
           input_key: "api_output"
           output_key: "code"
           strip: True
           assert_unique: True
-        - _target_: flows.data_transformations.EndOfInteraction
           end_of_interaction_string: "Final answer"
           input_key: "api_output"
           output_key: "end_of_interaction"
-        - _target_: flows.data_transformations.PrintPreviousMessages
       keys_to_select:
         - "code"
         - "end_of_interaction"
@@ -67,9 +67,9 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
@@ -77,7 +77,7 @@ topology:
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       keys_to_rename:
         fixed_reply: "code_reflect_message"

     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
       keys_to_rename:
         code_reflect_message: "query"
     ### 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"
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       keys_to_rename:
         fixed_reply: "code_reflect_message"

CF_CodeTesting.py CHANGED Viewed

@@ -1,7 +1,7 @@
 from typing import Any, Dict
-from flows import logging
-from flows.utils.general_helpers import validate_flow_config
 from .src.evaluation import testing_utils_codeforces
 from .CodeTesting import CodeTesting

 from typing import Any, Dict
+from aiflows import logging
+from aiflows.utils.general_helpers import validate_flow_config
 from .src.evaluation import testing_utils_codeforces
 from .CodeTesting import CodeTesting

CF_CodeWithPlan.py CHANGED Viewed

@@ -1,6 +1,6 @@
-from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
-class CF_CodeWithPlan(OpenAIChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

+from flow_modules.aiflows.ChatFlowModule import ChatAtomicFlow
+class CF_CodeWithPlan(ChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

CF_CodeWithPlan.yaml CHANGED Viewed

@@ -19,7 +19,7 @@ output_interface:
 # ~~~ Flow specification ~~~
 backend:
-  _target_: flows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
     openai: "gpt-4"
@@ -32,7 +32,7 @@ backend:
   presence_penalty: 0
 system_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to provide executable Python code that solves a competitive programming problem. The code should correctly handle all corner cases in order to pass the hidden test cases, which are used to evaluate the correctness of the solution.
@@ -50,14 +50,14 @@ system_message_prompt_template:
 human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

 # ~~~ Flow specification ~~~
 backend:
+  _target_: aiflows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
     openai: "gpt-4"
   presence_penalty: 0
 system_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to provide executable Python code that solves a competitive programming problem. The code should correctly handle all corner cases in order to pass the hidden test cases, which are used to evaluate the correctness of the solution.
 human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

CF_Plan.py CHANGED Viewed

@@ -1,6 +1,6 @@
-from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
-class CF_Plan(OpenAIChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

+from flow_modules.aiflows.ChatFlowModule import ChatAtomicFlow
+class CF_Plan(ChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

CF_Plan.yaml CHANGED Viewed

@@ -30,7 +30,7 @@ backend:
   presence_penalty: 0
 system_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to provide a high-level conceptual solution that, if implemented, will solve a given competitive programming problem.
@@ -48,14 +48,14 @@ system_message_prompt_template:
 human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

   presence_penalty: 0
 system_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to provide a high-level conceptual solution that, if implemented, will solve a given competitive programming problem.
 human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

CF_PlanCollab.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import CircularFlow
 class CF_PlanCollab(CircularFlow):


1	+ from aiflows.base_flows import CircularFlow
2
3
4	class CF_PlanCollab(CircularFlow):

CF_PlanCollab.yaml CHANGED Viewed

@@ -21,7 +21,7 @@ subflows_config:
   PlanGenerator:
     _target_: .CF_Plan.instantiate_from_default_config
     human_message_prompt_template:
-      _target_: flows.prompt_template.JinjaPrompt
       template: |2-
         # Feedback on the last proposed conceptual solution
         {{plan_feedback}}
@@ -48,18 +48,18 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: PlanGenerator
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=Conceptual solution)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
           regex_fallback:
             - '(?<=Conceptual solution:)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
@@ -67,27 +67,27 @@ topology:
           output_key: "plan"
           strip: True
           assert_unique: True
-        - _target_: flows.data_transformations.EndOfInteraction
           end_of_interaction_string: "Final answer"
           input_key: "api_output"
           output_key: "end_of_interaction"
-        - _target_: flows.data_transformations.PrintPreviousMessages
     reset: false
   # ~~~ Plan Critic ~~~
   - goal: "Provide feedback for the candidate solution."
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: PlanCritic
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       keys_to_rename:
         api_output: "plan_feedback"

   PlanGenerator:
     _target_: .CF_Plan.instantiate_from_default_config
     human_message_prompt_template:
+      _target_: aiflows.prompt_template.JinjaPrompt
       template: |2-
         # Feedback on the last proposed conceptual solution
         {{plan_feedback}}
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: PlanGenerator
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=Conceptual solution)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
           regex_fallback:
             - '(?<=Conceptual solution:)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
           output_key: "plan"
           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
     reset: false
   # ~~~ Plan Critic ~~~
   - goal: "Provide feedback for the candidate solution."
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: PlanCritic
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       keys_to_rename:
         api_output: "plan_feedback"

CF_PlanCollab_Code.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import SequentialFlow
 class CF_PlanCollab_Code(SequentialFlow):


1	+ from aiflows.base_flows import SequentialFlow
2
3
4	class CF_PlanCollab_Code(SequentialFlow):

CF_PlanCollab_Code.yaml CHANGED Viewed

@@ -26,16 +26,16 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: PlanGenerator
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       keys_to_select:
         - "plan"
@@ -44,24 +44,24 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeGenerator
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=```python)([\s\S]*?)(?=```)'
           regex_fallback: '(?<=```)([\s\S]*?)(?=```)'
           input_key: "api_output"
           output_key: "code"
           strip: True
           assert_unique: True
-        - _target_: flows.data_transformations.PrintPreviousMessages
       keys_to_select:
         - "code"

     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: PlanGenerator
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       keys_to_select:
         - "plan"
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### 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.PrintPreviousMessages
       keys_to_select:
         - "code"

CF_PlanCritic.py CHANGED Viewed

@@ -1,6 +1,6 @@
-from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
-class CF_PlanCritic(OpenAIChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

+from flow_modules.aiflows.ChatFlowModule import ChatAtomicFlow
+class CF_PlanCritic(ChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

CF_PlanCritic.yaml CHANGED Viewed

@@ -19,7 +19,7 @@ output_interface:
 # ~~~ Flow specification ~~~
 backend:
-  _target_: flows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
     openai: "gpt-4"
@@ -33,7 +33,7 @@ backend:
   presence_penalty: 0
 system_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
       Your goal is to identify potential issues with a conceptual solution to a given competitive programming problem.
@@ -52,14 +52,14 @@ system_message_prompt_template:
 human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

 # ~~~ Flow specification ~~~
 backend:
+  _target_: aiflows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
     openai: "gpt-4"
   presence_penalty: 0
 system_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
       Your goal is to identify potential issues with a conceptual solution to a given competitive programming problem.
 human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

CF_PlanReflect.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import CircularFlow
 class CF_PlanReflect(CircularFlow):


1	+ from aiflows.base_flows import CircularFlow
2
3
4	class CF_PlanReflect(CircularFlow):

CF_PlanReflect.yaml CHANGED Viewed

@@ -30,9 +30,9 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
       keys_to_rename:
         plan_reflect_message: "query"
@@ -41,9 +41,9 @@ topology:
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=Conceptual solution)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
           regex_fallback:
             - '(?<=Conceptual solution:)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
@@ -51,11 +51,11 @@ topology:
           output_key: "plan"
           strip: True
           assert_unique: True
-        - _target_: flows.data_transformations.EndOfInteraction
           end_of_interaction_string: "Final answer"
           input_key: "api_output"
           output_key: "end_of_interaction"
-        - _target_: flows.data_transformations.PrintPreviousMessages
       keys_to_select:
         - "plan"
         - "end_of_interaction"
@@ -67,16 +67,16 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: PlanReflectCritic
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       keys_to_rename:
         fixed_reply: "plan_reflect_message"

     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
       keys_to_rename:
         plan_reflect_message: "query"
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=Conceptual solution)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
           regex_fallback:
             - '(?<=Conceptual solution:)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
           output_key: "plan"
           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:
         - "plan"
         - "end_of_interaction"
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: PlanReflectCritic
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       keys_to_rename:
         fixed_reply: "plan_reflect_message"

CF_PlanReflect_Code.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import SequentialFlow
 class CF_PlanReflect_Code(SequentialFlow):


1	+ from aiflows.base_flows import SequentialFlow
2
3
4	class CF_PlanReflect_Code(SequentialFlow):

CF_PlanReflect_Code.yaml CHANGED Viewed

@@ -26,16 +26,16 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: PlanGenerator
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       keys_to_select:
         - "plan"
@@ -44,24 +44,24 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeGenerator
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=```python)([\s\S]*?)(?=```)'
           regex_fallback: '(?<=```)([\s\S]*?)(?=```)'
           input_key: "api_output"
           output_key: "code"
           strip: True
           assert_unique: True
-        - _target_: flows.data_transformations.PrintPreviousMessages
       keys_to_select:
         - "code"

     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: PlanGenerator
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       keys_to_select:
         - "plan"
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### 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.PrintPreviousMessages
       keys_to_select:
         - "code"

CF_Plan_Code.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import SequentialFlow
 class CF_Plan_Code(SequentialFlow):


1	+ from aiflows.base_flows import SequentialFlow
2
3
4	class CF_Plan_Code(SequentialFlow):

CF_Plan_Code.yaml CHANGED Viewed

@@ -26,18 +26,18 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: PlanGenerator
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=Conceptual solution)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
           regex_fallback:
             - '(?<=Conceptual solution:)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
@@ -45,7 +45,7 @@ topology:
           output_key: "plan"
           strip: True
           assert_unique: True
-        - _target_: flows.data_transformations.PrintPreviousMessages
       keys_to_select:
         - "plan"
@@ -54,24 +54,24 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeGenerator
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=```python)([\s\S]*?)(?=```)'
           regex_fallback: '(?<=```)([\s\S]*?)(?=```)'
           input_key: "api_output"
           output_key: "code"
           strip: True
           assert_unique: True
-        - _target_: flows.data_transformations.PrintPreviousMessages
       keys_to_select:
         - "code"

     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: PlanGenerator
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=Conceptual solution)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
           regex_fallback:
             - '(?<=Conceptual solution:)([\s\S]*?)(?=\n\n# [A-Z]|\Z)'
           output_key: "plan"
           strip: True
           assert_unique: True
+        - _target_: aiflows.data_transformations.PrintPreviousMessages
       keys_to_select:
         - "plan"
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### 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.PrintPreviousMessages
       keys_to_select:
         - "code"

CodeTesting.py CHANGED Viewed

@@ -1,7 +1,7 @@
 from typing import Any, Dict
-from flows import logging
-from flows.base_flows import AtomicFlow
 log = logging.get_logger(__name__)

 from typing import Any, Dict
+from aiflows import logging
+from aiflows.base_flows import AtomicFlow
 log = logging.get_logger(__name__)

LC_Code.py CHANGED Viewed

@@ -1,6 +1,6 @@
-from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
-class LC_Code(OpenAIChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

+from flow_modules.aiflows.ChatFlowModule import ChatAtomicFlow
+class LC_Code(ChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

LC_Code.yaml CHANGED Viewed

@@ -18,7 +18,7 @@ output_interface:
 # ~~~ Flow specification ~~~
 backend:
-  _target_: flows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
     openai: "gpt-4"
@@ -33,7 +33,7 @@ backend:
   presence_penalty: 0
 system_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to provide executable Python code that solves a coding interview problem. The code should correctly handle all corner cases in order to pass the hidden test cases, which are used to evaluate the correctness of the solution.
@@ -47,14 +47,14 @@ system_message_prompt_template:
 human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

 # ~~~ Flow specification ~~~
 backend:
+  _target_: aiflows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
     openai: "gpt-4"
   presence_penalty: 0
 system_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to provide executable Python code that solves a coding interview problem. The code should correctly handle all corner cases in order to pass the hidden test cases, which are used to evaluate the correctness of the solution.
 human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

LC_CodeCollab.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import CircularFlow
 class LC_CodeCollab(CircularFlow):


1	+ from aiflows.base_flows import CircularFlow
2
3
4	class LC_CodeCollab(CircularFlow):

LC_CodeCollab.yaml CHANGED Viewed

@@ -18,7 +18,7 @@ subflows_config:
     _target_: .LC_Code.instantiate_from_default_config
     name: "CodeGenerator"
     human_message_prompt_template:
-      _target_: flows.prompt_template.JinjaPrompt
       template: |2-
         # Feedback on the last proposed solution
         {{code_feedback}}
@@ -45,29 +45,29 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeGenerator
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=```python)([\s\S]*?)(?=```)'
           regex_fallback: '(?<=```)([\s\S]*?)(?=```)'
           input_key: "api_output"
           output_key: "code"
           strip: True
           assert_unique: True
-        - _target_: flows.data_transformations.EndOfInteraction
           end_of_interaction_string: "Final answer"
           input_key: "api_output"
           output_key: "end_of_interaction"
-        - _target_: flows.data_transformations.PrintPreviousMessages
     reset: false
   # ~~~ Code Critic ~~~
@@ -75,16 +75,16 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeCritic
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       keys_to_rename:
         api_output: "code_feedback"

     _target_: .LC_Code.instantiate_from_default_config
     name: "CodeGenerator"
     human_message_prompt_template:
+      _target_: aiflows.prompt_template.JinjaPrompt
       template: |2-
         # Feedback on the last proposed solution
         {{code_feedback}}
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### 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
     reset: false
   # ~~~ Code Critic ~~~
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeCritic
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       keys_to_rename:
         api_output: "code_feedback"

LC_CodeCritic.py CHANGED Viewed

@@ -1,6 +1,6 @@
-from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
-class LC_CodeCritic(OpenAIChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

+from flow_modules.aiflows.ChatFlowModule import ChatAtomicFlow
+class LC_CodeCritic(ChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

LC_CodeCritic.yaml CHANGED Viewed

@@ -19,7 +19,7 @@ output_interface:
 # ~~~ Flow specification ~~~
 backend:
-  _target_: flows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
     openai: "gpt-4"
@@ -35,7 +35,7 @@ backend:
   presence_penalty: 0
 system_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to identify potential issues with a competitive programming solution attempt.
@@ -52,14 +52,14 @@ system_message_prompt_template:
 human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

 # ~~~ Flow specification ~~~
 backend:
+  _target_: aiflows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
     openai: "gpt-4"
   presence_penalty: 0
 system_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to identify potential issues with a competitive programming solution attempt.
 human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

LC_CodeCriticWrongAttempt.py CHANGED Viewed

@@ -1,6 +1,6 @@
-from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
-class LC_CodeCriticWrongAttempt(OpenAIChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

+from flow_modules.aiflows.ChatFlowModule import ChatAtomicFlow
+class LC_CodeCriticWrongAttempt(ChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

LC_CodeCriticWrongAttempt.yaml CHANGED Viewed

@@ -20,7 +20,7 @@ output_interface:
 # ~~~ Flow specification ~~~
 backend:
-  _target_: flows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
     openai: "gpt-4"
@@ -35,7 +35,7 @@ backend:
   presence_penalty: 0
 system_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to identify the issues with an incorrect competitive programming solution attempt.
@@ -53,14 +53,14 @@ system_message_prompt_template:
 human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

 # ~~~ Flow specification ~~~
 backend:
+  _target_: aiflows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name:
     openai: "gpt-4"
   presence_penalty: 0
 system_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     Your goal is to identify the issues with an incorrect competitive programming solution attempt.
 human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: "{{query}}"
   input_variables:
     - "query"
 init_human_message_prompt_template:
+  _target_: aiflows.prompt_template.JinjaPrompt
   template: |2-
     # Problem statement
     {{problem_description}}

LC_CodeDebug.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flows.base_flows import CircularFlow
 class LC_CodeDebug(CircularFlow):


1	+ from aiflows.base_flows import CircularFlow
2
3
4	class LC_CodeDebug(CircularFlow):

LC_CodeDebug.yaml CHANGED Viewed

@@ -23,7 +23,7 @@ subflows_config:
     _target_: .LC_Code.instantiate_from_default_config
     name: "CodeGenerator"
     backend:
-      _target_: flows.backends.llm_lite.LiteLLMBackend
       api_infos: ???
       model_name:
         openai: "gpt-4"
@@ -54,25 +54,25 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeGenerator
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.RegexFirstOccurrenceExtractor
           regex: '(?<=```python)([\s\S]*?)(?=```)'
           regex_fallback: '(?<=```)([\s\S]*?)(?=```)'
           input_key: "api_output"
           output_key: "code"
           strip: True
           assert_unique: True
-        - _target_: flows.data_transformations.PrintPreviousMessages
       keys_to_select:
         - "code"
@@ -84,16 +84,16 @@ topology:
     ### Input Interface
     input_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeTestingCritic
     ### Output Interface
     output_interface:
-      _target_: flows.interfaces.KeyInterface
       additional_transformations:
         - _target_: .src.data_transformations.CorrectnessFlag
           input_key: "public_tests_results"

     _target_: .LC_Code.instantiate_from_default_config
     name: "CodeGenerator"
     backend:
+      _target_: aiflows.backends.llm_lite.LiteLLMBackend
       api_infos: ???
       model_name:
         openai: "gpt-4"
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### 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.PrintPreviousMessages
       keys_to_select:
         - "code"
     ### Input Interface
     input_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
+        - _target_: aiflows.data_transformations.KeyMatchInput
     ### Flow Specification
     flow: CodeTestingCritic
     ### Output Interface
     output_interface:
+      _target_: aiflows.interfaces.KeyInterface
       additional_transformations:
         - _target_: .src.data_transformations.CorrectnessFlag
           input_key: "public_tests_results"