diff --git a/pr_agent/tools/pr_add_docs.py b/pr_agent/tools/pr_add_docs.py index b8a9c53a..fbcfe3c0 100644 --- a/pr_agent/tools/pr_add_docs.py +++ b/pr_agent/tools/pr_add_docs.py @@ -13,19 +13,6 @@ from pr_agent.git_providers import BitbucketProvider, get_git_provider from pr_agent.git_providers.git_provider import get_main_pr_language -def get_docs_for_language(language): - if language.lower() == 'java': - return "javadocs" - elif language.lower() in ['python', 'lisp', 'clojure']: - return "docstrings" - elif language.lower() in ['javascript', 'typescript']: - return "jsdocs" - elif language.lower() == 'c++': - return "doxygen" - else: - return "docs" - - class PRAddDocs: def __init__(self, pr_url: str, cli_mode=False, args: list = None): @@ -129,12 +116,13 @@ class PRAddDocs: relevant_line = int(d['relevant line']) # absolute position documentation = d['documentation'] if documentation: - new_code_snippet = self.dedent_code(relevant_file, relevant_line, documentation, add_original_line=True) + new_code_snippet = self.dedent_code(relevant_file, relevant_line, documentation, + add_original_line=True) body = f"**Suggestion:** Proposed documentation\n```suggestion\n" + new_code_snippet + "\n```" code_suggestions.append({'body': body, 'relevant_file': relevant_file, - 'relevant_lines_start': relevant_line, - 'relevant_lines_end': relevant_line}) + 'relevant_lines_start': relevant_line, + 'relevant_lines_end': relevant_line}) except Exception: if get_settings().config.verbosity_level >= 2: logging.info(f"Could not parse code docs: {d}") @@ -192,4 +180,18 @@ class PRAddDocs: else: data.update(data_per_chunk) self.data = data - return data \ No newline at end of file + return data + + +def get_docs_for_language(language): + language = language.lower() + if language == 'java': + return "javadocs" + elif language in ['python', 'lisp', 'clojure']: + return "docstrings" + elif language in ['javascript', 'typescript']: + return "jsdocs" + elif language == 'c++': + return "doxygen" + else: + return "docs"