-
-
Notifications
You must be signed in to change notification settings - Fork 2.3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
2 changed files
with
26 additions
and
20 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -6,19 +6,19 @@ on: | |
|
||
jobs: | ||
notify-discord: | ||
if: ${{ (github.event.pull_request.merged == true) && (contains(github.event.pull_request.labels.*.name, 'area/ai-model')) }} | ||
# if: ${{ (github.event.pull_request.merged == true) && (contains(github.event.pull_request.labels.*.name, 'area/ai-model')) }} | ||
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:latest-aio-cpu run --debug $MODEL_NAME | ||
until [ "`docker inspect -f {{.State.Health.Status}} local-ai`"=="healthy" ]; do | ||
echo "Waiting for container to be ready" | ||
sleep 2; | ||
done; | ||
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/[email protected] | ||
id: git-diff-action | ||
|
@@ -32,18 +32,18 @@ jobs: | |
id: summarize | ||
run: | | ||
# Read the content of file.txt | ||
input="$DIFF" | ||
input="$(cat $DIFF)" | ||
# Define the OpenAI 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", | ||
model: "'$MODEL_NAME'", | ||
messages: [ | ||
{ | ||
role: "system", | ||
content: "Write a discord message to notify everyone about the new model." | ||
content: "Write a discord message to notify everyone about the new model from the git diff. Make it informal. An example can include: the URL of the model, the name, and a brief description of the model if exists." | ||
}, | ||
{ | ||
role: "user", | ||
|
@@ -58,13 +58,20 @@ jobs: | |
-d "$json_payload") | ||
# Extract the summary from the response | ||
summary=$(echo $response | jq -r '.choices[0].message.content') | ||
summary="$(echo $response | jq -r '.choices[0].message.content')" | ||
# Print the summary | ||
# -H "Authorization: Bearer $API_KEY" \ | ||
echo "Summary:" | ||
echo "$summary" | ||
echo "message=$summary\n" >> $GITHUB_OUTPUT | ||
echo "payload sent" | ||
echo "$json_payload" | ||
{ | ||
echo 'message<<EOF' | ||
echo $summary | ||
echo EOF | ||
} >> "$GITHUB_OUTPUT" | ||
docker logs --tail 10 local-ai | ||
- name: Discord notification | ||
env: | ||
DISCORD_WEBHOOK: ${{ secrets.DISCORD_WEBHOOK_URL }} | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters