@kubuild/ai 0.4.0 → 0.6.0

package/dist/server/index.js

@@ -64,6 +64,13 @@ var KubuildAiEngine = class {
 Tone: ${request.tone}`;
       if (request.locale) userPrompt += `
 Language/Locale: ${request.locale}`;
+      if (request.conversationHistory && request.conversationHistory.length > 0) {
+        const chatContext = request.conversationHistory.map((m) => `${m.role.toUpperCase()}: ${getMessageText(m)}`).join("\n");
+        userPrompt += `
+
+Prior Conversation Discussion Context:
+${chatContext}`;
+      }
       const jsonSchema = buildJsonSchemaForMode("full-page");
       const result = await this.options.adapter.generate({
         systemPrompt,
@@ -202,18 +209,18 @@ ${JSON.stringify(request.node, null, 2)}`;
       };
     }
   }
-  /**
-   * Progressive Section Streaming Generator.
-   * Emits structured SSE events: status -> metadata -> section (one by one) -> complete.
-   */
-  async *streamPage(request, context) {
-    try {
-      this.log("info", `[SSE] Starting streamPage for prompt: "${request.prompt}"`);
-      yield {
-        type: "status",
-        message: "Analyzing requirements and planning page sections..."
-      };
-      const planSystemPrompt = `
+  buildPlanPrompts(request) {
+    let sectionGuidance = "Plan between 4 to 6 cohesive, essential sections (e.g., hero, features, testimonials, pricing, cta, footer) that fulfill the request thoroughly.";
+    if (typeof request.sectionCount === "number") {
+      sectionGuidance = `Plan exactly ${request.sectionCount} cohesive sections that fulfill the request.`;
+    } else if (request.sectionCount?.min || request.sectionCount?.max) {
+      const min = request.sectionCount.min ?? 3;
+      const max = request.sectionCount.max ?? 8;
+      sectionGuidance = `Plan between ${min} and ${max} cohesive sections that fulfill the request.`;
+    } else {
+      sectionGuidance += " If the user request mentions a specific number or list of sections, respect the user request.";
+    }
+    const planSystemPrompt = `
 You are a web architect for the KUBUILD page builder.
 Given the user's prompt, plan the website structure. Output pure JSON (no markdown fences, no explanatory text):
 {
@@ -235,44 +242,113 @@ Given the user's prompt, plan the website structure. Output pure JSON (no markdo
     }
   ]
 }
-Plan 3 cohesive, essential sections (e.g., hero, features, and cta) that fulfill the request.
+${sectionGuidance}
 `;
-      let planUserPrompt = `User Request: ${request.prompt}`;
-      if (request.stylePreference) {
-        planUserPrompt += `
+    let planUserPrompt = `User Request: ${request.prompt}`;
+    if (request.stylePreference) {
+      planUserPrompt += `
 Style Preference: ${request.stylePreference}`;
-      }
-      if (request.tone) planUserPrompt += `
+    }
+    if (request.tone) planUserPrompt += `
 Tone: ${request.tone}`;
-      if (request.locale) planUserPrompt += `
+    if (request.locale) planUserPrompt += `
 Locale: ${request.locale}`;
-      this.log("info", "[SSE] Generating website layout plan...");
+    if (request.conversationHistory && request.conversationHistory.length > 0) {
+      const chatContext = request.conversationHistory.map((m) => `${m.role.toUpperCase()}: ${getMessageText(m)}`).join("\n");
+      planUserPrompt += `
+
+Prior Conversation Discussion Context:
+${chatContext}`;
+    }
+    return { systemPrompt: planSystemPrompt, userPrompt: planUserPrompt };
+  }
+  async planPage(request, context) {
+    let rawText = "";
+    try {
+      const { systemPrompt, userPrompt } = this.buildPlanPrompts(request);
+      this.log("info", `Planning website layout for: "${request.prompt}"`);
       const planResult = await this.options.adapter.generate({
-        systemPrompt: planSystemPrompt,
-        userPrompt: planUserPrompt,
+        systemPrompt,
+        userPrompt,
         signal: context?.signal
       });
-      this.log("debug", "[SSE] Raw plan response from model", planResult.text);
+      rawText = planResult.text;
+      const plan = extractJsonFromResponse(rawText);
+      return {
+        success: true,
+        data: plan,
+        usage: planResult.usage,
+        rawModelResponse: rawText
+      };
+    } catch (err) {
+      const message = err instanceof Error ? err.message : String(err);
+      this.log("warn", "Failed to generate plan JSON, using fallback plan", message);
+      const fallbackPlan = {
+        title: "AI Generated Page",
+        description: "Generated by KUBUILD AI",
+        sections: [
+          { type: "hero", title: "Hero Banner", prompt: `Hero section for: ${request.prompt}` },
+          { type: "features", title: "Key Features", prompt: `Features grid for: ${request.prompt}` },
+          { type: "testimonials", title: "Testimonials", prompt: `Social proof and customer reviews for: ${request.prompt}` },
+          { type: "cta", title: "Call To Action", prompt: `Call to action section for: ${request.prompt}` },
+          { type: "footer", title: "Footer", prompt: `Footer navigation and copyright for: ${request.prompt}` }
+        ]
+      };
+      return {
+        success: true,
+        data: fallbackPlan,
+        rawModelResponse: rawText || void 0
+      };
+    }
+  }
+  /**
+   * Progressive Section Streaming Generator.
+   * Emits structured SSE events: status -> metadata -> section (one by one) -> complete.
+   */
+  async *streamPage(request, context) {
+    try {
+      this.log("info", `[SSE] Starting streamPage for prompt: "${request.prompt}"`);
+      yield {
+        type: "status",
+        message: "Analyzing requirements and planning page sections..."
+      };
       let plan;
-      try {
-        plan = extractJsonFromResponse(planResult.text);
-        this.log("debug", "[SSE] Parsed plan successfully", plan);
-      } catch (parseErr) {
-        this.log("warn", "[SSE] Failed to parse plan JSON, using fallback plan", parseErr);
-        plan = {
-          title: "AI Generated Page",
-          description: "Generated by KUBUILD AI",
-          sections: [
-            { type: "hero", title: "Hero Banner", prompt: `Hero section for: ${request.prompt}` },
-            { type: "features", title: "Key Features", prompt: `Features grid for: ${request.prompt}` },
-            { type: "cta", title: "Call To Action", prompt: `Call to action section for: ${request.prompt}` }
-          ]
-        };
+      if (request.plan && Array.isArray(request.plan.sections) && request.plan.sections.length > 0) {
+        plan = request.plan;
+        this.log("info", `[SSE] Using approved pre-planned structure with ${plan.sections?.length ?? 0} sections`);
+      } else {
+        const { systemPrompt: planSystemPrompt, userPrompt: planUserPrompt } = this.buildPlanPrompts(request);
+        this.log("info", "[SSE] Generating website layout plan...");
+        const planResult = await this.options.adapter.generate({
+          systemPrompt: planSystemPrompt,
+          userPrompt: planUserPrompt,
+          signal: context?.signal
+        });
+        this.log("debug", "[SSE] Raw plan response from model", planResult.text);
+        try {
+          plan = extractJsonFromResponse(planResult.text);
+          this.log("debug", "[SSE] Parsed plan successfully", plan);
+        } catch (parseErr) {
+          this.log("warn", "[SSE] Failed to parse plan JSON, using fallback plan", parseErr);
+          plan = {
+            title: "AI Generated Page",
+            description: "Generated by KUBUILD AI",
+            sections: [
+              { type: "hero", title: "Hero Banner", prompt: `Hero section for: ${request.prompt}` },
+              { type: "features", title: "Key Features", prompt: `Features grid for: ${request.prompt}` },
+              { type: "testimonials", title: "Testimonials", prompt: `Social proof for: ${request.prompt}` },
+              { type: "cta", title: "Call To Action", prompt: `Call to action section for: ${request.prompt}` },
+              { type: "footer", title: "Footer", prompt: `Footer for: ${request.prompt}` }
+            ]
+          };
+        }
       }
       const sectionsToGenerate = Array.isArray(plan.sections) && plan.sections.length > 0 ? plan.sections : [
         { type: "hero", title: "Hero Section", prompt: `Hero banner for: ${request.prompt}` },
         { type: "features", title: "Features", prompt: `Features grid for: ${request.prompt}` },
-        { type: "cta", title: "Call To Action", prompt: `CTA section for: ${request.prompt}` }
+        { type: "testimonials", title: "Testimonials", prompt: `Social proof for: ${request.prompt}` },
+        { type: "cta", title: "Call To Action", prompt: `CTA section for: ${request.prompt}` },
+        { type: "footer", title: "Footer", prompt: `Footer section for: ${request.prompt}` }
       ];
       const metadata = DocumentMetadataSchema.parse({
         title: plan.title || "AI Generated Page",
@@ -1667,7 +1743,10 @@ async function processAiRequest(engine, body, signal, agent) {
         stylePreference: payload.stylePreference,
         tone: payload.tone,
         locale: payload.locale,
-        metadata: payload.metadata
+        metadata: payload.metadata,
+        conversationHistory: payload.conversationHistory ?? payload.messages,
+        sectionCount: payload.sectionCount,
+        plan: payload.plan
       },
       { signal }
     );
@@ -1775,13 +1854,42 @@ async function processAiRequest(engine, body, signal, agent) {
       response: { success: result.stoppedBy !== "error", data: result }
     };
   }
+  if (mode === "plan") {
+    if (!payload.prompt || typeof payload.prompt !== "string") {
+      return {
+        status: 400,
+        response: {
+          success: false,
+          error: {
+            code: "INVALID_PROMPT",
+            message: '"prompt" is required for planning mode'
+          }
+        }
+      };
+    }
+    const result = await engine.planPage(
+      {
+        prompt: payload.prompt,
+        stylePreference: payload.stylePreference,
+        tone: payload.tone,
+        locale: payload.locale,
+        sectionCount: payload.sectionCount,
+        conversationHistory: payload.conversationHistory ?? payload.messages
+      },
+      { signal }
+    );
+    return {
+      status: result.success ? 200 : 500,
+      response: result
+    };
+  }
   return {
     status: 400,
     response: {
       success: false,
       error: {
         code: "UNKNOWN_MODE",
-        message: `Unsupported mode: ${String(mode)}. Supported modes: 'full-page', 'section', 'refactor', 'chat', 'agent'`
+        message: `Unsupported mode: ${String(mode)}. Supported modes: 'full-page', 'section', 'refactor', 'chat', 'agent', 'plan'`
       }
     }
   };
@@ -1964,7 +2072,10 @@ function createAiHandler(engine, options) {
               stylePreference: body.stylePreference,
               tone: body.tone,
               locale: body.locale,
-              metadata: body.metadata
+              metadata: body.metadata,
+              conversationHistory: body.conversationHistory ?? body.messages,
+              sectionCount: body.sectionCount,
+              plan: body.plan
             },
             { signal: request.signal }
           ),