• This repository has been archived on 09/Dec/2022
  • Stars
    star
    326
  • Rank 129,027 (Top 3 %)
  • Language SCSS
  • License
    MIT License
  • Created over 5 years ago
  • Updated over 2 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Code For The Issue Label Bot, an App that automatically labels issues using machine learning, available on the GitHub Marketplace. This is also code for the blog article: "How to automate tasks on GitHub with machine learning for fun and profit"

Python 3.6 License: MIT Install App

⚠️ 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

Code for: "How to automate tasks on GitHub with machine learning for fun and profit"

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:

alt text

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.

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):

  1. Navigate to https://www.heroku.com/, and create an account if you don’t already have one.
  2. On Heroku’s Dashboard, click “New” and choose “Create new app.”
  3. Give your app a name, and click “Create app.”
  4. On your app’s “Overview” page, click the “Configure Add-ons” button.
  5. In the “Add-ons” section of the page, type in and select “Heroku Postgres.”
  6. 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.”
  7. Now, click the “Heroku Postgres :: Database” link.
  8. 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

  1. PRIVATE_KEY: this is the private key you use to authenticate as an app with the GitHub API.
  2. WEBHOOK_SECRET: this is used to verify that payloads received by your app are actually from GitHub. This is described here.
  3. 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>
  4. APP_ID: this is a unique identifier provided to you by GitHub when you register your app.
  5. FLASK_ENV: this is usually set to either production or development. You will want to use deployment for local testing.
  6. 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.
  7. 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

  1. 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.

  2. 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.

  3. 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 named hamelsmu/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

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.

More Repositories

1

actions-ml-cicd

A Collection of GitHub Actions That Facilitate MLOps
Jupyter Notebook
208
star
2

ml-template-azure

Template for getting started with automated ML Ops on Azure Machine Learning
Python
126
star
3

actions-chatops

Actions That Enables ChatOps In a PR Through a GitHub App
Python
74
star
4

actions-app-token

Impersonate a GitHub App Token inside Actions
Python
61
star
5

wandb-action

GitHub Action That Retrieves Model Runs From Weights & Biases
Python
60
star
6

gpr-docker-publish

GitHub Action That Publishes Docker Images to GPR
Shell
53
star
7

actions-argo

GitHub Action That Submits Argo Workflows From GitHub
Shell
38
star
8

mlops-dashboard

Prototype MLOps Dashboard Using GitHub Pages
Jupyter Notebook
23
star
9

pr-comment

Action that makes a pr comment from a file artifact
Ruby
20
star
10

hands-on-ml2

Notebooks from https://github.com/ageron/handson-ml2 as blog posts using fastpages
Jupyter Notebook
18
star
11

self-hosted-k8s-runner

Run a self-hosted Actions runner on Kubernetes.
Shell
17
star
12

gke-argo

GitHub Action That Submits Argo Workflows For Execution on Your GKE Cluster
Shell
16
star
13

gke-kubeconfig

GitHub Action that Fetches Kubeconfig From GKE and saves to `$GITHUB_WORKSPACE/.kube/config`
Shell
13
star
14

mystify

Jupyter backend for textual notebooks in MyST format
Python
12
star
15

website-docs

Docs for ml-ops.github.com
Ruby
8
star
16

IssuesLanguageModel

A Language model trained on a large corpus of GitHub Issues
Jupyter Notebook
7
star
17

mdparse

Parsing of markdown files for deep learning
Python
6
star
18

actions-chatops-deprecated

GitHub Action For ChatOps in Pull Requests
Shell
6
star
19

MLOps

Managing machine learning operations using GitHub and other cloud providers.
CSS
5
star
20

great-expectations-render

Jupyter Notebook
4
star
21

actions-impersonate

Have Your Action Impersonate a GitHub App In Order To Overcome Restrictions
Python
2
star
22

dvc-example

1
star
23

demo-videos

1
star
24

actions

Repository for Discussing and Building GitHub Actions for Machine Learning
1
star