50 lines
1.2 KiB
Bash
Executable File
50 lines
1.2 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -eu
|
|
set -o pipefail
|
|
|
|
ARGS=("${@-}")
|
|
|
|
pushd $(pwd) >/dev/null || exit 1
|
|
|
|
TMP_FILE="/tmp/aicommit"
|
|
|
|
|
|
if [[ -z "${GROQ_API_KEY}" ]]; then
|
|
echo "API key not set."
|
|
exit 1
|
|
fi
|
|
|
|
system="You are an expert programmer that values clear, unambiguous communication and are specialized in generating concise and informative git commit messages. Only reply with the commit message and nothing else."
|
|
|
|
diff=$(git diff --cached)
|
|
if [[ 0 -ne $? ]]; then
|
|
echo "Failed to get diff."
|
|
exit 1
|
|
fi
|
|
|
|
message=$(jq -n \
|
|
".messages = [{role: \"system\", content: $(jq -R -s '@json' <<< "$system")}, {role: \"user\", content: $(jq -R -s '@json' <<< "$diff")}] |
|
|
.model = \"llama-3.1-70b-versatile\" |
|
|
.temperature = 1 |
|
|
.max_tokens = 1024 |
|
|
.top_p = 1 |
|
|
.stream = false |
|
|
.stop = null"
|
|
)
|
|
|
|
curl -s "https://api.groq.com/openai/v1/chat/completions" \
|
|
-X POST \
|
|
-H "Content-Type: application/json" \
|
|
-H "Authorization: Bearer ${GROQ_API_KEY}" \
|
|
-d "$message" | jq -r '.choices[0].message.content | gsub("\""; "")' > ${TMP_FILE}
|
|
if [[ 0 -ne $? ]]; then
|
|
echo "Failed to get response."
|
|
exit 1
|
|
fi
|
|
|
|
|
|
git commit -e -m "$(cat ${TMP_FILE})" && rm ${TMP_FILE}
|
|
|
|
popd >/dev/null || exit 1
|