Merge pull request Codium-ai#385 from Codium-ai/hl/fix_add_docs_in_sc…

…ripts Add Blacklist for Non-Editable File Extensions in Documentation
yochail · Oct 19, 2023 · 086501b · 086501b
2 parents bca293e + 16a97ad
commit 086501b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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']
diff --git 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,