@xuda.io/ai_module 1.1.3952 → 1.1.3955

package/index.mjs

@@ -2449,15 +2449,15 @@ export const get_ai_agents = async function (req, job_id, headers) {
       user_agents = await db_module.find_app_couch_query(app_id, opt);
       if (!limit || ai_agent_id) {
         user_agents.total_docs = user_agents.docs.length;
-      }
-     else {
-      delete opt.sort;
-      delete opt.skip;
-      opt.limit = 9999;
-      opt.fields = ['_id'];
+      } else {
+        delete opt.sort;
+        delete opt.skip;
+        opt.limit = 9999;
+        opt.fields = ['_id'];
 
-      const counter = await db_module.find_app_couch_query(app_id, opt);
-      user_agents.total_docs = counter.docs.length;
+        const counter = await db_module.find_app_couch_query(app_id, opt);
+        user_agents.total_docs = counter.docs.length;
+      }
     }
 
     ////// FROM - IN
@@ -6370,6 +6370,11 @@ const record_chat_usage = async function (uid, input_tokens, output_tokens, sour
   try {
     const app_id = await account_module.get_account_default_project_id(uid);
 
+    const cost = _conf?.ai_models?.[model] || {
+      input: 1.0,
+      output: 1.0,
+    };
+
     const t = Date.now();
     let usage_doc = {
       _id: await _common.xuda_get_uuid('chat_usage'),
@@ -6383,6 +6388,7 @@ const record_chat_usage = async function (uid, input_tokens, output_tokens, sour
       stat: 3,
       metadata,
       model,
+      cost,
     };
     const save_ret = await db_module.save_app_couch_doc(app_id, usage_doc);
     // console.log(save_ret);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xuda.io/ai_module",
-  "version": "1.1.3952",
+  "version": "1.1.3955",
   "description": "Xuda AI Module",
   "main": "index.mjs",
   "type": "module",