f
10 months ago
parent
fefd04928c
commit
6cc82cf1e8
@ -0,0 +1,172 @@
|
||||
name: AI Bot
|
||||
|
||||
on:
|
||||
issues:
|
||||
types: [opened]
|
||||
issue_comment:
|
||||
types: [created]
|
||||
pull_request_review_comment:
|
||||
types: [created]
|
||||
|
||||
jobs:
|
||||
respond-to-commands:
|
||||
runs-on: ubuntu-latest
|
||||
if: |
|
||||
(github.event_name == 'issues' && contains(github.event.issue.body, '/ai')) ||
|
||||
(github.event_name == 'issue_comment' && contains(github.event.comment.body, '/ai')) ||
|
||||
(github.event_name == 'pull_request_review_comment' && contains(github.event.comment.body, '/ai'))
|
||||
permissions:
|
||||
contents: write
|
||||
pull-requests: write
|
||||
issues: write
|
||||
|
||||
steps:
|
||||
- uses: actions/checkout@v3
|
||||
with:
|
||||
fetch-depth: 0
|
||||
|
||||
- name: Setup Node.js
|
||||
uses: actions/setup-node@v3
|
||||
with:
|
||||
node-version: '18'
|
||||
|
||||
- name: Install dependencies
|
||||
run: npm install openai@^4.0.0 @octokit/rest@^19.0.0
|
||||
|
||||
- name: Process command
|
||||
id: process
|
||||
env:
|
||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||
OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
|
||||
run: |
|
||||
node << 'EOF'
|
||||
const OpenAI = require('openai');
|
||||
const { Octokit } = require('@octokit/rest');
|
||||
|
||||
async function main() {
|
||||
const openai = new OpenAI({
|
||||
apiKey: process.env.OPENAI_API_KEY
|
||||
});
|
||||
|
||||
const octokit = new Octokit({
|
||||
auth: process.env.GITHUB_TOKEN
|
||||
});
|
||||
|
||||
const eventName = process.env.GITHUB_EVENT_NAME;
|
||||
const eventPath = process.env.GITHUB_EVENT_PATH;
|
||||
const event = require(eventPath);
|
||||
|
||||
// Get command and context
|
||||
let command = '';
|
||||
let issueNumber = null;
|
||||
|
||||
if (eventName === 'issues') {
|
||||
command = event.issue.body;
|
||||
issueNumber = event.issue.number;
|
||||
} else if (eventName === 'issue_comment') {
|
||||
command = event.comment.body;
|
||||
issueNumber = event.issue.number;
|
||||
} else if (eventName === 'pull_request_review_comment') {
|
||||
command = event.comment.body;
|
||||
issueNumber = event.pull_request.number;
|
||||
}
|
||||
|
||||
if (!command.startsWith('/ai')) {
|
||||
return;
|
||||
}
|
||||
|
||||
// Extract the actual command after /ai
|
||||
const aiCommand = command.substring(3).trim();
|
||||
|
||||
// Generate response using OpenAI
|
||||
const completion = await openai.chat.completions.create({
|
||||
model: "gpt-3.5-turbo",
|
||||
messages: [
|
||||
{
|
||||
role: "system",
|
||||
content: "You are a helpful AI assistant that helps with GitHub repositories. You can suggest code changes, fix issues, and improve code quality."
|
||||
},
|
||||
{
|
||||
role: "user",
|
||||
content: aiCommand
|
||||
}
|
||||
],
|
||||
temperature: 0.7,
|
||||
max_tokens: 2000
|
||||
});
|
||||
|
||||
const response = completion.choices[0].message.content;
|
||||
|
||||
// If response contains code changes, create a new branch and PR
|
||||
if (response.includes('```')) {
|
||||
const branchName = `ai-bot/fix-${issueNumber}`;
|
||||
|
||||
// Create new branch
|
||||
const defaultBranch = event.repository.default_branch;
|
||||
const ref = await octokit.git.getRef({
|
||||
owner: event.repository.owner.login,
|
||||
repo: event.repository.name,
|
||||
ref: `heads/${defaultBranch}`
|
||||
});
|
||||
|
||||
await octokit.git.createRef({
|
||||
owner: event.repository.owner.login,
|
||||
repo: event.repository.name,
|
||||
ref: `refs/heads/${branchName}`,
|
||||
sha: ref.data.object.sha
|
||||
});
|
||||
|
||||
// Extract code changes and file paths from response
|
||||
const codeBlocks = response.match(/```[\s\S]*?```/g);
|
||||
for (const block of codeBlocks) {
|
||||
const [_, filePath, ...codeLines] = block.split('\n');
|
||||
const content = Buffer.from(codeLines.join('\n')).toString('base64');
|
||||
|
||||
await octokit.repos.createOrUpdateFileContents({
|
||||
owner: event.repository.owner.login,
|
||||
repo: event.repository.name,
|
||||
path: filePath,
|
||||
message: `AI Bot: Apply suggested changes for #${issueNumber}`,
|
||||
content,
|
||||
branch: branchName
|
||||
});
|
||||
}
|
||||
|
||||
// Create PR
|
||||
await octokit.pulls.create({
|
||||
owner: event.repository.owner.login,
|
||||
repo: event.repository.name,
|
||||
title: `AI Bot: Fix for #${issueNumber}`,
|
||||
body: `This PR was automatically generated in response to #${issueNumber}\n\nChanges proposed:\n${response}`,
|
||||
head: branchName,
|
||||
base: defaultBranch
|
||||
});
|
||||
}
|
||||
|
||||
// Add comment with response
|
||||
await octokit.issues.createComment({
|
||||
owner: event.repository.owner.login,
|
||||
repo: event.repository.name,
|
||||
issue_number: issueNumber,
|
||||
body: `AI Bot Response:\n\n${response}`
|
||||
});
|
||||
}
|
||||
|
||||
main().catch(error => {
|
||||
console.error('Error:', error);
|
||||
process.exit(1);
|
||||
});
|
||||
EOF
|
||||
|
||||
- name: Handle errors
|
||||
if: failure()
|
||||
uses: actions/github-script@v6
|
||||
with:
|
||||
script: |
|
||||
const issueNumber = context.issue.number || context.payload.pull_request.number;
|
||||
await github.rest.issues.createComment({
|
||||
owner: context.repo.owner,
|
||||
repo: context.repo.repo,
|
||||
issue_number: issueNumber,
|
||||
body: '❌ Sorry, there was an error processing your command. Please try again or contact the repository maintainers.'
|
||||
});
|
||||
@ -1,89 +0,0 @@
|
||||
name: Improve PR Title
|
||||
|
||||
on:
|
||||
pull_request:
|
||||
types: [opened, edited, synchronize]
|
||||
|
||||
jobs:
|
||||
improve-title:
|
||||
runs-on: ubuntu-latest
|
||||
permissions:
|
||||
pull-requests: write
|
||||
|
||||
steps:
|
||||
- uses: actions/checkout@v3
|
||||
|
||||
- name: Setup Node.js
|
||||
uses: actions/setup-node@v3
|
||||
with:
|
||||
node-version: '18'
|
||||
|
||||
- name: Install OpenAI SDK
|
||||
run: npm install openai@^4.0.0
|
||||
|
||||
- name: Get PR Content
|
||||
id: pr-content
|
||||
uses: actions/github-script@v6
|
||||
with:
|
||||
script: |
|
||||
const { data: pr } = await github.rest.pulls.get({
|
||||
owner: context.repo.owner,
|
||||
repo: context.repo.repo,
|
||||
pull_number: context.issue.number
|
||||
});
|
||||
const content = `${pr.title}\n\n${pr.body}`;
|
||||
core.setOutput('content', content);
|
||||
|
||||
- name: Generate Better Title
|
||||
id: generate-title
|
||||
run: |
|
||||
node << 'EOF'
|
||||
const OpenAI = require('openai');
|
||||
|
||||
async function generateTitle() {
|
||||
const openai = new OpenAI({
|
||||
apiKey: process.env.OPENAI_API_KEY
|
||||
});
|
||||
|
||||
const content = `${{ steps.pr-content.outputs.content }}`;
|
||||
|
||||
const completion = await openai.chat.completions.create({
|
||||
model: "gpt-3.5-turbo",
|
||||
messages: [
|
||||
{
|
||||
role: "system",
|
||||
content: "You are a helpful assistant that improves pull request titles. Make titles concise, descriptive, and following conventional commit message style. Return ONLY the new title, nothing else."
|
||||
},
|
||||
{
|
||||
role: "user",
|
||||
content: `Based on this pull request content, generate a better title:\n\n${content}`
|
||||
}
|
||||
],
|
||||
temperature: 0.7,
|
||||
max_tokens: 60,
|
||||
top_p: 1.0
|
||||
});
|
||||
|
||||
const newTitle = completion.choices[0].message.content.trim();
|
||||
console.log(`::set-output name=title::${newTitle}`);
|
||||
}
|
||||
|
||||
generateTitle().catch(error => {
|
||||
console.error('Error:', error);
|
||||
process.exit(1);
|
||||
});
|
||||
EOF
|
||||
env:
|
||||
OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
|
||||
|
||||
- name: Update PR Title
|
||||
if: steps.generate-title.outputs.title != ''
|
||||
uses: actions/github-script@v6
|
||||
with:
|
||||
script: |
|
||||
await github.rest.pulls.update({
|
||||
owner: context.repo.owner,
|
||||
repo: context.repo.repo,
|
||||
pull_number: context.issue.number,
|
||||
title: '${{ steps.generate-title.outputs.title }}'
|
||||
});
|
||||
Loading…
Reference in new issue