From 7d05a9d1572d8d7c2ae11b3ea535a7ed0e34abec Mon Sep 17 00:00:00 2001
From: Emily <adharaultimasheireen@gmail.com>
Date: Mon, 23 Dec 2024 14:37:19 +0100
Subject: [PATCH] fix chat limits update

---
 dashboard/pages/project_creation.vue         | 1 +
 dashboard/server/api/ai/send_message.post.ts | 4 ++++
 dashboard/server/services/AiService.ts       | 2 +-
 3 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/dashboard/pages/project_creation.vue b/dashboard/pages/project_creation.vue
index 3cb4ef3..70a28ba 100644
--- a/dashboard/pages/project_creation.vue
+++ b/dashboard/pages/project_creation.vue
@@ -44,6 +44,7 @@ async function createProject() {
             await actions.setActiveProject(newActiveProjectId);
         }
 
+        setPageLayout('dashboard');
         router.push('/');
 
     } catch (ex: any) {
diff --git a/dashboard/server/api/ai/send_message.post.ts b/dashboard/server/api/ai/send_message.post.ts
index 5e4b66b..e7dd86b 100644
--- a/dashboard/server/api/ai/send_message.post.ts
+++ b/dashboard/server/api/ai/send_message.post.ts
@@ -1,5 +1,6 @@
 import { sendMessageOnChat, updateChatStatus } from "~/server/services/AiService";
 import { getAiChatRemainings } from "./chats_remaining";
+import { ProjectLimitModel } from "@schema/project/ProjectsLimits";
 
 
 
@@ -15,6 +16,9 @@ export default defineEventHandler(async event => {
     const chatsRemaining = await getAiChatRemainings(pid);
     if (chatsRemaining <= 0) return setResponseStatus(event, 400, 'CHAT_LIMIT_REACHED');
 
+
+    await ProjectLimitModel.updateOne({ project_id: pid }, { $inc: { ai_messages: 1 } });
+
     const currentStatus: string[] = [];
 
     let responseSent = false;
diff --git a/dashboard/server/services/AiService.ts b/dashboard/server/services/AiService.ts
index 6e7772d..8a93ca3 100644
--- a/dashboard/server/services/AiService.ts
+++ b/dashboard/server/services/AiService.ts
@@ -191,7 +191,7 @@ export async function sendMessageOnChat(text: string, pid: string, time_offset:
 
     if (chatMessages && chatMessages.length > 0) {
         messages.push(...chatMessages);
-        await ProjectLimitModel.updateOne({ project_id: pid }, { $inc: { ai_messages: 1 } })
+
         await updateChatStatus(chat_id, '', false);
     } else {
         const roleMessage: OpenAI.Chat.Completions.ChatCompletionMessageParam = {