⚠️ The Isssue label bot is no longer live. You can still view the code, but we have taken down the bot because of related infrastructure costs
"How to automate tasks on GitHub with machine learning for fun and profit"
Code for:Table of Contents
Original Authors: @hamelsmu, @inc0, @jlewi
Issue Label Bot
Install this app from the GitHub marketplace
A GitHub App powered by machine learning, written in Python. A discussion of the motivation for building this app is described in this blog post.
When an issue is opened, the bot predicts if the label should be a: feature request
, bug
or question
and applies a label automatically if appropriate. Here is a screenshot of the bot in action:
More examples can be viewed on our app's homepage. It should be noted that the bot may not apply any label in circumstances where the prediction is uncertain. See the disclaimers section for more caveats.
Important links
- Issue Label Bot homepage. Provides a way to view example predictions as well as other information regarding this bot.
- GitHub App page for Issue Label Bot, where you can install the app. See disclaimers below before installing.
Files
- /notebooks: contains notebooks on how to train the model and interact with the GitHub API using a Python client.
- /flask_app: code for a flask app that listens for GitHub issue events and responds with predictions. This is the main application that the user will interact with.
- /argo: the code in this directory relates to the construction of Argo ML Pipelines for training and deploying ML workflows.
- /deployment: This directory contains files that are helpful in deploying the app.
- Dockerfile this is the definition of the container that is used to run the flask app. The build for this container is hosted on DockerHub at hamelsmu/mlapp.
- heroku.yml: this is used for deploying to Heroku.
- *.yaml: these files relate to a Kubernetes deployment.
Running This Code
Prerequisites
To utilize the code in this repository, you will need to register a GitHub App of your own and install this app on your desired repositories and store authentication secrets.
First, walk through the prerequisites section of this getting started guide except "The Ruby programming language" section as we will be using Python instead as the client that interfaces with the GitHub API.
Second, setup your development environment. Make sure you create a Webhook secret, even though this step is optional.
Next, set up a Postgres database. You can do this for free on Heroku. Detailed instructions (stolen shamelessly from here):
- Navigate to https://www.heroku.com/, and create an account if you don’t already have one.
- On Heroku’s Dashboard, click “New” and choose “Create new app.”
- Give your app a name, and click “Create app.”
- On your app’s “Overview” page, click the “Configure Add-ons” button.
- In the “Add-ons” section of the page, type in and select “Heroku Postgres.”
- Choose the “Hobby Dev - Free” plan, which will give you access to a free PostgreSQL database that will support up to 10,000 rows of data. Click “Provision.”
- Now, click the “Heroku Postgres :: Database” link.
- You should now be on your database’s overview page. Click on 8 “Settings”, and then “View Credentials.” This is the information you’ll need to log into your database.
Finally, you need to create environment variables for all the secrets, which is described below.
Environment Variables
PRIVATE_KEY
: this is the private key you use to authenticate as an app with the GitHub API.WEBHOOK_SECRET
: this is used to verify that payloads received by your app are actually from GitHub. This is described here.DATABASE_URL
: this is the URL that contains the login information for your POSTGRESQL database, usually in the form:postgres://<username>:<password>@<url>:5432/<database_name>
APP_ID
: this is a unique identifier provided to you by GitHub when you register your app.FLASK_ENV
: this is usually set to eitherproduction
ordevelopment
. You will want to usedeployment
for local testing.PORT
: this is the port your app will be serving on. Note that if you are deploying to Heroku, Heroku will override this variable with their own value when building your app. For local development, you will want this to match the port Smee is serving to.APP_URL
: this is the url for the homepage of your app that is provided to users as a link in issue comments. You can set this to an arbitrary value for local development.
Note: If you are using zsh, the dotenv plugin can be useful for managing environment variables.
Run Locally
-
Install Dependencies: Install requirements.txt into a virtual environment. If you are using pipenv install the necessary dependencies from Pipfile.lock by typing
pipenv install
in the root of this repository. -
Run the flask app: run
python flask_app/app.py
from the root of this repository. For this to work, you must correctly set the environment variables as described in the Environment Variables section. -
Optional - Run app as docker container. A Docker container that serves Issue-Label Bot can be built with the command
bash script/bootstrap
from the root of this repository. This script builds a Docker image namedhamelsmu/mlapp
, which is also available on Dockerhub. If you desire to run the Docker container locally for testing, you must pass the necessary environment variables to the Docker container at runtime, as well as expose necessary ports for the app. See the References section for more resources on using Docker.
Deploy As A Service
The assets in this repo allow you to deploy to Heroku (easier) or a Kubernetes cluster (more advanced).
In Heroku, secrets can be passed in as configuration variables. Furthermore, this documentation describes how you can set secrets in Kubernetes. Make sure you set the environment variable FLASK_ENV
to production
if you are going to deploy the app publicly.
Contributing
We welcome all forms of contributions. We are especially interested in the following:
- Bug fixes
- Enhancements or additional features
- Improvements to the model, or expansion of the dataset(s) used for training.
Roadmap
The authors of this project are interested in adding the following features in the near future:
- Constructing better labels and negative samples of items that do not belong in the label set to drive further improvements.
- Using the tools from fastai to explore:
- State of the art architectures, such as Multi-Head Attention
- Utilizing transfer learning to predict unique labels and/or enhance accuracy.
- Using GitHub Actions to trigger automated deploys of this code.
- Model pipeline orchestration on Argo pipelines.
References
-
The code in this repo and associated tutorial(s) assume familiarity with Docker. This blog post offers a gentle introduction to Docker for data scientists.
-
Need inspiration for other data products you can build using machine learning and public GitHub datasets? See these examples:
- GitHub issue summarization and recommendation.
- Natural language semantic code search.
-
Excellent course on flask: HarvardX CS50 Web.
-
MOOCs by fastai for machine learning and deep learning.
Disclaimers
Issue-Label Bot is for educational and demonstration purposes only. Our goal was to provide a minimal working example for the community with the least amount of complexity as possible. Therefore, we believe the model demonstrated has great room from improvement. Furthermore, this app only works on public repositories and will do nothing if installed on a private repo.