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