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

Update pr_description.py

Browse Source
This commit is contained in:
Kamakura
2024-06-04 11:19:18 +08:00
parent 86d901d5a6
commit b980168e75
1 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pr_agent/tools/pr_description.py
View File

@ -43,7 +43,6 @@ class PRDescription:
self.ai_handler = ai_handler() self.ai_handler = ai_handler()
self.ai_handler.main_pr_language = self.main_pr_language self.ai_handler.main_pr_language = self.main_pr_language
# Initialize the variables dictionary # Initialize the variables dictionary
self.vars = { self.vars = {
"title": self.git_provider.pr.title, "title": self.git_provider.pr.title,
@ -157,7 +156,7 @@ class PRDescription:
self.git_provider.remove_initial_comment() self.git_provider.remove_initial_comment()
except Exception as e: except Exception as e:
get_logger().error(f"Error generating PR description {self.pr_id}: {e}") get_logger().error(f"Error generating PR description {self.pr_id}: {e}")
return "" return ""
async def _prepare_prediction(self, model: str) -> None: async def _prepare_prediction(self, model: str) -> None:
@ -221,9 +220,6 @@ class PRDescription:
if 'pr_files' in self.data: if 'pr_files' in self.data:
self.data['pr_files'] = self.data.pop('pr_files') self.data['pr_files'] = self.data.pop('pr_files')
def _prepare_labels(self) -> List[str]: def _prepare_labels(self) -> List[str]:
pr_types = [] pr_types = []
@ -321,7 +317,7 @@ class PRDescription:
value = self.file_label_dict value = self.file_label_dict
else: else:
key_publish = key.rstrip(':').replace("_", " ").capitalize() key_publish = key.rstrip(':').replace("_", " ").capitalize()
if key_publish== "Type": if key_publish == "Type":
key_publish = "PR Type" key_publish = "PR Type"
# elif key_publish == "Description": # elif key_publish == "Description":
# key_publish = "PR Description" # key_publish = "PR Description"
@ -512,6 +508,7 @@ def insert_br_after_x_chars(text, x=70):
is_inside_code = False is_inside_code = False
return ''.join(new_text).strip() return ''.join(new_text).strip()
def replace_code_tags(text): def replace_code_tags(text):
""" """
Replace odd instances of ` with <code> and even instances of ` with </code> Replace odd instances of ` with <code> and even instances of ` with </code>
@ -519,4 +516,4 @@ def replace_code_tags(text):
parts = text.split('`') parts = text.split('`')
for i in range(1, len(parts), 2): for i in range(1, len(parts), 2):
parts[i] = '<code>' + parts[i] + '</code>' parts[i] = '<code>' + parts[i] + '</code>'
return ''.join(parts) return ''.join(parts)