From c12ea250bbba112f2c13d496c07263977b712187 Mon Sep 17 00:00:00 2001
From: Robert Field <robertfield@quadrimular.com>
Date: Tue, 7 Oct 2025 14:47:55 +0100
Subject: [PATCH 1/2] feat: copilot init commit

---
 platform/wab/package.json                     |   3 +-
 platform/wab/src/wab/server/AppServer.ts      |  22 ++
 .../wab/src/wab/server/app-backend-real.ts    |   2 +
 .../src/wab/server/copilot-backend-real.ts    |  17 ++
 .../wab/src/wab/server/copilot-backend.ts     |   7 +
 platform/wab/src/wab/server/copilot/llms.ts   |  66 ++--
 .../wab/server/copilot/ui-copilot-chain.ts    | 228 ++++++++++++++
 platform/wab/src/wab/server/routes/copilot.ts | 283 ++++++++++++++++++
 platform/wab/src/wab/server/simple-cache.ts   |  30 ++
 9 files changed, 624 insertions(+), 34 deletions(-)
 create mode 100644 platform/wab/src/wab/server/copilot-backend-real.ts
 create mode 100644 platform/wab/src/wab/server/copilot-backend.ts
 create mode 100644 platform/wab/src/wab/server/copilot/ui-copilot-chain.ts
 create mode 100644 platform/wab/src/wab/server/routes/copilot.ts

diff --git a/platform/wab/package.json b/platform/wab/package.json
index c270cfb1d..5fef42f1f 100644
--- a/platform/wab/package.json
+++ b/platform/wab/package.json
@@ -449,7 +449,8 @@
     "validator": "^13.1.1",
     "workerpool": "^6.1.4",
     "yargs": "^16.2.0",
