diff --git a/.github/workflows/pr-description.yml b/.github/workflows/pr-description.yml index e4514b4..7b08458 100644 --- a/.github/workflows/pr-description.yml +++ b/.github/workflows/pr-description.yml @@ -1,4 +1,4 @@ -name: 🤖 AI PR Description Commenter (100% Safe with jq) +name: 🤖 AI PR Description with HuggingFace on: pull_request: @@ -27,25 +27,21 @@ jobs: env: GH_TOKEN: ${{ secrets.GH_PERSONAL_TOKEN }} - - name: Generate PR Description with OpenAI (Safe JSON with jq) + - name: Generate PR Description with HuggingFace (DeepSeek) run: | REQUEST_BODY=$(jq -n \ - --arg model "gpt-4o" \ - --arg content "Given the following commit messages:\n\n${commits}\n\nGenerate a clear and professional pull request description." \ - '{ - model: $model, - messages: [{ role: "user", content: $content }] - }' + --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.openai.com/v1/chat/completions \ - -H "Authorization: Bearer $OPENAI_API_KEY" \ + RESPONSE=$(curl -s https://api-inference.huggingface.co/models/deepseek-ai/deepseek-coder-6.7b-instruct \ + -H "Authorization: Bearer $HUGGINGFACE_API_KEY" \ -H "Content-Type: application/json" \ -d "$REQUEST_BODY") - DESCRIPTION=$(echo "$RESPONSE" | jq -r '.choices[0].message.content') + DESCRIPTION=$(echo "$RESPONSE" | jq -r '.[0].generated_text') - echo "---------- OpenAI Raw Response ----------" + echo "---------- HuggingFace Raw Response ----------" echo "$RESPONSE" echo "---------- Extracted Description ----------" echo "$DESCRIPTION" @@ -54,7 +50,7 @@ jobs: echo "$DESCRIPTION" >> $GITHUB_ENV echo "EOF" >> $GITHUB_ENV env: - OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} + HUGGINGFACE_API_KEY: ${{ secrets.HUGGINGFACE_API_KEY }} commits: ${{ env.commits }} - name: Post AI Generated Description as Comment