@xuda.io/account_module 1.2.2156 → 1.2.2158

package/index.mjs

@@ -3381,6 +3381,98 @@ export const save_cache_hit = async function (_id) {
 ///////////////////////////////
 
 export const get_account_ai_usage = async function (req, job_id, headers) {
+  const d = new Date();
+  const curr_month = Number(String(d.getMonth() + 1).padStart(2, '0')); // months are 0-based
+  const curr_year = d.getFullYear();
+
+  const { uid, year_from = curr_year, month_from = curr_month, day_from = 0, year_to = curr_year, month_to = curr_month, day_to = 99 } = req;
+  try {
+    const app_id = await get_account_default_project_id(uid);
+
+    const timestamp_from = new Date(Number(year_from) || year, month - 1, day).getTime();
+
+    const ai_usage = await db_module.get_couch_view('xuda_usage', 'ai_usage', {
+      startkey: [uid, year_from || curr_year, month_from || curr_month, day_from || 0],
+      endkey: [uid, year_to || curr_year, month_to || curr_month, day_to || 99],
+      reduce: true,
+      group_level: 999,
+    });
+    // console.log('ai_usage', ai_usage.rows);
+    debugger;
+    let total_usage = 0;
+    let profile = {};
+
+    for (const row of ai_usage.data.rows) {
+      const user = row.key[0];
+      const year = row.key[1];
+      const month = row.key[2];
+      const day = row.key[3];
+      const account_profile_id = row.key[4];
+      const uid = row.key[5];
+      const model = row.key[6];
+      const input_tokens = row.value[0];
+      const output_tokens = row.value[1];
+      const input_credits = row.value[2];
+      const output_credits = row.value[3];
+      if (!profile[account_profile_id]) {
+        profile[account_profile_id] = {};
+      }
+      if (!profile[account_profile_id][uid]) {
+        profile[account_profile_id][uid] = 0;
+      }
+
+      profile[account_profile_id][uid] += input_credits + output_credits;
+      total_usage += input_credits + output_credits;
+    }
+
+    const ai_credits = await db_module.get_couch_view('xuda_billing', 'ai_credits', {
+      startkey: [uid, ''],
+      endkey: [uid, 'zzz'],
+      reduce: true,
+      group_level: 999,
+    });
+
+    let credits_obj = {},
+      total_credits = 0;
+    for (const row of ai_credits.data.rows) {
+      const user = row.key[0];
+      const source = row.key[1];
+
+      const credits = row.value;
+
+      total_credits += credits;
+
+      if (!credits_obj[source]) {
+        credits_obj[source] = 0;
+      }
+      credits_obj[source] += credits;
+    }
+
+    let data = { credits: { ...credits_obj, total: total_credits }, usage: { profile, projects: [], total: total_usage } };
+
+    if (job_id) {
+      data.profile_info = {};
+      data.user_info = {};
+      for (const [key, val] of Object.entries(data?.usage?.profile || {})) {
+        try {
+          data.profile_info[key] = (await db_module.get_app_couch_doc_native(app_id, key))?.profile_name;
+
+          for (const [user_id, val2] of Object.entries(val || {})) {
+            try {
+              data.user_info[user_id] = (await db_module.get_couch_doc_native('xuda_accounts', user_id))?.account_info?.full_name;
+            } catch (error) {}
+          }
+        } catch (error) {}
+      }
+    }
+
+    return { code: 55, data };
+  } catch (err) {
+    return { code: -55, data: err.message };
+  }
+};
+
+export const get_account_ai_usage_old = async function (req, job_id, headers) {
   const { uid, year_from, month_from, day_from, year_to, month_to, day_to } = req;
   try {
     const app_id = await get_account_default_project_id(uid);
@@ -3396,7 +3488,7 @@ export const get_account_ai_usage = async function (req, job_id, headers) {
       group_level: 999,
     });
     // console.log('ai_usage', ai_usage.rows);
-
+    debugger;
     let total_usage = 0;
     let profile = {};
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xuda.io/account_module",
-  "version": "1.2.2156",
+  "version": "1.2.2158",
   "description": "Xuda Account Server Module",
   "main": "index.mjs",
   "dependencies": {