From decd533e4ec083489a8fe41d2fd9f02c930dd437 Mon Sep 17 00:00:00 2001 From: ztimson Date: Wed, 14 Jan 2026 13:05:16 -0500 Subject: [PATCH 1/2] Added configurable labels to ticket refiner --- package.json | 2 +- src/refine.mjs | 58 ++++++++++++++++++++++++++++---------------------- 2 files changed, 33 insertions(+), 27 deletions(-) diff --git a/package.json b/package.json index 9826270..3789c7f 100644 --- a/package.json +++ b/package.json @@ -1,6 +1,6 @@ { "name": "@ztimson/ai-agents", - "version": "0.1.0", + "version": "0.1.1", "description": "AI agents", "keywords": ["ai", "review"], "author": "ztimson", diff --git a/src/refine.mjs b/src/refine.mjs index d61cc01..477c1a5 100644 --- a/src/refine.mjs +++ b/src/refine.mjs @@ -6,8 +6,8 @@ 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}); +dotenv.config({quiet: true, debug: false}); +dotenv.config({path: '.env.local', override: true, quiet: true, debug: false}); (async () => { let p = process.argv[process.argv.length - 1]; @@ -21,6 +21,10 @@ dotenv.config({path: '.env.local', override: true, quiet: true}); ticket = process.env['TICKET'], host = process.env['AI_HOST'], model = process.env['AI_MODEL'], + labelDupe = process.env['LABELS_DUPE'] || 'Review/Duplicate', + labelEnabled = process.env['LABEL_ENABLED'] || 'Review/AI', + labelsReq = process.env['LABELS_REQ'] || 'Kind/Aesthetic,Kind/Bug,Kind/DevOps,Kind/Document,Kind/Enhancement,Kind/Refactor,Kind/Security', + labelsOpt = process.env['LABELS_OPT'] || 'Breaking,Priority,QA', token = process.env['AI_TOKEN']; console.log(`Processing issue #${ticket}`); @@ -32,13 +36,13 @@ dotenv.config({path: '.env.local', override: true, quiet: true}); 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') { + if(issueData.labels?.length !== 1 || issueData.labels[0]?.name !== labelEnabled) { console.log('Skipping'); return process.exit(); } // Gather readme & template - let title = '', type = '', readme = '', readmeP = path.join(process.cwd(), 'README.md'); + let title = '', type = '', labels = [], 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 @@ -96,18 +100,27 @@ Implementation details, constraints, dependencies, design decisions 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 + name: 'add_label', + description: 'Add a label to the ticket', + args: {label: {type: 'string', description: 'Label name', required: true}}, + fn: async (args) => { + labels.push(args.label); + fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}/labels`, { + method: 'POST', + headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'}, + body: `{"labels":["${args.label}"]}` + }).then(async resp => { if(!resp.ok) throw new Error(`${resp.status} ${await resp.text()}`); }); + } }], 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 +1. Call \`title\` tool EXACTLY ONCE in format: "[Module] - [Verb] [subject]" (example: Storage - fix file uploads) +2. Identify one label from each group which best applies to the ticket: ${labelsReq.replace(',', ', ')} +3. Call the \`add_label\` tool ONCE FOR EVERY LABEL identified in the previous step +4. Filter the following labels to any that apply to this ticket: ${labelsOpt.replace(',', ', ')} +5. Call the \`add_label\` tool ONCE FOR EVERY LABEL identified in the previous step +6. Output the new ticket description in formated markdown matching the following rules: **TEMPLATE RULES:** - Use ## headers (match template exactly) @@ -128,11 +141,11 @@ Implementation details, constraints, dependencies, design decisions | **Total** | **0-15** | **SCORING:** -- Size: # of modules/layers/files changed +- Size: # of modules/layers/files affected - Complexity: Technical difficulty - Unknowns: Research/uncertainty needed -**README:** +**PROJECT README:** \`\`\`markdown ${readme.trim() || 'No README available'} \`\`\` @@ -165,11 +178,11 @@ Output ONLY markdown. No explanations, labels, or extra formatting.`}); 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}` + const hasDuplicates = (await ai.language.ask(`ID: ${issueData.id}\nTitle: ${title}\n\`\`\`markdown\n${body}\n\`\`\``, { + system: `Your job is to identify duplicates. Respond ONLY with the duplicate's ID number or "NONE" if no match exists\n\n${dupes}` }))?.pop()?.content; // Handle duplicates - if(!!hasDuplicates && (dupeId = dupeIds.find(id => new RegExp(`\\b${id}\\b`, 'm').test(hasDuplicates)))) { + if(hasDuplicates && hasDuplicates !== 'NONE' && (dupeId = dupeIds.find(id => id === hasDuplicates.trim()))) { await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}/comments`, { method: 'POST', headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'}, @@ -178,7 +191,7 @@ Output ONLY markdown. No explanations, labels, or extra formatting.`}); 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"]}' + body: `{"labels":["${labelDupe}"]}` }).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', @@ -195,15 +208,8 @@ Output ONLY markdown. No explanations, labels, or extra formatting.`}); 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}`); + console.log(`Title: ${title}\nLabels: ${labels.join(', ')}\nBody:\n${body}`); })().catch(err => { console.error(`Error: ${err.message || err.toString()}`); process.exit(1); From 5b9f8e0e131d9026d692bbce6ef19ab1ac012328 Mon Sep 17 00:00:00 2001 From: ztimson Date: Wed, 14 Jan 2026 13:14:03 -0500 Subject: [PATCH 2/2] Fixed minor bugs and typos --- src/refine.mjs | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/refine.mjs b/src/refine.mjs index 477c1a5..9ce3b9c 100644 --- a/src/refine.mjs +++ b/src/refine.mjs @@ -42,7 +42,7 @@ dotenv.config({path: '.env.local', override: true, quiet: true, debug: false}); } // Gather readme & template - let title = '', type = '', labels = [], readme = '', readmeP = path.join(process.cwd(), 'README.md'); + let title = '', labels = [], 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 @@ -105,7 +105,7 @@ Implementation details, constraints, dependencies, design decisions args: {label: {type: 'string', description: 'Label name', required: true}}, fn: async (args) => { labels.push(args.label); - fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}/labels`, { + return await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}/labels`, { method: 'POST', headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'}, body: `{"labels":["${args.label}"]}` @@ -116,11 +116,11 @@ Implementation details, constraints, dependencies, design decisions **MANDATORY STEPS:** 1. Call \`title\` tool EXACTLY ONCE in format: "[Module] - [Verb] [subject]" (example: Storage - fix file uploads) -2. Identify one label from each group which best applies to the ticket: ${labelsReq.replace(',', ', ')} +2. Identify one label from each group which best applies to the ticket: ${labelsReq.replaceAll(',', ', ')} 3. Call the \`add_label\` tool ONCE FOR EVERY LABEL identified in the previous step -4. Filter the following labels to any that apply to this ticket: ${labelsOpt.replace(',', ', ')} +4. Filter the following labels to any that apply to this ticket: ${labelsOpt.replaceAll(',', ', ')} 5. Call the \`add_label\` tool ONCE FOR EVERY LABEL identified in the previous step -6. Output the new ticket description in formated markdown matching the following rules: +6. Output the new ticket description in formatted markdown matching the following rules: **TEMPLATE RULES:** - Use ## headers (match template exactly) @@ -182,7 +182,7 @@ Output ONLY markdown. No explanations, labels, or extra formatting.`}); system: `Your job is to identify duplicates. Respond ONLY with the duplicate's ID number or "NONE" if no match exists\n\n${dupes}` }))?.pop()?.content; // Handle duplicates - if(hasDuplicates && hasDuplicates !== 'NONE' && (dupeId = dupeIds.find(id => id === hasDuplicates.trim()))) { + if(hasDuplicates && hasDuplicates !== 'NONE' && (dupeId = dupeIds.find(id => id == hasDuplicates.trim()))) { await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}/comments`, { method: 'POST', headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'},