diff --git a/pr_agent/git_providers/git_provider.py b/pr_agent/git_providers/git_provider.py index 9dced773..4651d47d 100644 --- a/pr_agent/git_providers/git_provider.py +++ b/pr_agent/git_providers/git_provider.py @@ -86,11 +86,11 @@ class GitProvider(ABC): def get_pr_description_full(self) -> str: pass - def get_pr_description(self) -> str: + def get_pr_description(self, *, full: bool = True) -> str: from pr_agent.config_loader import get_settings from pr_agent.algo.pr_processing import clip_tokens max_tokens = get_settings().get("CONFIG.MAX_DESCRIPTION_TOKENS", None) - description = self.get_pr_description_full() + description = self.get_pr_description_full() if full else self.get_user_description() if max_tokens: return clip_tokens(description, max_tokens) return description diff --git a/pr_agent/tools/pr_description.py b/pr_agent/tools/pr_description.py index acd272bc..c45917f4 100644 --- a/pr_agent/tools/pr_description.py +++ b/pr_agent/tools/pr_description.py @@ -36,7 +36,7 @@ class PRDescription: self.vars = { "title": self.git_provider.pr.title, "branch": self.git_provider.get_pr_branch(), - "description": self.git_provider.get_pr_description(), + "description": self.git_provider.get_pr_description(full=False), "language": self.main_pr_language, "diff": "", # empty diff for initial calculation "extra_instructions": get_settings().pr_description.extra_instructions,