fix: update AI comment extraction to include full match

This commit is contained in:
Paul Gauthier 2024-10-25 16:05:50 -07:00 committed by Paul Gauthier (aider)
parent 77f636f949
commit 3d8ec25a33

View file

@ -135,7 +135,6 @@ def watch_source_files(
result[file] = comments
yield result
def get_ai_comment(filepath, encoding="utf-8"):
"""Extract all AI comments from a file"""
comments = []
@ -143,6 +142,7 @@ def get_ai_comment(filepath, encoding="utf-8"):
with open(filepath, encoding=encoding, errors="ignore") as f:
for line in f:
if match := re.search(r"(?:#|//) *ai\b(.*)", line, re.IGNORECASE):
#ai return the whole thing, including the group0
comment = match.group(1).strip()
if comment:
comments.append(comment)