xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-15 10:10:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use ReviewHeaderTitle in lieu of PrReviewTitle

Browse Source
This commit is contained in:
KennyDizi
2024-06-27 07:11:57 +07:00
parent ba963149ac
commit 692904bb71
4 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pr_agent/algo/utils.py
View File

@ -23,7 +23,7 @@ class ModelType(str, Enum):
REGULAR = "regular"
TURBO = "turbo"
class PrReviewTitle(str, Enum):
class ReviewHeaderTitle(str, Enum):
REGULAR = "## PR Reviewer Guide"
INCREMENTAL = "## Incremental PR Reviewer Guide"
@ -90,9 +90,9 @@ def convert_to_markdown(output_data: dict, gfm_supported: bool = True, increment
}
markdown_text = ""
if not incremental_review:
markdown_text += f"{PrReviewTitle.REGULAR.value} 🔍\n\n"
markdown_text += f"{ReviewHeaderTitle.REGULAR.value} 🔍\n\n"
else:
markdown_text += f"{PrReviewTitle.INCREMENTAL.value} 🔍\n\n"
markdown_text += f"{ReviewHeaderTitle.INCREMENTAL.value} 🔍\n\n"
markdown_text += f"⏮️ Review for commits since previous PR-Agent review {incremental_review}.\n\n"
if gfm_supported:
markdown_text += "<table>\n"

6
pr_agent/git_providers/github_provider.py
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 PrReviewTitle, load_large_diff, clip_tokens, find_line_number_of_relevant_line_in_file
from ..algo.utils import ReviewHeaderTitle, 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({PrReviewTitle.REGULAR.value})
prefixes.append({ReviewHeaderTitle.REGULAR.value})
if incremental:
prefixes.append({PrReviewTitle.INCREMENTAL.value})
prefixes.append({ReviewHeaderTitle.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]

4
pr_agent/tools/pr_reviewer.py
View File

@ -8,7 +8,7 @@ from pr_agent.algo.ai_handlers.base_ai_handler import BaseAiHandler
from pr_agent.algo.ai_handlers.litellm_ai_handler import LiteLLMAIHandler
from pr_agent.algo.pr_processing import get_pr_diff, retry_with_fallback_models
from pr_agent.algo.token_handler import TokenHandler
from pr_agent.algo.utils import PrReviewTitle, convert_to_markdown, github_action_output, load_yaml, ModelType, \
from pr_agent.algo.utils import ReviewHeaderTitle, convert_to_markdown, github_action_output, load_yaml, ModelType, \
show_relevant_configurations
from pr_agent.config_loader import get_settings
from pr_agent.git_providers import get_git_provider, get_git_provider_with_context
@ -134,7 +134,7 @@ class PRReviewer:
if get_settings().pr_reviewer.persistent_comment and not self.incremental.is_incremental:
final_update_message = get_settings().pr_reviewer.final_update_message
self.git_provider.publish_persistent_comment(pr_review,
initial_header=f"{PrReviewTitle.REGULAR.value} 🔍",
initial_header=f"{ReviewHeaderTitle.REGULAR.value} 🔍",
update_header=True,
final_update_message=final_update_message, )
else: