Last active
August 28, 2024 07:06
-
-
Save Mustafa-khann/16fb113171f250ac0f61289d2c2d6366 to your computer and use it in GitHub Desktop.
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 | |
# gets the current staged changed diff | |
# sends that to the LLM, and generates a message. | |
# Copy paste this gist into your ~/.bashrc or ~/.zshrc to gain the `gcm` command. It: | |
# the `llm` CLI util is awesome, can get it here: https://llm.datasette.io/en/stable/ | |
# The command is "gcm" | |
gcm() { | |
# Function to generate commit message | |
generate_commit_message() { | |
git diff --cached | llm -m gemini-pro " | |
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." | |
} | |
# 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 "Generating AI-powered commit message using gemini-pro model..." | |
commit_message=$(generate_commit_message) | |
while true; do | |
echo -e "\nProposed commit message:" | |
echo "$commit_message" | |
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