@xuda.io/ai_module 1.1.3835 → 1.1.3837

package/index.mjs

@@ -2018,7 +2018,7 @@ export const unarchive_ai_chat = async function (req) {
 //   }
 // };
 
-const create_image = async function (prompt, model = 'gpt-image-1-mini', size = '1024x1024', n = 1, width = 256, height = 256) {
+const create_image = async function (uid, prompt, model = 'gpt-image-1-mini', size = '1024x1024', n = 1, width = 256, height = 256) {
   try {
     const response = await client.images.generate({
       model,
@@ -2061,6 +2061,9 @@ const create_image = async function (prompt, model = 'gpt-image-1-mini', size =
     const resized = await sharp(fullImgBuffer).resize(width, height).toFormat('png').toBuffer();
 
     const thumbnailBase64 = resized.toString('base64');
+
+    record_chat_usage(uid, response.usage.input_tokens, response.usage.output_tokens, 'create image', prompt);
+
     return thumbnailBase64;
   } catch (err) {
     console.error('create_image error:', err);
@@ -5459,7 +5462,7 @@ 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) {
   try {
     // 1. Generate the images
-    let result = await create_image(prompt, undefined, undefined, numGenerations);
+    let result = await create_image(uid, prompt, undefined, undefined, numGenerations);
 
     // Normalize to array
     const images = Array.isArray(result) ? result : [result];
@@ -5775,7 +5778,7 @@ export const convert_profile_image_to_avatar = async function (req, job_id, head
           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(prompt, undefined, undefined, 1);
+          imageBase64 = await create_image(uid, prompt, undefined, undefined, 1);
           avatar_source = 'fictional';
         };
         // personal account
@@ -6261,7 +6264,7 @@ const record_chat_usage = async function (uid, input_tokens, output_tokens, sour
       stat: 3,
     };
     const save_ret = await db_module.save_app_couch_doc(app_id, usage_doc);
-    console.log(save_ret);
+    // console.log(save_ret);
   } catch (err) {
     console.error(err);
   }

package/package.json

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