mirror of
https://github.com/mudler/LocalAI.git
synced 2024-12-29 08:58:52 +00:00
1f7cedf5ee
* minor cleanup to go.mod - importing ourself? Signed-off-by: Dave Lee <dave@gray101.com> * figured out why we were importing ourself and fixed it Signed-off-by: Dave Lee <dave@gray101.com> * set pull_request_target Signed-off-by: Dave Lee <dave@gray101.com> --------- Signed-off-by: Dave Lee <dave@gray101.com> Co-authored-by: Ettore Di Giacinto <mudler@users.noreply.github.com>
78 lines
2.7 KiB
YAML
78 lines
2.7 KiB
YAML
name: Comment PRs
|
|
on:
|
|
pull_request_target:
|
|
|
|
jobs:
|
|
comment-pr:
|
|
env:
|
|
MODEL_NAME: hermes-2-theta-llama-3-8b
|
|
runs-on: ubuntu-latest
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
with:
|
|
fetch-depth: 0 # needed to checkout all branches for this Action to work
|
|
- name: Start LocalAI
|
|
run: |
|
|
echo "Starting LocalAI..."
|
|
docker run -e -ti -d --name local-ai -p 8080:8080 localai/localai:master-ffmpeg-core run --debug $MODEL_NAME
|
|
until [ "`docker inspect -f {{.State.Health.Status}} local-ai`" == "healthy" ]; do echo "Waiting for container to be ready"; docker logs --tail 10 local-ai; sleep 2; done
|
|
# Check the PR diff using the current branch and the base branch of the PR
|
|
- uses: GrantBirki/git-diff-action@v2.7.0
|
|
id: git-diff-action
|
|
with:
|
|
json_diff_file_output: diff.json
|
|
raw_diff_file_output: diff.txt
|
|
file_output_only: "true"
|
|
- name: Summarize
|
|
env:
|
|
DIFF: ${{ steps.git-diff-action.outputs.raw-diff-path }}
|
|
id: summarize
|
|
run: |
|
|
input="$(cat $DIFF)"
|
|
|
|
# Define the LocalAI API endpoint
|
|
API_URL="http://localhost:8080/chat/completions"
|
|
|
|
# Create a JSON payload using jq to handle special characters
|
|
json_payload=$(jq -n --arg input "$input" '{
|
|
model: "'$MODEL_NAME'",
|
|
messages: [
|
|
{
|
|
role: "system",
|
|
content: "You are LocalAI-bot in Github that helps understanding PRs and assess complexity. Explain what has changed in this PR diff and why"
|
|
},
|
|
{
|
|
role: "user",
|
|
content: $input
|
|
}
|
|
]
|
|
}')
|
|
|
|
# Send the request to LocalAI
|
|
response=$(curl -s -X POST $API_URL \
|
|
-H "Content-Type: application/json" \
|
|
-d "$json_payload")
|
|
|
|
# Extract the summary from the response
|
|
summary="$(echo $response | jq -r '.choices[0].message.content')"
|
|
|
|
# Print the summary
|
|
# -H "Authorization: Bearer $API_KEY" \
|
|
echo "Summary:"
|
|
echo "$summary"
|
|
echo "payload sent"
|
|
echo "$json_payload"
|
|
{
|
|
echo 'message<<EOF'
|
|
echo "$summary"
|
|
echo EOF
|
|
} >> "$GITHUB_OUTPUT"
|
|
docker logs --tail 10 local-ai
|
|
- uses: mshick/add-pr-comment@v2
|
|
if: always()
|
|
with:
|
|
repo-token: ${{ secrets.UPDATE_BOT_TOKEN }}
|
|
message: ${{ steps.summarize.outputs.message }}
|
|
message-failure: |
|
|
Uh oh! Could not analyze this PR, maybe it's too big?
|