feat: implemnted the property usedTo on the controller

controllers/apiUsagePricing.js

@@ -28,6 +28,10 @@ const setApiPricing = async (req, res) => {
 
     let filter = { provider: normalizedProvider, product: normalizedProduct }
 
+    if(format){
+        filter = {...filter, format}
+    }
+
     if (type) {
         filter = { ...filter, type }
     }
@@ -197,7 +201,8 @@ const registerUsage = async (req, res) => {
         callerId,
         sessionId,
         companyId,
-        format
+        format,
+        usedTo
 
     } = req.body
 
@@ -213,6 +218,7 @@ const registerUsage = async (req, res) => {
     const apiPricing = await API_Pricing.findOne({
         provider: provider.trim().toLowerCase(),
         product: product.trim().toLowerCase(),
+        format: format?.trim().toLowerCase()
     })
 
     if (apiPricing) {
@@ -230,7 +236,8 @@ const registerUsage = async (req, res) => {
             billingUnit,
             companyId,
             total_cost: calculateApiUsage(price, billingUnit, usage),
-            format
+            format,
+            usedTo
         })
 
         return res.status(StatusCodes.OK).json({ apiUsage })

models/API_Usage.js

@@ -1,51 +1,61 @@
-const mongoose = require('../db/connect')
+const mongoose = require("../db/connect");
 
-const { Schema } = mongoose
+const { Schema } = mongoose;
 
-const apiUsage = new Schema({
+const apiUsage = new Schema(
+  {
     companyId: {
-        type: String,
+      type: String,
-        required: true,
+      required: true,
     },
     callerId: {
-        type: String,
+      type: String,
-        required: true,
+      required: true,
     },
     sessionId: {
-        type: String,
+      type: String,
-        required: true,
+      required: true,
-    },    
+    },
     provider: {
-        type: String,
+      type: String,
-        required: true,
+      required: true,
     },
     product: {
-        type: String,
+      type: String,
-        required: true,
+      required: true,
-    },    
+    },
     usage: {
-        type: String,
+      type: String,
-        required: true
+      required: true,
     },
     price: {
-        type: String,
+      type: String,
-        required: true
+      required: true,
     },
     billingBy: {
-        type: String, 
+      type: String,
-        required: true,
+      required: true,
     },
     billingUnit: {
-        type: Number,
+      type: Number,
-        required: true
+      required: true,
-    },     
+    },
     total_cost: {
-        type: String,
+      type: String,
-        required: true
+      required: true,
-    } 
+    },
+    format: {
+      type: String,
+      enum: ["text", "audio", "image", "video"],
+    },
+    usedTo: {
+      type: String,
+      enum: ["stt", "llm", "tts"],
+    },
+  },
+  { timestamps: true }
+);
 
-}, { timestamps: true })
+const API_Usage = mongoose.model("API_Usage", apiUsage);
- 
-const API_Usage = mongoose.model('API_Usage', apiUsage)
 
-module.exports = API_Usage
+module.exports = API_Usage;