@mongoosejs/studio 0.3.5 → 0.3.7

package/backend/actions/ChatThread/createChatMessage.js

@@ -1,6 +1,7 @@
 'use strict';
 
 const Archetype = require('archetype');
+const assert = require('assert');
 const authorize = require('../../authorize');
 const callLLM = require('../../integrations/callLLM');
 const getModelDescriptions = require('../../helpers/getModelDescriptions');
@@ -16,13 +17,17 @@ const CreateChatMessageParams = new Archetype({
   content: {
     $type: 'string'
   },
+  currentDateTime: {
+    $type: 'string',
+    $validate: v => assert.ok(v == null || v.match(/^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}$/))
+  },
   roles: {
     $type: ['string']
   }
 }).compile('CreateChatMessageParams');
 
 module.exports = ({ db, studioConnection, options }) => async function createChatMessage(params) {
-  const { chatThreadId, initiatedById, content, script, roles } = new CreateChatMessageParams(params);
+  const { chatThreadId, initiatedById, content, currentDateTime, script, roles } = new CreateChatMessageParams(params);
   const ChatThread = studioConnection.model('__Studio_ChatThread');
   const ChatMessage = studioConnection.model('__Studio_ChatMessage');
 
@@ -75,7 +80,12 @@ module.exports = ({ db, studioConnection, options }) => async function createCha
   }
 
   const modelDescriptions = getModelDescriptions(db);
