black fixes | prompt clarified to avoid method signature copying

Author: Tendo1904

osa_tool/operations/codebase/docstring_generation/docgen.py

@@ -286,7 +286,8 @@ async def generate_method_documentation(
             f"- Method Name: {method_details['method_name']}\n\n"
             "Method source code: You are given only the body of a single method, without its signature. "
             "All visible code, including any inner functions or nested logic, belongs to this single method. "
-            "Do not write separate docstrings for inner functions — they are part of the main method's logic.\n"
+            "Do NOT write separate docstrings for inner functions — they are part of the main method's logic.\n"
+            "Do NOT repeat the function signature or decorators.\n"
             "```\n"
             f"{method_details['source_code']}\n"
             "```\n\n"
@@ -368,6 +369,7 @@ async def update_method_documentation(
             f"The main idea of the project (for context only): {self.main_idea}\n\n"
             "Return only the updated docstring.\n"
             "DO NOT return code.\n"
+            "Do NOT repeat the function signature or decorators.\n"
             "DO NOT return other documentation sections.\n"
             "Return only the docstring without any quotation marks.\n"
         )

osa_tool/operations/codebase/docstring_generation/docstring_transformer.py

@@ -7,7 +7,6 @@
     IndentedBlock,
 )
 from typing import Sequence
-import json
 
 
 class DocstringTransformer(CSTTransformer):

tests/unit/osatreesitter/test_docgen.py

@@ -260,7 +260,7 @@ async def test_generate_method_documentation(mock_config_manager):
     docstring = await docgen.generate_method_documentation(method_details, semaphore)
 
     # Assert
-    assert docstring == 'Generated docstring'
+    assert docstring == "Generated docstring"
     docgen.model_handler.async_request.assert_called_once()
 
 
@@ -285,7 +285,7 @@ async def test_update_method_documentation(mock_config_manager):
     updated_doc = await docgen.update_method_documentation(method_details, semaphore)
 
     # Assert
-    assert updated_doc == 'Updated docstring'
+    assert updated_doc == "Updated docstring"
     docgen.model_handler.async_request.assert_called_once()