/pr-agent

๐Ÿš€CodiumAI PR-Agent: An AI-Powered ๐Ÿค– Tool for Automated Pull Request Analysis, Feedback, Suggestions and More! ๐Ÿ’ป๐Ÿ”

Primary LanguagePythonApache License 2.0Apache-2.0

logo
CodiumAI PR-Agent aims to help efficiently review and handle pull requests, by providing AI feedbacks and suggestions

GitHub license Static Badge Discord Twitter GitHub

  • See the Installation Guide for instructions on installing PR-Agent on different platforms.

  • See the Usage Guide for instructions on running PR-Agent tools via different interfaces, such as CLI, PR Comments, or by automatically triggering them when a new PR is opened.

  • See the Tools Guide for a detailed description of the different tools, and the available configurations for each tool.

Table of Contents

News and Updates

May 23, 2024

We released a new version of PR-Agent Chrome extension, with two new features:

  • PR-Agent filters ๐ŸŽจ
  • Code suggestions interactions ๐Ÿ”—

See more here

May 21, 2024

Check out CodiumAI new project, Cover-Agent, that can automatically generate qualified tests to enhance existing test suites, aiming to increase code and behavior coverage efficiently.

May 19, 2024

GPT-4o is now the default fast model ("Turbo"). This model will be used for all commands except review and improve, which will still use "GPT-4-2024-04-09", since they are harder and would still benefit from the larger model.

May 12, 2024

Inspired by AlphaCodium flow engineering scheme, PR-Agent now performs self-reflection on the code suggestions it provides, enabling to remove invalid suggestions, and score the valid ones. The suggestions will be presented sorted by their score, enabling to focus on the most important ones first.

You can also choose to automatically remove suggestions below a certain importance score threshold, by setting the pr_code_suggestions.suggestions_score_threshold configuration.

Overview

Supported commands per platform:

GitHub Gitlab Bitbucket Azure DevOps
TOOLS Review โœ… โœ… โœ… โœ…
โฎ‘ Incremental โœ…
โฎ‘ SOC2 Compliance ๐Ÿ’Ž โœ… โœ… โœ… โœ…
Describe โœ… โœ… โœ… โœ…
โฎ‘ Inline File Summary ๐Ÿ’Ž โœ…
Improve โœ… โœ… โœ… โœ…
โฎ‘ Extended โœ… โœ… โœ… โœ…
Ask โœ… โœ… โœ… โœ…
โฎ‘ Ask on code lines โœ… โœ…
Custom Prompt ๐Ÿ’Ž โœ… โœ… โœ… โœ…
Test ๐Ÿ’Ž โœ… โœ… โœ…
Reflect and Review โœ… โœ… โœ… โœ…
Update CHANGELOG.md โœ… โœ… โœ… โœ…
Find Similar Issue โœ…
Add PR Documentation ๐Ÿ’Ž โœ… โœ… โœ…
Custom Labels ๐Ÿ’Ž โœ… โœ… โœ…
Analyze ๐Ÿ’Ž โœ… โœ… โœ…
CI Feedback ๐Ÿ’Ž โœ…
Similar Code ๐Ÿ’Ž โœ…
USAGE CLI โœ… โœ… โœ… โœ…
App / webhook โœ… โœ… โœ… โœ…
Tagging bot โœ…
Actions โœ… โœ…
CORE PR compression โœ… โœ… โœ… โœ…
Repo language prioritization โœ… โœ… โœ… โœ…
Adaptive and token-aware file patch fitting โœ… โœ… โœ… โœ…
Multiple models support โœ… โœ… โœ… โœ…
Static code analysis ๐Ÿ’Ž โœ… โœ… โœ… โœ…
Global and wiki configurations ๐Ÿ’Ž โœ… โœ… โœ… โœ…
PR interactive actions ๐Ÿ’Ž โœ…
  • ๐Ÿ’Ž means this feature is available only in PR-Agent Pro

