From 61348aa351482edeffaa788004b9f75886cb2747 Mon Sep 17 00:00:00 2001
From: faris Aljohari <83524184+farisaljohari@users.noreply.github.com>
Date: Wed, 16 Jul 2025 00:08:08 -0600
Subject: [PATCH] fix: update workflow to use HuggingFace Falcon model for PR
 description generation

---
 .github/workflows/pr-description.yml | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

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"