generated from ztimson/template
Compare commits
23 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 99a1e55471 | |||
| 23cb66544e | |||
| 9e5372f37b | |||
| eb4486f196 | |||
| f3df34ec47 | |||
| f2936ae4dc | |||
| e62e11fb75 | |||
| 604e04559b | |||
| 8add830d2b | |||
| 57bbc1fdb4 | |||
| 078892297e | |||
| 3c2d6f7824 | |||
| 2d9662e86d | |||
| 91d22b8b16 | |||
| c7f8ffb32a | |||
| 27fad6a3d3 | |||
| 3daf5442d8 | |||
| 002e809ef6 | |||
| d1c0b7a872 | |||
| 1b026529fe | |||
| af09bd0f53 | |||
| 1f48b5a872 | |||
| 6bb78d0862 |
19
.github/issue_template/ai-refinement.md
vendored
Normal file
19
.github/issue_template/ai-refinement.md
vendored
Normal file
@@ -0,0 +1,19 @@
|
|||||||
|
---
|
||||||
|
|
||||||
|
name: "AI refinement"
|
||||||
|
about: "Use AI to refine ticket"
|
||||||
|
ref: "develop"
|
||||||
|
labels:
|
||||||
|
- Review/AI
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
Describe your request:
|
||||||
|
|
||||||
|
What are you trying to do and what's happening?
|
||||||
|
|
||||||
|
How can it be fixed or improved?
|
||||||
|
|
||||||
|
Steps to reproduce?
|
||||||
|
|
||||||
|
Anything other useful information, logs or screenshots?
|
||||||
@@ -1,4 +1,5 @@
|
|||||||
name: Code review
|
name: Code review
|
||||||
|
|
||||||
on:
|
on:
|
||||||
pull_request:
|
pull_request:
|
||||||
types: [opened, synchronize, reopened]
|
types: [opened, synchronize, reopened]
|
||||||
@@ -15,7 +16,7 @@ jobs:
|
|||||||
git fetch origin ${{ github.event.pull_request.base.ref }}
|
git fetch origin ${{ github.event.pull_request.base.ref }}
|
||||||
|
|
||||||
- name: Run AI Review
|
- name: Run AI Review
|
||||||
run: npx -y @ztimson/ai-agents@latest review
|
run: npx -y -p @ztimson/ai-agents@latest review $GITHUB_WORKSPACE
|
||||||
env:
|
env:
|
||||||
AI_HOST: anthropic
|
AI_HOST: anthropic
|
||||||
AI_MODEL: claude-sonnet-4-5
|
AI_MODEL: claude-sonnet-4-5
|
||||||
26
.github/workflows/ticket-refinement.yml
vendored
Normal file
26
.github/workflows/ticket-refinement.yml
vendored
Normal file
@@ -0,0 +1,26 @@
|
|||||||
|
name: Ticket refinement
|
||||||
|
|
||||||
|
on:
|
||||||
|
issues:
|
||||||
|
types: [labeled]
|
||||||
|
jobs:
|
||||||
|
format:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
container: node:22
|
||||||
|
steps:
|
||||||
|
- name: Fetch code
|
||||||
|
run: |
|
||||||
|
git clone "$(echo ${{github.server_url}}/${{github.repository}}.git | sed s%://%://${{github.token}}@% )" .
|
||||||
|
git checkout ${{ github.event.repository.default_branch }}
|
||||||
|
|
||||||
|
- name: Run AI Formatter
|
||||||
|
run: npx -y -p @ztimson/ai-agents@latest refine
|
||||||
|
env:
|
||||||
|
AI_HOST: anthropic
|
||||||
|
AI_MODEL: claude-sonnet-4-5
|
||||||
|
AI_TOKEN: ${{ secrets.ANTHROPIC_TOKEN }}
|
||||||
|
GIT_HOST: ${{ github.server_url }}
|
||||||
|
GIT_OWNER: ${{ github.repository_owner }}
|
||||||
|
GIT_REPO: ${{ github.event.repository.name }}
|
||||||
|
GIT_TOKEN: ${{ secrets.ASSISTANT_TOKEN }}
|
||||||
|
TICKET: ${{ github.event.issue.number }}
|
||||||
@@ -9,7 +9,7 @@
|
|||||||
### AI Agents
|
### AI Agents
|
||||||
|
|
||||||
<!-- Description -->
|
<!-- Description -->
|
||||||
Automated AI-powered agents for automated reviews and code assistance
|
AI-powered Gitea agents for automating reviews and administration
|
||||||
|
|
||||||
<!-- Repo badges -->
|
<!-- Repo badges -->
|
||||||
[](https://git.zakscode.com/ztimson/ai-agents/tags)
|
[](https://git.zakscode.com/ztimson/ai-agents/tags)
|
||||||
@@ -37,7 +37,9 @@ Automated AI-powered agents for automated reviews and code assistance
|
|||||||
|
|
||||||
## About
|
## About
|
||||||
|
|
||||||
Automated code agents that uses AI to analyze git diffs and provide inline comments on pull requests. Supports Anthropic, OpenAI, and Ollama models with tool-based reviewing for precise feedback.
|
Only supports Gitea
|
||||||
|
|
||||||
|
Use LLM models from Anthropic, OpenAI, or Ollama to automate ticket refinement, code reviews, and releases.
|
||||||
|
|
||||||
### Built With
|
### Built With
|
||||||
[](https://docker.com/)
|
[](https://docker.com/)
|
||||||
|
|||||||
@@ -1,12 +1,13 @@
|
|||||||
{
|
{
|
||||||
"name": "@ztimson/ai-agents",
|
"name": "@ztimson/ai-agents",
|
||||||
"version": "0.0.3",
|
"version": "0.1.0",
|
||||||
"description": "AI agents",
|
"description": "AI agents",
|
||||||
"keywords": ["ai", "review"],
|
"keywords": ["ai", "review"],
|
||||||
"author": "ztimson",
|
"author": "ztimson",
|
||||||
"license": "ISC",
|
"license": "ISC",
|
||||||
"type": "module",
|
"type": "module",
|
||||||
"bin": {
|
"bin": {
|
||||||
|
"refine": "./src/refine.mjs",
|
||||||
"review": "./src/review.mjs"
|
"review": "./src/review.mjs"
|
||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
|
|||||||
210
src/refine.mjs
Normal file
210
src/refine.mjs
Normal file
@@ -0,0 +1,210 @@
|
|||||||
|
#!/usr/bin/env node
|
||||||
|
|
||||||
|
import {Ai} from '@ztimson/ai-utils';
|
||||||
|
import * as os from 'node:os';
|
||||||
|
import * as dotenv from 'dotenv';
|
||||||
|
import * as fs from 'node:fs';
|
||||||
|
import * as path from 'node:path';
|
||||||
|
|
||||||
|
dotenv.config({quiet: true});
|
||||||
|
dotenv.config({path: '.env.local', override: true, quiet: true});
|
||||||
|
|
||||||
|
(async () => {
|
||||||
|
let p = process.argv[process.argv.length - 1];
|
||||||
|
if(p === 'refine' || p.endsWith('refine.mjs')) p = null;
|
||||||
|
if(!/^(\/|[A-Z]:)/m.test(p)) p = path.join(process.cwd(), p);
|
||||||
|
|
||||||
|
const git = process.env['GIT_HOST'],
|
||||||
|
owner = process.env['GIT_OWNER'],
|
||||||
|
repo = process.env['GIT_REPO'],
|
||||||
|
auth = process.env['GIT_TOKEN'],
|
||||||
|
ticket = process.env['TICKET'],
|
||||||
|
host = process.env['AI_HOST'],
|
||||||
|
model = process.env['AI_MODEL'],
|
||||||
|
token = process.env['AI_TOKEN'];
|
||||||
|
|
||||||
|
console.log(`Processing issue #${ticket}`);
|
||||||
|
|
||||||
|
// Fetch issue
|
||||||
|
const issueData = await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}`, {
|
||||||
|
headers: {'Authorization': `token ${auth}`}
|
||||||
|
}).then(async resp => {
|
||||||
|
if(resp.ok) return resp.json();
|
||||||
|
else throw new Error(`${resp.status} ${await resp.text()}`);
|
||||||
|
});
|
||||||
|
if(issueData.labels?.[0] !== 1 || issueData.labels?.[0]?.name !== 'Review/AI') {
|
||||||
|
console.log('Skipping');
|
||||||
|
return process.exit();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Gather readme & template
|
||||||
|
let title = '', type = '', readme = '', readmeP = path.join(process.cwd(), 'README.md');
|
||||||
|
if(fs.existsSync(readmeP)) readme = fs.readFileSync(readmeP, 'utf-8');
|
||||||
|
const template = p ? fs.readFileSync(p, 'utf-8') : `## Description
|
||||||
|
|
||||||
|
A clear explanation of the request
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Current Behavior
|
||||||
|
|
||||||
|
what's happening now or the current state/gap
|
||||||
|
|
||||||
|
## Expected Behavior
|
||||||
|
|
||||||
|
What should happen instead
|
||||||
|
|
||||||
|
## Steps to Reproduce / Desired Flow
|
||||||
|
|
||||||
|
1. First step
|
||||||
|
2. Second step
|
||||||
|
3. Third step
|
||||||
|
|
||||||
|
## Additional Context
|
||||||
|
|
||||||
|
Logs, screenshots, links, related issues
|
||||||
|
|
||||||
|
## Acceptance Criteria
|
||||||
|
|
||||||
|
- [ ] Todo requirement
|
||||||
|
- [X] Completed requirement
|
||||||
|
|
||||||
|
## Technical Notes
|
||||||
|
|
||||||
|
Implementation details, constraints, dependencies, design decisions
|
||||||
|
|
||||||
|
|
||||||
|
| Effort / Weight | Score |
|
||||||
|
|-----------------|----------|
|
||||||
|
| Size | 0-5 |
|
||||||
|
| Complexity | 0-5 |
|
||||||
|
| Unknowns | 0-5 |
|
||||||
|
| **Total** | **0-15** |
|
||||||
|
`;
|
||||||
|
|
||||||
|
// Create AI
|
||||||
|
let options = {ollama: {model, host}};
|
||||||
|
if(host === 'anthropic') options = {anthropic: {model, token}};
|
||||||
|
else if(host === 'openai') options = {openAi: {model, token}};
|
||||||
|
const ai = new Ai({
|
||||||
|
...options,
|
||||||
|
model: [host, model],
|
||||||
|
path: process.env['path'] || os.tmpdir(),
|
||||||
|
tools: [{
|
||||||
|
name: 'title',
|
||||||
|
description: 'Set the ticket title, must be called EXACTLY ONCE',
|
||||||
|
args: {title: {type: 'string', description: 'Ticket title, must match format: Module - Verb noun', required: true}},
|
||||||
|
fn: (args) => title = args.title
|
||||||
|
}, {
|
||||||
|
name: 'type',
|
||||||
|
description: 'Set the ticket type, must be called EXACTLY ONCE',
|
||||||
|
args: {type: {type: 'string', description: 'Ticket type', enum: ['Bug', 'DevOps', 'Document', 'Enhancement', 'Refactor', 'Security'], required: true}},
|
||||||
|
fn: (args) => type = args.type
|
||||||
|
}],
|
||||||
|
system: `Transform raw tickets into structured markdown following the template EXACTLY.
|
||||||
|
|
||||||
|
**MANDATORY STEPS:**
|
||||||
|
1. Identify ticket type: Bug, DevOps, Document, Enhancement, Refactor, or Security
|
||||||
|
2. Call \`type\` tool EXACTLY ONCE with the type from step 1
|
||||||
|
3. Call \`title\` tool EXACTLY ONCE in format: "[Module] - [Verb] [subject]" (example: Storage - fix file uploads)
|
||||||
|
4. Output formatted markdown matching template structure below
|
||||||
|
|
||||||
|
**TEMPLATE RULES:**
|
||||||
|
- Use ## headers (match template exactly)
|
||||||
|
- Description: Clear summary of the request
|
||||||
|
- Current Behavior: What's happening now (remove for Document tickets)
|
||||||
|
- Expected Behavior: What should happen (remove for Document tickets)
|
||||||
|
- Steps to Reproduce: Numbered list for bugs, flow for enhancements, remove if not applicable
|
||||||
|
- Additional Context: Logs, screenshots, links provided by user
|
||||||
|
- Acceptance Criteria: Convert to checkboxes (- [ ] format)
|
||||||
|
- Technical Notes: Implementation approach, constraints, dependencies
|
||||||
|
- Weight table (use exact format below):
|
||||||
|
|
||||||
|
| Effort / Weight | Score |
|
||||||
|
|-----------------|----------|
|
||||||
|
| Size | 0-5 |
|
||||||
|
| Complexity | 0-5 |
|
||||||
|
| Unknowns | 0-5 |
|
||||||
|
| **Total** | **0-15** |
|
||||||
|
|
||||||
|
**SCORING:**
|
||||||
|
- Size: # of modules/layers/files changed
|
||||||
|
- Complexity: Technical difficulty
|
||||||
|
- Unknowns: Research/uncertainty needed
|
||||||
|
|
||||||
|
**README:**
|
||||||
|
\`\`\`markdown
|
||||||
|
${readme.trim() || 'No README available'}
|
||||||
|
\`\`\`
|
||||||
|
|
||||||
|
**TEMPLATE:**
|
||||||
|
\`\`\`markdown
|
||||||
|
${template.trim()}
|
||||||
|
\`\`\`
|
||||||
|
|
||||||
|
Output ONLY markdown. No explanations, labels, or extra formatting.`});
|
||||||
|
|
||||||
|
// Format ticket with AI
|
||||||
|
const messages = await ai.language.ask(`Title: ${issueData.title}\n\nDescription:\n${issueData.body || 'No description provided'}`).catch(() => []);
|
||||||
|
const body = messages?.pop()?.content;
|
||||||
|
if(!body) throw new Error('Invalid response from AI');
|
||||||
|
|
||||||
|
// Check for duplicates
|
||||||
|
const repoInfo = await fetch(`${git}/api/v1/repos/${owner}/${repo}`, {headers: {'Authorization': `token ${auth}`},}).then(resp => resp.ok ? resp.json() : null);
|
||||||
|
const search = await fetch(`${git}/api/v1/repos/issues/search`, {
|
||||||
|
method: 'POST',
|
||||||
|
headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'},
|
||||||
|
body: JSON.stringify({
|
||||||
|
owner,
|
||||||
|
priority_repo_id: repoInfo.id,
|
||||||
|
type: 'issues',
|
||||||
|
limit: 3,
|
||||||
|
q: title
|
||||||
|
})
|
||||||
|
}).then(resp => resp.ok ? resp.json() : []);
|
||||||
|
let dupeId = null;
|
||||||
|
const dupeIds = search.map(t => t.id);
|
||||||
|
const dupes = search.map(t => `ID: ${t.id}\nTitle: ${t.title}\n\`\`\`markdown\n${t.body}\n\`\`\``).join('\n\n');
|
||||||
|
const hasDuplicates = (await ai.language.ask(`${title}\n\`\`\`markdown\n${body}\n\`\`\``, {
|
||||||
|
system: `Your job is to identify duplicates. Respond with the ID number of the duplicate or nothing if there are no matches \n\n${dupes}`
|
||||||
|
}))?.pop()?.content;
|
||||||
|
// Handle duplicates
|
||||||
|
if(!!hasDuplicates && (dupeId = dupeIds.find(id => new RegExp(`\\b${id}\\b`, 'm').test(hasDuplicates)))) {
|
||||||
|
await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}/comments`, {
|
||||||
|
method: 'POST',
|
||||||
|
headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'},
|
||||||
|
body: `{"body": "Duplicate of #${dupeId}"}`
|
||||||
|
}).then(async resp => { if(!resp.ok) throw new Error(`${resp.status} ${await resp.text()}`); });
|
||||||
|
await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}/labels`, {
|
||||||
|
method: 'POST',
|
||||||
|
headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'},
|
||||||
|
body: '{"labels":["Reviewed/Duplicate"]}'
|
||||||
|
}).then(async resp => { if(!resp.ok) throw new Error(`${resp.status} ${await resp.text()}`); });
|
||||||
|
await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}`, {
|
||||||
|
method: 'PATCH',
|
||||||
|
headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'},
|
||||||
|
body: '{"state": "closed"}'
|
||||||
|
}).then(async resp => { if(!resp.ok) throw new Error(`${resp.status} ${await resp.text()}`); });
|
||||||
|
console.log('Duplicate');
|
||||||
|
return process.exit();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update ticket
|
||||||
|
await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}`, {
|
||||||
|
method: 'PATCH',
|
||||||
|
headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'},
|
||||||
|
body: JSON.stringify({title, body})
|
||||||
|
}).then(async resp => { if(!resp.ok) throw new Error(`${resp.status} ${await resp.text()}`); });
|
||||||
|
if(type) { // Label
|
||||||
|
await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}/labels`, {
|
||||||
|
method: 'POST',
|
||||||
|
headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'},
|
||||||
|
body: `{"labels":["Reviewed/${type[0].toUpperCase() + type.slice(1).toLowerCase()}"]}`
|
||||||
|
}).then(async resp => { if(!resp.ok) throw new Error(`${resp.status} ${await resp.text()}`); });
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log(`Title: ${title}\nType: ${type}\nBody:\n${body}`);
|
||||||
|
})().catch(err => {
|
||||||
|
console.error(`Error: ${err.message || err.toString()}`);
|
||||||
|
process.exit(1);
|
||||||
|
});
|
||||||
@@ -35,19 +35,15 @@ dotenv.config({path: '.env.local', override: true, quiet: true, debug: false});
|
|||||||
return process.exit();
|
return process.exit();
|
||||||
}
|
}
|
||||||
|
|
||||||
let existingComments = '';
|
let existingComments = 'Existing Comments:\n';
|
||||||
if(git && pr) {
|
if(git && pr) {
|
||||||
const res = await fetch(`${git}/api/v1/repos/${owner}/${repo}/pulls/${pr}/reviews`, {
|
const reviews = await fetch(`${git}/api/v1/repos/${owner}/${repo}/pulls/${pr}/reviews`, {
|
||||||
headers: {'Authorization': `token ${auth}`}
|
headers: {'Authorization': `token ${auth}`}
|
||||||
});
|
}).then(resp => resp.ok ? resp.json() : []);
|
||||||
if(res.ok) {
|
const comments = await Promise.all(reviews.map(r => fetch(`${git}/api/v1/repos/${owner}/${repo}/pulls/${pr}/reviews/${r.id}/comments`, {
|
||||||
const reviews = await res.json();
|
headers: {'Authorization': `token ${auth}`}
|
||||||
const allComments = reviews.flatMap(r => r.comments || []);
|
}).then(resp => resp.ok ? resp.json() : [])));
|
||||||
if(allComments.length) {
|
existingComments += comments.flat().map(c => `${c.path}:${c.position}\n${c.body}`).join('\n\n');
|
||||||
existingComments = '\n\nExisting review comments (DO NOT repeat these):\n' +
|
|
||||||
allComments.map(c => `- ${c.path}:${c.line || c.position}: ${c.body}`).join('\n');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
let options = {ollama: {model, host}};
|
let options = {ollama: {model, host}};
|
||||||
@@ -110,4 +106,7 @@ dotenv.config({path: '.env.local', override: true, quiet: true, debug: false});
|
|||||||
if(!res.ok) throw new Error(`${res.status} ${await res.text()}`);
|
if(!res.ok) throw new Error(`${res.status} ${await res.text()}`);
|
||||||
}
|
}
|
||||||
console.log(comments.map(c => `${c.path}${c.new_position ? `:${c.new_position}` : ''}\n${c.body}`).join('\n\n') + '\n\n' + summary);
|
console.log(comments.map(c => `${c.path}${c.new_position ? `:${c.new_position}` : ''}\n${c.body}`).join('\n\n') + '\n\n' + summary);
|
||||||
})();
|
})().catch(err => {
|
||||||
|
console.error(`Error: ${err.message || err.toString()}`);
|
||||||
|
process.exit(1);
|
||||||
|
});
|
||||||
|
|||||||
Reference in New Issue
Block a user