@aiready/agents 0.3.1 → 0.3.2

package/.turbo/turbo-test.log

@@ -1,18 +1,16 @@
 
 
-> @aiready/agents@0.3.0 test /Users/pengcao/projects/aiready/packages/agents
+> @aiready/agents@0.3.1 test /Users/pengcao/projects/aiready/packages/agents
 > vitest run
 
 
  RUN  v1.6.1 /Users/pengcao/projects/aiready/packages/agents
 
- ✓ src/__tests__/smoke.test.ts (1)
-   ✓ Agents Smoke Test (1)
-     ✓ should pass
+ ✓ src/__tests__/smoke.test.ts  (1 test) 1ms
 
  Test Files  1 passed (1)
       Tests  1 passed (1)
-   Start at  17:20:21
-   Duration  396ms (transform 50ms, setup 0ms, collect 47ms, tests 1ms, environment 0ms, prepare 58ms)
+   Start at  02:00:27
+   Duration  246ms (transform 46ms, setup 0ms, collect 43ms, tests 1ms, environment 0ms, prepare 54ms)
 
 [?25h

package/package.json

@@ -1,14 +1,14 @@
 {
   "name": "@aiready/agents",
-  "version": "0.3.1",
+  "version": "0.3.2",
   "description": "Agentic remediation and analysis system for AIReady",
   "main": "index.js",
   "type": "module",
   "dependencies": {
-    "mastra": "^0.18.0",
+    "@mastra/core": "^0.24.0",
     "zod": "^3.0.0",
-    "@aiready/core": "0.23.1",
-    "@aiready/cli": "0.14.1"
+    "@aiready/cli": "0.14.3",
+    "@aiready/core": "0.23.2"
   },
   "devDependencies": {
     "typescript": "^5.0.0",
@@ -17,6 +17,7 @@
   },
   "scripts": {
     "test": "vitest run",
-    "type-check": "tsc --noEmit"
+    "type-check": "tsc --noEmit",
+    "lint": "eslint ."
   }
 }

package/src/impact-agent.ts

@@ -1,4 +1,4 @@
-import { Agent } from 'mastra';
+import { Agent } from '@mastra/core';
 import { z } from 'zod';
 
 export const ImpactAgent = new Agent({
@@ -19,10 +19,7 @@ export const ImpactAgent = new Agent({
     - confidenceScore: number (0.0 to 1.0)
     - breakdown: string (explanation of the math)
   `,
-  model: {
-    provider: 'openai',
-    name: 'gpt-4o',
-  },
+  model: 'openai/gpt-4o',
 });
 
 export const ImpactSchema = z.object({

package/src/prioritization-agent.ts

@@ -1,4 +1,4 @@
-import { Agent } from 'mastra';
+import { Agent } from '@mastra/core';
 import { z } from 'zod';
 
 export const PrioritizationAgent = new Agent({
@@ -16,10 +16,7 @@ export const PrioritizationAgent = new Agent({
     - rank: 'P0' | 'P1' | 'P2' | 'P3'
     - reasoning: string explanation of the ranking decision.
   `,
-  model: {
-    provider: 'openai',
-    name: 'gpt-4o',
-  },
+  model: 'openai/gpt-4o',
 });
 
 export const PrioritySchema = z.object({

package/src/refactor-agent.ts

@@ -1,4 +1,4 @@
-import { Agent } from 'mastra';
+import { Agent } from '@mastra/core';
 import { z } from 'zod';
 
 export const RefactorAgent = new Agent({
@@ -19,10 +19,7 @@ export const RefactorAgent = new Agent({
     - Update the original call sites to use the consolidated logic.
     - Verify with type checking if possible.
   `,
-  model: {
-    provider: 'openai',
-    name: 'gpt-4o',
-  },
+  model: 'openai/gpt-4o',
 });
 
 export const RefactorResultSchema = z.object({

package/src/rename-agent.ts

@@ -1,4 +1,4 @@
-import { Agent } from 'mastra';
+import { Agent } from '@mastra/core';
 import { z } from 'zod';
 
 export const RenameAgent = new Agent({
@@ -18,10 +18,7 @@ export const RenameAgent = new Agent({
     - Propose a mapping of old -> new names.
     - Execute the rename across all affected files.
   `,
-  model: {
-    provider: 'openai',
-    name: 'gpt-4o',
-  },
+  model: 'openai/gpt-4o',
 });
 
 export const RenameResultSchema = z.object({

package/src/restructure-agent.ts

@@ -1,4 +1,4 @@
-import { Agent } from 'mastra';
+import { Agent } from '@mastra/core';
 import { z } from 'zod';
 
 export const RestructureAgent = new Agent({
@@ -17,10 +17,7 @@ export const RestructureAgent = new Agent({
     - strategy: string (explanation of the architectural vision)
     - impact: { cognitiveLoadReduction: number (0-100), fragmentationReduction: number (0-100) }
   `,
-  model: {
-    provider: 'openai',
-    name: 'gpt-4o',
-  },
+  model: 'openai/gpt-4o',
 });
 
 export const RestructureSchema = z.object({

package/src/risk-agent.ts

@@ -1,4 +1,4 @@
-import { Agent } from 'mastra';
+import { Agent } from '@mastra/core';
 import { z } from 'zod';
 
 export const RiskAssessmentAgent = new Agent({
@@ -19,10 +19,7 @@ export const RiskAssessmentAgent = new Agent({
     - autoApprove: boolean (true only for 'low' risk)
     - reasoning: string
   `,
-  model: {
-    provider: 'openai',
-    name: 'gpt-4o',
-  },
+  model: 'openai/gpt-4o',
 });
 
 export const RiskSchema = z.object({

package/src/tools/github.ts

@@ -1,4 +1,4 @@
-import { createTool } from 'mastra';
+import { createTool } from '@mastra/core';
 import { z } from 'zod';
 
 // Note: In a real implementation, this would use Octokit with appropriate auth
@@ -15,14 +15,14 @@ export const githubTools = {
       success: z.boolean(),
       branchUrl: z.string().optional(),
     }),
-    execute: async ({ input }) => {
+    execute: async (context) => {
       console.log(
-        `[GitHubTool] Creating branch ${input.branchName} from ${input.baseBranch}`
+        `[GitHubTool] Creating branch ${context.context.branchName} from ${context.context.baseBranch}`
       );
       // Mock implementation
       return {
         success: true,
-        branchUrl: `https://github.com/placeholder/${input.branchName}`,
+        branchUrl: `https://github.com/placeholder/${context.context.branchName}`,
       };
     },
   }),
@@ -42,8 +42,8 @@ export const githubTools = {
       prNumber: z.number().optional(),
       prUrl: z.string().optional(),
     }),
-    execute: async ({ input }) => {
-      console.log(`[GitHubTool] Creating PR: ${input.title}`);
+    execute: async (context) => {
+      console.log(`[GitHubTool] Creating PR: ${context.context.title}`);
       // Mock implementation
       return {
         success: true,

package/src/validation-agent.ts

@@ -1,4 +1,4 @@
-import { Agent } from 'mastra';
+import { Agent } from '@mastra/core';
 import { z } from 'zod';
 
 export const ValidationAgent = new Agent({
@@ -17,10 +17,7 @@ export const ValidationAgent = new Agent({
     - errors: array of { file: string, message: string, line?: number }
     - feedbackForRefactor: string (detailed instructions for the RefactorAgent on how to fix these errors)
   `,
-  model: {
-    provider: 'openai',
-    name: 'gpt-4o',
-  },
+  model: 'openai/gpt-4o',
 });
 
 export const ValidationSchema = z.object({

package/src/workflows/remediation-swarm.ts

@@ -1,4 +1,5 @@
-import { Workflow } from 'mastra';
+import { Workflow } from '@mastra/core';
+import { z } from 'zod';
 import { PrioritizationAgent } from '../prioritization-agent';
 import { ImpactAgent } from '../impact-agent';
 import { RestructureAgent } from '../restructure-agent';
@@ -6,34 +7,12 @@ import { RefactorAgent } from '../refactor-agent';
 import { ValidationAgent } from '../validation-agent';
 
 export const RemediationSwarm = new Workflow({
-  name: 'Remediation Swarm',
-  steps: [
-    {
-      id: 'prioritize',
-      agent: PrioritizationAgent,
-      description: 'Calculating ROI and Risk hierarchy',
-    },
-    {
-      id: 'impact',
-      agent: ImpactAgent,
-      description: 'Predicting monetary token savings',
-    },
-    {
-      id: 'restructure',
-      agent: RestructureAgent,
-      description: 'Optimizing architecture and folder structure',
-    },
-    {
-      id: 'refactor',
-      agent: RefactorAgent,
-      description: 'Executing the code consolidation',
-    },
-    {
-      id: 'validate',
-      agent: ValidationAgent,
-      description: 'Verifying types and automated tests',
-    },
-  ],
-  // Note: Complex looping and conditional logic is typically handled
-  // in the workflow's trigger or a custom execution handler in Mastra.
-});
+  id: 'remediation-swarm',
+  inputSchema: z.any(),
+  outputSchema: z.any(),
+})
+  .then(PrioritizationAgent as any)
+  .then(ImpactAgent as any)
+  .then(RestructureAgent as any)
+  .then(RefactorAgent as any)
+  .then(ValidationAgent as any);