-
-
Save gobilo/2af3b7e5480b77f5901be2ab860fccc0 to your computer and use it in GitHub Desktop.
Git Commit Message AI
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # ----------------------------------------------------------------------------- | |
| # AI-powered Git Commit Function by Andrej Karpathy. Edited by Stefan Goebel for use with a local instance of OLLAMA. | |
| # Check the MODEL variable and adjust it to the model you want to use. | |
| # Copy paste this gist into your ~/.bashrc or ~/.zshrc to gain the `gcm` command. Source it first to test it. | |
| # It: | |
| # 1) gets the current staged changed diff | |
| # 2) sends it to Ollama to write the git commit message | |
| # 3) allows you to easily accept, edit, regenerate, cancel | |
| # Read and edit the code however you like. | |
| # Remark: Andrej's orignal version of the script uses the `llm` CLI util which you can get here: https://llm.datasette.io/en/stable/ | |
| gcm() { | |
| # Function to generate commit message | |
| generate_commit_message() { | |
| MODEL="llama3:latest" | |
| DIFF=$(git diff --cached) | |
| ollama run llama3:latest " | |
| Below is a diff of all staged changes, coming from the command: | |
| \`\`\` | |
| git diff --cached | |
| \`\`\` | |
| Please generate a concise, one-line commit message for these changes. Only return the commit message and nothing else. | |
| " + "${DIFF}" | |
| } | |
| # Function to read user input compatibly with both Bash and Zsh | |
| read_input() { | |
| if [ -n "$ZSH_VERSION" ]; then | |
| echo -n "$1" | |
| read -r REPLY | |
| else | |
| read -p "$1" -r REPLY | |
| fi | |
| } | |
| # Main script | |
| echo -e "Generating AI-powered commit message...\n" | |
| commit_message=$(generate_commit_message) | |
| while true; do | |
| # echo -e "\nProposed commit message:" | |
| echo -e "$commit_message\n" | |
| read_input "Do you want to (a)ccept, (e)dit, (r)egenerate, or (c)ancel? " | |
| choice=$REPLY | |
| case "$choice" in | |
| a|A ) | |
| if git commit -m "$commit_message"; then | |
| echo "Changes committed successfully!" | |
| return 0 | |
| else | |
| echo "Commit failed. Please check your changes and try again." | |
| return 1 | |
| fi | |
| ;; | |
| e|E ) | |
| read_input "Enter your commit message: " | |
| commit_message=$REPLY | |
| if [ -n "$commit_message" ] && git commit -m "$commit_message"; then | |
| echo "Changes committed successfully with your message!" | |
| return 0 | |
| else | |
| echo "Commit failed. Please check your message and try again." | |
| return 1 | |
| fi | |
| ;; | |
| r|R ) | |
| echo "Regenerating commit message..." | |
| commit_message=$(generate_commit_message) | |
| ;; | |
| c|C ) | |
| echo "Commit cancelled." | |
| return 1 | |
| ;; | |
| * ) | |
| echo "Invalid choice. Please try again." | |
| ;; | |
| esac | |
| done | |
| } |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment