@mongoosejs/studio 0.0.75 → 0.0.76

package/backend/actions/ChatThread/createChatMessage.js

@@ -1,6 +1,7 @@
 'use strict';
 
 const Archetype = require('archetype');
+const getModelDescriptions = require('../../helpers/getModelDescriptions');
 const mongoose = require('mongoose');
 
 const CreateChatMessageParams = new Archetype({
@@ -18,17 +19,21 @@ const CreateChatMessageParams = new Archetype({
 const systemPrompt = `
 You are a data querying assistant who writes scripts for users accessing MongoDB data using Node.js and Mongoose.
 
-Keep scripts concise. Avoid unnecessary comments and error handling, unless explicitly asked for by the user.
+Keep scripts concise. Avoid unnecessary comments, error handling, and temporary variables.
+
+Do not write any imports or require() statements, that will cause the script to break.
 
 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.
 
+Think carefully about the user's input and identify the models referred to by the user's query.
+
 Format output as Markdown, including code fences for any scripts the user requested.
 
 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 };\`
+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.
 
 Example output:
 
@@ -42,136 +47,6 @@ return { numUsers: users.length };
 -----------
 
 Here is a description of the user's models. Assume these are the only models available in the system unless explicitly instructed otherwise by the user.
-
-Vehicle (collection name: Vehicle)
-  - _id: ObjectId
-  - userId: ObjectId (ref User)
-  - authorizedUsers: Array
-  - status: String
-  - _previousStatus: String
-  - make: String
-  - model: String
-  - year: Number
-  - color: String
-  - nickName: String
-  - tollTagId: String
-  - availabilityDate: String
-  - vehicleFeatures: Array
-  - vehicleIdentityId: ObjectId (ref VehicleIdentity)
-  - images: Object
-  - thumbnail: String
-  - photos: Array
-  - registrationPhoto: String
-  - smartcarId: String
-  - smartcarPermissions: Array
-  - percentRemaining: Number
-  - range: Number
-  - location: Object
-  - locationTimestamp: Date
-  - pricePerMonth: Number
-  - pricePerWeek: Number
-  - returnAddress: String
-  - returnLocation: Object
-  - description: String
-  - maxUnlockRadiusMiles: Number
-  - returnRadiusKM: Number
-  - numReviews: Number
-  - numCompletedTrips: Number
-  - totalReviewScore: Number
-  - totalRevenueGenerated: Number
-  - bookingsCount: Number
-  - odometer: Number
-  - pickupInstructions: String
-  - returnInstructions: String
-  - unlockInstructions: String
-  - chargingState: String
-  - isPluggedIn: Boolean
-  - currentBookingId: ObjectId (ref Booking)
-  - maxTripLengthDays: Number
-  - smartcarWebhookIds: Array
-  - lastErrorMessage: String
-  - vinNumber: String
-  - licensePlate: String
-  - licensePlateState: String
-  - registrationExpirationDate: String
-  - style: Object
-  - driveUnit: Object
-  - efficiencyPackage: String
-  - performancePackage: String
-  - connectivityProvider: String
-  - abiId: String
-  - platformFeePercentage: Number
-  - promotionsEnabled: Boolean
-  - modelDescription: String
-  - isConcierge: Boolean
-  - lastLoadedSuperchargerInvoicesAt: Date
-  - privateBookingNotes: String
-  - publicMarketingNotes: String
-  - fleetKeyState: Object
-  - weakBattery: Boolean
-  - fleetITconnectionStartDate: Date
-  - fleetITId: Number
-  - interiorTrim: String
-  - skipLoadingSuperchargerInvoices: Boolean
-  - chargeKey: Object
-  - securityDepositPerUser: Array
-
-User (collection name User)
-  - _id: ObjectId
-  - firstName: String
-  - lastName: String
-  - email: String (required, lowercase)
-  - picture: String
-  - zipCode: String
-  - telephone: String
-  - verifiedTelephone: Boolean
-  - verifiedEmail: Boolean
-  - googleUserId: String
-  - appleUserId: String
-  - smartcarUserId: String
-  - stripeCustomerId: String
-  - roles: Array of String (enum: 'root', 'admin', 'manager', 'host', 'verified-driver', 'internal-tester', 'authorized-payer')
-  - accountPreference: String (enum: 'host', 'guest', 'both')
-  - defaultPaymentMethodId: ObjectId (ref PaymentMethod)
-  - personaInquiryId: String
-  - personaAccountId: String
-  - personaInquiryStatus: String (enum: 'created', 'approved', 'declined')
-  - personaHasDriverLicense: Boolean
-  - driverLicenseState: String
-  - numReviews: Number
-  - totalReviewScore: Number
-  - needsSmartcarReconnect: Boolean
-  - isFrozen: Boolean
-  - instantTransactions: Boolean
-  - publicNotes: String
-  - bookingNotes: String
-  - hostOnboarding: Object
-  - guestOnboarding: Object
-  - canUseApp: Boolean
-  - useWalletForChargeReceipt: Boolean
-  - appOnboardingToken: Object
-  - addedToLaunchlist: Boolean
-  - stripeAccountId: String
-  - isStripeConnectedAccountConfirmed: Boolean
-  - requestedAccountDeletion: Boolean
-  - qrcodeUrl: String
-  - checkrCandidateId: String
-  - checkrInvitationId: String
-  - checkrInvitationUrl: String
-  - checkrInvitationStatus: String
-  - referralCode: String
-  - referralUrl: String
-  - referralConversionId: ObjectId (ref ReferralConversion)
-  - abiOwnerId: String
-  - abiRenterId: String
-  - driverLicense: Object
-  - securityDeposit: Number
-  - tripcityOwnerId: String
-  - bannedUserId: ObjectId (ref BannedUser)
-  - isConcierge: Boolean
-  - lastHostDashboardLoginAt: Date
-  - internalSlackChannel: String
-  - skipLoadingSuperchargerInvoices: Boolean
 `.trim();
 
 module.exports = ({ db }) => async function createChatMessage(params) {
@@ -208,7 +83,7 @@ module.exports = ({ db }) => async function createChatMessage(params) {
 
   llmMessages.unshift({
     role: 'system',
-    content: systemPrompt
+    content: systemPrompt + getModelDescriptions(db)
   });
 
   // Create the chat message and get OpenAI response in parallel

package/backend/helpers/getModelDescriptions.js

@@ -0,0 +1,18 @@
+'use strict';
+
+const listModelPaths = Model => [
+  ...Object.entries(Model.schema.paths).map(
+    ([path, schemaType]) => `- ${path}: ${schemaType.instance}`
+      + (schemaType.options?.ref ? ' (ref: ' + schemaType.options.ref + ')' : '')
+  ),
+  ...Object.entries(Model.schema.virtuals).filter(([path, virtual]) => virtual.options?.ref).map(
+    ([path, virtual]) => `- ${path}: Virtual (ref: ${virtual.options.ref})`
+  )
+].join('\n');
+
+const getModelDescriptions = db => Object.values(db.models).filter(Model => !Model.modelName.startsWith('__Studio')).map(Model => `
+${Model.modelName} (collection: ${Model.collection.collectionName})
+${listModelPaths(Model)}
+`.trim()).join('\n\n');
+
+module.exports = getModelDescriptions;

package/backend/index.js

@@ -8,12 +8,15 @@ const chatMessageSchema = require('./db/chatMessageSchema');
 const chatThreadSchema = require('./db/chatThreadSchema');
 const dashboardSchema = require('./db/dashboardSchema');
 
+const getModelDescriptions = require('./helpers/getModelDescriptions');
+
 module.exports = function backend(db) {
   db = db || mongoose.connection;
 
   const Dashboard = db.model('__Studio_Dashboard', dashboardSchema, 'studio__dashboards');
   const ChatMessage = db.model('__Studio_ChatMessage', chatMessageSchema, 'studio__chatMessages');
   const ChatThread = db.model('__Studio_ChatThread', chatThreadSchema, 'studio__chatThreads');
+
   const actions = applySpec(Actions, { db });
   return actions;
 };

package/frontend/public/app.js

@@ -485,7 +485,8 @@ module.exports = app => app.component('chat', {
       return message.role === 'user' ? 'bg-gray-100' : '';
     },
     async createNewThread() {
-
+      const { chatThread } = await api.ChatThread.createChatThread();
+      this.$router.push('/chat/' + chatThread._id);
     }
   },
   async mounted() {
@@ -3678,7 +3679,7 @@ module.exports = "<button v-bind=\"attrsToBind\" :disabled=\"isDisabled\" @click
 /***/ ((module) => {
 
 "use strict";
-module.exports = "<div class=\"relative border rounded bg-gray-100 text-black text-sm overflow-hidden\">\n  <div class=\"flex border-b pt-[1px] text-xs font-medium bg-gray-200\">\n    <button\n      class=\"px-3 py-1 border-r border-gray-300 hover:bg-green-300\"\n      :class=\"{'bg-gray-300': activeTab === 'code', 'bg-green-300': activeTab === 'code'}\"\n      @click=\"activeTab = 'code'\">\n      Code\n    </button>\n    <button\n      class=\"px-3 py-1 hover:bg-green-300\"\n      :class=\"{'bg-green-300': activeTab === 'output'}\"\n      @click=\"activeTab = 'output'\">\n      Output\n    </button>\n    <div class=\"ml-auto mr-1 flex\">\n      <button\n        v-if=\"activeTab === 'output'\"\n        class=\"px-2 py-1 mr-1 text-xs bg-blue-500 text-white border-none rounded cursor-pointer hover:bg-blue-600 transition-colors flex items-center\"\n        @click=\"openDetailModal\">\n        <svg xmlns=\"http://www.w3.org/2000/svg\" class=\"h-3 w-3\" fill=\"none\" viewBox=\"0 0 24 24\" stroke=\"currentColor\">\n          <path stroke-linecap=\"round\" stroke-linejoin=\"round\" stroke-width=\"2\" d=\"M4 8V4m0 0h4M4 4l5 5m11-1V4m0 0h-4m4 0l-5 5M4 16v4m0 0h4m-4 0l5-5m11 1v4m0 0h-4m4 0l-5-5\" />\n        </svg>\n      </button>\n      <async-button\n        class=\"px-2 py-1 text-xs bg-green-500 text-white border-none rounded cursor-pointer hover:bg-green-600 transition-colors disabled:bg-gray-400\"\n        @click=\"executeScript(message, script)\">\n        Execute\n      </async-button>\n    </div>\n  </div>\n\n  <pre class=\"p-3 whitespace-pre-wrap max-h-[30vh] overflow-y-auto\" v-show=\"activeTab === 'code'\"><code v-text=\"script\" ref=\"code\" :class=\"'language-' + language\"></code></pre>\n\n  <pre class=\"p-3 whitespace-pre-wrap max-h-[30vh] overflow-y-auto bg-white border-t\" v-show=\"activeTab === 'output'\">\n    <dashboard-chart v-if=\"message.executionResult?.output?.$chart\" :value=\"message.executionResult?.output\" />\n    <div v-else>\n      {{ message.executionResult?.output ? JSON.stringify(message.executionResult.output, null, 2) : 'No output' }}\n    </div>\n  </pre>\n\n  <modal ref=\"outputModal\" v-if=\"showDetailModal\" containerClass=\"!h-[90vh] !w-[90vw]\">\n    <template #body>\n      <div class=\"absolute font-mono right-1 top-1 cursor-pointer text-xl\" @click=\"showDetailModal = false;\">&times;</div>\n      <div class=\"h-full overflow-auto\">\n        <dashboard-chart v-if=\"message.executionResult?.output?.$chart\" :value=\"message.executionResult?.output\" />\n        <pre v-else class=\"whitespace-pre-wrap\">\n          {{ message.executionResult?.output ? JSON.stringify(message.executionResult.output, null, 2) : 'No output' }}\n        </pre>\n      </div>\n    </template>\n  </modal>\n</div>\n";
+module.exports = "<div class=\"relative border rounded bg-gray-100 text-black text-sm overflow-hidden\">\n  <div class=\"flex border-b pt-[1px] text-xs font-medium bg-gray-200\">\n    <button\n      class=\"px-3 py-1 border-r border-gray-300 hover:bg-green-300\"\n      :class=\"{'bg-gray-300': activeTab === 'code', 'bg-green-300': activeTab === 'code'}\"\n      @click=\"activeTab = 'code'\">\n      Code\n    </button>\n    <button\n      class=\"px-3 py-1 hover:bg-green-300\"\n      :class=\"{'bg-green-300': activeTab === 'output'}\"\n      @click=\"activeTab = 'output'\">\n      Output\n    </button>\n    <div class=\"ml-auto mr-1 flex\">\n      <button\n        v-if=\"activeTab === 'output'\"\n        class=\"px-2 py-1 mr-1 text-xs bg-blue-500 text-white border-none rounded cursor-pointer hover:bg-blue-600 transition-colors flex items-center\"\n        @click=\"openDetailModal\">\n        <svg xmlns=\"http://www.w3.org/2000/svg\" class=\"h-3 w-3\" fill=\"none\" viewBox=\"0 0 24 24\" stroke=\"currentColor\">\n          <path stroke-linecap=\"round\" stroke-linejoin=\"round\" stroke-width=\"2\" d=\"M4 8V4m0 0h4M4 4l5 5m11-1V4m0 0h-4m4 0l-5 5M4 16v4m0 0h4m-4 0l5-5m11 1v4m0 0h-4m4 0l-5-5\" />\n        </svg>\n      </button>\n      <async-button\n        class=\"px-2 py-1 text-xs bg-green-500 text-white border-none rounded cursor-pointer hover:bg-green-600 transition-colors disabled:bg-gray-400\"\n        @click=\"executeScript(message, script)\">\n        Execute\n      </async-button>\n    </div>\n  </div>\n\n  <pre class=\"p-3 whitespace-pre-wrap max-h-[30vh] overflow-y-auto\" v-show=\"activeTab === 'code'\"><code v-text=\"script\" ref=\"code\" :class=\"'language-' + language\"></code></pre>\n\n  <pre class=\"p-3 whitespace-pre-wrap max-h-[30vh] overflow-y-auto bg-white border-t\" v-show=\"activeTab === 'output'\">\n    <dashboard-chart v-if=\"message.executionResult?.output?.$chart\" :value=\"message.executionResult?.output\" />\n    <div v-else>{{ message.executionResult?.output ? JSON.stringify(message.executionResult.output, null, 2) : 'No output' }}</div>\n  </pre>\n\n  <modal ref=\"outputModal\" v-if=\"showDetailModal\" containerClass=\"!h-[90vh] !w-[90vw]\">\n    <template #body>\n      <div class=\"absolute font-mono right-1 top-1 cursor-pointer text-xl\" @click=\"showDetailModal = false;\">&times;</div>\n      <div class=\"h-full overflow-auto\">\n        <dashboard-chart v-if=\"message.executionResult?.output?.$chart\" :value=\"message.executionResult?.output\" />\n        <pre v-else class=\"whitespace-pre-wrap\">\n          {{ message.executionResult?.output ? JSON.stringify(message.executionResult.output, null, 2) : 'No output' }}\n        </pre>\n      </div>\n    </template>\n  </modal>\n</div>\n";
 
 /***/ }),
 
@@ -3700,7 +3701,7 @@ module.exports = "<div class=\"relative flex items-start space-x-3\" :class=\"{'
 /***/ ((module) => {
 
 "use strict";
-module.exports = "<div class=\"flex\" style=\"height: calc(100vh - 55px)\">\n    <!-- Sidebar: Chat Threads -->\n  <aside class=\"w-64 bg-gray-100 border-r overflow-y-auto h-full\">\n    <div class=\"p-4 font-bold text-lg border-b\">Chat Threads</div>\n    <div class=\"p-4\">\n      <button\n        @click=\"createNewThread\"\n        class=\"w-full bg-blue-600 text-white px-4 py-2 rounded hover:bg-blue-700\"\n      >\n        Create New Thread\n      </button>\n    </div>\n    <div v-if=\"status === 'loaded' && chatThreads.length === 0\" class=\"p-4 text-sm text-gray-700\">\n      No threads yet\n    </div>\n    <ul vif=\"status === 'loaded'\">\n      <li\n        v-for=\"thread in chatThreads\"\n        :key=\"thread._id\"\n        @click=\"selectThread(thread._id)\"\n        class=\"p-4 hover:bg-gray-200 cursor-pointer\"\n        :class=\"{ 'bg-gray-300': thread._id === chatThreadId }\"\n      >\n        {{ thread.title || 'Untitled Thread' }}\n      </li>\n    </ul>\n  </aside>\n\n  <!-- Main Chat Area -->\n  <main class=\"flex-1 flex flex-col\">\n    <div class=\"flex-1 overflow-y-auto p-6 space-y-4\" ref=\"messagesContainer\">\n      <ul role=\"list\" class=\"space-y-4\">\n        <div v-if=\"true\">\n          <div class=\"flex items-center justify-center py-3 mb-4\">\n            <div class=\"bg-gray-300 h-px flex-grow max-w-xs\"></div>\n            <p class=\"mx-4 text-sm font-medium text-gray-500\">This is the beginning of the message thread</p>\n            <div class=\"bg-gray-300 h-px flex-grow max-w-xs\"></div>\n          </div>\n        </div>\n        <li v-for=\"message in chatMessages\" :key=\"message._id\">\n          <chat-message :message=\"message\"></chat-message>\n        </li>\n      </ul>\n    </div>\n\n\n    <!-- Input Area -->\n    <div class=\"border-t p-4\">\n      <form @submit.prevent=\"sendMessage\" :disabled=\"sendingMessage\" class=\"flex gap-2\">\n        <input\n          v-model=\"newMessage\"\n          placeholder=\"Ask something...\"\n          class=\"flex-1 border rounded px-4 py-2\"\n        />\n        <button class=\"bg-blue-600 text-white px-4 py-2 rounded disabled:bg-gray-600\" :disabled=\"sendingMessage\">\n          <svg v-if=\"sendingMessage\" style=\"height: 1em\" viewBox=\"0 0 24 24\" xmlns=\"http://www.w3.org/2000/svg\">\n            <g>\n              <circle cx=\"12\" cy=\"12\" r=\"10\" fill=\"none\" stroke=\"currentColor\" stroke-width=\"2\" opacity=\"0.3\" />\n              <path d=\"M12 2a10 10 0 0 1 10 10\" fill=\"none\" stroke=\"currentColor\" stroke-width=\"2\" stroke-linecap=\"round\">\n                <animateTransform attributeName=\"transform\" type=\"rotate\" from=\"0 12 12\" to=\"360 12 12\" dur=\"1s\" repeatCount=\"indefinite\" />\n              </path>\n            </g>\n          </svg>\n          <span v-else>Send</span>\n        </button>\n      </form>\n    </div>\n  </main>\n</div>\n";
+module.exports = "<div class=\"flex\" style=\"height: calc(100vh - 55px)\">\n    <!-- Sidebar: Chat Threads -->\n  <aside class=\"w-64 bg-gray-100 border-r overflow-y-auto h-full\">\n    <div class=\"p-4 font-bold text-lg border-b\">Chat Threads</div>\n    <div class=\"p-4\">\n      <async-button\n        @click=\"createNewThread\"\n        class=\"w-full bg-blue-600 text-white px-4 py-2 rounded hover:bg-blue-700\"\n      >\n        Create New Thread\n      </async-button>\n    </div>\n    <div v-if=\"status === 'loaded' && chatThreads.length === 0\" class=\"p-4 text-sm text-gray-700\">\n      No threads yet\n    </div>\n    <ul vif=\"status === 'loaded'\">\n      <li\n        v-for=\"thread in chatThreads\"\n        :key=\"thread._id\"\n        @click=\"selectThread(thread._id)\"\n        class=\"p-4 hover:bg-gray-200 cursor-pointer\"\n        :class=\"{ 'bg-gray-300': thread._id === chatThreadId }\"\n      >\n        {{ thread.title || 'Untitled Thread' }}\n      </li>\n    </ul>\n  </aside>\n\n  <!-- Main Chat Area -->\n  <main class=\"flex-1 flex flex-col\">\n    <div class=\"flex-1 overflow-y-auto p-6 space-y-4\" ref=\"messagesContainer\">\n      <ul role=\"list\" class=\"space-y-4\">\n        <div v-if=\"true\">\n          <div class=\"flex items-center justify-center py-3 mb-4\">\n            <div class=\"bg-gray-300 h-px flex-grow max-w-xs\"></div>\n            <p class=\"mx-4 text-sm font-medium text-gray-500\">This is the beginning of the message thread</p>\n            <div class=\"bg-gray-300 h-px flex-grow max-w-xs\"></div>\n          </div>\n        </div>\n        <li v-for=\"message in chatMessages\" :key=\"message._id\">\n          <chat-message :message=\"message\"></chat-message>\n        </li>\n      </ul>\n    </div>\n\n\n    <!-- Input Area -->\n    <div class=\"border-t p-4\">\n      <form @submit.prevent=\"sendMessage\" :disabled=\"sendingMessage\" class=\"flex gap-2\">\n        <input\n          v-model=\"newMessage\"\n          placeholder=\"Ask something...\"\n          class=\"flex-1 border rounded px-4 py-2\"\n        />\n        <button class=\"bg-blue-600 text-white px-4 py-2 rounded disabled:bg-gray-600\" :disabled=\"sendingMessage\">\n          <svg v-if=\"sendingMessage\" style=\"height: 1em\" viewBox=\"0 0 24 24\" xmlns=\"http://www.w3.org/2000/svg\">\n            <g>\n              <circle cx=\"12\" cy=\"12\" r=\"10\" fill=\"none\" stroke=\"currentColor\" stroke-width=\"2\" opacity=\"0.3\" />\n              <path d=\"M12 2a10 10 0 0 1 10 10\" fill=\"none\" stroke=\"currentColor\" stroke-width=\"2\" stroke-linecap=\"round\">\n                <animateTransform attributeName=\"transform\" type=\"rotate\" from=\"0 12 12\" to=\"360 12 12\" dur=\"1s\" repeatCount=\"indefinite\" />\n              </path>\n            </g>\n          </svg>\n          <span v-else>Send</span>\n        </button>\n      </form>\n    </div>\n  </main>\n</div>\n";
 
 /***/ }),
 

package/frontend/src/chat/chat-message-script/chat-message-script.html

@@ -33,9 +33,7 @@
 
   <pre class="p-3 whitespace-pre-wrap max-h-[30vh] overflow-y-auto bg-white border-t" v-show="activeTab === 'output'">
     <dashboard-chart v-if="message.executionResult?.output?.$chart" :value="message.executionResult?.output" />
-    <div v-else>
-      {{ message.executionResult?.output ? JSON.stringify(message.executionResult.output, null, 2) : 'No output' }}
-    </div>
+    <div v-else>{{ message.executionResult?.output ? JSON.stringify(message.executionResult.output, null, 2) : 'No output' }}</div>
   </pre>
 
   <modal ref="outputModal" v-if="showDetailModal" containerClass="!h-[90vh] !w-[90vw]">

package/frontend/src/chat/chat.html

@@ -3,12 +3,12 @@
   <aside class="w-64 bg-gray-100 border-r overflow-y-auto h-full">
     <div class="p-4 font-bold text-lg border-b">Chat Threads</div>
     <div class="p-4">
-      <button
+      <async-button
         @click="createNewThread"
         class="w-full bg-blue-600 text-white px-4 py-2 rounded hover:bg-blue-700"
       >
         Create New Thread
-      </button>
+      </async-button>
     </div>
     <div v-if="status === 'loaded' && chatThreads.length === 0" class="p-4 text-sm text-gray-700">
       No threads yet

package/frontend/src/chat/chat.js

@@ -60,7 +60,8 @@ module.exports = app => app.component('chat', {
       return message.role === 'user' ? 'bg-gray-100' : '';
     },
     async createNewThread() {
-
+      const { chatThread } = await api.ChatThread.createChatThread();
+      this.$router.push('/chat/' + chatThread._id);
     }
   },
   async mounted() {

package/package.json

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