@ztimson/ai-agents 0.0.3 → 0.0.4

package/LICENSE

@@ -1,11 +1,11 @@
-Copyright (c) 2023 Zakary Timson
-
-All Rights Reserved.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-THE SOFTWARE.
+Copyright (c) 2023 Zakary Timson
+
+All Rights Reserved.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

package/README.md

@@ -1,81 +1,84 @@
-<!-- Header -->
-<div id="top" align="center">
-  <br />
-
-  <!-- Logo -->
-  <img src="https://git.zakscode.com/repo-avatars/d2c56ffd0220751c2f4a9f6fc1e8125a9a63aeed452ae4f4ab696c084330faa2" alt="Logo" width="200" height="200">
-
-  <!-- Title -->
-### AI Agents
-
-  <!-- Description -->
-Automated AI-powered agents for automated reviews and code assistance
-
-  <!-- Repo badges -->
-[![Version](https://img.shields.io/badge/dynamic/json.svg?label=Version&style=for-the-badge&url=https://git.zakscode.com/api/v1/repos/ztimson/ai-agents/tags&query=$[0].name)](https://git.zakscode.com/ztimson/ai-agents/tags)
-[![Pull Requests](https://img.shields.io/badge/dynamic/json.svg?label=Pull%20Requests&style=for-the-badge&url=https://git.zakscode.com/api/v1/repos/ztimson/ai-agents&query=open_pr_counter)](https://git.zakscode.com/ztimson/ai-agents/pulls)
-[![Issues](https://img.shields.io/badge/dynamic/json.svg?label=Issues&style=for-the-badge&url=https://git.zakscode.com/api/v1/repos/ztimson/ai-agents&query=open_issues_count)](https://git.zakscode.com/ztimson/ai-agents/issues)
-
-<!-- Links -->
----
-  <div>
-    <a href="https://git.zakscode.com/ztimson/ai-agents/releases" target="_blank">Release Notes</a>
-    • <a href="https://git.zakscode.com/ztimson/ai-agents/issues/new?template=.github%2fissue_template%2fbug.md" target="_blank">Report a Bug</a>
-    • <a href="https://git.zakscode.com/ztimson/ai-agents/issues/new?template=.github%2fissue_template%2fenhancement.md" target="_blank">Request a Feature</a>
-  </div>
-
-  ---
-</div>
-
-## Table of Contents
-- [AI Agents](#top)
-	- [About](#about)
-		- [Built With](#built-with)
-	- [Setup](#setup)
-		- [Production](#production)
-	- [License](#license)
-
-## 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.
-
-### Built With
-[![Docker](https://img.shields.io/badge/Docker-384d54?style=for-the-badge&logo=docker)](https://docker.com/)
-[![JavaScript](https://img.shields.io/badge/JavaScript-000000?style=for-the-badge&logo=javascript)](https://javascript.com/)
-[![Node](https://img.shields.io/badge/Node.js-000000?style=for-the-badge&logo=nodedotjs)](https://nodejs.org/)
-
-## Setup
-
-<details>
-<summary>
-  <h3 id="production" style="display: inline">
-    Production
-  </h3>
-</summary>
-
-#### Instructions
-1. Run using npx: `npx @ztimson/ai-agents review`
-
-</details>
-
-<details>
-<summary>
-  <h3 id="development" style="display: inline">
-    Development
-  </h3>
-</summary>
-
-#### Prerequisites
-- [Node.js](https://nodejs.org/en/download)
-
-#### Instructions
-1. Install the dependencies: `npm i`
-2. Build library: `npm run review`
-
-</details>
-
-## License
-
-Copyright © 2025 Zakary Timson | All Rights Reserved | Available under MIT Licensing
-
-See the [license](./LICENSE) for more information.
+<!-- Header -->
+<div id="top" align="center">
+  <br />
+
+  <!-- Logo -->
+  <img src="https://git.zakscode.com/repo-avatars/309c233243bcd1c1e9b3f359ec3f59769bb01b655e8ed7b32587781be4c8b21c" alt="Logo" width="200" height="200">
+
+  <!-- Title -->
+### AI Agents
+
+  <!-- Description -->
+Automated AI-powered agents for automated reviews and code assistance
+
+  <!-- Repo badges -->
+[![Version](https://img.shields.io/badge/dynamic/json.svg?label=Version&style=for-the-badge&url=https://git.zakscode.com/api/v1/repos/ztimson/ai-agents/tags&query=$[0].name)](https://git.zakscode.com/ztimson/ai-agents/tags)
+[![Pull Requests](https://img.shields.io/badge/dynamic/json.svg?label=Pull%20Requests&style=for-the-badge&url=https://git.zakscode.com/api/v1/repos/ztimson/ai-agents&query=open_pr_counter)](https://git.zakscode.com/ztimson/ai-agents/pulls)
+[![Issues](https://img.shields.io/badge/dynamic/json.svg?label=Issues&style=for-the-badge&url=https://git.zakscode.com/api/v1/repos/ztimson/ai-agents&query=open_issues_count)](https://git.zakscode.com/ztimson/ai-agents/issues)
+
+<!-- Links -->
+---
+  <div>
+    <a href="https://git.zakscode.com/ztimson/ai-agents/releases" target="_blank">Release Notes</a>
+    • <a href="https://git.zakscode.com/ztimson/ai-agents/issues/new?template=.github%2fissue_template%2fbug.md" target="_blank">Report a Bug</a>
+    • <a href="https://git.zakscode.com/ztimson/ai-agents/issues/new?template=.github%2fissue_template%2fenhancement.md" target="_blank">Request a Feature</a>
+  </div>
+
+  ---
+</div>
+
+## Table of Contents
+- [AI Agents](#top)
+	- [About](#about)
+		- [Built With](#built-with)
+	- [Setup](#setup)
+		- [Production](#production)
+	- [License](#license)
+
+## 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.
+
+### Built With
+[![Docker](https://img.shields.io/badge/Docker-384d54?style=for-the-badge&logo=docker)](https://docker.com/)
+[![JavaScript](https://img.shields.io/badge/JavaScript-000000?style=for-the-badge&logo=javascript)](https://javascript.com/)
+[![Node](https://img.shields.io/badge/Node.js-000000?style=for-the-badge&logo=nodedotjs)](https://nodejs.org/)
+
+## Setup
+
+<details>
+<summary>
+  <h3 id="production" style="display: inline">
+    Production
+  </h3>
+</summary>
+
+#### Prerequisites
+- [Node.js](https://nodejs.org/en/download)
+
+#### Instructions
+1. Run using npx: `npx -y @ztimson/ai-agents@latest review`
+
+</details>
+
+<details>
+<summary>
+  <h3 id="development" style="display: inline">
+    Development
+  </h3>
+</summary>
+
+#### Prerequisites
+- [Node.js](https://nodejs.org/en/download)
+
+#### Instructions
+1. Install the dependencies: `npm i`
+2. Build library: `npm run review`
+
+</details>
+
+## License
+
+Copyright © 2025 Zakary Timson | All Rights Reserved | Available under MIT Licensing
+
+See the [license](./LICENSE) for more information.

package/package.json

@@ -1,12 +1,13 @@
 {
 	"name": "@ztimson/ai-agents",
-	"version": "0.0.3",
+	"version": "0.0.4",
 	"description": "AI agents",
 	"keywords": ["ai", "review"],
 	"author": "ztimson",
 	"license": "ISC",
 	"type": "module",
 	"bin": {
+		"refine": "./src/refine.mjs",
 		"review": "./src/review.mjs"
 	},
 	"dependencies": {

package/src/refine.mjs

@@ -0,0 +1,110 @@
+#!/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);
+
+	if(!p || !fs.existsSync(p)) throw new Error('Please provide a template');
+
+	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 issueRes = await fetch(`${git}/api/v1/repos/${owner}/${repo}/issues/${ticket}`, {
+		headers: {'Authorization': `token ${auth}`}
+	});
+	if(!issueRes.ok) throw new Error(`${issueRes.status} ${await issueRes.text()}`);
+	const issueData = await issueRes.json();
+	if(!issueData.labels?.some(l => l.name === 'Review/AI')) {
+		console.log('Skipping');
+		return process.exit();
+	}
+
+	let readme = '', readmeP = path.join(process.cwd(), 'README.md');
+	if(fs.existsSync(readmeP)) readme = fs.readFileSync(readmeP, 'utf-8');
+	const template = fs.readFileSync(p, 'utf-8');
+
+	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(),
+		system: `You are a ticket formatter. Transform raw issue descriptions into structured tickets.
+
+**CRITICAL RULES:**
+1. Identify the ticket type (Bug, DevOps, Enhancement, Refactor, Security)
+2. Output MUST only contain the new ticket information in markdown, no extra fluff
+3. Follow the template structure EXACTLY:
+   - Title format: [Module] - [Verb] [noun]
+     Example: Storage - Fix file uploads
+   - Fill in the identified ticket type
+   - Write a clear description
+   - For bugs: fill Steps to Reproduce with numbered list
+   - For enhancements/refactors: REMOVE the Steps to Reproduce section entirely
+   - Acceptance Criteria: convert requirements into checkboxes (- [ ])
+   - Weight scoring (0-5 each):
+     * Size: Number of modules, layers & files affected by change
+     * Complexity: Technical difficulty to implement
+     * Unknowns: Research/uncertainty in work estimation
+     * Calculate Total as sum of the three
+   - Remove sections that are not applicable based on ticket type
+   - Use proper markdown headers (##)
+
+**README:**
+\`\`\`markdown
+${readme.trim() || 'No README available'}
+\`\`\`
+
+**TEMPLATE:**
+\`\`\`markdown
+${template.trim()}
+\`\`\`
+
+Output ONLY the formatted ticket, no explanation.`
+	})
+
+	const messages = await ai.language.ask(`Title: ${issueData.title}\n\nDescription:\n${issueData.body || 'No description provided'}`).catch(() => []);
+	const content = messages?.pop()?.content;
+	if(!content) {
+		console.log('Invalid response from AI');
+		return process.exit(1);
+	}
+	const title = /^# (.+)$/m.exec(content)?.[1] || issueData.title;
+	const typeMatch = /^## Type:\s*(.+)$/m.exec(content);
+	const type = typeMatch?.[1]?.split('/')[0]?.trim() || 'Unassigned';
+	const body = content.replace(/^# .+$/m, '').replace(/^## Type:.+$/m, '').trim();
+	const updateRes = 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,
+			labels: type?.length ? [`Kind/${type[0].toUpperCase() + type.slice(1).toLowerCase()}`] : []
+		})
+	});
+	if(!updateRes.ok) throw new Error(`${updateRes.status} ${await updateRes.text()}`);
+	console.log(body);
+})();

package/src/review.mjs

@@ -30,6 +30,26 @@ dotenv.config({path: '.env.local', override: true, quiet: true, debug: false});
 	const commit = await $`cd ${root} && git log -1 --pretty=format:%H`;
 	const gitDiff = await $`cd ${root} && git diff ${branch}`;
 
+	if(!gitDiff) {
+		console.warn('No diff found');
+		return process.exit();
+	}
+
+	let existingComments = '';
+	if(git && pr) {
+		const res = await fetch(`${git}/api/v1/repos/${owner}/${repo}/pulls/${pr}/reviews`, {
+			headers: {'Authorization': `token ${auth}`}
+		});
+		if(res.ok) {
+			const reviews = await res.json();
+			const allComments = reviews.flatMap(r => r.comments || []);
+			if(allComments.length) {
+				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}};
 	if(host === 'anthropic') options = {anthropic: {model, token}};
 	else if(host === 'openai') options = {openAi: {model, token}};
@@ -37,7 +57,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. After making all recommendations, provide a brief bullet point summary in markdown.`,
+		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}`,
 		tools: [{
 			name: 'read_file',
 			description: 'Read contents of a file',
@@ -71,11 +91,6 @@ dotenv.config({path: '.env.local', override: true, quiet: true, debug: false});
 		}]
 	});
 
-	if(!gitDiff) {
-		console.warn('No diff found');
-		return process.exit();
-	}
-
 	const messages = await ai.language.ask(gitDiff);
 	const summary = messages.pop().content;
 	if(git) {