NaLCoS – NAtural Language COmmit Search
Search commit messages in your repository in natural language.
NaLCoS (NAtural Language COmmit Search) is a command-line tool for searching commit messages in your repository in natural language.
The key features are:
- Search commit messages in both local and remote GitHub repositories.
- Search for commits in a specific branch.
- Restrict the number of commits to look back in history while searching.
- Increase the number of retrieved results.
Internally, NaLCoS uses Sentence Transformers with pre-trained weights from multi-qa-MiniLM-L6-cos-v1. I chose this particular model because it has a good Performance vs Speed tradeoff. Since this model was designed for semantic search and has been pre-trained on 215M (question, answer) pairs from diverse sources, it is a good choice for tasks such as finding similarity between two sentences.
NaLCoS encodes the query string and all the commits into their corresponding vector embeddings and computes the cosine similarity between the query and all the commits. This is then used to rank the commits.
Why did I build this?
Most of the times when I’ve used Machine Learning till now, has been in dedicated environments such as Google Colab or Kaggle. I had been learning Natural Language Processing for a while and wanted to use transformers to build something different that is not very resource (read GPU) intensive and can be used like an everyday tool.
Though many Transformer models are far from fitting this description, I found that distilled models are not as hungry as their older siblings are infamous for. Searching for Git commits using natural language was something on which I could not find any pre-existing tool and thus decided to give this a shot.
Though there are various improvements left, I’m happy with what this initially turned out to be. I’m eager to see what further enhancements can be made to this to make it more efficient and useful.
Requirements
NaLCoS uses the following packages:
- Sentence Transformers for the Transformer model.
- Git Python for local Git operations.
- GitHub API for interacting with GitHub repos.
- Rich for well-formatted CLI output.
Installation
Installing with pip (Recommended)
Install with pip or your favourite PyPi manager:
$ pip install nalcos
Run NaLCoS with the --help flag to see all the available options:
$ nalcos --help
Note: When you run the nalcos command for the first time, it will, download the model which would be cached and used the next time you run NaLCoS.
Installing bleeding edge from the GitHub repository
- Clone the repository:
$ git clone https://github.com/thepushkarp/nalcos.git
This also downloads the model weights stored in the nalcos/models directory so you don’t have to download them while running the model for the first time.
- Create a virtual environment (click here to read about activating virtualenv):
$ virtualenv venv
- Activate virtualenv (for Linux and MacOS):
$ source ./venv/bin/activate
- Activate virtualenv (for Windows):
$ cd venv/Scripts/
$ activate
- Install the requirements:
$ pip install -r requirements.txt
- Change directory to the
nalcosdirectory:
$ cd nalcos/
- Run NaLCoS with the
--helpflag to see all the available options:
$ python nalcos.py --help
Usage
A detailed information about the usage of NaLCoS can be found below:
usage: nalcos [-h] [-g] [-n N_MATCHES] [-b BRANCH] [-l LOOK_PAST] [-v] query location
Search a commit in your git repository using natural language.
positional arguments:
query The query to search for similar commit messages.
location The repository path to search in. If `-g` flag is not passed, searches locally in the path specified, else
takes in a remote GitHub repository name in the format '{owner}/{repo_name}'
optional arguments:
-h, --help show this help message and exit
-g, --github Flag to search on GitHub instead of searching in a local repository. Due to API limits currently this
allows for around 15 lookups per hour from your IP.
-n N_MATCHES, --n-matches N_MATCHES
The number of matching results to return. Default 10.
-b BRANCH, --branch BRANCH
The branch to search in. If not specified, the current branch will be used by default.
-l LOOK_PAST, --look-past LOOK_PAST
Look back this many commits. Default 100.
-v, --version show program's version number and exit
Examples
- Input:
$ python nalcos.py "improve language" "github/docs" --github
- Output:
Found 100 commits.
Commits related to "improve language" in "github/docs"
┏━━━━━┳━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━┓
┃ No. ┃ Commit ID ┃ Commit Message ┃ Commit Author ┃ Commit Date ┃
┡━━━━━╇━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━┩
│ 1. │ 51bfdbb95 │ Merge branch 'main' into fatenhealy-fix-supportedlanguage │ Faten Healy │ 2021-09-12T22:26:31Z │
│ 2. │ a9c2c8eea │ fix deprecation label spelling (#21474) │ Rachael Sewell │ 2021-09-13T18:12:03Z │
│ 3. │ 94e3c092d │ English search sync (#21446) │ Rachael Sewell │ 2021-09-13T17:30:08Z │
│ 4. │ b048e27e9 │ Merge pull request #9909 from github/fatenhealy-fix-supportedlanguage │ Ramya Parimi │ 2021-09-12T22:35:19Z │
│ 5. │ 73c2717f7 │ Fix typo │ Adrian Mato │ 2021-09-13T06:35:27Z │
│ 6. │ 86b571982 │ Export changes to a branch for codespaces (#21462) │ Matthew Isabel │ 2021-09-13T14:55:50Z │
│ 7. │ 969288662 │ Update diff limit to 500KB (#20616) │ jjkennedy3 │ 2021-09-11T09:12:38Z │
│ 8. │ f28ee46d4 │ Update OpenAPI Descriptions (#21447) │ github-openapi-bot │ 2021-09-11T09:22:28Z │
│ 9. │ 92af3a469 │ update search indexes │ GitHub Actions │ 2021-09-12T09:50:46Z │
│ 10. │ e6018f2aa │ update search indexes │ GitHub Actions │ 2021-09-11T02:05:19Z │
└─────┴───────────┴───────────────────────────────────────────────────────────────────────┴────────────────────┴──────────────────────┘
Future plans
Please visit the NaLCoS To Do Project Board to see current status and future plans.
Known issues
Not all retrieved results are always relevant. I could think of two primary reasons for this:
- The data the model was pre-trained on is not representative of how people write commit messages. Since commit messages usually contain technical jargon, merge commit messages, abbreviations and other non-common terms, the model (which has a limited vocabulary) is not able to generalize well to this data.
- Two commits may be related even when their commit messages may not be similar and similarly two commit messages maybe unrelated even when their commit messages are similar. We often need more metadata (such as lines changes, files changed) etc. to make the predictions more accurate.
Contributing
Any suggestions, improvements or bug reports are welcome.
- If you want to discuss any aspect of the project, please use the Discussions Tab.
- You can submit your idea by opening an issue or creating a Pull Request.
- If you’d like to improve the code, make sure you stick to the existing code style and naming conventions.
Contributors
Thanks goes to these wonderful people (emoji key):
 Pushkar Patel ? ? ? |
This project follows the specification. Contributions of any kind welcome!
License
This project is licensed under the terms of the MIT license.
