From 16a97ade090f15fe4fdb98671860ad20d3c24dc0 Mon Sep 17 00:00:00 2001 From: "Hussam.lawen" Date: Tue, 17 Oct 2023 18:50:39 +0300 Subject: [PATCH] Add docs editable blacklist of file extensions like sql, yaml... --- pr_agent/settings/language_extensions.toml | 3 +++ pr_agent/tools/pr_add_docs.py | 5 +++++ 2 files changed, 8 insertions(+) diff --git a/pr_agent/settings/language_extensions.toml b/pr_agent/settings/language_extensions.toml index 0cc31f0ef..eadb80c8f 100644 --- a/pr_agent/settings/language_extensions.toml +++ b/pr_agent/settings/language_extensions.toml @@ -433,3 +433,6 @@ reStructuredText = [".rst", ".rest", ".rest.txt", ".rst.txt", ] wisp = [".wisp", ] xBase = [".prg", ".prw", ] +[docs_blacklist_extensions] +# Disable docs for these extensions of text files and scripts that are not programming languages of function, classes and methods +docs_blacklist = ['sql', 'txt', 'yaml', 'json', 'xml', 'md', 'rst', 'rest', 'rest.txt', 'rst.txt', 'mdpolicy', 'mdown', 'markdown', 'mdwn', 'mkd', 'mkdn', 'mkdown', 'sh'] \ No newline at end of file diff --git a/pr_agent/tools/pr_add_docs.py b/pr_agent/tools/pr_add_docs.py index c6499bb52..eec75b9cb 100644 --- a/pr_agent/tools/pr_add_docs.py +++ b/pr_agent/tools/pr_add_docs.py @@ -65,6 +65,11 @@ async def run(self): async def _prepare_prediction(self, model: str): get_logger().info('Getting PR diff...') + + # Disable adding docs to scripts and other non-relevant text files + from pr_agent.algo.language_handler import bad_extensions + bad_extensions += get_settings().docs_blacklist_extensions.docs_blacklist + self.patches_diff = get_pr_diff(self.git_provider, self.token_handler, model,