Update CodeGeneratorAtomicFlow.py

CodeGeneratorAtomicFlow.py

@@ -6,7 +6,23 @@ from flow_modules.aiflows.ChatFlowModule import ChatAtomicFlow
 
 
 class CodeGeneratorAtomicFlow(ChatAtomicFlow):
-    """Generates one function with docstrings to finish the given goal (from the controller).
+    """This class wraps around the Chat API to generate code from a goal. One thing worth noting is that we need to
+    make sure the code generator does not write repetitive code that is present in the library, so we need to inject
+    the function signatures in the library to the system prompts.
+
+    *Input Interface Non Initialized*:
+    - `goal`
+    - `code_library`
+    - `memory_files`
+
+    *Input Interface Initialized*:
+    - `goal`
+    - `code_library`
+    - `memory_files`
+
+    *Output Interface*:
+    - `code`
+    - `language_of_code`
     """
     def __init__(self, **kwargs):
         super().__init__(**kwargs)