@mongoosejs/studio 0.0.76 → 0.0.78

package/backend/actions/ChatMessage/executeScript.js

@@ -27,7 +27,7 @@ module.exports = ({ db }) => async function executeScript(params) {
 
   // Create a sandbox with the db object
   const logs = [];
-  const sandbox = { db, console: {} };
+  const sandbox = { db, console: {}, ObjectId: mongoose.Types.ObjectId };
 
   // Capture console logs
   sandbox.console.log = function() {

package/backend/actions/ChatThread/createChatMessage.js

@@ -13,6 +13,10 @@ const CreateChatMessageParams = new Archetype({
   },
   content: {
     $type: String
+  },
+  authorization: {
+    $type: 'string',
+    $required: true
   }
 }).compile('CreateChatMessageParams');
 
@@ -23,6 +27,8 @@ Keep scripts concise. Avoid unnecessary comments, error handling, and temporary
 
 Do not write any imports or require() statements, that will cause the script to break.
 
+If the user approves the script, the script will run in the Node.js server and then send the response via JSON to the client. Be aware that the result of the query will be serialized to JSON before being displayed to the user.
+
 Assume the user has pre-defined schemas and models. Do not define any new schemas or models for the user.
 
 Use async/await where possible. Assume top-level await is allowed.
@@ -33,7 +39,7 @@ Format output as Markdown, including code fences for any scripts the user reques
 
 Add a brief text description of what the script does.
 
-If the user's query is best answered with a chart, return a Chart.js 4 configuration as \`return { $chart: chartJSConfig };\`. Disable ChartJS animation by default unless user asks for it.
+If the user's query is best answered with a chart, return a Chart.js 4 configuration as \`return { $chart: chartJSConfig };\`. Disable ChartJS animation by default unless user asks for it. Set responsive: true, maintainAspectRatio: false options unless the user explicitly asks.
 
 Example output:
 
@@ -50,7 +56,7 @@ Here is a description of the user's models. Assume these are the only models ava
 `.trim();
 
 module.exports = ({ db }) => async function createChatMessage(params) {
-  const { chatThreadId, userId, content, script } = new CreateChatMessageParams(params);
+  const { chatThreadId, userId, content, script, authorization } = new CreateChatMessageParams(params);
   const ChatThread = db.model('__Studio_ChatThread');
   const ChatMessage = db.model('__Studio_ChatMessage');
 
@@ -74,8 +80,8 @@ module.exports = ({ db }) => async function createChatMessage(params) {
     getChatCompletion([
       { role: 'system', content: 'Summarize the following chat thread in 6 words or less, as a helpful thread title' },
       ...llmMessages
-    ]).then(res => {
-      const title = res.choices?.[0]?.message?.content;
+    ], authorization).then(res => {
+      const title = res.response;
       chatThread.title = title;
       return chatThread.save();
     }).catch(() => {});
@@ -95,9 +101,8 @@ module.exports = ({ db }) => async function createChatMessage(params) {
       script,
       executionResult: null
     }),
-    getChatCompletion(llmMessages).then(res => {
-      const content = res.choices?.[0]?.message?.content || '';
-      console.log('Content', content, res);
+    getChatCompletion(llmMessages, authorization).then(res => {
+      const content = res.response;
       return ChatMessage.create({
         chatThreadId,
         role: 'assistant',
@@ -109,20 +114,27 @@ module.exports = ({ db }) => async function createChatMessage(params) {
   return { chatMessages };
 };
 
-async function getChatCompletion(messages, options = {}) {
-  const response = await fetch('https://api.openai.com/v1/chat/completions', {
+async function getChatCompletion(messages, authorization) {
+  const response = await fetch('https://mongoose-js.netlify.app/.netlify/functions/createChatMessage', {
     method: 'POST',
     headers: {
-      Authorization: `Bearer ${process.env.OPENAI_API_KEY}`,
+      Authorization: authorization,
       'Content-Type': 'application/json'
     },
     body: JSON.stringify({
-      model: 'gpt-4o',
-      max_tokens: 2500,
-      ...options,
       messages
     })
+  }).then(response => {
+    if (response.status < 200 || response.status >= 400) {
+      return response.json().then(data => {
+        throw new Error(`Mongoose Studio chat completion error: ${data.message}`);
+      });
+    }
+    return response;
   });
 
-  return await response.json();
-};
+  return await response.json().then(res => {
+    console.log('Response', res);
+    return res;
+  });
+}

package/express.js

@@ -63,6 +63,7 @@ module.exports = async function(apiUrl, conn, options) {
           if (!user || !roles) {
             throw new Error('Not authorized');
           }
+          req.params.authorization = authorization;
           req.params.initiatedById = user._id;
           req.params.roles = roles;
 

package/frontend/public/app.js

@@ -3767,7 +3767,7 @@ module.exports = "<div>\n  <div class=\"mb-2\">\n    <textarea class=\"border bo
 /***/ ((module) => {
 
 "use strict";
-module.exports = "<div class=\"py-2\">\n  <div v-if=\"header\" class=\"border-b border-gray-100 px-2 pb-2 text-xl font-bold\">\n    {{header}}\n  </div>\n  <div class=\"text-xl py-2\">\n    <canvas ref=\"chart\"></canvas>\n  </div>\n</div>";
+module.exports = "<div class=\"py-2 h-full\">\n  <div v-if=\"header\" class=\"border-b border-gray-100 px-2 pb-2 text-xl font-bold\">\n    {{header}}\n  </div>\n  <div class=\"text-xl py-2 h-full\">\n    <canvas ref=\"chart\"></canvas>\n  </div>\n</div>\n";
 
 /***/ }),
 

package/frontend/src/dashboard-result/dashboard-chart/dashboard-chart.html

@@ -1,8 +1,8 @@
-<div class="py-2">
+<div class="py-2 h-full">
   <div v-if="header" class="border-b border-gray-100 px-2 pb-2 text-xl font-bold">
     {{header}}
   </div>
-  <div class="text-xl py-2">
+  <div class="text-xl py-2 h-full">
     <canvas ref="chart"></canvas>
   </div>
-</div>
+</div>

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mongoosejs/studio",
-  "version": "0.0.76",
+  "version": "0.0.78",
   "dependencies": {
     "archetype": "0.13.1",
     "csv-stringify": "6.3.0",