@ztimson/ai-agents 0.1.1 → 0.1.2

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@ztimson/ai-agents",
-	"version": "0.1.1",
+	"version": "0.1.2",
 	"description": "AI agents",
 	"keywords": ["ai", "review"],
 	"author": "ztimson",

package/src/refine.mjs

@@ -181,8 +181,9 @@ Output ONLY markdown. No explanations, labels, or extra formatting.`});
 	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 && hasDuplicates !== 'NONE' && (dupeId = dupeIds.find(id => id == hasDuplicates.trim()))) {
+	if(hasDuplicates && !hasDuplicates.toUpperCase().includes('NONE') && (dupeId = dupeIds.find(id => id == hasDuplicates.trim())) != null && dupeId != issueData.id) {
 		await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}/comments`, {
 			method: 'POST',
 			headers: {'Authorization': `token ${auth}`, 'Content-Type': 'application/json'},

package/src/review.mjs

@@ -19,12 +19,23 @@ dotenv.config({path: '.env.local', override: true, quiet: true, debug: false});
 		owner = process.env['GIT_OWNER'],
 		repo = process.env['GIT_REPO'],
 		auth = process.env['GIT_TOKEN'],
+		labelEnabled = process.env['LABEL_ENABLED'] || 'Review/AI',
 		pr = process.env['PULL_REQUEST'],
 		host = process.env['AI_HOST'],
 		model = process.env['AI_MODEL'],
 		token = process.env['AI_TOKEN'];
 
 	console.log(`Reviewing: ${root}\n`);
+	const info = await fetch(`${git}/api/v1/repos/${owner}/${repo}/pulls/${pr}`)
+		.then(async resp => {
+			if(resp.ok) return resp.json();
+			throw new Error(`${resp.status} ${await resp.text()}`);
+		});
+	if(!info.labels.some(l => l.name === labelEnabled)) {
+		console.log('Skipping');
+		return process.exit();
+	}
+
 	const branch = process.env['GIT_BRANCH'] || await $`cd ${root} && git symbolic-ref refs/remotes/origin/HEAD`;
 	const comments = [];
 	const commit = await $`cd ${root} && git log -1 --pretty=format:%H`;
@@ -53,7 +64,7 @@ dotenv.config({path: '.env.local', override: true, quiet: true, debug: false});
 		...options,
 		model: [host, model],
 		path: process.env['path'] || os.tmpdir(),
-		system: `You are a code reviewer. Analyze the git diff and use the \`recommend\` tool for EACH issue you find. You must call \`recommend\` exactly once for every bug or improvement opportunity directly related to changes. Ignore formatting recommendations. After making all recommendations, provide some concluding remarks about the overall state of the changes.${existingComments}`,
+		system: `You are a code reviewer. Analyze the git diff and use the \`recommend\` tool for EACH issue you find. You must call \`recommend\` exactly once for every bug or improvement opportunity directly related to changes. Ignore formatting recommendations. After making all recommendations, provide a quick 75 words or less sitrep.${existingComments}`,
 		tools: [{
 			name: 'read_file',
 			description: 'Read contents of a file',
@@ -87,7 +98,16 @@ dotenv.config({path: '.env.local', override: true, quiet: true, debug: false});
 		}]
 	});
 
-	const messages = await ai.language.ask(gitDiff);
+	const messages = await ai.language.ask(`Title: ${info.title || 'None'}
+Description:
+\`\`\`md
+${info.body || 'None'}
+\`\`\`
+
+Git Diff:
+\`\`\`
+${gitDiff}
+\`\`\``);
 	const summary = messages.pop().content;
 	if(git) {
 		const res = await fetch(`${git}/api/v1/repos/${owner}/${repo}/pulls/${pr}/reviews`, {