@myvillage/cli 1.69.0 → 1.71.0

package/package.json

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

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

@@ -0,0 +1,41 @@
+/**
+ * Registry of native built-in agent tools — plain data only (no `ai`/`zod`/
+ * tfjs imports), so the Studio can read tool metadata cheaply to show built-ins
+ * in its tool palette / allowlist alongside MCP tools.
+ *
+ * Single source of truth for built-in tool name + description + tier:
+ *   - the agent loop's builtin-tools.js pulls descriptions from here
+ *   - the Studio (via cli-bridge) imports this to list selectable built-ins
+ *
+ * `builtin: true` marks them so the UI can distinguish them from MCP tools.
+ * Permission `tier` reuses the notification send-tier model.
+ */
+
+export const DATASET_APPEND_DESCRIPTION =
+  'Append labeled training example(s) to a LOCAL dataset on this machine. ' +
+  'The dataset must already exist (created via the Studio Lab or ' +
+  '`myvillage dataset create`). For classification provide `label`; for ' +
+  '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 = [
+  {
+    name: 'dataset_append',
+    description: DATASET_APPEND_DESCRIPTION,
+    tier: 'villager',
+    builtin: true,
+  },
+  {
+    name: 'model_predict',
+    description: MODEL_PREDICT_DESCRIPTION,
+    tier: 'villager',
+    builtin: true,
+  },
+];

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

@@ -15,6 +15,9 @@
 import { tool } from 'ai';
 import { z } from 'zod';
 import { appendRows, datasetExists } from '../utils/dataset-workspace.js';
+import { DATASET_APPEND_DESCRIPTION, MODEL_PREDICT_DESCRIPTION } from './builtin-tools-registry.js';
+
+export { BUILTIN_TOOL_DEFS } from './builtin-tools-registry.js';
 
 /**
  * Build the native built-in tools for an agent.
@@ -25,12 +28,7 @@ import { appendRows, datasetExists } from '../utils/dataset-workspace.js';
 export function getBuiltinTools(agentName, config) {
   return {
     dataset_append: tool({
-      description:
-        'Append labeled training example(s) to a LOCAL dataset on this machine. ' +
-        'The dataset must already exist (created via the Studio Lab or ' +
-        '`myvillage dataset create`). For classification provide `label`; for ' +
-        'regression provide a numeric `target`. Data stays local; provenance ' +
-        'is recorded automatically.',
+      description: DATASET_APPEND_DESCRIPTION,
       inputSchema: z.object({
         datasetId: z.string().describe('Slug of the local dataset to append to'),
         rows: z
@@ -73,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 };
+        }
+      },
+    }),
   };
 }