Added fix for supercoder (TransformerOptimus#521)

* Added logic for requirements and dependencies

* Added generate logic prompt

* Refactored pytest for unittest

---------

Co-authored-by: COLONAYUSH <ayushkaps9462@gmail.com>

superagi/tools/code/prompts/generate_logic.txt

@@ -0,0 +1,11 @@
+You typically always place distinct classes in separate files.
+Always create a run.sh file which act as the entrypoint of the program, create it intellligently after analyzing the file types
+For Python, always generate a suitable requirements.txt file.
+For NodeJS, consistently produce an appropriate package.json file.
+Always include a brief comment that describes the purpose of the function definition.
+Attempt to provide comments that explain complicated logic.
+Consistently adhere to best practices for the specified languages, ensuring code is defined as a package or project.
+
+Preferred Python toolbelt:
+- pytest
+- dataclasses

superagi/tools/code/write_code.py

@@ -61,7 +61,7 @@ def _execute(self, code_description: str) -> str:
         Returns:
             Generated codes files or error message.
         """
-        prompt = PromptReader.read_tools_prompt(__file__, "write_code.txt")
+        prompt = PromptReader.read_tools_prompt(__file__, "write_code.txt") + "\nUseful to know:\n" + PromptReader.read_tools_prompt(__file__, "generate_logic.txt")
         prompt = prompt.replace("{goals}", AgentPromptBuilder.add_list_items_to_string(self.goals))
         prompt = prompt.replace("{code_description}", code_description)
         spec_response = self.tool_response_manager.get_last_response("WriteSpecTool")

superagi/tools/code/write_test.py

@@ -26,7 +26,7 @@ class WriteTestSchema(BaseModel):
 
 class WriteTestTool(BaseTool):
     """
-    Used to generate pytest unit tests based on the specification.
+    Used to generate unit tests based on the specification.
 
     Attributes:
         llm: LLM used for test generation.
@@ -62,7 +62,7 @@ def _execute(self, test_description: str, test_file_name: str) -> str:
             test_file_name: The name of the file where the generated tests will be saved.
 
         Returns:
-            Generated pytest unit tests or error message.
+            Generated unit tests or error message.
         """
         prompt = PromptReader.read_tools_prompt(__file__, "write_test.txt")
         prompt = prompt.replace("{goals}", AgentPromptBuilder.add_list_items_to_string(self.goals))