Apply PrReviewTitles enum for github provider file

This commit is contained in:
KennyDizi
2024-06-27 07:03:25 +07:00
parent c185b7c610
commit 2d21df61c7

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 load_large_diff, clip_tokens, find_line_number_of_relevant_line_in_file
from ..algo.utils import PrReviewTitles, 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("## PR Reviewer Guide")
prefixes.append({PrReviewTitles.REGULAR})
if incremental:
prefixes.append("## Incremental PR Reviewer Guide")
prefixes.append({PrReviewTitles.INCREMENTAL})
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]