diff --git a/frontend/app/components/Kai/KaiStore.ts b/frontend/app/components/Kai/KaiStore.ts index d4d85526a..f1c9071e0 100644 --- a/frontend/app/components/Kai/KaiStore.ts +++ b/frontend/app/components/Kai/KaiStore.ts @@ -211,7 +211,13 @@ class KaiStore { bumpUsage = () => { this.usage.used += 1; - this.usage.percent = (this.usage.used / this.usage.total) * 100; + this.usage.percent = Math.max( + (this.usage.used / this.usage.total) * 100, + 100, + ); + if (this.usage.used >= this.usage.total) { + toast.error('You have reached the daily limit for queries.'); + } }; sendMessage = (message: string) => { diff --git a/frontend/app/components/Kai/components/ChatMsg.tsx b/frontend/app/components/Kai/components/ChatMsg.tsx index 27f89a660..2f2439e90 100644 --- a/frontend/app/components/Kai/components/ChatMsg.tsx +++ b/frontend/app/components/Kai/components/ChatMsg.tsx @@ -53,7 +53,7 @@ function ChatMsg({ const onCancelEdit = () => { kaiStore.setQueryText(''); kaiStore.setReplacing(null); - } + }; const onFeedback = (feedback: 'like' | 'dislike', messageId: string) => { kaiStore.sendMsgFeedback(feedback, messageId, siteId); }; @@ -108,16 +108,11 @@ function ChatMsg({ } }; return ( -