โ€ฃ Auto Description (/describe): Automatically generating PR description - title, type, summary, code walkthrough and labels.
โ€ฃ Auto Review (/review): Adjustable feedback about the PR, possible issues, security concerns, review effort and more.
โ€ฃ Code Suggestions (/improve): Code suggestions for improving the PR.
โ€ฃ Question Answering (/ask ...): Answering free-text questions about the PR.
โ€ฃ Update Changelog (/update_changelog): Automatically updating the CHANGELOG.md file with the PR changes.
โ€ฃ Find Similar Issue (/similar_issue): Automatically retrieves and presents similar issues.
โ€ฃ Add Documentation ๐Ÿ’Ž (/add_docs): Generates documentation to methods/functions/classes that changed in the PR.
โ€ฃ Generate Custom Labels ๐Ÿ’Ž (/generate_labels): Generates custom labels for the PR, based on specific guidelines defined by the user.
โ€ฃ Analyze ๐Ÿ’Ž (/analyze): Identify code components that changed in the PR, and enables to interactively generate tests, docs, and code suggestions for each component.
โ€ฃ Custom Prompt ๐Ÿ’Ž (/custom_prompt): Automatically generates custom suggestions for improving the PR code, based on specific guidelines defined by the user.
โ€ฃ Generate Tests ๐Ÿ’Ž (/test component_name): Generates unit tests for a selected component, based on the PR code changes.
โ€ฃ CI Feedback ๐Ÿ’Ž (/checks ci_job): Automatically generates feedback and analysis for a failed CI job.
โ€ฃ Similar Code ๐Ÿ’Ž (/find_similar_component): Retrieves the most similar code components from inside the organization's codebase, or from open-source code.


Example results





Try it now

Try the GPT-4 powered PR-Agent instantly on your public GitHub repository. Just mention @CodiumAI-Agent and add the desired command in any PR comment. The agent will generate a response based on your command. For example, add a comment to any pull request with the following text:

@CodiumAI-Agent /review

and the agent will respond with a review of your PR

Review generation process

To set up your own PR-Agent, see the Installation section below. Note that when you set your own PR-Agent or use CodiumAI hosted PR-Agent, there is no need to mention @CodiumAI-Agent .... Instead, directly start with the command, e.g., /ask ....


PR-Agent Pro ๐Ÿ’Ž

PR-Agent Pro is a hosted version of PR-Agent, provided by CodiumAI. It is available for a monthly fee, and provides the following benefits:

  1. Fully managed - We take care of everything for you - hosting, models, regular updates, and more. Installation is as simple as signing up and adding the PR-Agent app to your GitHub\GitLab\BitBucket repo.
  2. Improved privacy - No data will be stored or used to train models. PR-Agent Pro will employ zero data retention, and will use an OpenAI account with zero data retention.
  3. Improved support - PR-Agent Pro users will receive priority support, and will be able to request new features and capabilities.
  4. Extra features -In addition to the benefits listed above, PR-Agent Pro will emphasize more customization, and the usage of static code analysis, in addition to LLM logic, to improve results. See here for a list of features available in PR-Agent Pro.

How it works

The following diagram illustrates PR-Agent tools and their flow:

PR-Agent Tools

Check out the PR Compression strategy page for more details on how we convert a code diff to a manageable LLM prompt

Why use PR-Agent?

A reasonable question that can be asked is: "Why use PR-Agent? What makes it stand out from existing tools?"

Here are some advantages of PR-Agent:

  • We emphasize real-life practical usage. Each tool (review, improve, ask, ...) has a single GPT-4 call, no more. We feel that this is critical for realistic team usage - obtaining an answer quickly (~30 seconds) and affordably.
  • Our PR Compression strategy is a core ability that enables to effectively tackle both short and long PRs.
  • Our JSON prompting strategy enables to have modular, customizable tools. For example, the '/review' tool categories can be controlled via the configuration file. Adding additional categories is easy and accessible.
  • We support multiple git providers (GitHub, Gitlab, Bitbucket), multiple ways to use the tool (CLI, GitHub Action, GitHub App, Docker, ...), and multiple models (GPT-4, GPT-3.5, Anthropic, Cohere, Llama2).

Data privacy

Self-hosted PR-Agent

CodiumAI-hosted PR-Agent Pro ๐Ÿ’Ž

  • When using PR-Agent Pro ๐Ÿ’Ž, hosted by CodiumAI, we will not store any of your data, nor will we use it for training. You will also benefit from an OpenAI account with zero data retention.

  • For certain clients, CodiumAI-hosted PR-Agent Pro will use CodiumAIโ€™s proprietary models โ€” if this is the case, you will be notified.

  • No passive collection of Code and Pull Requestsโ€™ data โ€” PR-Agent will be active only when you invoke it, and it will then extract and analyze only data relevant to the executed command and queried pull request.

PR-Agent Chrome extension

  • The PR-Agent Chrome extension serves solely to modify the visual appearance of a GitHub PR screen. It does not transmit any user's repo or pull request code. Code is only sent for processing when a user submits a GitHub comment that activates a PR-Agent tool, in accordance with the standard privacy policy of PR-Agent.

Links

Join our Discord community