Why don’t use available solution? All of them using ChatGPT. But I’m out of credits ;) Of course I want to learn something!
How to generate git commit message?
Git allows you to create hooks. Let’s use global one. Global hooks works without modifying every git repo.
Create a directory for hooks:
$ mkdir ~/.config/git/hooks/Let git knows where hooks are:
$ git config core.hooksPath ~/.config/git/hooks/Long story short the prepare-commit-msg is the one we need. The file we need to update is passed as first parameter. Create a simple script:
#!/bin/sh
echo "Fancy commit message" > $1Make it executable:
$ chmod +z ~/.confog/git/hooks/prepare-commit-msgIs it works? Let’s commit something … Yep, we have a message at the end of the commit message.
Let’s generate something:
Generating commit message
Let’s build something that’s works offline. AI? Yes, let’s use AI!
We need model right?
Let’s look at huggingface!
There it is: https://huggingface.co/mamiksik/T5-commit-message-generation but there is no docs :( But if you’ll look deeper you’ll find https://huggingface.co/spaces/mamiksik/commit-message-generator
We can use this https://huggingface.co/spaces/mamiksik/commit-message-generator/blob/main/app.py with a little modifications.
As we can use any shell script in a hook, let’s use python.
Let’s take a look what’s there:
import re
import gradio as gr
import torch
from transformers import T5ForConditionalGeneration, RobertaTokenizer
tokenizer = RobertaTokenizer.from_pretrained("mamiksik/CommitPredictorT5PL", revision="fb08d01")
model = T5ForConditionalGeneration.from_pretrained("mamiksik/CommitPredictorT5PL", revision="fb08d01")
def parse_files(patch):
accumulator = []
lines = patch.splitlines()
filename_before = None
for line in lines:
if line.startswith("index") or line.startswith("diff"):
continue
if line.startswith("---"):
filename_before = line.split(" ", 1)[1][1:]
continue
if line.startswith("+++"):
filename_after = line.split(" ", 1)[1][1:]
if filename_before == filename_after:
accumulator.append(f"" , 1)
elif line[0] == "-":
line = line.replace("-", "", 1)
else:
line = f"Everything we need is here! We need to:
- fetch gitmessage file to update
- fetch git diff
- use current script make predictions
- prepend commit message to gitmessage file
File that we need to update is passed as first parameter so
Heh that was easy.
Fetch git diff
import subprocess
subprocess.run(['git', 'diff', '--cached'], capture_output=True).stdout.decode('utf-8')Easy peasy!
Use current script to make predictions
max_message = 40
min_message = 5
num_beams = 10
num_predictions = 1
msg = predict(diff, max_message, min_message, num_beams, num_predictions)Prepend our message to gitmessage file
with open(sys.argv[1], 'r+') as f:
content = f.read()
f.seek(0)
f.write(msg + 'n' + content)
f.close()It’s just like that. With little cleanups this is our final script.
#!/usr/bin/env python
print("Generating commit message", end="", flush=True)
import sys
import re
import subprocess
import torch
from transformers import T5ForConditionalGeneration, RobertaTokenizer
def parse_files(patch):
accumulator = []
lines = patch.splitlines()
filename_before = None
for line in lines:
print(".", end="", flush=True)
if line.startswith("index") or line.startswith("diff"):
continue
if line.startswith("---"):
filename_before = line.split(" ", 1)[1][1:]
continue
if line.startswith("+++"):
filename_after = line.split(" ", 1)[1][1:]
if filename_before == filename_after:
accumulator.append(f"" , 1)
elif line[0] == "-":
line = line.replace("-", "", 1)
else:
line = f"It’s fast on cpu, but loading model take a lot of times. Anyway 3s is OK. That’s all. It works. At least for me.
