From d4ca9d438caa413e4190246fb27ef4bfec94007e Mon Sep 17 00:00:00 2001
From: KernelDeimos <eric.alex.dube@gmail.com>
Date: Thu, 12 Dec 2024 09:51:32 -0500
Subject: [PATCH] dev: fix issue with prompt validation

---
 src/backend/src/modules/puterai/AIChatService.js   | 1 +
 src/backend/src/modules/puterai/FakeChatService.js | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/src/backend/src/modules/puterai/AIChatService.js b/src/backend/src/modules/puterai/AIChatService.js
index 9dce80f0..8b2b867a 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 33080553..6855aedd 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"
                 }