Update app.js

app.js

@@ -29,6 +29,7 @@ const simpleId = (prefix = "") => prefix + Math.random().toString(36).slice(2, 9
 
 /**
  * Action vocabulary — keep plugin and server in sync.
+ * We also accept some common synonyms from the model and map them to canonical allowed actions.
  */
 const VALID_ACTIONS = [
   "create_part",
@@ -55,17 +56,65 @@ const VALID_ACTIONS = [
 ];
 
 /**
- * Validate/prepare model output tasks (ensure allowed actions, add ids)
+ * Map of alternate action names the model might output -> canonical action.
+ * Add more mappings here if you see similar synonyms from the model.
+ */
+const ACTION_SYNONYMS = {
+  // common variants
+  "add_bone": "create_rig",
+  "add_bones": "create_rig",
+  "create_bone": "create_rig",
+  "attach_script_to": "attach_script",
+  "modify_script": "edit_script",
+  "move": "move_to",
+  "rotate": "rotate_to",
+  "set_parent_to": "set_parent",
+  // geometry synonyms
+  "union": "create_union",
+  "make_union": "create_union",
+  "mesh_part": "create_meshpart",
+  "add_script": "create_script",
+  "insert_script": "create_script",
+  "append_script": "attach_script",
+  "add_animation": "apply_animation"
+};
+
+/**
+ * Validate/prepare model output tasks (ensure allowed actions, add ids).
+ * Also maps synonyms to canonical actions (records the originalAction in the returned object).
  */
 function validateAndPrepareTasks(arr) {
   if (!Array.isArray(arr)) throw new Error("Model output must be a JSON array of tasks.");
   return arr.map((t) => {
     if (!t || typeof t !== "object") throw new Error("Each task must be an object.");
     if (!t.action) throw new Error("Each task must have an action field.");
-    if (!VALID_ACTIONS.includes(t.action)) throw new Error(`Invalid action from model: ${t.action}`);
+
+    // Normalize action: trim and coerce to string
+    let actionRaw = String(t.action).trim();
+    const actionLower = actionRaw.toLowerCase();
+
+    // Map synonyms (case-insensitive)
+    let canonicalAction = null;
+    // First try exact match against VALID_ACTIONS (case-sensitive)
+    if (VALID_ACTIONS.includes(actionRaw)) canonicalAction = actionRaw;
+    // Then try case-insensitive match
+    if (!canonicalAction) {
+      canonicalAction = VALID_ACTIONS.find(a => a.toLowerCase() === actionLower) || null;
+    }
+    // Then try synonyms map
+    if (!canonicalAction && ACTION_SYNONYMS[actionLower]) {
+      canonicalAction = ACTION_SYNONYMS[actionLower];
+    }
+
+    if (!canonicalAction) {
+      // Not recognized — include the offending action in the error to help debugging
+      throw new Error(`Invalid action from model: ${actionRaw}`);
+    }
+
     const id = t.id || simpleId("cmd_");
     const params = t.params && typeof t.params === "object" ? t.params : {};
-    return { id, action: t.action, params, target_ref: t.target_ref || null };
+    // Attach originalAction for traceability in history/logs
+    return { id, action: canonicalAction, originalAction: actionRaw, params, target_ref: t.target_ref || null };
   });
 }
 
@@ -79,6 +128,7 @@ async function generateTasks(project) {
     commandId: h.commandId,
     status: h.status,
     message: h.message ? (typeof h.message === "string" ? (h.message.length > 120 ? h.message.slice(0, 117) + "..." : h.message) : null) : null,
+    originalAction: h.originalAction || null,
     target_ref: h.target_ref || null,
     ts: h.ts
   }));
