fix: update workflow to use HuggingFace Falcon model for PR description generation

This commit is contained in:
faris Aljohari
2025-07-16 00:08:08 -06:00
parent dea942f11e
commit 61348aa351

View File

@ -1,4 +1,4 @@
name: 🤖 AI PR Description with HuggingFace
name: 🤖 AI PR Description with HuggingFace Falcon
on:
pull_request:
@ -27,22 +27,23 @@ jobs:
env:
GH_TOKEN: ${{ secrets.GH_PERSONAL_TOKEN }}
- name: Generate PR Description with HuggingFace (DeepSeek)
- name: Generate PR Description with HuggingFace Falcon
run: |
REQUEST_BODY=$(jq -n \
--arg inputs "Given the following commit messages:\n\n${commits}\n\nGenerate a clear and professional pull request description." \
'{ inputs: $inputs }'
)
RESPONSE=$(curl -s https://api-inference.huggingface.co/models/deepseek-ai/deepseek-coder-6.7b-instruct \
RESPONSE=$(curl -s https://api-inference.huggingface.co/models/tiiuae/falcon-7b-instruct \
-H "Authorization: Bearer $HUGGINGFACE_API_KEY" \
-H "Content-Type: application/json" \
-d "$REQUEST_BODY")
DESCRIPTION=$(echo "$RESPONSE" | jq -r '.[0].generated_text')
echo "---------- HuggingFace Raw Response ----------"
echo "$RESPONSE"
DESCRIPTION=$(echo "$RESPONSE" | jq -r 'if type=="array" then .[0].generated_text else .error else "Error: Unexpected response" end')
echo "---------- Extracted Description ----------"
echo "$DESCRIPTION"