Fix formatting

Author: Ekarerina-Ulyanova

osa_tool/operations/codebase/docstring_generation/docgen.py

@@ -524,9 +524,7 @@ def _get_method_generation_prompt_small(self, method_details: dict, context_code
                 if context_code
                 else ""
             ),
-            constructor_rules=(
-                self._render_prompt("method_generation_small_constructor") if is_constructor else ""
-            ),
+            constructor_rules=(self._render_prompt("method_generation_small_constructor") if is_constructor else ""),
         )
 
     async def update_method_documentation(
@@ -543,9 +541,7 @@ async def update_method_documentation(
         docstring = method_details["docstring"]
 
         if language in ("javascript", "typescript"):
-            prompt = self._get_method_update_prompt_large(
-                method_details, docstring, context_code, class_name, language
-            )
+            prompt = self._get_method_update_prompt_large(method_details, docstring, context_code, class_name, language)
         elif self.is_small_model:
             prompt = self._get_method_update_prompt_small(method_details, docstring, context_code, class_name)
         else:
@@ -606,11 +602,7 @@ def _get_method_update_prompt_large(
             class_location=f" (located inside {class_name} class)" if class_name else "",
             decorators=method_details["decorators"],
             source_code=method_details["source_code"],
-            context=(
-                self._render_prompt("method_update_standard_context", context_code=context_code)
-                if context_code
-                else ""
-            ),
+            context=(self._render_prompt("method_update_standard_context", context_code=context_code) if context_code else ""),
             main_idea=self.main_idea,
         )
 
@@ -629,11 +621,7 @@ def _get_method_update_prompt_small(
             constructor_rules=self._render_prompt(
                 "method_update_small_constructor" if is_constructor else "method_update_small_non_constructor"
             ),
-            context=(
-                self._render_prompt("method_update_small_context", context_code=context_code)
-                if context_code
-                else ""
-            ),
+            context=(self._render_prompt("method_update_small_context", context_code=context_code) if context_code else ""),
             main_idea=self.main_idea,
         )
 
@@ -709,8 +697,10 @@ def clean_docstring(docstring: str) -> str:
                 while next_index < len(lines) and not lines[next_index].strip():
                     next_index += 1
                 next_line = lines[next_index].strip().lower() if next_index < len(lines) else ""
-                if not next_line or next_line == "none" or re.match(
-                    r"^(Args|Returns|Raises|Attributes):", next_line, re.IGNORECASE
+                if (
+                    not next_line
+                    or next_line == "none"
+                    or re.match(r"^(Args|Returns|Raises|Attributes):", next_line, re.IGNORECASE)
                 ):
                     index = next_index + (next_line == "none")
                     continue
@@ -1458,12 +1448,14 @@ async def generate_the_main_idea(self, parsed_structure: dict, top_n: int = 5) -
                 else:
                     docstring = component["details"]["docstring"] if component["details"]["docstring"] else ""
 
-                prompt_structure.append(f"""
+                prompt_structure.append(
+                    f"""
                     {_type.capitalize()} name: {component["name"] if _type == "class" else component["details"]["method_name"]}
                     Component description: {docstring}
                     Component place in hierarchy: {file}
                     Component importance score: {score}
-                    """)
+                    """
+                )
 
         logger.info(f"Generating the main idea of the project...")
 
@@ -1492,7 +1484,6 @@ async def summarize_submodules(self, project_structure: dict[str, Any], rate_lim
 
         semaphore = asyncio.Semaphore(rate_limit)
 
-
         _summaries = {}
 
         async def summarize_directory(name: str, file_summaries: List[str], submodule_summaries: List[str]) -> str:

tests/unit/operations/codebase/docstring_generation/test_docgen.py

@@ -563,7 +563,7 @@ def test_extract_pure_docstring_preserves_escaped_triple_quotes_in_content(mock_
 
     result = docgen.extract_pure_docstring('"""Return \\"\\"\\" as the delimiter."""')
 
-    assert r'\"\"\"' in result
+    assert r"\"\"\"" in result
     assert "as the delimiter." in result