@xuda.io/ai_module 1.1.4926 → 1.1.4928

package/index_ms.mjs

@@ -3,261 +3,200 @@
 //   Queue target : ai_module
 // =====================================================
 
-import amqplib from 'amqplib';
-import { randomUUID } from 'crypto';
+import path from 'node:path';
 
-let queue = {};
+//const _ms_broker = await import(path.join(process.env.XUDA_HOME, 'common', 'ms_broker.mjs'));
+const { MessageBroker } = await import(path.join(process.env.XUDA_HOME, 'common', 'ms_broker.mjs'));
 
-// --- AMQP Connection & Channel ---
-const conn = await amqplib.connect('amqp://localhost');
-const channel = await conn.createChannel();
-await channel.assertQueue(`_ai_module-${global.module_in}_ch`, { durable: false });
+const broker = new MessageBroker();
 
-channel.consume(`_ai_module-${global.module_in}_ch`, async (msg) => {
-  try {
-    if (msg === null) {
-      console.log('Consumer cancelled by server');
-      return;
-    }
-
-    const content = JSON.parse(msg.content.toString());
-
-    queue[content.queue_id].resolve(content.ret);
-
-    channel.ack(msg);
-  } catch (err) {
-    channel.ack(msg); // Acknowledge to prevent requeue loops on error
-    console.error(module_in, content.method, err.message, err);
-    // debugger;
-  }
+await broker.init({
+  queueName: 'ai_module',
+  module_in: global.module_in,
 });
 
-const send_to_queue = async function (ms_method, ...args) {
-  const queue_id = randomUUID();
-  return new Promise((resolve, reject) => {
-    queue[queue_id] = { resolve, reject };
-
-    // Set timeout to prevent hanging
-    const timeout = setTimeout(() => {
-      delete queue[queue_id];
-      reject(new Error(`Queue timeout for ${ms_method}`));
-    }, 300000); // 300 second timeout
-
-    // Wrap resolve to clean up
-    queue[queue_id].resolve = (value) => {
-      clearTimeout(timeout);
-      delete queue[queue_id];
-      resolve(value);
-    };
-
-    // Wrap reject to clean up
-    queue[queue_id].reject = (error) => {
-      clearTimeout(timeout);
-      delete queue[queue_id];
-      reject(error);
-    };
-
-    try {
-      global[`_${global.module_in}_ch`].sendToQueue(
-        'ai_module',
-        Buffer.from(
-          JSON.stringify({
-            ms_method,
-            data: args,
-            cb: `_ai_module-${global.module_in}_ch`,
-            queue_id,
-          }),
-        ),
-      );
-    } catch (err) {
-      debugger;
-    }
-  });
-};
-
-// You must define this somewhere before using these wrappers:
-const module_name = 'ai_module';
+//_ms_broker.init({queueName:'ai_module'})
 
 export const get_chat_conversation = async function (...args) {
-  return await send_to_queue('get_chat_conversation', ...args);
+  return await broker.send_to_queue('get_chat_conversation', ...args);
 };
 
 export const get_ai_chats = async function (...args) {
-  return await send_to_queue('get_ai_chats', ...args);
+  return await broker.send_to_queue('get_ai_chats', ...args);
 };
 
 export const delete_ai_chat = async function (...args) {
-  return await send_to_queue('delete_ai_chat', ...args);
+  return await broker.send_to_queue('delete_ai_chat', ...args);
 };
 
 export const archive_ai_chat = async function (...args) {
-  return await send_to_queue('archive_ai_chat', ...args);
+  return await broker.send_to_queue('archive_ai_chat', ...args);
 };
 
 export const unarchive_ai_chat = async function (...args) {
-  return await send_to_queue('unarchive_ai_chat', ...args);
+  return await broker.send_to_queue('unarchive_ai_chat', ...args);
 };
 
 export const get_ai_agent_info = async function (...args) {
-  return await send_to_queue('get_ai_agent_info', ...args);
+  return await broker.send_to_queue('get_ai_agent_info', ...args);
 };
 
 export const get_ai_agents = async function (...args) {
-  return await send_to_queue('get_ai_agents', ...args);
+  return await broker.send_to_queue('get_ai_agents', ...args);
 };
 
 export const delete_ai_agent = async function (...args) {
-  return await send_to_queue('delete_ai_agent', ...args);
+  return await broker.send_to_queue('delete_ai_agent', ...args);
 };
 
 export const uninstall_ai_agent = async function (...args) {
-  return await send_to_queue('uninstall_ai_agent', ...args);
+  return await broker.send_to_queue('uninstall_ai_agent', ...args);
 };
 
 export const unarchive_ai_agent = async function (...args) {
-  return await send_to_queue('unarchive_ai_agent', ...args);
+  return await broker.send_to_queue('unarchive_ai_agent', ...args);
 };
 
 export const archive_ai_agent = async function (...args) {
-  return await send_to_queue('archive_ai_agent', ...args);
+  return await broker.send_to_queue('archive_ai_agent', ...args);
 };
 
 export const archive_mini_app = async function (...args) {
-  return await send_to_queue('archive_mini_app', ...args);
+  return await broker.send_to_queue('archive_mini_app', ...args);
 };
 
 export const unarchive_mini_app = async function (...args) {
-  return await send_to_queue('unarchive_mini_app', ...args);
+  return await broker.send_to_queue('unarchive_mini_app', ...args);
 };
 
 export const delete_mini_app = async function (...args) {
-  return await send_to_queue('delete_mini_app', ...args);
+  return await broker.send_to_queue('delete_mini_app', ...args);
 };
 
 export const update_ai_agent = async function (...args) {
-  return await send_to_queue('update_ai_agent', ...args);
+  return await broker.send_to_queue('update_ai_agent', ...args);
 };
 
 export const get_apps = async function (...args) {
-  return await send_to_queue('get_apps', ...args);
+  return await broker.send_to_queue('get_apps', ...args);
 };
 
 export const create_ai_agent = async function (...args) {
-  return await send_to_queue('create_ai_agent', ...args);
+  return await broker.send_to_queue('create_ai_agent', ...args);
 };
 
 export const update_ai_agent_properties = async function (...args) {
-  return await send_to_queue('update_ai_agent_properties', ...args);
+  return await broker.send_to_queue('update_ai_agent_properties', ...args);
 };
 
 export const update_thumbnail = async function (...args) {
-  return await send_to_queue('update_thumbnail', ...args);
+  return await broker.send_to_queue('update_thumbnail', ...args);
 };
 
 export const upload_prompt_attachment = async function (...args) {
-  return await send_to_queue('upload_prompt_attachment', ...args);
+  return await broker.send_to_queue('upload_prompt_attachment', ...args);
 };
 
 export const delete_prompt_attachment = async function (...args) {
-  return await send_to_queue('delete_prompt_attachment', ...args);
+  return await broker.send_to_queue('delete_prompt_attachment', ...args);
 };
 
 export const submit_chat_gpt_prompt = async function (...args) {
-  return await send_to_queue('submit_chat_gpt_prompt', ...args);
+  return await broker.send_to_queue('submit_chat_gpt_prompt', ...args);
 };
 
 export const create_conversation = async function (...args) {
-  return await send_to_queue('create_conversation', ...args);
+  return await broker.send_to_queue('create_conversation', ...args);
 };
 
 export const submit_chat_conversation = async function (...args) {
-  return await send_to_queue('submit_chat_conversation', ...args);
+  return await broker.send_to_queue('submit_chat_conversation', ...args);
 };
 
 export const create_avatar = async function (...args) {
-  return await send_to_queue('create_avatar', ...args);
+  return await broker.send_to_queue('create_avatar', ...args);
 };
 
 export const get_profile_avatar = async function (...args) {
-  return await send_to_queue('get_profile_avatar', ...args);
+  return await broker.send_to_queue('get_profile_avatar', ...args);
 };
 
 export const get_profile_picture = async function (...args) {
-  return await send_to_queue('get_profile_picture', ...args);
+  return await broker.send_to_queue('get_profile_picture', ...args);
 };
 
 export const conversation_actions = async function (...args) {
-  return await send_to_queue('conversation_actions', ...args);
+  return await broker.send_to_queue('conversation_actions', ...args);
 };
 
 export const modify_profile_avatar = async function (...args) {
-  return await send_to_queue('modify_profile_avatar', ...args);
+  return await broker.send_to_queue('modify_profile_avatar', ...args);
 };
 
 export const get_ai_workspace_pending_counts = async function (...args) {
-  return await send_to_queue('get_ai_workspace_pending_counts', ...args);
+  return await broker.send_to_queue('get_ai_workspace_pending_counts', ...args);
 };
 
 export const create_mini_app = async function (...args) {
-  return await send_to_queue('create_mini_app', ...args);
+  return await broker.send_to_queue('create_mini_app', ...args);
 };
 
 export const pin_ai_agent = async function (...args) {
-  return await send_to_queue('pin_ai_agent', ...args);
+  return await broker.send_to_queue('pin_ai_agent', ...args);
 };
 
 export const unpin_ai_agent = async function (...args) {
-  return await send_to_queue('unpin_ai_agent', ...args);
+  return await broker.send_to_queue('unpin_ai_agent', ...args);
 };
 
 export const get_transcript = async function (...args) {
-  return await send_to_queue('get_transcript', ...args);
+  return await broker.send_to_queue('get_transcript', ...args);
 };
 
 export const add_transcript_conversation_item = async function (...args) {
-  return await send_to_queue('add_transcript_conversation_item', ...args);
+  return await broker.send_to_queue('add_transcript_conversation_item', ...args);
 };
 
 export const is_spam_email = async function (...args) {
-  return await send_to_queue('is_spam_email', ...args);
+  return await broker.send_to_queue('is_spam_email', ...args);
 };
 
 export const is_business_contact = async function (...args) {
-  return await send_to_queue('is_business_contact', ...args);
+  return await broker.send_to_queue('is_business_contact', ...args);
 };
 
 export const is_business_contact_has_person = async function (...args) {
-  return await send_to_queue('is_business_contact_has_person', ...args);
+  return await broker.send_to_queue('is_business_contact_has_person', ...args);
 };
 
 export const get_name_from_email_addr = async function (...args) {
-  return await send_to_queue('get_name_from_email_addr', ...args);
+  return await broker.send_to_queue('get_name_from_email_addr', ...args);
 };
 
 export const categorize_email_subject = async function (...args) {
-  return await send_to_queue('categorize_email_subject', ...args);
+  return await broker.send_to_queue('categorize_email_subject', ...args);
 };
 
 export const categorize_ai_prompt = async function (...args) {
-  return await send_to_queue('categorize_ai_prompt', ...args);
+  return await broker.send_to_queue('categorize_ai_prompt', ...args);
 };
 
 export const categorize_text_message = async function (...args) {
-  return await send_to_queue('categorize_text_message', ...args);
+  return await broker.send_to_queue('categorize_text_message', ...args);
 };
 
 export const categorize_contact_note = async function (...args) {
-  return await send_to_queue('categorize_contact_note', ...args);
+  return await broker.send_to_queue('categorize_contact_note', ...args);
 };
 
 export const get_business_info = async function (...args) {
-  return await send_to_queue('get_business_info', ...args);
+  return await broker.send_to_queue('get_business_info', ...args);
 };
 
 export const get_person_info = async function (...args) {
-  return await send_to_queue('get_person_info', ...args);
+  return await broker.send_to_queue('get_person_info', ...args);
 };
 
 export const open_ai_alive_check = async function (...args) {
-  return await send_to_queue('open_ai_alive_check', ...args);
+  return await broker.send_to_queue('open_ai_alive_check', ...args);
 };

package/package.json

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