name : AI Bot
on :
issues :
types : [ opened]
issue_comment :
types : [ created]
pull_request_review_comment :
types : [ created]
pull_request :
types : [ opened, edited, synchronize]
jobs :
respond-to-commands :
runs-on : ubuntu-latest
if : |
(github.actor == 'f') &&
((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')) ||
(github.event_name == 'pull_request' && contains(github.event.pull_request.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 :
GH_SECRET : ${{ secrets.GH_SECRET }}
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.GH_SECRET
});
const eventName = process.env.GITHUB_EVENT_NAME;
const eventPath = process.env.GITHUB_EVENT_PATH;
const event = require(eventPath);
// Double check user authorization
const actor = event.sender?.login || event.pull_request?.user?.login || event.issue?.user?.login;
if (actor !== 'f') {
console.log('Unauthorized user attempted to use the bot:', actor);
return;
}
// Get command and context
let command = '';
let issueNumber = null;
let isPullRequest = false;
if (eventName === 'issues') {
command = event.issue.body;
issueNumber = event.issue.number;
} else if (eventName === 'issue_comment') {
command = event.comment.body;
issueNumber = event.issue.number;
isPullRequest = !!event.issue.pull_request;
} else if (eventName === 'pull_request_review_comment') {
command = event.comment.body;
issueNumber = event.pull_request.number;
isPullRequest = true;
} else if (eventName === 'pull_request') {
command = event.pull_request.body;
issueNumber = event.pull_request.number;
isPullRequest = true;
}
if (!command.startsWith('/ai')) {
return;
}
// Extract the actual command after /ai
const aiCommand = command.substring(3).trim();
// Handle resolve conflicts command
if (aiCommand === 'resolve' || aiCommand === 'fix conflicts') {
if (!isPullRequest) {
await octokit.issues.createComment({
owner : event.repository.owner.login,
repo : event.repository.name,
issue_number : issueNumber,
body : '❌ The resolve command can only be used on pull requests.'
});
return;
}
try {
// Get PR details
const { data: pr } = await octokit.pulls.get({
owner : event.repository.owner.login,
repo : event.repository.name,
pull_number : issueNumber
});
// Get the PR diff to extract the new prompt
const { data: files } = await octokit.pulls.listFiles({
owner : event.repository.owner.login,
repo : event.repository.name,
pull_number : issueNumber
});
// Extract prompt from changes
let newPrompt = '';
let actName = '';
let contributorInfo = '';
for (const file of files) {
if (file.filename === 'README.md') {
const patch = file.patch || '';
const addedLines = patch.split('\n')
.filter(line => line.startsWith('+'))
.map(line => line.substring(1))
.join('\n');
const promptMatch = addedLines.match(/## Act as (?:a |an )?([^\n]+)\n(?:Contributed by:[^\n]*\n)?(?:> )?([^#]+?)(?=\n\n|$)/);
if (promptMatch) {
actName = `Act as ${promptMatch[1].trim()}`;
newPrompt = promptMatch[2].trim();
const contributorLine = addedLines.match(/Contributed by : \[@([^\]]+)\]\(https:\/\/github\.com\/([^\)]+)\)/);
if (contributorLine) {
contributorInfo = `Contributed by : [ @${contributorLine[1]}](https://github.com/${contributorLine[2]})`;
}
}
}
}
if (!actName || !newPrompt) {
await octokit.issues.createComment({
owner : event.repository.owner.login,
repo : event.repository.name,
issue_number : issueNumber,
body : '❌ Could not extract prompt information from changes'
});
return;
}
// Get content from main branch as reference
const { data: readmeFile } = await octokit.repos.getContent({
owner : event.repository.owner.login,
repo : event.repository.name,
path : 'README.md' ,
ref : 'main'
});
const { data: csvFile } = await octokit.repos.getContent({
owner : event.repository.owner.login,
repo : event.repository.name,
path : 'prompts.csv' ,
ref : 'main'
});
// Format the new prompt section
const newSection = `## ${actName}\n${contributorInfo ? contributorInfo + '\n' : ''}\n> ${newPrompt}\n\n`;
// Insert the new section before Contributors in README
let readmeContent = Buffer.from(readmeFile.content, 'base64').toString('utf-8');
const contributorsIndex = readmeContent.indexOf('## Contributors');
if (contributorsIndex === -1) {
readmeContent += newSection; // Append if Contributors section not found
} else {
readmeContent = readmeContent.slice(0, contributorsIndex) + newSection + readmeContent.slice(contributorsIndex);
}
// Prepare CSV content
const csvContent = Buffer.from(csvFile.content, 'base64').toString('utf-8') +
`\n"${actName.replace(/"/g, '""')}","${newPrompt.replace(/"/g, '""')}"`;
// Create a tree with both files
const { data: mainTree } = await octokit.git.getTree({
owner : event.repository.owner.login,
repo : event.repository.name,
tree_sha : pr.head.sha,
recursive : true
});
// Create blobs for both files
const [readmeBlob, csvBlob] = await Promise.all([
octokit.git.createBlob({
owner : event.repository.owner.login,
repo : event.repository.name,
content : Buffer.from(readmeContent).toString('base64'),
encoding : 'base64'
}),
octokit.git.createBlob({
owner : event.repository.owner.login,
repo : event.repository.name,
content : Buffer.from(csvContent).toString('base64'),
encoding : 'base64'
})
] );
// Create a new tree
const { data: newTree } = await octokit.git.createTree({
owner : event.repository.owner.login,
repo : event.repository.name,
base_tree : mainTree.sha,
tree : [
{
path : 'README.md' ,
mode : '100644' ,
type : 'blob' ,
sha : readmeBlob.data.sha
},
{
path : 'prompts.csv' ,
mode : '100644' ,
type : 'blob' ,
sha : csvBlob.data.sha
}
]
});
// Create a commit
const { data: newCommit } = await octokit.git.createCommit({
owner : event.repository.owner.login,
repo : event.repository.name,
message: `feat : Add "${actName}" to prompts`,
tree : newTree.sha,
parents : [ pr.head.sha]
});
// Update the reference
await octokit.git.updateRef({
owner : event.repository.owner.login,
repo : event.repository.name,
ref : `heads/${pr.head.ref}`,
sha : newCommit.sha
});
await octokit.issues.createComment({
owner : event.repository.owner.login,
repo : event.repository.name,
issue_number : issueNumber,
body : `✨ Added "${actName}" to both files`
});
} catch (error) {
console.error('Error:', error);
await octokit.issues.createComment({
owner : event.repository.owner.login,
repo : event.repository.name,
issue_number : issueNumber,
body: `❌ Error while trying to update files : ${error.message}`
});
}
return;
}
// Handle rename command specifically
if (aiCommand.startsWith('rename') || aiCommand === 'suggest title') {
if (!isPullRequest) {
await octokit.issues.createComment({
owner : event.repository.owner.login,
repo : event.repository.name,
issue_number : issueNumber,
body : '❌ The rename command can only be used on pull requests.'
});
return;
}
// Get PR details for context
const { data: pr } = await octokit.pulls.get({
owner : event.repository.owner.login,
repo : event.repository.name,
pull_number : issueNumber
});
// Get the list of files changed in the PR
const { data: files } = await octokit.pulls.listFiles({
owner : event.repository.owner.login,
repo : event.repository.name,
pull_number : issueNumber
});
// Process file changes
const fileChanges = await Promise.all(files.map(async file => {
if (file.status === 'removed') {
return `Deleted : ${file.filename}`;
}
// Get file content for added or modified files
if (file.status === 'added' || file.status === 'modified') {
const patch = file.patch || '';
return `${file.status === 'added' ? 'Added' : 'Modified'}: ${file.filename}\nChanges:\n${patch}`;
}
return `${file.status}: ${file.filename}`;
}));
const completion = await openai.chat.completions.create({
model : "gpt-3.5-turbo" ,
messages : [
{
role : "system" ,
content : "You are a helpful assistant that generates clear and concise pull request titles. Follow these rules:\n1. Use conventional commit style (feat:, fix:, docs:, etc.)\n2. Focus on WHAT changed, not HOW or WHERE\n3. Keep it short and meaningful\n4. Don't mention file names or technical implementation details\n5. Return ONLY the new title, nothing else\n\nGood examples:\n- feat: Add \"Act as a Career Coach\"\n- fix: Correct typo in Linux Terminal prompt\n- docs: Update installation instructions\n- refactor: Improve error handling"
},
{
role : "user" ,
content : `Based on these file changes, generate a concise PR title:\n\n${fileChanges.join('\n\n')}`
}
] ,
temperature : 0.5 ,
max_tokens : 60
});
const newTitle = completion.choices[0].message.content.trim();
// Update PR title
await octokit.pulls.update({
owner : event.repository.owner.login,
repo : event.repository.name,
pull_number : issueNumber,
title : newTitle
});
// Add comment about the rename
await octokit.issues.createComment({
owner : event.repository.owner.login,
repo : event.repository.name,
issue_number : issueNumber,
body: `✨ Updated PR title to : "${newTitle}" \n\nBased on the following changes:\n\`\`\`diff\n${fileChanges.join('\n')}\n\`\`\``
});
return;
}
// Handle other commands
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.'
});