-    "zod": "3.25.28"
+    "zod": "3.25.28",
+    "zod-to-json-schema": "^3.24.6"
   },
   "resolutionsComments": {
     "ini": "CVE-2020-7788 for 1.3.5"
diff --git a/platform/wab/src/wab/server/AppServer.ts b/platform/wab/src/wab/server/AppServer.ts
index 962815cbe..a486589d6 100644
--- a/platform/wab/src/wab/server/AppServer.ts
+++ b/platform/wab/src/wab/server/AppServer.ts
@@ -79,6 +79,13 @@ import {
   updateTable,
 } from "@/wab/server/routes/cmse";
 import { addCommentsRoutes } from "@/wab/server/routes/comments";
+import {
+  queryCopilot,
+  queryCopilotFeedback,
+  queryPublicUiCopilot,
+  queryUiCopilot,
+  sendCopilotFeedback,
+} from "@/wab/server/routes/copilot";
 import {
   ROUTES_WITH_TIMING,
   addInternalRoutes,
@@ -1241,6 +1248,21 @@ export function addCodegenRoutes(app: express.Application) {
   );
 }
 
+export function addCopilotRoutes(app: express.Application) {
+  // Main copilot endpoint for code/chat/sql/debug
+  app.post("/api/v1/copilot", apiAuth, withNext(queryCopilot));
+
+  // UI copilot endpoint for HTML/token generation
+  app.post("/api/v1/copilot/ui", apiAuth, withNext(queryUiCopilot));
+
+  // Public UI copilot endpoint (no auth required)
+  app.post("/api/v1/copilot/ui/public", withNext(queryPublicUiCopilot));
+
+  // Copilot feedback endpoints
+  app.post("/api/v1/copilot-feedback", apiAuth, withNext(sendCopilotFeedback));
+  app.get("/api/v1/copilot-feedback", apiAuth, withNext(queryCopilotFeedback));
+}
+
 export function addMainAppServerRoutes(
   app: express.Application,
   config: Config
diff --git a/platform/wab/src/wab/server/app-backend-real.ts b/platform/wab/src/wab/server/app-backend-real.ts
index 943afc230..bc31ede09 100644
--- a/platform/wab/src/wab/server/app-backend-real.ts
+++ b/platform/wab/src/wab/server/app-backend-real.ts
@@ -2,6 +2,7 @@
 // the top so that we know if we are running production and want newrelic.
 import {
   addCodegenRoutes,
+  addCopilotRoutes,
   addMainAppServerRoutes,
   createApp,
 } from "@/wab/server/AppServer";
@@ -58,6 +59,7 @@ export async function runAppServer(config: Config) {
         // protected against.
         logger().info("Adding codegen routes");
         addCodegenRoutes(application);
+        addCopilotRoutes(application);
       }
     },
     (application) => {
diff --git a/platform/wab/src/wab/server/copilot-backend-real.ts b/platform/wab/src/wab/server/copilot-backend-real.ts
new file mode 100644
index 000000000..33148c532
--- /dev/null
+++ b/platform/wab/src/wab/server/copilot-backend-real.ts
@@ -0,0 +1,17 @@
+import { addCopilotRoutes, createApp } from "@/wab/server/AppServer";
+import { Config } from "@/wab/server/config";
+import { ensureDbConnections } from "@/wab/server/db/DbCon";
+import { runExpressApp, setupServerCli } from "@/wab/server/server-common";
+import "core-js";
+
+async function runAppServer(config: Config) {
+  await ensureDbConnections(config.databaseUri);
+
+  const { app } = await createApp("copilot", config, addCopilotRoutes);
+  return runExpressApp(app);
+}
+
+export async function copilotBackendMain() {
+  const { opts, config } = setupServerCli();
+  await runAppServer(config);
+}
diff --git a/platform/wab/src/wab/server/copilot-backend.ts b/platform/wab/src/wab/server/copilot-backend.ts
new file mode 100644
index 000000000..23063ad43
--- /dev/null
+++ b/platform/wab/src/wab/server/copilot-backend.ts
@@ -0,0 +1,7 @@
+// tslint:disable:ordered-imports
+import { spawn } from "@/wab/shared/common";
+import { copilotBackendMain } from "@/wab/server/copilot-backend-real";
+
+if (require.main === module) {
+  spawn(copilotBackendMain());
+}
diff --git a/platform/wab/src/wab/server/copilot/llms.ts b/platform/wab/src/wab/server/copilot/llms.ts
index 34eeda244..a3ab6fcb9 100644
--- a/platform/wab/src/wab/server/copilot/llms.ts
+++ b/platform/wab/src/wab/server/copilot/llms.ts
@@ -9,7 +9,11 @@ import {
   getDynamoDbSecrets,
   getOpenaiApiKey,
 } from "@/wab/server/secrets";
-import { DynamoDbCache, SimpleCache } from "@/wab/server/simple-cache";
+import {
+  DynamoDbCache,
+  InMemoryCache,
+  SimpleCache,
+} from "@/wab/server/simple-cache";
 import { last, mkShortId } from "@/wab/shared/common";
 import {
   ChatCompletionRequestMessageRoleEnum,
@@ -42,9 +46,6 @@ export class OpenAIWrapper {
     createChatCompletionRequest: CreateChatCompletionRequest,
     options?: CreateChatCompletionRequestOptions
   ) => {
-    if (verbose) {
-      logger().debug(showCompletionRequest(createChatCompletionRequest));
-    }
     const key = hash(
       JSON.stringify([
         "OpenAI.createChatCompletion",
@@ -178,35 +179,34 @@ export function getOpenAI() {
   return new OpenAI({ apiKey: openaiApiKey });
 }
 
+function createCache(): SimpleCache {
+  // Only use DynamoDB if we have valid credentials
+  if (
+    dynamoDbCredentials?.accessKeyId &&
+    dynamoDbCredentials?.secretAccessKey
+  ) {
+    try {
+      return new DynamoDbCache(
+        new DynamoDBClient({
+          credentials: {
+            ...dynamoDbCredentials,
+          },
+          region: "us-west-2",
+        })
+      );
+    } catch (error) {
+      console.warn(
+        "Failed to create DynamoDB cache, falling back to in-memory cache:",
+        error
+      );
+    }
+  }
+  // Use in-memory cache as fallback
+  return new InMemoryCache();
+}
+
 export const createOpenAIClient = (_?: DbMgr) =>
-  new OpenAIWrapper(
-    getOpenAI(),
-    new DynamoDbCache(
-      new DynamoDBClient({
-        ...(dynamoDbCredentials
-          ? {
-              credentials: {
-                ...dynamoDbCredentials,
-              },
-            }
-          : {}),
-        region: "us-west-2",
-      })
-    )
-  );
+  new OpenAIWrapper(getOpenAI(), createCache());
 
 export const createAnthropicClient = (_?: DbMgr) =>
-  new AnthropicWrapper(
-    new DynamoDbCache(
-      new DynamoDBClient({
-        ...(dynamoDbCredentials
-          ? {
-              credentials: {
-                ...dynamoDbCredentials,
-              },
-            }
-          : {}),
-        region: "us-west-2",
-      })
-    )
-  );
+  new AnthropicWrapper(createCache());
diff --git a/platform/wab/src/wab/server/copilot/ui-copilot-chain.ts b/platform/wab/src/wab/server/copilot/ui-copilot-chain.ts
new file mode 100644
index 000000000..e0b2c03b7
--- /dev/null
+++ b/platform/wab/src/wab/server/copilot/ui-copilot-chain.ts
@@ -0,0 +1,228 @@
+import { createOpenAIClient } from "@/wab/server/copilot/llms";
+import { DbMgr } from "@/wab/server/db/DbMgr";
+import {
+  CopilotUiActions,
+  CopilotUiActionsSchema,
+  CopilotUiChainProps,
+  CreateChatCompletionRequest,
+  WholeChatCompletionResponse,
+} from "@/wab/shared/copilot/prompt-utils";
+import { zodToJsonSchema } from "zod-to-json-schema";
+
+const SYSTEM_PROMPT = `You are an expert UI designer and developer assistant for Plasmic, a visual web builder. 
+  Your task is to help users create UI components by generating clean, semantic HTML with inline styles.
+  
+  When users ask for complex layouts like "landing page", "website", or "homepage", generate a COMPLETE multi-section layout, not just a single section.
+  
+  Guidelines:
+  - Generate modern, responsive, and accessible HTML fragments
+  - Use semantic HTML5 elements (div, section, button, etc.) but NO document-level tags
+  - Apply inline styles using style attributes
+  - When design tokens are provided, use CSS variables like var(--token-primary) instead of hardcoded values
+  - Ensure proper color contrast for accessibility
+  - Use flexbox or grid for layouts when appropriate
+  - Keep the HTML structure simple and maintainable
+  - Include placeholder content that makes sense in context
+  - Use rem/em units for better scalability
+  - Add appropriate aria-labels for accessibility
+  - NEVER include <html>, <body>, <head>, <meta>, <link>, or <script> tags
+  
+  IMPORTANT: You must respond with a JSON object with an "actions" array. Each action must have:
+  - "name": must be exactly "insert-html" for HTML generation or "add-token" for design tokens
+  - "data": for "insert-html", must have an "html" field with the HTML string
+  
+  Example response for HTML generation:
+  {
+    "actions": [
+      {
+        "name": "insert-html",
+        "data": {
+          "html": "<div>Your HTML here</div>"
+        }
+      }
+    ]
+  }`;
+
+const UI_GENERATION_PROMPT = `Based on the user's request, generate appropriate UI elements.
+  
+  User request: {goal}
+  
+  {tokensContext}
+  
+  {imagesContext}
+  
+  Generate HTML that fulfills this request. The HTML should:
+  1. Only include the inner content (no <html>, <head>, <body>, <meta>, or <script> tags)
+  2. Use inline styles on each element
+  3. Be responsive and modern
+  4. Follow best practices for accessibility
+  5. Be a self-contained component that can be inserted into an existing page
+  
+  For complex requests like "landing page" or "website", generate ALL necessary sections:
+  - Hero section
+  - Features/Services section
+  - About section
+  - Call-to-action section
+  - Any other relevant sections
+  
+  IMPORTANT: 
+  - Generate the COMPLETE component(s) requested, not just one part
+  - For multi-section requests, include all sections in a single HTML response
+  - Do NOT include document-level tags like html, body, head, etc.
+  
+  Respond with actions array containing either:
+  - "insert-html" action with the generated HTML fragment
+  - "add-token" action(s) if new design tokens should be created
+  `;
+
+function formatTokensContext(tokens?: CopilotUiChainProps["tokens"]): string {
+  if (!tokens || tokens.length === 0) {
+    return "";
+  }
+
+  return `Available design tokens:
+  ${tokens
+    .map(
+      (t) =>
+        `- ${t.name} (${t.type}): ${t.value} - Use as: var(--token-${t.name})`
+    )
+    .join("\n")}
+  
+  IMPORTANT: Reference these tokens in your styles using CSS variables:
+  - For a token named "primary", use: var(--token-primary)
+  - For a token named "spacing-lg", use: var(--token-spacing-lg)
+  - Example: style="background-color: var(--token-primary); padding: var(--token-spacing-lg);"
+  
+  Always use the CSS variable format var(--token-NAME) instead of the raw value when a matching token exists.`;
+}
+
+function formatImagesContext(images?: CopilotUiChainProps["images"]): string {
+  if (!images || images.length === 0) {
+    return "";
+  }
+
+  return `The user has provided ${images.length} image(s) as reference. Consider these when designing the UI.`;
+}
+
+export async function executeUiCopilotChain(
+  props: CopilotUiChainProps,
+  mgr: DbMgr
+): Promise<CopilotUiActions> {
+  const { goal, tokens, images, isPublicMode } = props;
+
+  console.log("Executing UI copilot chain with goal:", goal);
+
+  // For public mode, use simpler prompts and models
+  const model = isPublicMode ? "gpt-3.5-turbo" : "gpt-4";
+
+  const prompt = UI_GENERATION_PROMPT.replace("{goal}", goal)
+    .replace("{tokensContext}", formatTokensContext(tokens))
+    .replace("{imagesContext}", formatImagesContext(images));
+
+  const request: CreateChatCompletionRequest = {
+    model,
+    messages: [
+      {
+        role: "system",
+        content: SYSTEM_PROMPT,
+      },
+      {
+        role: "user",
+        content: prompt,
+      },
+    ],
+    temperature: 0.7,
+    max_tokens: 4000, // Increased for comprehensive landing pages
+    tools: [
+      {
+        type: "function",
+        function: {
+          name: "generate_ui",
+          description: "Generate UI elements based on user request",
+          parameters: zodToJsonSchema(CopilotUiActionsSchema) as any,
+        },
+      },
+    ],
+    tool_choice: { type: "function", function: { name: "generate_ui" } },
+  };
+
+  try {
+    const client = createOpenAIClient(mgr);
+    const response: WholeChatCompletionResponse =
+      await client.createChatCompletion(request);
+
+    const toolCalls = response.choices[0]?.message?.tool_calls;
+    if (!toolCalls || toolCalls.length === 0) {
+      throw new Error("No tool calls in response");
+    }
+
+    const toolCall = toolCalls[0];
+    if (toolCall.type !== "function") {
+      throw new Error("Unexpected tool call type");
+    }
+
+    const result = JSON.parse(toolCall.function.arguments || "{}");
+
+    // Validate the response matches our schema
+    const validated = CopilotUiActionsSchema.parse(result);
+
+    return validated;
+  } catch (error: any) {
+    console.error("Error in UI copilot chain:", error);
+    console.error("Error details:", {
+      message: error.message,
+      stack: error.stack,
+      response: error.response?.data,
+    });
+
+    // Fallback response with error message
+    return {
+      actions: [
+        {
+          name: "insert-html",
+          data: {
+            html: `<div style="padding: 20px; background-color: #f0f0f0; border-radius: 8px; text-align: center;">
+                <p style="color: #666;">Unable to generate UI. Error: ${error.message}</p>
+              </div>`,
+          },
+        },
+      ],
+    };
+  }
+}
+
+// Example function to generate UI based on common patterns
+export function generateExampleUI(type: string): string {
+  const examples: Record<string, string> = {
+    button: `<button style="background-color: #4CAF50; color: white; padding: 12px 24px; border: none; border-radius: 4px; cursor: pointer; font-size: 16px; font-weight: 500; transition: background-color 0.3s;">Click me</button>`,
+
+    card: `<div style="background-color: white; border-radius: 8px; box-shadow: 0 2px 4px rgba(0,0,0,0.1); padding: 24px; max-width: 400px;">
+        <h3 style="margin: 0 0 12px 0; color: #333; font-size: 24px;">Card Title</h3>
+        <p style="margin: 0 0 16px 0; color: #666; line-height: 1.6;">This is a card component with some example content. You can customize it as needed.</p>
+        <button style="background-color: #007bff; color: white; padding: 8px 16px; border: none; border-radius: 4px; cursor: pointer;">Learn more</button>
+      </div>`,
+
+    form: `<form style="max-width: 500px; padding: 24px;">
+        <div style="margin-bottom: 16px;">
+          <label style="display: block; margin-bottom: 8px; color: #333; font-weight: 500;">Name</label>
+          <input type="text" style="width: 100%; padding: 8px 12px; border: 1px solid #ddd; border-radius: 4px; font-size: 16px;" placeholder="Enter your name">
+        </div>
+        <div style="margin-bottom: 16px;">
+          <label style="display: block; margin-bottom: 8px; color: #333; font-weight: 500;">Email</label>
+          <input type="email" style="width: 100%; padding: 8px 12px; border: 1px solid #ddd; border-radius: 4px; font-size: 16px;" placeholder="Enter your email">
+        </div>
+        <button type="submit" style="background-color: #28a745; color: white; padding: 10px 20px; border: none; border-radius: 4px; cursor: pointer; font-size: 16px;">Submit</button>
+      </form>`,
+
+    hero: `<section style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color: white; padding: 80px 20px; text-align: center;">
+        <h1 style="font-size: 48px; margin: 0 0 20px 0; font-weight: bold;">Welcome to Our Site</h1>
+        <p style="font-size: 20px; margin: 0 0 32px 0; opacity: 0.9; max-width: 600px; margin-left: auto; margin-right: auto;">Discover amazing features and build something great today.</p>
+        <div style="display: flex; gap: 16px; justify-content: center;">
+          <button style="background-color: white; color: #667eea; padding: 12px 32px; border: none; border-radius: 4px; cursor: pointer; font-size: 16px; font-weight: 500;">Get Started</button>
+          <button style="background-color: transparent; color: white; padding: 12px 32px; border: 2px solid white; border-radius: 4px; cursor: pointer; font-size: 16px; font-weight: 500;">Learn More</button>
+        </div>
+      </section>`,
+  };
+
+  return examples[type] || examples.card;
+}
diff --git a/platform/wab/src/wab/server/routes/copilot.ts b/platform/wab/src/wab/server/routes/copilot.ts
new file mode 100644
index 000000000..4948f33cd
--- /dev/null
+++ b/platform/wab/src/wab/server/routes/copilot.ts
@@ -0,0 +1,283 @@
+import {
+  createAnthropicClient,
+  createOpenAIClient,
+} from "@/wab/server/copilot/llms";
+import { executeUiCopilotChain } from "@/wab/server/copilot/ui-copilot-chain";
+import { DbMgr } from "@/wab/server/db/DbMgr";
+import { getUser, superDbMgr, userDbMgr } from "@/wab/server/routes/util";
+import {
+  CopilotInteractionId,
+  PublicCopilotInteractionId,
+  PublicQueryCopilotUiRequest,
+  PublicQueryCopilotUiResponse,
+  QueryCopilotFeedbackRequest,
+  QueryCopilotRequest,
+  QueryCopilotResponse,
+  QueryCopilotUiRequest,
+  QueryCopilotUiResponse,
+  SendCopilotFeedbackRequest,
+} from "@/wab/shared/ApiSchema";
+import { mkShortId } from "@/wab/shared/common";
+import { Request, Response } from "express";
+
+// Handler for main copilot queries (chat, code, sql, debug)
+export async function queryCopilot(req: Request, res: Response) {
+  const mgr = userDbMgr(req);
+  const user = getUser(req);
+  const request = req.body as QueryCopilotRequest;
+
+  try {
+    // Check rate limit and track usage
+    await mgr.useCopilotAndCheckRateLimit();
+
+    let response: QueryCopilotResponse;
+
+    // Route based on request type
+    switch (request.type) {
+      case "chat":
+        response = await handleChatQuery(mgr, request, user);
+        break;
+      case "code":
+        response = await handleCodeQuery(mgr, request, user);
+        break;
+      case "code-sql":
+        response = await handleSqlQuery(mgr, request, user);
+        break;
+      case "debug":
+        response = await handleDebugQuery(mgr, request, user);
+        break;
+      default:
+        return res.status(400).json({ error: "Invalid request type" });
+    }
+
+    res.json(response);
+  } catch (error: any) {
+    if (error.name === "CopilotRateLimitExceededError") {
+      return res.status(429).json({
+        error: "Rate limit exceeded",
+        message: error.message || "You have reached your copilot usage limit",
+      });
+    }
+    console.error("Copilot error:", error);
+    return res.status(500).json({ error: "Internal server error" });
+  }
+}
+
+// Handler for UI copilot queries
+export async function queryUiCopilot(req: Request, res: Response) {
+  const mgr = userDbMgr(req);
+  const user = getUser(req);
+  const request = req.body as QueryCopilotUiRequest;
+
+  try {
+    // Check rate limit and track usage
+    await mgr.useCopilotAndCheckRateLimit();
+
+    // Generate UI based on request
+    const copilotInteractionId = mkShortId() as CopilotInteractionId;
+    const data = await handleUiQuery(mgr, request, user);
+
+    // Store interaction
+    await mgr.createCopilotInteraction({
+      userPrompt: request.goal,
+      response: JSON.stringify(data),
+      model: request.useClaude ? "claude" : "gpt",
+      projectId: request.projectId,
+      request: { messages: [], model: "gpt-3.5-turbo" }, // TODO: Add proper request object
+    });
+
+    // TODO: Track usage
+
+    const response: QueryCopilotUiResponse = {
+      data,
+      copilotInteractionId,
+    };
+
+    res.json(response);
+  } catch (error: any) {
+    if (error.name === "CopilotRateLimitExceededError") {
+      return res.status(429).json({
+        error: "Rate limit exceeded",
+        message: error.message || "You have reached your copilot usage limit",
+      });
+    }
+    console.error("UI Copilot error:", error);
+    return res.status(500).json({ error: "Internal server error" });
+  }
+}
+
+// Handler for public UI copilot queries
+export async function queryPublicUiCopilot(req: Request, res: Response) {
+  // Public endpoint doesn't require authentication, use SUPER_USER
+  const mgr = superDbMgr(req);
+  const request = req.body as PublicQueryCopilotUiRequest;
+
+  try {
+    // Simple rate limiting for public endpoint (by IP)
+    // TODO: Implement proper rate limiting
+
+    const id = mkShortId() as PublicCopilotInteractionId;
+    const data = await handlePublicUiQuery(mgr, request);
+
+    // Store public interaction
+    await mgr.createPublicCopilotInteraction({
+      userPrompt: request.goal,
+      response: JSON.stringify(data),
+      model: "gpt", // Default to GPT for public
+      request: { messages: [], model: "gpt-3.5-turbo" }, // TODO: Add proper request object
+    });
+
+    const response: PublicQueryCopilotUiResponse = {
+      data,
+      id,
+    };
+
+    res.json(response);
+  } catch (error) {
+    console.error("Public UI Copilot error:", error);
+    res.status(500).json({ error: "Internal server error" });
+  }
+}
+
+// Handler for copilot feedback submission
+export async function sendCopilotFeedback(req: Request, res: Response) {
+  const mgr = userDbMgr(req);
+  const user = getUser(req);
+  const request = req.body as SendCopilotFeedbackRequest;
+
+  try {
+    await mgr.saveCopilotFeedback({
+      copilotInteractionId: request.id,
+      projectId: request.projectId,
+      feedback: request.feedback,
+      feedbackDescription: request.feedbackDescription,
+    });
+
+    res.json({ success: true });
+  } catch (error) {
+    console.error("Feedback error:", error);
+    res.status(500).json({ error: "Internal server error" });
+  }
+}
+
+// Handler for querying copilot feedback
+export async function queryCopilotFeedback(req: Request, res: Response) {
+  const mgr = userDbMgr(req);
+  const user = getUser(req);
+  const request = req.query as unknown as QueryCopilotFeedbackRequest;
+
+  try {
+    // Only admins can query feedback
+    // TODO: Check if user is admin
+    // if (!user.isAdmin) {
+    //   return res.status(403).json({ error: "Forbidden" });
+    // }
+
+    const result = await mgr.queryCopilotFeedback({
+      query: request.query || "",
+      pageSize: request.pageSize,
+      pageIndex: request.pageIndex,
+    });
+
+    res.json(result);
+    return res.json(result);
+  } catch (error) {
+    console.error("Query feedback error:", error);
+    return res.status(500).json({ error: "Internal server error" });
+  }
+}
+
+// Helper functions for handling different query types
+async function handleChatQuery(
+  mgr: DbMgr,
+  request: any,
+  user: any
+): Promise<QueryCopilotResponse> {
+  // TODO: Implement chat query logic
+  // TODO: Implement chat query logic
+  const client = request.useClaude
+    ? createAnthropicClient(mgr)
+    : createOpenAIClient(mgr);
+
+  // For now, return a placeholder response
+  return {
+    response: "Chat query handler not yet implemented",
+  };
+}
+
+async function handleCodeQuery(
+  mgr: DbMgr,
+  request: any,
+  user: any
+): Promise<QueryCopilotResponse> {
+  // TODO: Implement code generation logic
+  return {
+    response: "Code query handler not yet implemented",
+  };
+}
+
+async function handleSqlQuery(
+  mgr: DbMgr,
+  request: any,
+  user: any
+): Promise<QueryCopilotResponse> {
+  // TODO: Implement SQL generation logic
+  return {
+    response: "SQL query handler not yet implemented",
+  };
+}
+
+async function handleDebugQuery(
+  mgr: DbMgr,
+  request: any,
+  user: any
+): Promise<QueryCopilotResponse> {
+  // TODO: Implement debug query logic
+  return {
+    response: "Debug query handler not yet implemented",
+  };
+}
+
+async function handleUiQuery(
+  mgr: DbMgr,
+  request: QueryCopilotUiRequest,
+  user: any
+): Promise<any> {
+  // Execute the UI copilot chain to generate UI
+  const result = await executeUiCopilotChain(
+    {
+      goal: request.goal,
+      tokens: request.tokens,
+      images: request.images,
+      isPublicMode: false,
+    },
+    mgr
+  );
+
+  return result;
+}
+
+async function handlePublicUiQuery(
+  mgr: DbMgr,
+  request: PublicQueryCopilotUiRequest
+): Promise<any> {
+  return {
+    actions: [
+      {
+        name: "insert-html",
+        data: {
+          html: "<div>Public UI generation not yet implemented</div>",
+        },
+      },
+    ],
+  };
+  // // Execute the UI copilot chain for public requests
+  // const result = await executeUiCopilotChain({
+  //   goal: request.goal,
+  //   tokens: undefined, // No tokens for public mode
+  //   images: undefined, // No images for public mode
+  //   isPublicMode: true,
+  // }, mgr);
+
+  // return result;
+}
diff --git a/platform/wab/src/wab/server/simple-cache.ts b/platform/wab/src/wab/server/simple-cache.ts
index dff484bfe..2c793faa6 100644
--- a/platform/wab/src/wab/server/simple-cache.ts
+++ b/platform/wab/src/wab/server/simple-cache.ts
@@ -48,3 +48,33 @@ export class DbMgrCache implements SimpleCache {
     return this.dbMgr.setKeyValue("copilot-cache", key, value);
   }
 }
+
+// Simple in-memory cache for development
+export class InMemoryCache implements SimpleCache {
+  private cache = new Map<string, { value: string; expires: number }>();
+  private ttlMs: number;
+
+  constructor(ttlMs: number = 15 * 60 * 1000) {
+    // 15 minutes default
+    this.ttlMs = ttlMs;
+  }
+
+  async get(key: string): Promise<string | undefined> {
+    const entry = this.cache.get(key);
+    if (!entry) {
+      return undefined;
+    }
+    if (Date.now() > entry.expires) {
+      this.cache.delete(key);
+      return undefined;
+    }
+    return entry.value;
+  }
+
+  async put(key: string, value: string): Promise<void> {
+    this.cache.set(key, {
+      value,
+      expires: Date.now() + this.ttlMs,
+    });
+  }
+}

From a4f33ffdb37a9481e3ad5837acca2bb762e83773 Mon Sep 17 00:00:00 2001
From: Robert Field <robertfield@quadrimular.com>
Date: Tue, 7 Oct 2025 14:50:34 +0100
Subject: [PATCH 2/2] build: latest lock

---
 platform/wab/yarn.lock | 33 ++++-----------------------------
 1 file changed, 4 insertions(+), 29 deletions(-)

diff --git a/platform/wab/yarn.lock b/platform/wab/yarn.lock
index 7edc10e0c..eb12d9a04 100644
--- a/platform/wab/yarn.lock
+++ b/platform/wab/yarn.lock
@@ -20961,16 +20961,7 @@ string-length@^5.0.1:
     char-regex "^2.0.0"
     strip-ansi "^7.0.1"
 
-"string-width-cjs@npm:string-width@^4.2.0":
-  version "4.2.3"
-  resolved "https://registry.yarnpkg.com/string-width/-/string-width-4.2.3.tgz#269c7117d27b05ad2e536830a8ec895ef9c6d010"
-  integrity sha512-wKyQRQpjJ0sIp62ErSZdGsjMJWsap5oRNihHhu6G7JVO/9jIB6UyevL+tXuOqrng8j/cxKTWyWUwvSTriiZz/g==
-  dependencies:
-    emoji-regex "^8.0.0"
-    is-fullwidth-code-point "^3.0.0"
-    strip-ansi "^6.0.1"
-
-"string-width@^1.0.2 || 2 || 3 || 4", string-width@^4.1.0, string-width@^4.2.0, string-width@^4.2.3:
+"string-width-cjs@npm:string-width@^4.2.0", "string-width@^1.0.2 || 2 || 3 || 4", string-width@^4.1.0, string-width@^4.2.0, string-width@^4.2.3:
   version "4.2.3"
   resolved "https://registry.yarnpkg.com/string-width/-/string-width-4.2.3.tgz#269c7117d27b05ad2e536830a8ec895ef9c6d010"
   integrity sha512-wKyQRQpjJ0sIp62ErSZdGsjMJWsap5oRNihHhu6G7JVO/9jIB6UyevL+tXuOqrng8j/cxKTWyWUwvSTriiZz/g==
@@ -21020,7 +21011,7 @@ stringify-object@^3.2.1:
     is-obj "^1.0.1"
     is-regexp "^1.0.0"
 
-"strip-ansi-cjs@npm:strip-ansi@^6.0.1":
+"strip-ansi-cjs@npm:strip-ansi@^6.0.1", strip-ansi@^6.0.0, strip-ansi@^6.0.1:
   version "6.0.1"
   resolved "https://registry.yarnpkg.com/strip-ansi/-/strip-ansi-6.0.1.tgz#9e26c63d30f53443e9489495b2105d37b67a85d9"
   integrity sha512-Y38VPSHcqkFrCpFnQ9vuSXmquuv5oXOKpGeT6aGrr3o3Gc9AlVa6JBfUSOCnbxGGZF+/0ooI7KrPuUSztUdU5A==
@@ -21034,13 +21025,6 @@ strip-ansi@^5.0.0, strip-ansi@^5.1.0, strip-ansi@^5.2.0:
   dependencies:
     ansi-regex "^4.1.0"
 
-strip-ansi@^6.0.0, strip-ansi@^6.0.1:
-  version "6.0.1"
-  resolved "https://registry.yarnpkg.com/strip-ansi/-/strip-ansi-6.0.1.tgz#9e26c63d30f53443e9489495b2105d37b67a85d9"
-  integrity sha512-Y38VPSHcqkFrCpFnQ9vuSXmquuv5oXOKpGeT6aGrr3o3Gc9AlVa6JBfUSOCnbxGGZF+/0ooI7KrPuUSztUdU5A==
-  dependencies:
-    ansi-regex "^5.0.1"
-
 strip-ansi@^7.0.1:
   version "7.0.1"
   resolved "https://registry.yarnpkg.com/strip-ansi/-/strip-ansi-7.0.1.tgz#61740a08ce36b61e50e65653f07060d000975fb2"
@@ -22709,7 +22693,7 @@ workerpool@^6.1.4:
   resolved "https://registry.yarnpkg.com/workerpool/-/workerpool-6.1.4.tgz#6a972b6df82e38d50248ee2820aa98e2d0ad3090"
   integrity sha512-jGWPzsUqzkow8HoAvqaPWTUPCrlPJaJ5tY8Iz7n1uCz3tTp6s3CDG0FF1NsX42WNlkRSW6Mr+CDZGnNoSsKa7g==
 
-"wrap-ansi-cjs@npm:wrap-ansi@^7.0.0":
+"wrap-ansi-cjs@npm:wrap-ansi@^7.0.0", wrap-ansi@^7.0.0:
   version "7.0.0"
   resolved "https://registry.yarnpkg.com/wrap-ansi/-/wrap-ansi-7.0.0.tgz#67e145cff510a6a6984bdf1152911d69d2eb9e43"
   integrity sha512-YVGIj2kamLSTxw6NsZjoBxfSwsn0ycdesmc4p+Q21c5zPuZ1pl+NfxVdxPtdHvmNVOQ6XSYG4AUtyt/Fi7D16Q==
@@ -22736,15 +22720,6 @@ wrap-ansi@^6.0.1, wrap-ansi@^6.2.0:
     string-width "^4.1.0"
     strip-ansi "^6.0.0"
 
-wrap-ansi@^7.0.0:
-  version "7.0.0"
-  resolved "https://registry.yarnpkg.com/wrap-ansi/-/wrap-ansi-7.0.0.tgz#67e145cff510a6a6984bdf1152911d69d2eb9e43"
-  integrity sha512-YVGIj2kamLSTxw6NsZjoBxfSwsn0ycdesmc4p+Q21c5zPuZ1pl+NfxVdxPtdHvmNVOQ6XSYG4AUtyt/Fi7D16Q==
-  dependencies:
-    ansi-styles "^4.0.0"
-    string-width "^4.1.0"
-    strip-ansi "^6.0.0"
-
 wrap-ansi@^8.1.0:
   version "8.1.0"
   resolved "https://registry.yarnpkg.com/wrap-ansi/-/wrap-ansi-8.1.0.tgz#56dc22368ee570face1b49819975d9b9a5ead214"
@@ -23016,7 +22991,7 @@ zen-observable@0.8.15:
   resolved "https://registry.yarnpkg.com/zen-observable/-/zen-observable-0.8.15.tgz#96415c512d8e3ffd920afd3889604e30b9eaac15"
   integrity sha512-PQ2PC7R9rslx84ndNBZB/Dkv8V8fZEpk83RLgXtYd0fwUgEjseMn1Dgajh2x6S8QbZAFa9p2qVCEuYZNgve0dQ==
 
-zod-to-json-schema@^3.24.1:
+zod-to-json-schema@^3.24.1, zod-to-json-schema@^3.24.6:
   version "3.24.6"
   resolved "https://registry.yarnpkg.com/zod-to-json-schema/-/zod-to-json-schema-3.24.6.tgz#5920f020c4d2647edfbb954fa036082b92c9e12d"
   integrity sha512-h/z3PKvcTcTetyjl1fkj79MHNEjm+HpD6NXheWjzOekY7kV+lwDYnHw+ivHkijnCSMz1yJaWBD9vu/Fcmk+vEg==