diff --git a/pr_agent/settings/pr_code_suggestions_prompts.toml b/pr_agent/settings/pr_code_suggestions_prompts.toml index 2fb224c7..ee311054 100644 --- a/pr_agent/settings/pr_code_suggestions_prompts.toml +++ b/pr_agent/settings/pr_code_suggestions_prompts.toml @@ -51,6 +51,7 @@ The output must be a YAML object equivalent to type $PRCodeSuggestions, accordin ===== class CodeSuggestion(BaseModel): relevant_file: str = Field(description="the relevant file full path") + language: str = Field(description="the code language of the relevant file") suggestion_content: str = Field(description="an actionable suggestion for meaningfully improving the new code introduced in the PR") {%- if summarize_mode %} existing_code: str = Field(description="a short code snippet from a '__new hunk__' section to illustrate the relevant existing code. Don't show the line numbers.") @@ -74,6 +75,8 @@ Example output: code_suggestions: - relevant_file: |- src/file1.py + language: |- + python suggestion_content: |- Add a docstring to func1() {%- if summarize_mode %} @@ -105,11 +108,6 @@ user="""PR Info: Title: '{{title}}' -{%- if language %} - -Main PR language: '{{ language }}' -{%- endif %} - The PR Diff: ====== diff --git a/pr_agent/settings/pr_description_prompts.toml b/pr_agent/settings/pr_description_prompts.toml index b9c5ce39..b36b0183 100644 --- a/pr_agent/settings/pr_description_prompts.toml +++ b/pr_agent/settings/pr_description_prompts.toml @@ -39,6 +39,7 @@ class PRType(str, Enum): Class FileDescription(BaseModel): filename: str = Field(description="the relevant file full path") + language: str = Field(description="the relevant file language") changes_summary: str = Field(description="concise summary of the changes in the relevant file, in bullet points (1-4 bullet points).") changes_title: str = Field(description="an informative title for the changes in the files, describing its main theme (5-10 words).") label: str = Field(description="a single semantic label that represents a type of code changes that occurred in the File. Possible values (partial list): 'bug fix', 'tests', 'enhancement', 'documentation', 'error handling', 'configuration changes', 'dependencies', 'formatting', 'miscellaneous', ...") @@ -67,6 +68,8 @@ type: pr_files: - filename: | ... + language: | + ... changes_summary: | ... changes_title: | @@ -104,10 +107,7 @@ Previous description: {%- endif %} Branch: '{{branch}}' -{%- if language %} -Main PR language: '{{ language }}' -{%- endif %} {%- if commit_messages_str %} Commit messages: diff --git a/pr_agent/settings/pr_reviewer_prompts.toml b/pr_agent/settings/pr_reviewer_prompts.toml index 736fb247..5312832c 100644 --- a/pr_agent/settings/pr_reviewer_prompts.toml +++ b/pr_agent/settings/pr_reviewer_prompts.toml @@ -115,6 +115,9 @@ PR Feedback: relevant file: type: string description: the relevant file full path + language: + type: string + description: the language of the relevant file suggestion: type: string description: |- @@ -166,6 +169,8 @@ PR Feedback: Code feedback: - relevant file: |- directory/xxx.py + language: |- + python suggestion: |- xxx [important] relevant line: |- @@ -195,10 +200,6 @@ Description: ====== {%- endif %} -{%- if language %} - -Main PR language: '{{ language }}' -{%- endif %} {%- if commit_messages_str %} Commit messages: