From 617fb491789b237e43c29ccb62ce05cef2bcebee Mon Sep 17 00:00:00 2001 From: KernelDeimos Date: Mon, 24 Mar 2025 15:50:39 -0400 Subject: [PATCH] fix: allow fake-chat to report usage for costly model Modified the event handler in AIChatService to only skip usage reporting for fake-chat when it's not using the costly model, allowing the costly model to properly report its usage and costs. ai: true --- src/backend/src/modules/puterai/AIChatService.js | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/backend/src/modules/puterai/AIChatService.js b/src/backend/src/modules/puterai/AIChatService.js index 67e57b47c..0d39a42e6 100644 --- a/src/backend/src/modules/puterai/AIChatService.js +++ b/src/backend/src/modules/puterai/AIChatService.js @@ -78,7 +78,8 @@ class AIChatService extends BaseService { const svc_event = this.services.get('event'); svc_event.on('ai.prompt.report-usage', async (_, details) => { - if ( details.service_used === 'fake-chat' ) return; + // Only skip usage reporting for fake-chat if it's not using the costly model + if ( details.service_used === 'fake-chat' && details.model_used !== 'costly' ) return; const values = { user_id: details.actor?.type?.user?.id,