move enable custom labels to custom labels function

rhyst · Oct 27, 2023 · f53ec1d · f53ec1d
1 parent a7cb59c
commit f53ec1d
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 4 deletions.
diff --git a/pr_agent/settings/configuration.toml b/pr_agent/settings/configuration.toml
@@ -37,7 +37,6 @@ extra_instructions = ""
 # markers
 use_description_markers=false
 include_generated_by_header=true
-enable_custom_labels=false
 
 #custom_labels = ['Bug fix', 'Tests', 'Bug fix with tests', 'Refactoring', 'Enhancement', 'Documentation', 'Other']
 

diff --git a/pr_agent/settings/custom_labels.toml b/pr_agent/settings/custom_labels.toml
@@ -1,3 +1,5 @@
+enable_custom_labels=true
+
 [custom_labels."Bug fix"]
 description = "Fixes a bug in the code"
 [custom_labels."Tests"]

diff --git a/pr_agent/tools/pr_description.py b/pr_agent/tools/pr_description.py
@@ -43,7 +43,7 @@ def __init__(self, pr_url: str, args: list = None):
             "use_bullet_points": get_settings().pr_description.use_bullet_points,
             "extra_instructions": get_settings().pr_description.extra_instructions,
             "commit_messages_str": self.git_provider.get_commit_messages(),
-            "enable_custom_labels": get_settings().pr_description.enable_custom_labels,
+            "enable_custom_labels": get_settings().enable_custom_labels,
             "custom_labels": "",
             "custom_labels_examples": "",
         }

diff --git a/pr_agent/tools/pr_generate_labels.py b/pr_agent/tools/pr_generate_labels.py
@@ -45,7 +45,7 @@ def __init__(self, pr_url: str, args: list = None):
             "commit_messages_str": self.git_provider.get_commit_messages(),
             "custom_labels": "",
             "custom_labels_examples": "",
-            "enable_custom_labels": get_settings().pr_description.enable_custom_labels,
+            "enable_custom_labels": get_settings().enable_custom_labels,
         }
 
         # Initialize the token handler

diff --git a/pr_agent/tools/pr_reviewer.py b/pr_agent/tools/pr_reviewer.py
@@ -63,7 +63,8 @@ def __init__(self, pr_url: str, is_answer: bool = False, is_auto: bool = False,
             'answer_str': answer_str,
             "extra_instructions": get_settings().pr_reviewer.extra_instructions,
             "commit_messages_str": self.git_provider.get_commit_messages(),
-            "custom_labels": ""
+            "custom_labels": "",
+            "enable_custom_labels": get_settings().enable_custom_labels,
         }
 
         self.token_handler = TokenHandler(