Update utils.py

utils.py

@@ -6,7 +6,6 @@ import os
 def parse_file_content(string: str) -> tuple[str | None, str | None]:
     """
     Parses the content of a file and returns the action and description.
-
     :param string: The content of a file.
     :return: A tuple containing the action and description.
     """
@@ -25,7 +24,6 @@ def parse_file_content(string: str) -> tuple[str | None, str | None]:
 def parse_action(string: str) -> tuple[str, str | None]:
     """
     Parses the action from a string.
-
     :param string: The string to parse the action from.
     :return: A tuple containing the action and action input.
     """
@@ -39,7 +37,6 @@ def parse_action(string: str) -> tuple[str, str | None]:
 def extract_imports(file_contents: str) -> tuple[list[str], list[ast.FunctionDef], list[ast.ClassDef]]:
     """
     Extracts imports, functions, and classes from a file's contents.
-
     :param file_contents: The contents of a file.
     :return: A tuple containing the imports, functions, and classes.
     """
@@ -67,7 +64,6 @@ def extract_imports(file_contents: str) -> tuple[list[str], list[ast.FunctionDef
 def read_python_module_structure(path: str) -> tuple[str, dict[str, str], dict[str, list[str]]]:
     """
     Reads the structure of a Python module and returns a prompt, content, and internal imports map.
-
     :param path: The path to the Python module.
     :return: A tuple containing the structure prompt, content, and internal imports map.
     """
@@ -102,4 +98,6 @@ def read_python_module_structure(path: str) -> tuple[str, dict[str, str], dict[s
 
         structure_prompt += f"{fn}\n"
         for function in functions:
-            structure_prompt += "  {}({})
+            structure_prompt += f"  {function.name}({function.args})\n"
+
+    return structure_prompt, content, internal_imports_map