From 2d21df61c703d6c3cc235db85edb2f91543cbae9 Mon Sep 17 00:00:00 2001 From: KennyDizi Date: Thu, 27 Jun 2024 07:03:25 +0700 Subject: [PATCH] Apply PrReviewTitles enum for github provider file --- pr_agent/git_providers/github_provider.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pr_agent/git_providers/github_provider.py b/pr_agent/git_providers/github_provider.py index fc87c84d..f69648b6 100644 --- a/pr_agent/git_providers/github_provider.py +++ b/pr_agent/git_providers/github_provider.py @@ -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]