@xuda.io/ai_module 1.1.5587 → 1.1.5589

package/index.mjs

@@ -1112,6 +1112,7 @@ export const get_chat_conversation = async function (req) {
     });
 
     for (let conversation_item_doc of conversation_items.docs || []) {
+      conversation_item_doc.metadata = conversation_item_doc.metadata || {};
       if (!conversation_item_doc.read) {
         conversation_item_doc.read = {};
       }
@@ -1539,6 +1540,7 @@ export const archive_ai_chat = async function (req) {
     conversation_doc.stat = 5;
     conversation_doc.ts = Date.now();
     const save_ret = await db_module.save_app_couch_doc(account_profile_info.app_id, conversation_doc);
+    await update_conversation_items_stat(account_profile_info.app_id, 5, conversation_id);
 
     return save_ret;
   } catch (err) {
@@ -2282,10 +2284,8 @@ export const unarchive_ai_agent = async function (req) {
   }
 };
 
-const update_agent_conversation_items_stat = async function (app_id, agent_id, stat, conversation_id) {
-  const target_conversation_id = conversation_id || agent_id;
+const update_conversation_items_stat = async function (app_id, stat, target_conversation_id) {
   const ts = Date.now();
-  debugger;
   const items_ret = await db_module.find_app_couch_query(app_id, {
     selector: {
       docType: 'chat_conversation_item',
@@ -2308,6 +2308,32 @@ const update_agent_conversation_items_stat = async function (app_id, agent_id, s
   return docs.length;
 };
 
+const update_conversation_stat = async function (app_id, agent_id, stat, conversation_id) {
+  const conversation_ids = [];
+
+  if (conversation_id) {
+    conversation_ids.push(conversation_id);
+  } else {
+    const conversations_ret = await db_module.find_app_couch_query(app_id, {
+      selector: {
+        docType: 'chat_conversation',
+        reference_id: agent_id,
+      },
+      fields: ['_id'],
+      limit: 9999,
+    });
+
+    conversation_ids.push(...(conversations_ret.docs || []).map((doc) => doc._id).filter(Boolean));
+  }
+
+  let updated = 0;
+  for (const target_conversation_id of conversation_ids) {
+    updated += await update_conversation_items_stat(app_id, stat, target_conversation_id);
+  }
+
+  return updated;
+};
+
 export const archive_ai_agent = async function (req) {
   let { agent_id, uid, conversation_id } = req;
   const account_profile_info = await get_active_account_profile_info(uid);
@@ -2320,11 +2346,10 @@ export const archive_ai_agent = async function (req) {
     if (agent_doc.studio_meta.createdByUid !== uid && agent_doc?.studio_meta?.account_profile_info?.uid !== uid) {
       throw new Error('Operation not allowed');
     }
-    debugger;
     agent_doc.stat = 5;
     agent_doc.ts = Date.now();
     const save_ret = await db_module.save_app_couch_doc(account_profile_info.app_id, agent_doc);
-    const updated_conversation_items = await update_agent_conversation_items_stat(account_profile_info.app_id, agent_id, 5, conversation_id);
+    const updated_conversation_items = await update_conversation_stat(account_profile_info.app_id, agent_id, 5, conversation_id);
 
     return { code: 1, data: { save_ret, updated_conversation_items } };
   } catch (err) {
@@ -3893,7 +3918,7 @@ const contactGuardrailAgent = new Agent({
 // };
 
 export const submit_chat_conversation = async function (req, job_id, headers) {
-  const { profile_id, uid } = req;
+  const { profile_id, uid, metadata = {} } = req;
   let { conversation_id } = req;
   try {
     if (!conversation_id) throw new Error('missing conversation_id');
@@ -3901,6 +3926,7 @@ export const submit_chat_conversation = async function (req, job_id, headers) {
 
     let conversation_doc = await db_module.get_app_couch_doc_native(account_profile_info.app_id, conversation_id);
     req.conversation_doc = conversation_doc;
+    req.metadata = metadata || {};
     let ret;
     if (conversation_doc.conversation_type === 'studio') {
       ret = await chat_studio(req, job_id, headers);
@@ -4880,7 +4906,8 @@ const load_ai_agent_doc = async function (app_id, agent_id) {
 const ai_chat_conversation = async function (req, job_id, headers) {
   const { uid, gtp_token, profile_id, conversation_item_id } = req;
   const account_profile_info = await get_active_account_profile_info(uid, profile_id);
-  let { prompt, conversation_doc, attachments = [], ai_agents, output_format = 'md', stream = true, with_context = true, ai_model = _conf.default_ai_model } = req;
+  let { prompt, conversation_doc, attachments = [], ai_agents, output_format = 'md', stream = true, with_context = true, ai_model = _conf.default_ai_model, metadata = {} } = req;
+  metadata = metadata || {};
 
   let local_ai_agents = _.cloneDeep(ai_agents);
 
@@ -4919,6 +4946,7 @@ const ai_chat_conversation = async function (req, job_id, headers) {
     direction: 'out',
     role: 'user',
     job_id,
+    metadata,
   };
   let out_conversation_item_save_ret = await db_module.save_app_couch_doc(account_profile_info.app_id, out_conversation_item_obj);
   if (!conversation_item_id) {

package/package.json

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