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

feat: Implement label case conversion and update label descriptions in settings files

Browse Source
This commit is contained in:
mrT23
2023-12-18 12:29:06 +02:00
parent 54891ad1d2
commit 1c4e64333c
5 changed files with 37 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pr_agent/tools/pr_generate_labels.py
View File

@ -135,6 +135,7 @@ class PRGenerateLabels:
environment = Environment(undefined=StrictUndefined)
set_custom_labels(variables, self.git_provider)
self.variables = variables
system_prompt = environment.from_string(get_settings().pr_custom_labels_prompt.system).render(variables)
user_prompt = environment.from_string(get_settings().pr_custom_labels_prompt.user).render(variables)
@ -170,4 +171,14 @@ class PRGenerateLabels:
elif type(self.data['labels']) == str:
pr_types = self.data['labels'].split(',')
# convert lowercase labels to original case
try:
if "labels_minimal_to_labels_dict" in self.variables:
d: dict = self.variables["labels_minimal_to_labels_dict"]
for i, label_i in enumerate(pr_types):
if label_i in d:
pr_types[i] = d[label_i]
except Exception as e:
get_logger().error(f"Error converting labels to original case {self.pr_id}: {e}")
return pr_types