@xuda.io/ai_module 1.1.3851 → 1.1.3853
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/index.mjs +3 -2
- package/package.json +1 -1
package/index.mjs
CHANGED
|
@@ -5071,7 +5071,7 @@ const ai_chat_conversation = async function (req, job_id, headers) {
|
|
|
5071
5071
|
conversation_doc.stat = 3;
|
|
5072
5072
|
conversation_doc.messages.push(obj);
|
|
5073
5073
|
await db_module.save_app_couch_doc_native(app_id, conversation_doc);
|
|
5074
|
-
record_chat_usage(uid, output.state._context.usage.inputTokens, output.state._context.usage.outputTokens, 'conversation chat', prompt, model, { conversation_id });
|
|
5074
|
+
record_chat_usage(uid, output.state._context.usage.inputTokens, output.state._context.usage.outputTokens, 'conversation chat', prompt, model, { conversation_id, ai_agent_id: obj.ai_agent_id });
|
|
5075
5075
|
// if (!conversation_doc.chat_image && (!conversation_doc.thumbnail_request_ts || Date.now() - conversation_doc.thumbnail_request_ts > 120000)) {
|
|
5076
5076
|
// update_thumbnail('conversation', conversation_doc, app_id, uid, job_id, headers);
|
|
5077
5077
|
// }
|
|
@@ -6251,7 +6251,7 @@ const create_ai_agent_image = async function (req, job_id, headers) {
|
|
|
6251
6251
|
}
|
|
6252
6252
|
};
|
|
6253
6253
|
|
|
6254
|
-
const record_chat_usage = async function (uid, input_tokens, output_tokens, source, prompt, metadata = {}) {
|
|
6254
|
+
const record_chat_usage = async function (uid, input_tokens, output_tokens, source, prompt, model, metadata = {}) {
|
|
6255
6255
|
try {
|
|
6256
6256
|
const app_id = await account_module.get_account_default_project_id(uid);
|
|
6257
6257
|
|
|
@@ -6267,6 +6267,7 @@ const record_chat_usage = async function (uid, input_tokens, output_tokens, sour
|
|
|
6267
6267
|
output_tokens,
|
|
6268
6268
|
stat: 3,
|
|
6269
6269
|
metadata,
|
|
6270
|
+
model,
|
|
6270
6271
|
};
|
|
6271
6272
|
const save_ret = await db_module.save_app_couch_doc(app_id, usage_doc);
|
|
6272
6273
|
// console.log(save_ret);
|