feat(Tx): Add timestamps to transaction schema (danny-avila#1123)

aitok-ai · Oct 30, 2023 · 4101fdd · 4101fdd
1 parent d6f30e7
commit 4101fdd
Showing 1 changed file with 33 additions and 28 deletions.
diff --git a/models/schema/transaction.js b/models/schema/transaction.js
@@ -1,34 +1,39 @@
 const mongoose = require('mongoose');
 
-const transactionSchema = mongoose.Schema({
-  user: {
-    type: mongoose.Schema.Types.ObjectId,
-    ref: 'User',
-    index: true,
-    required: true,
+const transactionSchema = mongoose.Schema(
+  {
+    user: {
+      type: mongoose.Schema.Types.ObjectId,
+      ref: 'User',
+      index: true,
+      required: true,
+    },
+    conversationId: {
+      type: String,
+      ref: 'Conversation',
+      index: true,
+    },
+    tokenType: {
+      type: String,
+      enum: ['prompt', 'completion', 'credits'],
+      required: true,
+    },
+    model: {
+      type: String,
+    },
+    context: {
+      type: String,
+    },
+    valueKey: {
+      type: String,
+    },
+    rate: Number,
+    rawAmount: Number,
+    tokenValue: Number,
   },
-  conversationId: {
-    type: String,
-    ref: 'Conversation',
-    index: true,
+  {
+    timestamps: true,
   },
-  tokenType: {
-    type: String,
-    enum: ['prompt', 'completion', 'credits'],
-    required: true,
-  },
-  model: {
-    type: String,
-  },
-  context: {
-    type: String,
-  },
-  valueKey: {
-    type: String,
-  },
-  rate: Number,
-  rawAmount: Number,
-  tokenValue: Number,
-});
+);
 
 module.exports = transactionSchema;