- made inclusion of "import" statements optional;

src/coverup/prompt.py

@@ -213,7 +213,7 @@ def initial_prompt(self, segment: CodeSegment) -> T.List[dict]:
 calling into pytest.main or the test itself.
 Respond ONLY with the Python code enclosed in backticks, without any explanation.
 ```python
-{segment.get_excerpt(tag_lines=False)}
+{segment.get_excerpt(tag_lines=False, include_imports=False)}
 ```
 """)
         ]

src/coverup/segment.py

@@ -37,13 +37,14 @@ def identify(self) -> str:
     def __str__(self) -> str:
         return self.identify()
 
-    def get_excerpt(self, tag_lines=True):
+    def get_excerpt(self, tag_lines=True, include_imports=True):
         excerpt = []
         with open(self.filename, "r") as src:
             code = src.readlines()
 
-            for imp in self.imports:
-                excerpt.extend([f"{'':10}  {imp}\n"])
+            if include_imports:
+                for imp in self.imports:
+                    excerpt.extend([f"{'':10}  {imp}\n"])
 
             for b, e in self.context:
                 for i in range(b, e):