diff --git a/.github/workflows/pr-description.yml b/.github/workflows/pr-description.yml index 7b08458..b37a658 100644 --- a/.github/workflows/pr-description.yml +++ b/.github/workflows/pr-description.yml @@ -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"