@xuda.io/account_module 1.2.2123 → 1.2.2124

package/index.mjs

@@ -1786,101 +1786,6 @@ export const ts_contact = async function (uid, contact_id) {
   }
 };
 
-export const get_account_ai_usage = 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);
-
-    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 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);
-
-    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;
-    }
-
-    // let membership_plan = _conf.PLAN_OBJ?.[acc_obj?.ai_workspace_plan]?.features?.ai_credits || 0;
-    // let ai_workspace_plan = _conf.PLAN_OBJ?.[acc_obj?.membership_plan]?.features?.ai_credits || 0;
-    // let contact_connection = 0;
-    // let total_credits = membership_plan + ai_workspace_plan + contact_connection;
-
-    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 };
-  }
-};
-
 const set_account_profile_picture = async function (uid, account_uid, metadata, job_id, headers, account_profile_info) {
   await update_account_profile_picture_status(account_uid, 1);
   try {
@@ -3405,6 +3310,101 @@ export const save_cache_hit = async function (_id) {
 
 ///////////////////////////////
 
+export const get_account_ai_usage = 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);
+
+    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 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);
+
+    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;
+    }
+
+    // let membership_plan = _conf.PLAN_OBJ?.[acc_obj?.ai_workspace_plan]?.features?.ai_credits || 0;
+    // let ai_workspace_plan = _conf.PLAN_OBJ?.[acc_obj?.membership_plan]?.features?.ai_credits || 0;
+    // let contact_connection = 0;
+    // let total_credits = membership_plan + ai_workspace_plan + contact_connection;
+
+    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 record_ai_usage = async function (uid, input_tokens, output_tokens, source, prompt, model, metadata = {}, account_profile_info, tools) {
   try {
     if (!account_profile_info) throw new Error('missing account_profile_info');

package/package.json

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