Fix the JSON Error

Author: Shashikant86

app/api/chat/route.ts

@@ -8,14 +8,24 @@ export const maxDuration = 300;
 export async function POST(req: Request) {
   const {
     message,
+    mode,
     plannerModel,
     workerModel,
     sandboxProvider,
     repoUrl,
   } = await req.json();
 
   const resolvedApiKey = process.env.GOOGLE_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY;
-  const useReal = !!resolvedApiKey && process.env.SIMULATION_MODE !== 'true';
+  const realBlockedBySimulation = process.env.SIMULATION_MODE === 'true';
+  const canUseReal = !!resolvedApiKey && !realBlockedBySimulation;
+  const requestedMode = mode === 'real'
+    ? 'real'
+    : mode === 'simulation'
+      ? 'simulation'
+      : canUseReal
+        ? 'real'
+        : 'simulation';
+  const useReal = requestedMode === 'real' && canUseReal;
 
   const config: VrlmConfig = {
     ...DEFAULT_CONFIG,
@@ -35,6 +45,21 @@ export async function POST(req: Request) {
         controller.enqueue(encoder.encode(`data: ${JSON.stringify(event)}\n\n`));
       }
 
+      if (requestedMode === 'real' && !canUseReal) {
+        const reason = !resolvedApiKey
+          ? 'GOOGLE_API_KEY is not set'
+          : 'SIMULATION_MODE is true';
+
+        send({
+          type: 'error',
+          data: {
+            message: `Real mode was requested, but ${reason}. The runtime never started, so no Docker sandboxes were created.`,
+          },
+        });
+        controller.close();
+        return;
+      }
+
       if (useReal) {
         const emitter = new VrlmEventEmitter();
         const runtime = new VrlmRuntime(emitter, config);

app/demo/page.tsx

@@ -264,6 +264,7 @@ export default function DemoPage() {
           headers: { 'Content-Type': 'application/json' },
           body: JSON.stringify({
             message,
+            mode,
             sandboxProvider: settings.sandboxProvider,
             repoUrl: settings.repoUrl,
             plannerModel: settings.plannerModel,

lib/vrlm/runtime.ts

@@ -1,4 +1,4 @@
-import { generateText, stepCountIs, ToolLoopAgent } from 'ai';
+import { generateObject, generateText, stepCountIs, ToolLoopAgent } from 'ai';
 import { z } from 'zod';
 import { gateway } from '../gateway';
 import { createAgentTools } from './tools';
@@ -143,17 +143,15 @@ export class VrlmRuntime {
   // ── Planner ─────────────────────────────────────────────────────────────────
 
   private async runPlanner(goal: string): Promise<WorkerPlan> {
-    const { text } = await generateText({
+    const { object } = await generateObject({
       model: gateway(this.config.plannerModel, this.config.googleApiKey),
       system: PLANNER_SYSTEM_PROMPT,
       prompt: buildPlannerPrompt(goal),
+      schema: WorkerPlanSchema,
+      schemaName: 'WorkerPlan',
       maxOutputTokens: 2000,
     });
-
-    // Strip markdown code fences if model wrapped in ```json ... ```
-    const clean = text.replace(/^```(?:json)?\n?/m, '').replace(/\n?```$/m, '').trim();
-    const raw = JSON.parse(clean);
-    return WorkerPlanSchema.parse(raw);
+    return WorkerPlanSchema.parse(object);
   }
 
   // ── Worker ──────────────────────────────────────────────────────────────────

packages/cli/package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "agentnetes",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "Zero to a self-organizing AI agency. On demand.",
   "keywords": [
     "ai",