@xuda.io/ai_module 1.1.3850 → 1.1.3852

package/index.mjs

@@ -2018,7 +2018,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) {
+const create_image = async function (uid, prompt, model = 'gpt-image-1-mini', size = '1024x1024', n = 1, width = 256, height = 256, metadata = {}) {
   try {
     const response = await client.images.generate({
       model,
@@ -2062,7 +2062,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);
+    record_chat_usage(uid, response.usage.input_tokens, response.usage.output_tokens, 'create image', prompt, model, metadata);
 
     return thumbnailBase64;
   } catch (err) {
@@ -5462,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(uid, prompt, undefined, undefined, numGenerations);
+    let result = await create_image(uid, prompt, undefined, undefined, numGenerations, undefined, undefined, { drive_type, file_path, file_name });
 
     // Normalize to array
     const images = Array.isArray(result) ? result : [result];
@@ -6251,7 +6251,7 @@ const create_ai_agent_image = async function (req, job_id, headers) {
   }
 };
 
-const record_chat_usage = async function (uid, input_tokens, output_tokens, source, prompt, metadata = {}) {
+const record_chat_usage = async function (uid, input_tokens, output_tokens, source, prompt, model, metadata = {}) {
   try {
     const app_id = await account_module.get_account_default_project_id(uid);
 
@@ -6267,6 +6267,7 @@ const record_chat_usage = async function (uid, input_tokens, output_tokens, sour
       output_tokens,
       stat: 3,
       metadata,
+      model,
     };
     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.3850",
+  "version": "1.1.3852",
   "description": "Xuda AI Module",
   "main": "index.mjs",
   "type": "module",