Allow keeping the original user description

This commit is contained in:
zmeir
2023-08-17 15:40:24 +03:00
parent fda98643c2
commit 31e91edebc
7 changed files with 34 additions and 19 deletions

View File

@ -82,9 +82,26 @@ class GitProvider(ABC):
pass
@abstractmethod
def get_pr_description(self):
def get_pr_description_full(self):
pass
def get_pr_description(self):
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()
if max_tokens:
return clip_tokens(description, max_tokens)
return description
def get_user_description(self):
description = (self.get_pr_description_full() or "").strip()
if not description.startswith("## PR Type"):
return description
if "## User Description:" not in description:
return ""
return description.split("## User Description:", 1)[1].strip()
@abstractmethod
def get_issue_comments(self):
pass