UNPKG

tinyagent-ts

Version:

Modern TypeScript framework for building AI agents with pluggable tools and ReAct reasoning

121 lines 5.03 kB
"use strict"; var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) { if (k2 === undefined) k2 = k; var desc = Object.getOwnPropertyDescriptor(m, k); if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) { desc = { enumerable: true, get: function() { return m[k]; } }; } Object.defineProperty(o, k2, desc); }) : (function(o, m, k, k2) { if (k2 === undefined) k2 = k; o[k2] = m[k]; })); var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) { Object.defineProperty(o, "default", { enumerable: true, value: v }); }) : function(o, v) { o["default"] = v; }); var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) { var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d; if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc); else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r; return c > 3 && r && Object.defineProperty(target, key, r), r; }; var __importStar = (this && this.__importStar) || (function () { var ownKeys = function(o) { ownKeys = Object.getOwnPropertyNames || function (o) { var ar = []; for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k; return ar; }; return ownKeys(o); }; return function (mod) { if (mod && mod.__esModule) return mod; var result = {}; if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]); __setModuleDefault(result, mod); return result; }; })(); var __metadata = (this && this.__metadata) || function (k, v) { if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v); }; Object.defineProperty(exports, "__esModule", { value: true }); const dotenv = __importStar(require("dotenv")); dotenv.config(); const zod_1 = require("zod"); const decorators_1 = require("../src/decorators"); const multiStepAgent_1 = require("../src/multiStepAgent"); /** * CalculatorAgent demonstrates the ReAct loop with math tools. */ let CalculatorAgent = class CalculatorAgent extends multiStepAgent_1.MultiStepAgent { add({ a, b }) { return String(a + b); } subtract({ a, b }) { return String(a - b); } multiply({ a, b }) { return String(a * b); } divide({ a, b }) { return b === 0 ? 'Error: Division by zero' : String(a / b); } }; __decorate([ (0, decorators_1.tool)('Add two numbers', zod_1.z.object({ a: zod_1.z.number(), b: zod_1.z.number() })), __metadata("design:type", Function), __metadata("design:paramtypes", [Object]), __metadata("design:returntype", String) ], CalculatorAgent.prototype, "add", null); __decorate([ (0, decorators_1.tool)('Subtract two numbers', zod_1.z.object({ a: zod_1.z.number(), b: zod_1.z.number() })), __metadata("design:type", Function), __metadata("design:paramtypes", [Object]), __metadata("design:returntype", String) ], CalculatorAgent.prototype, "subtract", null); __decorate([ (0, decorators_1.tool)('Multiply two numbers', zod_1.z.object({ a: zod_1.z.number(), b: zod_1.z.number() })), __metadata("design:type", Function), __metadata("design:paramtypes", [Object]), __metadata("design:returntype", String) ], CalculatorAgent.prototype, "multiply", null); __decorate([ (0, decorators_1.tool)('Divide two numbers', zod_1.z.object({ a: zod_1.z.number(), b: zod_1.z.number() })), __metadata("design:type", Function), __metadata("design:paramtypes", [Object]), __metadata("design:returntype", String) ], CalculatorAgent.prototype, "divide", null); CalculatorAgent = __decorate([ (0, decorators_1.model)('openai/gpt-4.1') ], CalculatorAgent); function displaySteps(pad) { const last = pad.getSteps()[pad.getSteps().length - 1]; if (last.type === 'thought') { console.log(`🤔 ${last.text}`); } else if (last.type === 'action') { if (last.mode === 'json') { console.log(`🛠️ ${last.tool}(${JSON.stringify(last.args)})`); } else { console.log('🛠️ [code action]'); } } else if (last.type === 'observation') { console.log(`👁️ ${last.text}`); } } async function runDemo() { const agent = new CalculatorAgent(); const question = 'What is (5 * 3) + 10?'; console.log(`❓ ${question}`); const result = await agent.run(question, { trace: true, onStep: displaySteps }); console.log('✅ Final Answer:', result); } if (require.main === module) { runDemo(); } //# sourceMappingURL=react-calculator.js.map