@blinkk/root-cms 3.0.1-beta.3 → 3.0.1-beta.4

package/dist/plugin.js

@@ -12,17 +12,18 @@ import {
   serializeAiConfig,
   summarizeDiff,
   translateString
-} from "./chunk-NXEXOHBD.js";
+} from "./chunk-MVS2NLZM.js";
 import {
-  SearchIndexService,
   runCronJobs
-} from "./chunk-TRM4MQHU.js";
+} from "./chunk-BGTUWIV6.js";
+import {
+  SearchIndexService
+} from "./chunk-C245C557.js";
 import {
   RootCMSClient,
   parseDocId,
   unmarshalData
-} from "./chunk-F4SODS5S.js";
-import "./chunk-CRK7N6RR.js";
+} from "./chunk-2BSW7SIH.js";
 import "./chunk-MLKGABMK.js";
 
 // core/plugin.ts
@@ -546,11 +547,6 @@ function api(server, options) {
       return;
     }
     const body = req.body || {};
-    const messages = body.messages || [];
-    if (!Array.isArray(messages) || messages.length === 0) {
-      res.status(400).json({ success: false, error: "MISSING_MESSAGES" });
-      return;
-    }
     const model = findModel(aiConfig, body.modelId);
     if (!model) {
       res.status(400).json({ success: false, error: "UNKNOWN_MODEL" });
@@ -561,10 +557,12 @@ function api(server, options) {
     const store = new ChatStore(cmsClient, req.user.email);
     const requestedChatId = typeof body.chatId === "string" ? body.chatId.trim() : "";
     let chatId = "";
+    let storedMessages = [];
     if (requestedChatId) {
       const existing = await store.getChat(requestedChatId);
       if (existing) {
         chatId = existing.id;
+        storedMessages = existing.messages || [];
       } else {
         const created = await store.createChat({
           id: requestedChatId,
@@ -576,6 +574,15 @@ function api(server, options) {
       const created = await store.createChat({ modelId: model.id });
       chatId = created.id;
     }
+    let messages;
+    if (body.message && typeof body.message === "object") {
+      messages = [...storedMessages, body.message];
+    } else if (Array.isArray(body.messages) && body.messages.length > 0) {
+      messages = body.messages;
+    } else {
+      res.status(400).json({ success: false, error: "MISSING_MESSAGES" });
+      return;
+    }
     try {
       const streamResponse = await runChatStream({
         rootConfig: req.rootConfig,
@@ -585,7 +592,9 @@ function api(server, options) {
         messages,
         chatId,
         user: req.user.email,
-        executionMode
+        executionMode,
+        loadCollection: (collectionId) => getCollectionSchema(req, collectionId),
+        loadAllCollections: async () => (await options.getRenderer(req)).getCollections()
       });
       streamResponse.headers.set("x-root-cms-chat-id", chatId);
       await pipeWebResponse(streamResponse, res);
@@ -628,10 +637,14 @@ function api(server, options) {
       try {
         const streamResponse = await runEditObjectStream({
           rootConfig: req.rootConfig,
+          cmsClient: new RootCMSClient(req.rootConfig),
+          user: req.user.email,
           config: aiConfig,
           model,
           messages,
-          editData: body.editData
+          editData: body.editData,
+          loadCollection: (collectionId) => getCollectionSchema(req, collectionId),
+          loadAllCollections: async () => (await options.getRenderer(req)).getCollections()
         });
         await pipeWebResponse(streamResponse, res);
       } catch (err) {
@@ -1442,6 +1455,10 @@ function cmsPlugin(options) {
      * Attaches CMS-specific middleware to the Root.js server.
      */
     configureServer: async (server, serverOptions) => {
+      server.use(
+        ["/cms/api/ai.chat", "/cms/api/ai.edit_object"],
+        bodyParser.json({ limit: "4mb" })
+      );
       server.use(bodyParser.json());
       server.use(
         (err, req, res, next) => {