Change the data structure for prefixes to a list to preserve order

This commit is contained in:
KennyDizi
2024-06-27 07:16:26 +07:00
parent 692904bb71
commit b20f364b15

View File

@ -10,7 +10,7 @@ from starlette_context import context
from ..algo.file_filter import filter_ignored
from ..algo.language_handler import is_valid_file
from ..algo.utils import ReviewHeaderTitle, load_large_diff, clip_tokens, find_line_number_of_relevant_line_in_file
from ..algo.utils import PRReviewHeader, load_large_diff, clip_tokens, find_line_number_of_relevant_line_in_file
from ..config_loader import get_settings
from ..log import get_logger
from ..servers.utils import RateLimitExceeded
@ -96,9 +96,9 @@ class GithubProvider(GitProvider):
self.comments = list(self.pr.get_issue_comments())
prefixes = []
if full:
prefixes.append({ReviewHeaderTitle.REGULAR.value})
prefixes.append(PRReviewHeader.REGULAR.value)
if incremental:
prefixes.append({ReviewHeaderTitle.INCREMENTAL.value})
prefixes.append(PRReviewHeader.INCREMENTAL.value)
for index in range(len(self.comments) - 1, -1, -1):
if any(self.comments[index].body.startswith(prefix) for prefix in prefixes):
return self.comments[index]