fix for vercel

Author: vkarpov15

backend/actions/Model/createChatMessage.js

@@ -0,0 +1,54 @@
+'use strict';
+
+const Archetype = require('archetype');
+const authorize = require('../../authorize');
+const callLLM = require('../../integrations/callLLM');
+const getModelDescriptions = require('../../helpers/getModelDescriptions');
+
+const CreateChatMessageParams = new Archetype({
+  model: {
+    $type: 'string',
+    $required: true
+  },
+  content: {
+    $type: 'string',
+    $required: true
+  },
+  documentData: {
+    $type: 'string'
+  },
+  roles: {
+    $type: ['string']
+  }
+}).compile('CreateChatMessageParams');
+
+module.exports = ({ db, options }) => async function createChatMessage(params) {
+  const { model, content, documentData, roles } = new CreateChatMessageParams(params);
+
+  await authorize('Model.createChatMessage', roles);
+
+  const Model = db.models[model];
+  if (Model == null) {
+    throw new Error(`Model ${model} not found`);
+  }
+
+  const modelDescriptions = getModelDescriptions({ models: { [Model.modelName]: Model } });
+  const context = [
+    modelDescriptions,
+    'Current draft document:\n' + (documentData || '')
+  ].join('\n\n');
+  const system = systemPrompt + '\n\n' + context + (options?.context ? '\n\n' + options.context : '');
+
+  const llmMessages = [{ role: 'user', content: [{ type: 'text', text: content }] }];
+  const res = await callLLM(llmMessages, system, options);
+
+  return { text: res.text };
+};
+
+const systemPrompt = `
+  You are a helpful assistant that drafts MongoDB documents for the user.
+
+  Use the model description and the current draft document to refine the user's intent.
+
+  Return only the updated document body as a JavaScript object literal. Do not use Markdown or code fences.
+`.trim();

backend/actions/Model/index.js

@@ -1,6 +1,7 @@
 'use strict';
 
 exports.addField = require('./addField');
+exports.createChatMessage = require('./createChatMessage');
 exports.createDocument = require('./createDocument');
 exports.deleteDocument = require('./deleteDocument');
 exports.deleteDocuments = require('./deleteDocuments');

frontend/src/api.js

@@ -147,54 +147,13 @@ if (window.MONGOOSE_STUDIO_CONFIG.isLambda) {
     updateDocument: function updateDocument(params) {
       return client.post('', { action: 'Model.updateDocument', ...params }).then(res => res.data);
     },
+    createChatMessage(params) {
+      return client.post('', { action: 'Model.createChatMessage', ...params }).then(res => res.data);
+    },
     streamChatMessage: async function* streamChatMessage(params) {
-      const accessToken = window.localStorage.getItem('_mongooseStudioAccessToken') || null;
-      const url = window.MONGOOSE_STUDIO_CONFIG.baseURL + '?' + new URLSearchParams({ action: 'Model.streamChatMessage', ...params }).toString();
-
-      const response = await fetch(url, {
-        method: 'GET',
-        headers: {
-          Authorization: `${accessToken}`,
-          Accept: 'text/event-stream'
-        }
-      });
-
-      if (!response.ok) {
-        throw new Error(`HTTP error! Status: ${response.status}`);
-      }
-
-      const reader = response.body.getReader();
-      const decoder = new TextDecoder('utf-8');
-      let buffer = '';
-
-      while (true) {
-        const { done, value } = await reader.read();
-        if (done) break;
-        buffer += decoder.decode(value, { stream: true });
-
-        let eventEnd;
-        while ((eventEnd = buffer.indexOf('\n\n')) !== -1) {
-          const eventStr = buffer.slice(0, eventEnd);
-          buffer = buffer.slice(eventEnd + 2);
-
-          // Parse SSE event
-          const lines = eventStr.split('\n');
-          let data = '';
-          for (const line of lines) {
-            if (line.startsWith('data:')) {
-              data += line.slice(5).trim();
-            }
-          }
-          if (data) {
-            try {
-              const res = JSON.parse(data);
-              yield res;
-            } catch (err) {
-              yield data;
-            }
-          }
-        }
-      }
+      // Don't stream on Next.js or Netlify for now.
+      const data = await client.post('', { action: 'Model.createChatMessage', ...params }).then(res => res.data);
+      yield { textPart: data.text };
     },
     updateDocuments: function updateDocuments(params) {
       return client.post('', { action: 'Model.updateDocuments', ...params }).then(res => res.data);
@@ -299,6 +258,9 @@ if (window.MONGOOSE_STUDIO_CONFIG.isLambda) {
     createChart: function(params) {
       return client.post('/Model/createChart', params).then(res => res.data);
     },
+    createChatMessage: function(params) {
+      return client.post('/Model/createChatMessage', params).then(res => res.data);
+    },
     createDocument: function(params) {
       return client.post('/Model/createDocument', params).then(res => res.data);
     },