@xuda.io/ai_module 1.1.4143 → 1.1.4145

package/index.mjs

@@ -2021,7 +2021,7 @@ export const unarchive_ai_chat = async function (req) {
 //   }
 // };
 
-const create_image = async function (uid, prompt, model = 'gpt-image-1-mini', size = '1024x1024', n = 1, width = 256, height = 256, metadata = {}) {
+const create_image = async function (uid, prompt, model = 'gpt-image-1-mini', size = '1024x1024', n = 1, width = 256, height = 256, metadata = {}, account_profile_info) {
   try {
     const response = await client.images.generate({
       model,
@@ -2065,7 +2065,7 @@ const create_image = async function (uid, prompt, model = 'gpt-image-1-mini', si
 
     const thumbnailBase64 = resized.toString('base64');
 
-    record_chat_usage(uid, response.usage.input_tokens, response.usage.output_tokens, 'create image', prompt, model, metadata);
+    record_chat_usage(uid, response.usage.input_tokens, response.usage.output_tokens, 'create image', prompt, model, metadata, account_profile_info);
 
     return thumbnailBase64;
   } catch (err) {
@@ -3309,7 +3309,7 @@ export const update_thumbnail = async function (type, doc, app_id, uid, job_id,
       if (xuda_db) {
         return await db_module.get_couch_doc_native(xuda_db, doc._id);
       } else {
-        return await db_module.get_app_couch_doc_native(app_id, doc._id);
+        return await db_module.get_app_couch_doc_native(account_profile_info.app_id, doc._id);
       }
     };
 
@@ -3375,7 +3375,7 @@ export const update_thumbnail = async function (type, doc, app_id, uid, job_id,
       case 'conversation': {
         prompt = `create image: ${doc.title}`;
 
-        const images_arr = await create_and_upload_image_to_drive('user', 'Chat Thumbnails', doc._id, prompt, 1, uid, app_id, job_id, headers, true, tags);
+        const images_arr = await create_and_upload_image_to_drive('user', 'Chat Thumbnails', doc._id, prompt, 1, uid, app_id, job_id, headers, true, tags, account_profile_info);
         db_doc = await get_db_doc();
         db_doc.thumbnail_request_ts = Date.now();
         db_doc.chat_image = images_arr;
@@ -3386,7 +3386,7 @@ export const update_thumbnail = async function (type, doc, app_id, uid, job_id,
         if (doc.studio_meta.agent_image) return;
         prompt = `create image: ${doc.properties.menuName}`;
 
-        const images_arr = await create_and_upload_image_to_drive('studio', 'Progs Thumbnails', doc._id, prompt, 1, uid, app_id, job_id, headers, false, tags);
+        const images_arr = await create_and_upload_image_to_drive('studio', 'Progs Thumbnails', doc._id, prompt, 1, uid, app_id, job_id, headers, false, tags, account_profile_info);
         db_doc = await get_db_doc();
         db_doc.studio_meta.thumbnail_request_ts = Date.now();
         db_doc.studio_meta.agent_image = images_arr;
@@ -3399,7 +3399,7 @@ export const update_thumbnail = async function (type, doc, app_id, uid, job_id,
           doc?.context || ''
         }"} and create realistic fictional profile picture transparent background ,person in the center of the picture  add top margin the person should cover the whole picture`;
 
-        const images_arr = await create_and_upload_image_to_drive('user', 'Profile Images', doc._id, prompt, 1, uid, app_id, job_id, headers, true, tags, 1024, 1024);
+        const images_arr = await create_and_upload_image_to_drive('user', 'Profile Images', doc._id, prompt, 1, uid, app_id, job_id, headers, true, tags, account_profile_info, 1024, 1024);
         await update_profile_avatar_status(doc._id, 3);
         db_doc = await get_db_doc();
         db_doc.image_request_ts = Date.now();
@@ -3415,7 +3415,7 @@ export const update_thumbnail = async function (type, doc, app_id, uid, job_id,
           doc?.context || ''
         }"} and create realistic fictional profile picture transparent background person in the center of the picture  add top margin the person should cover the whole picture`;
 
-        const images_arr = await create_and_upload_image_to_drive('user', 'Profile Images', doc._id, prompt, 1, uid, app_id, job_id, headers, true, tags, 1024, 1024);
+        const images_arr = await create_and_upload_image_to_drive('user', 'Profile Images', doc._id, prompt, 1, uid, app_id, job_id, headers, true, tags, account_profile_info, 1024, 1024);
         db_doc = await get_db_doc();
         db_doc.image_request_ts = Date.now();
         db_doc.account_info.profile_avatar = images_arr[0].file_url;
@@ -3435,7 +3435,7 @@ export const update_thumbnail = async function (type, doc, app_id, uid, job_id,
     if (xuda_db) {
       save_ret = await db_module.save_couch_doc_native(xuda_db, db_doc);
     } else {
-      save_ret = await db_module.save_app_couch_doc_native(app_id, db_doc);
+      save_ret = await db_module.save_app_couch_doc_native(account_profile_info.app_id, db_doc);
     }
     return save_ret;
   } catch (error) {
@@ -5489,10 +5489,10 @@ const get_plugin_tool = async function (app_id, uid, plugin_name, method, prop_d
   }
 };
 
-const create_and_upload_image_to_drive = async function (drive_type, file_path, file_name, prompt, numGenerations, uid, app_id, job_id, headers = {}, is_system, tags = [], width, height) {
+const create_and_upload_image_to_drive = async function (drive_type, file_path, file_name, prompt, numGenerations, uid, app_id, job_id, headers = {}, is_system, tags = [], account_profile_info) {
   try {
     // 1. Generate the images
-    let result = await create_image(uid, prompt, undefined, undefined, numGenerations, undefined, undefined, { drive_type, file_path, file_name });
+    let result = await create_image(uid, prompt, undefined, undefined, numGenerations, undefined, undefined, { drive_type, file_path, file_name }, account_profile_info);
 
     // Normalize to array
     const images = Array.isArray(result) ? result : [result];
@@ -5829,7 +5829,7 @@ export const create_profile_avatar = async function (req, job_id, headers) {
           const prompt = `analyze the the person info :  ${JSON.stringify({ first_name, last_name, bio, country, mainCategory, subCategory })} 
             and create realistic fictional profile picture transparent background ,person in the center of the picture return only a centered head-and-shoulders portrait facing the camera. include no labels or text add top margin the person should cover the whole picture`;
 
-          imageBase64 = await create_image(uid, prompt, undefined, undefined, 1, undefined, undefined, { url });
+          imageBase64 = await create_image(uid, prompt, undefined, undefined, 1, undefined, undefined, { url }, account_profile_info);
           avatar_source = 'fictional';
         };
         // personal account
@@ -6150,7 +6150,7 @@ const create_ai_agent_image = async function (req, job_id, headers) {
     return { is_transparent, image_blob: new Blob([buffer], { type: 'image/png' }) };
   }
 
-  const { app_id, uid, ai_agent_id, tags } = req;
+  const { app_id, uid, ai_agent_id, tags, account_profile_info } = req;
 
   const ai_agent_doc = await db_module.get_app_couch_doc_native(app_id, ai_agent_id);
   const account_doc = await db_module.get_couch_doc_native('xuda_accounts', uid);
@@ -6190,7 +6190,7 @@ const create_ai_agent_image = async function (req, job_id, headers) {
 
         `;
 
-    const images_arr = await create_and_upload_image_to_drive('studio', 'Progs Thumbnails', ai_agent_id, prompt, 1, uid, app_id, job_id, headers, false, tags);
+    const images_arr = await create_and_upload_image_to_drive('studio', 'Progs Thumbnails', ai_agent_id, prompt, 1, uid, app_id, job_id, headers, false, tags, account_profile_info);
     return { code: 1, data: images_arr[0] };
   }
 

package/package.json

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