Skip to content

Automate code reviews, patching and documentation with LLM workflows.

License

Notifications You must be signed in to change notification settings

patched-codes/patchwork

Repository files navigation

Patchwork logo

⭐ As part of Github Constellation community, Patched is contributing INR 100 (USD 1.25) to open-source projects for every new star the patchwork repository receives in June. Join us in giving back to the open source community that we have all benefited from!

PatchWork

An open-source framework for automating development chores using large language models. PatchWork allows you to automate workflows like PR reviews, bug fixing, security patching, and more using a self-hosted CLI agent and your preferred LLMs.

Key Components

  • Steps: Reusable atomic actions like Create PR, commit changes, call an LLM etc
  • Prompt Templates: Customizable LLM prompts optimized for a chore like library updates, code generation, issue analysis etc.
  • Patchflows: LLM-assisted automations such as PR reviews, code fixing, documentation etc. built by combining steps and prompts.

Patchflows can be run locally in your CLI and IDE, or as part of your CI/CD pipeline. There are several patchflows available out of the box, and you can always create your own.

Quickstart

Patchwork CLI Quickstart

Installation

Using Pip

PatchWork is available on PyPI and can be installed using pip:

pip install patchwork-cli --upgrade

Using Poetry

If you'd like to build from source using poetry, please see detailed documentation here .

PatchWork CLI

The CLI runs Patchflows, as follows:

patchwork <PatchFlow> <?Arguments>

Where

  • Arguments: Allow for overriding default/optional attributes of the Patchflow in the format of key=value. If key does not have any value, it is considered a boolean True flag.

Example

For an AutoFix patchflow which patches vulnerabilities based on a scan using Semgrep:

patchwork AutoFix openai_api_key=<YOUR_OPENAI_API_KEY> github_api_key=<YOUR_GITHUB_TOKEN>

The above command will default to patching code in the current directory, by running Semgrep to identify the vulnerabilities. You can take a look at the default.yml file for the list of configurations you can set to manage the AutoFix patchflow.

You can replace the OpenAI key with a key from our managed service by signing in at https://app.patched.codes/signin and generating an API key from the integrations tab. You can then call the patchflow with the key as follows:

patchwork AutoFix patched_api_key=<YOUR_PATCHED_API_KEY> github_api_key=<YOUR_GITHUB_TOKEN>

Similarly, to use Google's models you can set the google_api_key and model, this is useful if you want to work with large contexts as the gemini-pro-1.5 model supports an input context length of 1 million tokens.

The patchwork-template repository contains the default configuration and prompts for all the patchflows. You can clone that repo and pass it as a flag to the CLI:

patchwork AutoFix --config /path/to/patchwork-configs/patchflows

Patchflows

Patchwork comes with a set of predefined patchflows, and more will be added over time. Below is a sample list of patchflows:

  • GenerateDocstring: Generate docstrings for methods in your code.
  • AutoFix: Generate and apply fixes to code vulnerabilities in a repository.
  • PRReview: On PR creation, extract code diff, summarize changes, and comment on PR.
  • GenerateREADME: Create a README markdown file for a given folder, to add documentation to your repository.
  • [Experimental] DependencyUpgrade: Update your dependencies from vulnerable to fixed versions.
  • [Experimental] ResolveIssue: Identify the files in your repository that need to be updated to resolve an issue (or bug) and create a PR to fix it.

Prompt Templates

Prompt templates are used by patchflows and passed as queries to LLMs. Templates contain prompts with placeholder variables enclosed by {{}} which are replaced by the data from the steps or inputs on every run.

Below is a sample prompt template:

{
  "id": "diffreview_summary",
    "prompts": [
      {
        "role": "user",
        "content": "Summarize the following code change descriptions in 1 paragraph. {{diffreviews}}"
      }
    ]
}

Each patchflow comes with an optimized default prompt template. But you can specify your own using the prompt_template_file=/path/to/prompt/template/file option.

Contributing

Contributions for new patchflows and steps, or even to the core framework are welcome. Please look at open issues for details.

We also provide chat assistants to help you create new steps and patchflows easily. Fair warning: they suffer from the same limitations as their underlying model.

Roadmap

Short Term Q2 '24

  • Expand patchflow library and integration options
  • Patchflow debugger and validation module
  • Bug fixing and performance improvements
  • Refactor code and documentation

Long Term Q3/Q4 '24

  • Support large-scale code embeddings in patchflows
  • Support parallelization and branching
  • Fine-tuned models that can be self-hosted
  • Open-source GUI

License

Patchwork is licensed under AGPL-3.0 terms. However, custom patchflows and steps can be created and shared using the patchwork template repository which is licensed under Apache-2.0 terms.