From 81c38f9646c43a068e7618c25723ce3d4fa39ec0 Mon Sep 17 00:00:00 2001 From: zmeir Date: Mon, 21 Aug 2023 09:22:58 +0300 Subject: [PATCH] Added type hints --- pr_agent/git_providers/git_provider.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pr_agent/git_providers/git_provider.py b/pr_agent/git_providers/git_provider.py index e7796d1e..9db0a5bd 100644 --- a/pr_agent/git_providers/git_provider.py +++ b/pr_agent/git_providers/git_provider.py @@ -82,10 +82,10 @@ class GitProvider(ABC): pass @abstractmethod - def get_pr_description_full(self): + def get_pr_description_full(self) -> str: pass - def get_pr_description(self): + def get_pr_description(self) -> 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) @@ -94,7 +94,7 @@ class GitProvider(ABC): return clip_tokens(description, max_tokens) return description - def get_user_description(self): + def get_user_description(self) -> str: description = (self.get_pr_description_full() or "").strip() if not description.startswith("## PR Type"): return description