@xuda.io/ai_module 1.1.4250 → 1.1.4252

package/index.mjs

@@ -5120,7 +5120,7 @@ const ai_chat_conversation = async function (req, job_id, headers) {
     const output = await runner.run(_agent, prompt, opt);
     const done = async function (output) {
       try {
-        const obj = { id: output?.output?.[0]?.id, ts: Date.now(), ai_agent_id: output.state._currentAgent.name, attachments, conversation_type: 'ai_chat' };
+        const obj = { id: output.state._lastTurnResponse.responseId, ts: Date.now(), ai_agent_id: output.state._currentAgent.name, attachments, conversation_type: 'ai_chat' };
         conversation_doc = await db_module.get_app_couch_doc_native(account_profile_info.app_id, conversation_id);
 
         conversation_doc.ts = Date.now();
@@ -5134,11 +5134,22 @@ const ai_chat_conversation = async function (req, job_id, headers) {
           inputTokens += item.usage.inputTokens;
           outputTokens += item.usage.outputTokens;
         }
-        debugger;
+
         await db_module.save_app_couch_doc_native(account_profile_info.app_id, conversation_doc);
-        // await db_module.save_app_couch_doc_native(account_profile_info.app_id, { _id: obj.id, stat: 3, docType: 'chat_conversation_item', uid, conversation_type: obj.conversation_type, date_created_ts: obj.ts, conversation_id, response: output.output });
+        await db_module.save_app_couch_doc_native(account_profile_info.app_id, {
+          _id: obj.id,
+          stat: 3,
+          docType: 'chat_conversation_item',
+          uid,
+          conversation_type: obj.conversation_type,
+          date_created_ts: obj.ts,
+          conversation_id,
+          output: output.output,
+          prompt,
+          response: output.state._currentStep.output,
+        });
 
-        record_chat_usage(uid, inputTokens, outputTokens, 'conversation chat', prompt, model, { conversation_id, ai_agent_id: obj.ai_agent_id, item_id: obj.id });
+        record_chat_usage(uid, inputTokens, outputTokens, 'conversation chat', prompt, model, { conversation_id, ai_agent_id: obj.ai_agent_id, item_id: obj.id }, account_profile_info);
         //   if (!conversation_doc.chat_image && (!conversation_doc.thumbnail_request_ts || Date.now() - conversation_doc.thumbnail_request_ts > 120000)) {
         //     update_thumbnail('conversation', conversation_doc, app_id, uid, job_id, headers);
         //   }
@@ -5764,7 +5775,7 @@ export const create_profile_avatar = async function (req, job_id, headers) {
         background: 'transparent',
       });
       imageBase64 = ai_avatar_response.data[0].b64_json;
-      record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'profile avatar', prompt, model, { url });
+      record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'profile avatar', prompt, model, { url }, account_profile_info);
     } else {
       //////////////////
       imageBase64 = Buffer.from(await images[0].image_blob.arrayBuffer()).toString('base64');
@@ -5842,7 +5853,7 @@ export const create_profile_avatar = async function (req, job_id, headers) {
           The background should be fully removed, remove trees,clouds,sky or any landscape objects`,
           background: 'transparent',
         });
-        record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'profile avatar', prompt, model, { url });
+        record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'profile avatar', prompt, model, { url }, account_profile_info);
         imageBase64 = ai_avatar_response.data[0].b64_json;
         avatar_source = 'fictional';
       } else {
@@ -5859,7 +5870,7 @@ export const create_profile_avatar = async function (req, job_id, headers) {
             Context: ${JSON.stringify({ first_name, last_name, bio, country, mainCategory, subCategory })}`,
             background: 'transparent',
           });
-          record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'profile avatar', prompt, model, { url });
+          record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'profile avatar', prompt, model, { url }, account_profile_info);
           imageBase64 = ai_avatar_response.data[0].b64_json;
           avatar_source = 'ai profile';
         };
@@ -6313,7 +6324,7 @@ const create_ai_agent_image = async function (req, job_id, headers) {
       background: 'transparent',
     });
     imageBase64 = ai_avatar_response.data[0].b64_json;
-    record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'agent avatar', prompt, model, { ai_agent_id });
+    record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'agent avatar', prompt, model, { ai_agent_id }, account_profile_info);
     console.log('Normalizing final avatar to 1024 with transparent padding...');
     imageBase64 = await normalizeBase64To1024(imageBase64);
 

package/package.json

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