-  const system = systemPrompt + '\n\n' + modelDescriptions + (options?.context ? '\n\n' + options.context : '');
+  const system = [
+    systemPrompt,
+    currentDateTime ? `Current date: ${currentDateTime}` : null,
+    modelDescriptions,
+    options?.context
+  ].filter(Boolean).join('\n\n');
 
   // Create the chat message and get LLM response in parallel
   const chatMessages = await Promise.all([

package/backend/actions/ChatThread/streamChatMessage.js

@@ -1,6 +1,7 @@
 'use strict';
 
 const Archetype = require('archetype');
+const assert = require('assert');
 const authorize = require('../../authorize');
 const callLLM = require('../../integrations/callLLM');
 const streamLLM = require('../../integrations/streamLLM');
@@ -17,13 +18,17 @@ const CreateChatMessageParams = new Archetype({
   content: {
     $type: 'string'
   },
+  currentDateTime: {
+    $type: 'string',
+    $validate: v => assert.ok(v == null || v.match(/^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}$/))
+  },
   roles: {
     $type: ['string']
   }
 }).compile('CreateChatMessageParams');
 
 module.exports = ({ db, studioConnection, options }) => async function* createChatMessage(params) {
-  const { chatThreadId, initiatedById, content, script, roles } = new CreateChatMessageParams(params);
+  const { chatThreadId, initiatedById, content, currentDateTime, script, roles } = new CreateChatMessageParams(params);
   const ChatThread = studioConnection.model('__Studio_ChatThread');
   const ChatMessage = studioConnection.model('__Studio_ChatMessage');
 
@@ -76,7 +81,12 @@ module.exports = ({ db, studioConnection, options }) => async function* createCh
   }
 
   const modelDescriptions = getModelDescriptions(db);
-  const system = systemPrompt + '\n\n' + modelDescriptions + (options?.context ? '\n\n' + options.context : '');
+  const system = [
+    systemPrompt,
+    currentDateTime ? `Current date: ${currentDateTime}` : null,
+    modelDescriptions,
+    options?.context
+  ].filter(Boolean).join('\n\n');
 
   const userChatMessage = await ChatMessage.create({
     chatThreadId,

package/backend/actions/Model/createChatMessage.js

@@ -1,6 +1,7 @@
 'use strict';
 
 const Archetype = require('archetype');
+const assert = require('assert');
 const authorize = require('../../authorize');
 const callLLM = require('../../integrations/callLLM');
 const getModelDescriptions = require('../../helpers/getModelDescriptions');
@@ -17,13 +18,17 @@ const CreateChatMessageParams = new Archetype({
   documentData: {
     $type: 'string'
   },
+  currentDateTime: {
+    $type: 'string',
+    $validate: v => assert.ok(v == null || v.match(/^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}$/))
+  },
   roles: {
     $type: ['string']
   }
 }).compile('CreateChatMessageParams');
 
 module.exports = ({ db, options }) => async function createChatMessage(params) {
-  const { model, content, documentData, roles } = new CreateChatMessageParams(params);
+  const { model, content, documentData, currentDateTime, roles } = new CreateChatMessageParams(params);
 
   await authorize('Model.createChatMessage', roles);
 
@@ -37,7 +42,12 @@ module.exports = ({ db, options }) => async function createChatMessage(params) {
     modelDescriptions,
     'Current draft document:\n' + (documentData || '')
   ].join('\n\n');
-  const system = systemPrompt + '\n\n' + context + (options?.context ? '\n\n' + options.context : '');
+  const system = [
+    systemPrompt,
+    currentDateTime ? `Current date: ${currentDateTime}` : null,
+    context,
+    options?.context
+  ].filter(Boolean).join('\n\n');
 
   const llmMessages = [{ role: 'user', content: [{ type: 'text', text: content }] }];
   const res = await callLLM(llmMessages, system, options);

package/backend/actions/Model/streamChatMessage.js

@@ -1,6 +1,7 @@
 'use strict';
 
 const Archetype = require('archetype');
+const assert = require('assert');
 const authorize = require('../../authorize');
 const streamLLM = require('../../integrations/streamLLM');
 const getModelDescriptions = require('../../helpers/getModelDescriptions');
@@ -17,13 +18,18 @@ const StreamChatMessageParams = new Archetype({
   documentData: {
     $type: 'string'
   },
+  currentDateTime: {
+    $type: 'string',
+    $transform: v => v == null ? null : decodeURIComponent(v),
+    $validate: v => assert.ok(v == null || v.match(/^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}$/))
+  },
   roles: {
     $type: ['string']
   }
 }).compile('StreamChatMessageParams');
 
 module.exports = ({ db, options }) => async function* streamChatMessage(params) {
-  const { model, content, documentData, roles } = new StreamChatMessageParams(params);
+  const { model, content, documentData, currentDateTime, roles } = new StreamChatMessageParams(params);
 
   await authorize('Model.streamChatMessage', roles);
 
@@ -37,7 +43,12 @@ module.exports = ({ db, options }) => async function* streamChatMessage(params)
     modelDescriptions,
     'Current draft document:\n' + (documentData || '')
   ].join('\n\n');
-  const system = systemPrompt + '\n\n' + context + (options?.context ? '\n\n' + options.context : '');
+  const system = [
+    systemPrompt,
+    currentDateTime ? `Current date: ${currentDateTime}` : null,
+    context,
+    options?.context
+  ].filter(Boolean).join('\n\n');
 
   const llmMessages = [{ role: 'user', content: [{ type: 'text', text: content }] }];
   const textStream = streamLLM(llmMessages, system, options);

package/backend/actions/Task/getTasksOverTime.js

@@ -0,0 +1,66 @@
+'use strict';
+
+const Archetype = require('archetype');
+
+const GetTasksOverTimeParams = new Archetype({
+  start: { $type: Date },
+  end: { $type: Date },
+  bucketSizeMs: { $type: 'number' }
+}).compile('GetTasksOverTimeParams');
+
+const TRACKED_STATUSES = ['succeeded', 'failed', 'cancelled'];
+
+module.exports = ({ db }) => async function getTasksOverTime(params) {
+  params = new GetTasksOverTimeParams(params);
+  const { Task } = db.models;
+  const { start, end, bucketSizeMs } = params;
+
+  const bucketMs = (bucketSizeMs != null && bucketSizeMs > 0) ? bucketSizeMs : 5 * 60 * 1000;
+
+  const match = { status: { $in: TRACKED_STATUSES } };
+  if (start != null && end != null) {
+    match.scheduledAt = { $gte: start, $lt: end };
+  } else if (start != null) {
+    match.scheduledAt = { $gte: start };
+  }
+
+  const pipeline = [
+    { $match: match },
+    {
+      $project: {
+        status: 1,
+        bucket: {
+          $toDate: {
+            $multiply: [
+              { $floor: { $divide: [{ $toLong: '$scheduledAt' }, bucketMs] } },
+              bucketMs
+            ]
+          }
+        }
+      }
+    },
+    {
+      $group: {
+        _id: { bucket: '$bucket', status: '$status' },
+        count: { $sum: 1 }
+      }
+    },
+    {
+      $group: {
+        _id: '$_id.bucket',
+        counts: { $push: { status: '$_id.status', count: '$count' } }
+      }
+    },
+    { $sort: { _id: 1 } }
+  ];
+
+  const results = await Task.aggregate(pipeline);
+
+  return results.map(r => {
+    const bucket = { timestamp: r._id, succeeded: 0, failed: 0, cancelled: 0 };
+    for (const { status, count } of r.counts) {
+      if (status in bucket) bucket[status] = count;
+    }
+    return bucket;
+  });
+};

package/backend/actions/Task/index.js

@@ -4,5 +4,6 @@ exports.cancelTask = require('./cancelTask');
 exports.createTask = require('./createTask');
 exports.getTasks = require('./getTasks');
 exports.getTaskOverview = require('./getTaskOverview');
+exports.getTasksOverTime = require('./getTasksOverTime');
 exports.rescheduleTask = require('./rescheduleTask');
 exports.runTask = require('./runTask');