@ztimson/ai-agents 0.0.0

package/LICENSE

@@ -0,0 +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.

package/README.md

@@ -0,0 +1,81 @@
+<!-- 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.

package/package.json

@@ -0,0 +1,20 @@
+{
+	"name": "@ztimson/ai-agents",
+	"version": "0.0.0",
+	"description": "AI agents",
+	"keywords": ["ai", "review"],
+	"author": "ztimson",
+	"license": "ISC",
+	"bin": {
+		"review": "./src/review.mjs"
+	},
+	"dependencies": {
+		"@ztimson/ai-utils": "^0.2.4",
+		"@ztimson/node-utils": "^1.0.7",
+		"@ztimson/utils": "^0.28.3",
+		"dotenv": "^17.2.3"
+	},
+	"files": [
+		"src/"
+	]
+}

package/src/review.mjs

@@ -0,0 +1,94 @@
+import {Ai} from '@ztimson/ai-utils';
+import {$} from '@ztimson/node-utils';
+import * as os from 'node:os';
+import * as path from 'node:path';
+import * as fs from 'node:fs';
+import * as dotenv from 'dotenv';
+
+dotenv.config({quiet: true, debug: false});
+dotenv.config({path: '.env.local', override: true, quiet: true, debug: false});
+
+(async () => {
+	const root = process.argv[2] || process.cwd(),
+		branch = process.env['GIT_BRANCH'] || await $`cd ${root} && git symbolic-ref refs/remotes/origin/HEAD`,
+		git = process.env['GIT_HOST'],
+		owner = process.env['GIT_OWNER'],
+		repo = process.env['GIT_REPO'],
+		auth = process.env['GIT_TOKEN'],
+		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 comments = [];
+	const commit = await $`cd ${root} && git log -1 --pretty=format:%H`;
+	const gitDiff = await $`cd ${root} && git diff ${branch}`;
+
+
+	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 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.`,
+		tools: [{
+			name: 'read_file',
+			description: 'Read contents of a file',
+			args: {
+				path: {type: 'string', description: 'Path to file relative to project root'}
+			},
+			fn: (args) => fs.readFileSync(path.join(root, args.path), 'utf-8')
+		}, {
+			name: 'get_diff',
+			description: 'Check a file for differences using git',
+			args: {
+				path: {type: 'string', description: 'Path to file relative to project root'}
+			},
+			fn: async (args) => await $`git diff HEAD^..HEAD -- ${path.join(root, args.path)}`
+		}, {
+			name: 'recommend',
+			description: 'REQUIRED: Call this once for every bug, improvement, or concern identified in the review.',
+			args: {
+				file: {type: 'string', description: 'File path'},
+				line: {type: 'number', description: 'Line number in new file'},
+				comment: {type: 'string', description: 'Review comment explaining the issue'}
+			},
+			fn: (args) => {
+				comments.push({
+					path: args.file,
+					new_position: args.line,
+					body: args.comment,
+				});
+				return 'Comment recorded, continue reviewing';
+			}
+		}]
+	});
+
+	if(!gitDiff) {
+		console.warn('No diff found');
+		return process.exit();
+	}
+
+	const messages = await ai.language.ask(gitDiff);
+	const summary = messages.pop().content;
+	if(git) {
+		const res = await fetch(`${git}/api/v1/repos/${owner}/${repo}/pulls/${pr}/reviews`, {
+			method: 'POST',
+			headers: {
+				'Authorization': `token ${auth}`,
+				'Content-Type': 'application/json'
+			},
+			body: JSON.stringify({
+				body: summary,
+				commit_id: commit,
+				event: 'COMMENT',
+				comments,
+			})
+		});
+		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);
+})();