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

Added type hints

Browse Source
This commit is contained in:
zmeir
2023-08-21 09:22:58 +03:00
parent b3749d08e2
commit 81c38f9646
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pr_agent/git_providers/git_provider.py
View File

@ -82,10 +82,10 @@ class GitProvider(ABC):
pass pass
@abstractmethod @abstractmethod
def get_pr_description_full(self): def get_pr_description_full(self) -> str:
pass pass
def get_pr_description(self): def get_pr_description(self) -> str:
from pr_agent.config_loader import get_settings from pr_agent.config_loader import get_settings
from pr_agent.algo.pr_processing import clip_tokens from pr_agent.algo.pr_processing import clip_tokens
max_tokens = get_settings().get("CONFIG.MAX_DESCRIPTION_TOKENS", None) max_tokens = get_settings().get("CONFIG.MAX_DESCRIPTION_TOKENS", None)
@ -94,7 +94,7 @@ class GitProvider(ABC):
return clip_tokens(description, max_tokens) return clip_tokens(description, max_tokens)
return description return description
def get_user_description(self): def get_user_description(self) -> str:
description = (self.get_pr_description_full() or "").strip() description = (self.get_pr_description_full() or "").strip()
if not description.startswith("## PR Type"): if not description.startswith("## PR Type"):
return description return description