@mongoosejs/studio 0.3.5 → 0.3.6

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);