diff --git a/src/backend/src/modules/puterai/AIChatService.js b/src/backend/src/modules/puterai/AIChatService.js index 9dce80f08..8b2b867af 100644 --- a/src/backend/src/modules/puterai/AIChatService.js +++ b/src/backend/src/modules/puterai/AIChatService.js @@ -279,6 +279,7 @@ class AIChatService extends BaseService { intended_service, parameters }; + await svc_event.emit('ai.prompt.validate', event); if ( ! event.allow ) { test_mode = true; } diff --git a/src/backend/src/modules/puterai/FakeChatService.js b/src/backend/src/modules/puterai/FakeChatService.js index 330805530..6855aedd2 100644 --- a/src/backend/src/modules/puterai/FakeChatService.js +++ b/src/backend/src/modules/puterai/FakeChatService.js @@ -65,6 +65,10 @@ class FakeChatService extends BaseService { "output_tokens": 1 } }, + "usage": { + "input_tokens": 0, + "output_tokens": 1 + }, "logprobs": null, "finish_reason": "stop" }