@myvillage/cli 1.70.0 → 1.71.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@myvillage/cli",
-  "version": "1.70.0",
+  "version": "1.71.0",
   "description": "MyVillageOS CLI for community developers",
   "type": "module",
   "bin": {

package/src/agent-runtime/builtin-tools-registry.js

@@ -18,6 +18,12 @@ export const DATASET_APPEND_DESCRIPTION =
   'regression provide a numeric `target`. Data stays local; provenance ' +
   'is recorded automatically.';
 
+export const MODEL_PREDICT_DESCRIPTION =
+  'Run a prediction with a LOCAL trained model on this machine. ' +
+  'Returns the predicted class with per-class confidence scores ' +
+  '(classification) or a numeric value (regression). The model must already ' +
+  'exist (trained in the Studio Lab or via `myvillage model train`).';
+
 /** All native built-in tools available to every local agent. */
 export const BUILTIN_TOOL_DEFS = [
   {
@@ -26,5 +32,10 @@ export const BUILTIN_TOOL_DEFS = [
     tier: 'villager',
     builtin: true,
   },
-  // model_predict lands here when L3 (use a trained model as a tool) ships.
+  {
+    name: 'model_predict',
+    description: MODEL_PREDICT_DESCRIPTION,
+    tier: 'villager',
+    builtin: true,
+  },
 ];

package/src/agent-runtime/builtin-tools.js

@@ -15,7 +15,7 @@
 import { tool } from 'ai';
 import { z } from 'zod';
 import { appendRows, datasetExists } from '../utils/dataset-workspace.js';
-import { DATASET_APPEND_DESCRIPTION } from './builtin-tools-registry.js';
+import { DATASET_APPEND_DESCRIPTION, MODEL_PREDICT_DESCRIPTION } from './builtin-tools-registry.js';
 
 export { BUILTIN_TOOL_DEFS } from './builtin-tools-registry.js';
 
@@ -71,5 +71,37 @@ export function getBuiltinTools(agentName, config) {
         }
       },
     }),
+
+    model_predict: tool({
+      description: MODEL_PREDICT_DESCRIPTION,
+      inputSchema: z.object({
+        modelId: z.string().describe('Slug of the local trained model'),
+        input: z
+          .union([z.string(), z.array(z.number())])
+          .describe('Text, or a numeric vector for tabular models'),
+      }),
+      execute: async ({ modelId, input }) => {
+        try {
+          // Lazy-load the model/tfjs code so agent startup stays light — tfjs
+          // only loads the first time an agent actually predicts.
+          const { modelExists, readModelManifest, loadModel } = await import(
+            '../utils/model-workspace.js'
+          );
+          if (!modelExists(modelId)) {
+            return {
+              error:
+                `Model "${modelId}" not found locally. Train one first ` +
+                '(Studio Lab or `myvillage model train`).',
+            };
+          }
+          const { predictWithModel } = await import('../utils/model-trainer.js');
+          const record = readModelManifest(modelId);
+          const model = await loadModel(modelId);
+          return predictWithModel(model, record, input);
+        } catch (e) {
+          return { error: e.message };
+        }
+      },
+    }),
   };
 }