@@ -99,7 +149,7 @@ async function generateTasks(project) {
   const systemPrompt = `
 You are a precise assistant that converts a high-level Roblox project description into an ordered JSON array of low-level editor tasks. Be spatially aware: reason about coordinates (x,y,z), sizes, rotations (euler degrees as {x,y,z}), orientations, and parent/child relationships. For every object you create, assign a stable "target_ref" string (short alphanumeric) that later tasks can reference.
 
-Return ONLY valid JSON — a single array, no commentary.
+Return ONLY valid JSON — a single array, no commentary, no extra text.
 
 Allowed actions (use exactly these): ${VALID_ACTIONS.join(", ")}.
 
@@ -112,9 +162,9 @@ Task object shape:
 
 Important rules:
 - Spatial parameters use explicit numeric fields: position: { x, y, z }, size: { x, y, z }, rotation: { x, y, z } (degrees).
-- When referencing an earlier-created object, use params.target_ref exactly as the ref string returned earlier.
+- When referencing an earlier-created object, use params.target_ref exactly as the ref string you returned earlier.
 - Do not reference Roblox-specific paths (like Workspace.Model.Part).
-- Use only the allowed actions listed above.
+- Use only the allowed actions listed above (exactly those names).
 - If you need the plugin to evaluate the scene and return properties (for feedback), use actions "feedback_request" or "list_workspace".
 - End the sequence with an action "finish" when the project is complete.
 
@@ -171,7 +221,7 @@ app.post("/projects", async (req, res) => {
       commandQueue: [],
       pendingResults: new Map(),
       running: false,
-      history: [], // { commandId, status, message, extra, target_ref, ts }
+      history: [], // { commandId, status, message, extra, target_ref, originalAction, ts }
       _refs: {}    // mapping: target_ref -> { commandId, extra, ts }
     };
 
@@ -191,7 +241,7 @@ app.post("/projects", async (req, res) => {
 });
 
 /**
- * Tweak / prompt endpoint — now includes full project context so the model knows what has been done.
+ * Tweak / prompt endpoint — includes project context so the model knows what has been done.
  * Body: { prompt: string }
  */
 app.post("/projects/:projectId/prompt", async (req, res) => {
@@ -206,7 +256,7 @@ app.post("/projects/:projectId/prompt", async (req, res) => {
       status: project.status,
       queuedCount: project.commandQueue.length,
       tasksCount: project.tasks.length,
-      recentHistory: (project.history || []).slice(-50).map(h => ({ commandId: h.commandId, status: h.status, target_ref: h.target_ref })),
+      recentHistory: (project.history || []).slice(-50).map(h => ({ commandId: h.commandId, status: h.status, target_ref: h.target_ref, originalAction: h.originalAction })),
       refsKeys: Object.keys(project._refs || {}).slice(0, 200)
     };
 
@@ -282,8 +332,15 @@ app.post("/projects/:projectId/result", (req, res) => {
     message: message || null,
     extra: extra || null,
     target_ref: target_ref || null,
+    originalAction: null, // we'll try to attach if we can find the task in history/tasks
     ts: Date.now()
   };
+
+  // Attempt to find the originalAction from project's tasks or prior mapping
+  // (task may have been dispatched earlier; search tasks and history)
+  const possibleTask = (project.tasks || []).find(t => t.id === commandId) || (project.commandQueue || []).find(t => t.id === commandId);
+  if (possibleTask && possibleTask.originalAction) entry.originalAction = possibleTask.originalAction;
+
   project.history = project.history || [];
   project.history.push(entry);
 
@@ -356,6 +413,18 @@ async function runProject(project) {
       // push to queue
       project.commandQueue.push(task);
 
+      // record that we dispatched this task (for easier traceability)
+      project.history = project.history || [];
+      project.history.push({
+        commandId: task.id,
+        status: "dispatched",
+        message: null,
+        extra: null,
+        target_ref: task.target_ref || null,
+        originalAction: task.originalAction || null,
+        ts: Date.now()
+      });
+
       // Wait for plugin result for this task
       const result = await new Promise((resolve, reject) => {
         const timeoutMs = 3 * 60 * 1000; // 3 minutes