Upload 8 files

CodeTestingAtomicFlow.py

@@ -34,6 +34,7 @@ class CodeTestingAtomicFlow(InterpreterAtomicFlow):
         last_timestamp = input_data["temp_code_file_written_timestamp"]
         self._wait_for_file_update(file_loc, last_timestamp)
         self._prepare_code_and_lang(input_data)
+        self._process_input_data(input_data)
         response = self._call()
         self._delete_file(file_loc)
         if response.strip() == '':

TestCodeFileEditAtomicFlow.py

@@ -5,7 +5,8 @@ from flow_modules.Tachi67.CodeFileEditFlowModule import CodeFileEditAtomicFlow
 class TestCodeFileEditAtomicFlow(CodeFileEditAtomicFlow):
     def _generate_import_statement(self, code_lib_location):
         module_dir = os.path.dirname(code_lib_location)
-        module_name = os.path.basename(code_lib_location).rstrip('.py')
+        module_name = os.path.splitext(os.path.basename(code_lib_location))[0]
+
         import_code = (
             f"import sys\n"
             f"sys.path.insert(0, '{module_dir}')\n"
@@ -25,7 +26,7 @@ class TestCodeFileEditAtomicFlow(CodeFileEditAtomicFlow):
                 "# You could also assert the output, anyway, the test results will be informed to JARVIS. \n"
                 "# If you do not write anything, JARVIS just checks if the syntax is alright. \n"
                 "###########\n"
-                "\n # Test Code:\n" +
+                "# Test Code:\n" +
                 "\n############\n"
         )
         return content
@@ -37,11 +38,11 @@ class TestCodeFileEditAtomicFlow(CodeFileEditAtomicFlow):
 
     def _check_input(self, input_data: Dict[str, Any]):
         assert "code" in input_data, "code is not passed to TestCodeFileEditAtomicFlow"
-        assert "code_library" in input_data, "code_library is not passed to TestCodeFileEditAtomicFlow"
+        assert "memory_files" in input_data, "memory_files is not passed to TestCodeFileEditAtomicFlow"
 
     def _generate_input_to_writer(self, input_data: Dict[str, Any]):
         code_str = input_data['code']
-        code_lib_location = input_data["code_library"]
+        code_lib_location = input_data["memory_files"]["code_library"]
         content_to_write = self._generate_content(code_lib_location, code_str)
         file_location_to_write = self._generate_temp_file_location(code_lib_location)
         return content_to_write, file_location_to_write

TestCodeFileEditAtomicFlow.yaml

@@ -3,7 +3,7 @@ description: "A flow that writes tests code to a temp file"
 
 input_interface:
   - "code"
-  - "code_library"
+  - "memory_files"
 
 output_interface:
   - "code_editor_output"

TestCodeFlow.yaml

@@ -9,12 +9,9 @@ input_interface:
   - "code"
 
 output_interface:
-  - "test_results"
+  - "feedback"
 
 subflows_config:
-  MemoryReading:
-    _target_: Tachi67.MemoryReadingFlowModule.MemoryReadingAtomicFlow.instantiate_from_default_config
-
   TestCodeFileEdit:
     _target_: Tachi67.TestCodeFlowModule.TestCodeFileEditAtomicFlow.instantiate_from_default_config
 
@@ -24,14 +21,6 @@ subflows_config:
 early_exit_key: "EARLY_EXIT"
 
 topology:
-  - goal: "Read in necessary memory"
-    input_interface:
-      _target_: flows.interfaces.KeyInterface
-      additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
-    flow: MemoryReading
-    reset: false
-
   - goal: "Write the code & instructions for testing to a temp file"
     input_interface:
       _target_: flows.interfaces.KeyInterface

__init__.py

@@ -1,5 +1,4 @@
 dependencies = [
-    {"url": "Tachi67/MemoryReadingFlowModule", "revision": "main"},
     {"url": "Tachi67/CodeFileEditFlowModule", "revision": "main"},
     {"url": "Tachi67/InterpreterFlowModule", "revision": "main"},
 ]