From 0a9474d00c0a418c3cddc97dda45e585038249fa Mon Sep 17 00:00:00 2001
From: Emily <adharaultimasheireen@gmail.com>
Date: Mon, 16 Dec 2024 16:57:52 +0100
Subject: [PATCH] adjust ai

---
 dashboard/package.json                        |   3 +-
 dashboard/pages/analyst.vue                   |  63 ++++++++++++++----
 dashboard/public/favicon.ico                  | Bin 4661 -> 9662 bytes
 dashboard/server/ai/functions/AI_Events.ts    |   8 +--
 dashboard/server/ai/functions/AI_Visits.ts    |  62 ++++++++++++-----
 .../server/api/ai/[chat_id]/get_messages.ts   |  10 +--
 dashboard/server/api/ai/send_message.post.ts  |   4 +-
 dashboard/server/services/AiService.ts        |  33 ++++-----
 pnpm-lock.yaml                                |  14 +++-
 9 files changed, 134 insertions(+), 63 deletions(-)

diff --git a/dashboard/package.json b/dashboard/package.json
index 1a74478..b97d66b 100644
--- a/dashboard/package.json
+++ b/dashboard/package.json
@@ -35,7 +35,8 @@
     "vue-chart-3": "^3.1.8",
     "vue-markdown-render": "^2.2.1",
     "vue-router": "^4.3.0",
-    "winston": "^3.14.2"
+    "winston": "^3.14.2",
+    "zod": "^3.24.1"
   },
   "devDependencies": {
     "@nuxt/ui": "^2.15.2",
diff --git a/dashboard/pages/analyst.vue b/dashboard/pages/analyst.vue
index 8fde690..3a37cff 100644
--- a/dashboard/pages/analyst.vue
+++ b/dashboard/pages/analyst.vue
@@ -4,13 +4,17 @@ import VueMarkdown from 'vue-markdown-render';
 
 definePageMeta({ layout: 'dashboard' });
 
+
+const debugModeAi = ref<boolean>(false);
+
+const { userRoles } = useLoggedUser();
+
 const { project } = useProject();
 
 const { data: chatsList, refresh: reloadChatsList } = useFetch(`/api/ai/chats_list`, {
     headers: useComputedHeaders({ useSnapshotDates: false })
 });
 
-
 const viewChatsList = computed(() => (chatsList.value || []).toReversed());
 
 const { data: chatsRemaining, refresh: reloadChatsRemaining } = useFetch(`/api/ai/chats_remaining`, {
@@ -21,14 +25,14 @@ const currentText = ref<string>("");
 const loading = ref<boolean>(false);
 
 const currentChatId = ref<string>("");
-const currentChatMessages = ref<{ role: string, content: string, charts?: any[] }[]>([]);
+const currentChatMessages = ref<{ role: string, content: string, charts?: any[], tool_calls?: any }[]>([]);
 const currentChatMessageDelta = ref<string>('');
 
 const currentChatMessageDeltaHtml = computed(() => {
     const lastData = currentChatMessageDelta.value.match(/\[(data:(.*?))\]/g);
     const cleanMessage = currentChatMessageDelta.value.replace(/\[(data:(.*?))\]/g, '');
     if (!lastData || lastData.length == 0) return cleanMessage;
-    return `<div> <span>LOADER HERE: ${lastData.at(-1)}</span>  ${cleanMessage} </div>`;
+    return `<div class="flex items-center gap-1"> <i class="fas fa-loader animate-spin"></i> <div> ${lastData.at(-1)}</div> </div> <div> ${cleanMessage} </div>`;
 });
 
 const scroller = ref<HTMLDivElement | null>(null);
@@ -36,7 +40,7 @@ const scroller = ref<HTMLDivElement | null>(null);
 
 async function pollSendMessageStatus(chat_id: string, times: number, updateStatus: (status: string) => any) {
 
-    if (times > 20) return;
+    if (times > 100) return;
 
     const res = await $fetch(`/api/ai/${chat_id}/status`, {
         headers: useComputedHeaders({
@@ -48,14 +52,22 @@ async function pollSendMessageStatus(chat_id: string, times: number, updateStatu
     updateStatus(res.status);
 
     if (res.completed === false) {
-        setTimeout(() => pollSendMessageStatus(chat_id, times + 1, updateStatus), 200);
+        setTimeout(() => pollSendMessageStatus(chat_id, times + 1, updateStatus), (times > 20 ? 1000 : 500));
     } else {
-        currentChatMessages.value.push({
-            role: 'assistant',
-            content: currentChatMessageDelta.value.replace(/\[data:.*?\]/g, ''),
+
+        const messages = await $fetch(`/api/ai/${chat_id}/get_messages`, {
+            headers: useComputedHeaders({ useSnapshotDates: false }).value
         });
+        if (!messages) return;
+
+        currentChatMessages.value = messages.map(e => ({ ...e, charts: e.charts.map(k => JSON.parse(k)) })) as any;
         currentChatMessageDelta.value = '';
 
+        // currentChatMessages.value.push({
+        //     role: 'assistant',
+        //     content: currentChatMessageDelta.value.replace(/\[data:.*?\]/g, ''),
+        // });
+
     }
 
 }
@@ -67,7 +79,7 @@ async function sendMessage() {
 
     loading.value = true;
 
-    const body: any = { text: currentText.value }
+    const body: any = { text: currentText.value, timeOffset: new Date().getTimezoneOffset() }
     if (currentChatId.value) body.chat_id = currentChatId.value
 
     currentChatMessages.value.push({ role: 'user', content: currentText.value });
@@ -92,6 +104,8 @@ async function sendMessage() {
             currentChatMessageDelta.value = status;
         });
 
+
+
     } catch (ex: any) {
 
         if (ex.message.includes('CHAT_LIMIT_REACHED')) {
@@ -179,6 +193,8 @@ async function deleteChat(chat_id: string) {
 
 const { visible: pricingDrawerVisible } = usePricingDrawer()
 
+
+
 </script>
 
 <template>
@@ -208,7 +224,7 @@ const { visible: pricingDrawerVisible } = usePricingDrawer()
 
                 <div ref="scroller" class="flex flex-col w-full gap-6 px-6 xl:px-28 overflow-y-auto pb-20">
 
-                    <div class="flex w-full flex-col" v-for="message of currentChatMessages">
+                    <div class="flex w-full flex-col" v-for="(message, messageIndex) of currentChatMessages">
 
                         <div class="flex justify-end w-full poppins text-[1.1rem]" v-if="message.role === 'user'">
                             <div class="bg-lyx-widget-light px-5 py-3 rounded-lg">
@@ -217,15 +233,32 @@ const { visible: pricingDrawerVisible } = usePricingDrawer()
                         </div>
 
                         <div class="flex items-center gap-3 justify-start w-full poppins text-[1.1rem]"
-                            v-if="message.role === 'assistant' && message.content">
+                            v-if="message.role === 'assistant' && (debugModeAi ? true : message.content)">
                             <div class="flex items-center justify-center shrink-0">
                                 <img class="h-[3.5rem] w-auto" :src="'analyst.png'">
                             </div>
                             <div class="max-w-[70%] text-text/90 ai-message">
-                                <vue-markdown :source="message.content" :options="{
+
+                                <vue-markdown v-if="message.content" :source="message.content" :options="{
                                     html: true,
                                     breaks: true,
                                 }" />
+
+
+
+                                <div v-if="debugModeAi && !message.content">
+                                    <div class="flex flex-col"
+                                        v-if="message.tool_calls && message.tool_calls.length > 0">
+                                        <div> {{ message.tool_calls[0].function.name }}</div>
+                                        <div> {{ message.tool_calls[0].function.arguments }} </div>
+                                    </div>
+                                </div>
+
+                                <div v-if="debugModeAi && !message.content"
+                                    class="text-[.8rem] flex gap-1 items-center w-fit hover:text-[#CCCCCC] cursor-pointer">
+                                    <i class="fas fa-info text-[.7rem]"></i>
+                                    <div class="mt-1">Debug</div>
+                                </div>
                             </div>
                         </div>
 
@@ -295,6 +328,9 @@ const { visible: pricingDrawerVisible } = usePricingDrawer()
                     </div>
                 </div>
 
+                <div :class="{ '!text-green-500': debugModeAi }" class="cursor-pointer text-red-500 w-fit"
+                    v-if="userRoles.isAdmin.value" @click="debugModeAi = !debugModeAi"> Debug mode </div>
+
                 <div class="flex justify-between items-center pt-3">
                     <div class="flex items-center gap-2">
                         <div class="bg-accent w-5 h-5 rounded-full animate-pulse">
@@ -358,6 +394,9 @@ const { visible: pricingDrawerVisible } = usePricingDrawer()
         color: white;
     }
 
+    p:last-of-type {
+        margin-bottom: 0;
+    }
 
     p {
         line-height: 1.8;
diff --git a/dashboard/public/favicon.ico b/dashboard/public/favicon.ico
index 2db10370e731be8e960454983d4e30071b0bca0f..e2b945a4d02057aed417c3e46be06c80be6bd537 100644
GIT binary patch
literal 9662
zcmeI2O(<<q6vxk#ggYRYCka=*S2vNbNEs+2WoF3al_Cc6H4>4Dk_nN|qA0IO6eCKA
ziI^xK17C%Fi%byfo&VCl-Qyna@w(i7>ujBK?p=GIbAD^>wbx#IyNdEd{Yyzv)U!?b
z`&&_dDvDxP4^^c6Q4h$u{8ALF<@I<K@R`)Jd6+gZeZceqhY#58cE@L5i4R#>Su`{>
zL|a>1)X~x5GyDk&3AC}XL89E;+)z|hluz(CG&C^w{{B8yR#wu(!vnRnwD<&jQ&SVO
zx3{-bSXdYp6&3OK<m6=O`D<%ynH4&S@d-W7&(F!v&(9$fbuwKO6BFs><%RnC`ZSyv
zyE{8O8ktPof@OYwp02O2DKs=x!#X}bPRq;78kx99mTP)?I{Qs|dASSc=;$b|uCBVs
zWnl+LTU#65-Q7`8P>>5}XJ;p!ot?SJW#V63TcgRzNnOU0k`nHln3x#ddKvV%xVWI!
z)>d6Uo6W}j#l^+C_1@+W4i09%#l=Ocsi~o)q$C&4nVA{FT;}iZ?^5S&{=mRMT3TA-
z7>W3a0$+0W7wCjiS63$;zq9=iJHd$YdT?++&U0>ecQ?-u85tQ;>i~V=^Y9nwfX|SH
zhlkVd?k?+=nVIP`I)GX9QDkH!>+tyaD7`O%IVUHF{oXkrvREuMFfhP!<ky(<-p<)V
zU*y}RrKJ=Q5a6(ZsQa$#;NT$L-riDdY^;VMFE5W)R#sS#`}=#^+}xy*krC?c?WL)y
zDOz7&=kc<)x2L)9yST-BMEq}VZr0_%nzFvWo`#2qnR$DAoAafyu`$A$XJTSP_dfCb
z@3?~P=;-KEIW)#Q;&DYqh1dB-epgUXV3?(_u#kN#EiKLPJzv=uh^NS<jlKhU>&eN9
z;r2$K`@ik5uDQCpGQ@*;gfZ_NxBmCNp1LnTKcD>}J3HI04&Z)zdZOs)Xt!-W#pmuh
zAt50=9;>UXKV-uEfPD5dzUJ=To~mD1SfIJNxexgp8ygMxlc%_hKBudzi%w5Z-{nWn
zdU<&%rR||(Sy>s|HX<TIR|n+X7`Jh8ak}+J?etaca&vRv^$p}fC{<NeQnJG)R;!g6
z->xOGzQG>j^z^i}?4nPfpPw~x9X^k_?eOrB*9fAWtXJ&Ci9I^32f%)Od`$82@zUEK
zHl3ZF<rs>*6uKbZq^73&1UvYV<6~V3yJNo*|0j^uSM<2BpNj&1u{PFgC#$T#zn^+~
idNlhhva;%N`J&DB^e|=S?=wsv_%a_*O=}i!S>PYI$&LpA

literal 4661
zcma)A_dgW=AAjG#*^caWPEtlLQdXRtJ))4zb7e$kNJj4LokCVJ5|TY4^U6#J*%g<a
zeJUgC+vh*{zQ*JAd_7)2y&kV$p06hWkbnN4K>!K}vjf2O562iAX)`nMG5mwfx;nV)
z|4si(1pHsw3B<4e*LipwnL<A2Kga|CIjM_NH}#)d&kjsCToCHsO|t*?8P~_ADatw2
znaz<OXT<$8_C`rj!LY7)eL;bdqJQJcvVe{7&jOAi#m0uzr>9KLnY^3Pz&ZA$Z$2Ei
zmFxT4;yUB8`u*3@#>wKOYsJH{-J0pu&7-3r)!6*A|8H!+HD1w3H3SlQf+g&vYDTPK
zQ26Dcr|Z~TG$?*MsKw*JQw+FVCa5vLUHf&BX$|qWFb7tm(jJT06e-F$4v``aysP-S
zO~WUhKl$ZKL5zi6e~f5-i;TtPmV!Yxev}!JzQh#;b$}ATR5H6~CUD_$rGV!pzjm9M
zIazelF{tgXKon#*FlL0KS|H1@L2z@3kcK=uX#iN0+7X3Un?FG8m0*L{`j0ez3SLA*
zf8K3lg$bH-{~$_2g<mQnb*_O8Zsud>K2J?NuO$b`2aa!UU({Gmyl2p{c^(lI#eze9
z2R-CZEXlhE58yQz_O^GZO1|tEYyiYlVA{{3nbGO}@qydK&rmYo=v6k-MQE-BS9C(I
zmi9|{Mb{Z__*gzaQb!oL{E&=lkt|a@8Nv^_TP%c<AB&MhQ})5g&4hrv(pt;{{HVW!
zERxo277;r*)D_^Im?JngP(5}W{0gZsE-ZJp4!Xof@&UQYe7k95cT&RcXD(~#OL?i|
z=`-0OzaY!!91pK4NlR<xMo2>Md!mt;VBX?3jBoVC-)EuvHf-sMQ*iP<AjGB&x%@Md
zOVrcPYp>x(vT+0=or$d0adoQ`@!L>%Dun^1_V&RRB2g4t2=x>wR9yS^#(;$>PX3Pd
z8xc=X?{yk40Y<5>+xrZi<MHbetZDX5k>t;N97qfYv61j<li$^Ni7^ptx8_AJ?4}Dh
zfKz;^s-LwT!ujOy;MrJQ5Pg~~5DY7Efbivk-cN^IBH|8#DbX2m$ao7!u2sYe+s{u7
za{aA%6Zw%3w7q#o`cAY5Ok8JJn7Xkiseeo#<gwDqoQ66=Fz@Xt_SU?Yfa>#wYP06P
zOiea$0~8Z82%NdnAELT5**AvS()&`??}NQCi<ud`CTt(APO+8tPZ5QL0R{S1!9{5^
z9o$&N9s8$awD)S}BF=8Y28{$5vtKJk=x8JVP-7y=Q9CBGy{j5zSCeJTMn=y)$qia1
zLQN!@$Mp&c;V($c`|vgpEnmh%f=u;NgsiWEW>)g)3qegP-e2%be8K9jFx8vcFyz)n
zka}U<e$rjCPfjm|ud#zMLHi+^;sCrmt`4}2#PNX~m+f8W?_^Cl`4i~*hW%A0_Pn?$
z+1tNwL1&*HM{W`&+7ggX?6Pd1Hb;?BeWi$Yx`wm!HXstwv=P@SF)<lA?=@WJX9VdD
z1a~N67w8($jzD&+c#PSg?y5mFaTgSyG2imDRJXZMFQI5r;J~|T2$~VZZ*&&+%oDSN
zRo0QgQ^;y-G^GVeeu(B-%f20T_4eQI40}am%${eL)xrFe$|`DQ&&j8vSvu<1RWDk|
z79aQ=OK;WI0HL(o5R&Qr6W@eRL$T*!o0?5~JaFZ~JLLe+t;7ywAcMvNR7=k9vYuG@
z`Ju?r6D=_`C7T8Ng3f=uT0F5WC*~#%I2Z~8{&Yl=CD0b=r_SVYzIEA;<UPNF{I^HG
z!t8hD4Y0TMVEq)86Hev_BOQbbIAwmShl3I4es^hNp#*XZ3S{Lbyx*APp-P!_G+=+I
zL%@W3dC}4YJuC0}>;>`Qd3@@W0?|rC2;iSrE>y%|I(<JqY|Flzt|#{IK%eb<J(e>u
zfY13>M4^QfYcy5@;*6f%3#qL>5-Y9aEia>gD+*q(gYr{(c#rN@R-SqHgxaz^r<bf(
zUe4Now5ac&f=_QEz1o=lVsA;v1Jn0W1Q$Q`P>6ycKMk}UfnAwCaYRd-e$WEio3~g#
z(}QQS)+)Z>8>$-G!Z{g3Dc+cbv_*rFRRq=r^b8LCVl^t|>~x|lifb~`(1a#203$Z~
zoLw>1j4)x+8&91MRL-YMC{95vw}}wpVP#*gSF?w7kjNBq2#(?eDyVz@s=W8TPqL$8
zltVu1!Ct3B2XyZUV529yxw>0eWtC9-n<0#AZ13J@o$iK*5dNOaXZ9B{wX~F&2H(V_
z(lbh{8dkAe;?FxSp)$aoSaemT++Q(fqCk+SQZ7F|R(9s3FrRzc;pVaaYzS(OZMrhH
z-}qy|;gQL_|KYFvegu2_X`h@4uHM`HNsTO&<9XGf5KPIhtvzHtpMz<Bp}mzRLy?~@
z9XTqBCcL7CLuc%zaKWqlWs&nI+}SwqDS79fODNvQt+9UAax{V4$3_&nDj9!Hh)!45
zxXB%={t6SBniG#i8KJ;O|Fzq7a!R*!`a`@S+nSy|R5*&NkCDVFMMGXc+&oNWX!AXk
zY0HOwCE{R%+Rx#ZIVJn`pWct~{NlKtC;U*;jfV1SPnaeSbn|N(PyOzoeMLR;5ch^I
zeNBCIqXAuIbB{qD1!96I*P?ero|y3>ULQOziH;51_p4KvxUjQymuBY1)KX(%B9A&?
z->r;tRy;5=#9m73QQh|B(}vJL2j|CScN|d1Z*LlZ*MrsK#I3_uJsI7*vm0PXUK{~=
z%2RVp^ZA+Cl7ll6Zd{#9_YqhEXt0TI*zMq>-KcDlFnUkZyHa!kre8^`TqPVeA{eJv
zrf_ODH*?ja+eNCwbaZbfKCy(^@@(jr3{8y4&N#c|rul7t-?y#TyPWdQ1<ak3B5PHZ
zb6aZmZU}ca!%NCgfF4k|s(>k*&FS*FY>%)!e>?df5z$3w*ok{z?X2dZu*KYflvGGr
z@${n0F~;usc3hgOo}<t8TM!Bh^_kFzKPG0hb(g$kI`2qRZ>!V^iE|M5rtDbl?#ef#
zEr62CBf)PafBy!ZHm4iR$i@aaEsA!Es)o1zP@TQLIFqNf8=({FE(WOZ+>En`+nrz>
zv5Dl)vSU8ldY6u)y`Gn8TW%tnSWkrtbz=@`U@3X^=YnYM;aziP%~1|6U6pnnVrgSH
z;HXDCPA+vpj2{?VBrdh7H2$#hO6)dlvrm-`K2@&adeN9D=p-%sE=blGIq4j)HtQJQ
zevf8JeZAgMOv23U7M$Ykh4`if=h7<NAT$|$;1KEht2vPmcoWa})YTj-s)=yL)(j%j
z?3uWfHmccu>9Ij;6>n1A4>WIaemH)q-+0qCOgnmnWiwaJsGd8;=aZ;~c4^|<k}wNn
zfwBhi#^KSFpk{aZAaTo~U4)xK`AW<#-OrzGUG2SvdQbR;;o&aM6||*{RaSZ0jB}X3
zZ8}ztNAyJz4k6kmdczMhLt0nb?i1=&0j>PV9>oC4b}*-6Vs#{GdV3r7OwjGG2bZ$X
zEAgcJUI{Ic>6OnvsG4)dci%3JUXnkU*62^#88=R@he6itN(#f|os)R`Pk>uM-Ia+Q
z89f56EhA=TCO=?FAk!oE$fn9Lt2_a5Pe|{n;%9I96HMVt&i>2Y$Bt-NZcD>c!4Jod
z5ZXssUiH5IDT`Tzkl^*LqPliJBgBD~^W?$3{+l=CS`TO^L!i^c27UF(GN>&xC(8Zt
z3nw^N!^Fj`+b2{Pwxzf4p7M?yCs&deO1`@cTg2>S{n=@nRFm`+8kk*}J!!PAGwI)*
z{LSxh5|h42nBOM5hIBp&mT9v7b7i1t-*Iy0x<GYx-55(X%fw+?w{sfmvB%)Eovb^L
zXwTo=EG-xGoA*eXZmakd`I(`%@aW8BPQsJ&z@4Tc@s8nZlBHkDE!~`$ULF-TA7pQ}
zuFkHz|7NrgaeE^3&Gz>@;!YN!l{Mjl*8HLB+l`sa;YSNw4Su7x$_^^)yRCoBFD%-Z
zyz?8kY?6I2n2@87E5}G0el=E3e7s#v9qTBAb={R$czjp)<E&fKOz)5pgQem@jdR<0
zpPx|EfVxvWS!MmZbhg*7nu1t-?&G1U)VR`2Qtb>sFx3*u+*30gJ!};X$ZcDxa>6?#
zoxKuB*~ROUwTzpmT(UZq*Njb|6lS3G+BWFTNey$;;nG_B5W(7-Jx+M!Sju2@D&a@%
zj6U$WuPBAI{<?M6TO(`vMC+RL{A!qKVQjun@_@EFd~vh=R4u*JT%iIqB$4a;=l1k2
zXeIX-&nIRZ=hoV7$L|XED!d_ZLom!lw7gMzw3?7|Ci2<QXq|Ijsx$7Og%+EZ>$JsK
zWzw=+pHHotOS8Ww6Ub1TNO7OqyaJS>L-pRSA5^mZ+A47A$eKW^B6B+56gV*0hAP3y
zfuPnjS<T76%h(|>@FH%(`SSpo*7dRfiAT|qEgW^t^v^+#imUKw#Y#qK+N4NM$~Y*F
z5*E~wj;|}e;ng!C0U_h3HVr{F_0=+Tr3*q-i=$NdN{ud?0a@w`izQb+Il#FU`Rp7j
zY$uR|$n<2qJ4ZpfYgg@>p-{GBaO1C|T?rstWrcK4RHTcN8rbF?KWSn5$2Z64E<lZ0
zDxz8N7uR=Xp>CSDMl-5T8~xG-!Qz*a21J1^ZEh=K-_pTtr)x0Z$0WG*J`E;oCy<vI
z>wmMkDx1)|9wHo9n|X;LiGF$`0YcpNw^@_?Edp%ma?{RkZJF10rW{6frNtn~j&SlS
zV%|R4qFz~X#fIoOyVpaIwBlKC1JxZa|JW?QZ%mrL=BYt_YuLp+4xaN`5E-jHE0d$)
zmf5t(N*D%0WFbi7-7dbjI<^bY6M!^@f^;I*@X5~uor^T1hhi@Z<AQ!H4L_2bBAQqp
zK1w}{CY8+Hcp_Ua0>QwDW9YBDsy+!&q5u+6ZD7c_Ph`%Z?-?nRnUa4X`ZraJa8>au
zgIH0YGt8acL`AvuSotc*oO7xwFns~i|Is>gXU?`ynmkO?w1JvJpd88*PS&v8P8&l)
z2X6_`21h1ilYs&=e=SWO1Jdkh?TisDK^!=WU~Ivb<oQGdabvb}-vWx~0{u9JBC~xt
z<xmb1JJAbHz6ci@mknTI*$q!>3(!$ztjT1?zIeuQuO^!vUAnqp@@0vh_z=M~TumBa
zkSVnaUXt<}`TfV)kMs2i)Kf)uxRFHHuDe>?Is~`sH2J~+C#%2%q89{QZ*Fc^ZbL`A
z`JNyvmHvA=w*9Z`^Hs6;Ylu+PTpCBbMp=iV(r>)RQE+LEwl(X7MD>ckQzDDXp0p55
z9sfj`$X1EX$ICsw+r@WD+3YW5S&2;XBaSd_%=Ex1uK7l~PI(xD_!=?d`I=!*S?g;C
z6lObWg>Qd_3PbjAM5#y#b`<9#dYFGqniKqc4M32>HQe5Wyy}VhQ}w=u5^EDng7iyh
zzZNAR4Q&!djc>VxILeGg*GdU<S94(aNu=}daB*9Oe;oB)ZR(S?*g*}lm`G>+1|c=_
zE0Vh?`sh2gWmk~_BS?z%pWr0<HoQ;d@kXH@XS#R(ya(&k#;3U2$H4sg{^-@ij-}pj
z#%|1-L-DH63?8oB*g>ad`y3SZ<AbPJ4}&<dRkj6Ac<&t7nvW))^+h#na(;pFh|mx{
zMi-aahX|N00;x2unkP?Ri<HdW7Gl446K*+|3=jfcLDmkhGR<?4manjdu?GC`mAMFF
z=K{k>!KTY7wDwdCfx+~vIvA!UNqvjyhkWW2Fcl=Dg36iUl1S2Tb#j#79=R?hNY}d|
z%(DLiSG>L)qI$H$Qy(r*XBRi0&g-D1JjifEv{-o}6%+`=Q@xl~nTL5(#^QUv3>crO
zh1HGbdMd_Vg&Gs5;V2uCo=&V3@|=@zFjAsAU6wagX}M^T_8=6E3GJNw!XH|)97=B?
zSSdY0l%vNsY9x_pm*YO}>4rz9%rnpWR5D@l>q3`sGAaq06F<|%a4EjwvVySHAa%x3
zA(fc@3Df2%OpcX?11n6p1jU**E7I4o>r$x$dZiE<F5yYtha?>7aoGQ_pU4`X&^<Ic
WZ-|q0dhk#20lJz-xbmy`(EkAyp=nJ3

diff --git a/dashboard/server/ai/functions/AI_Events.ts b/dashboard/server/ai/functions/AI_Events.ts
index d8de4c8..96c15cd 100644
--- a/dashboard/server/ai/functions/AI_Events.ts
+++ b/dashboard/server/ai/functions/AI_Events.ts
@@ -12,8 +12,8 @@ const getEventsCountTool: AIPlugin_TTool<'getEventsCount'> = {
         parameters: {
             type: 'object',
             properties: {
-                from: { type: 'string', description: 'ISO string of start date including hours' },
-                to: { type: 'string', description: 'ISO string of end date including hours' },
+                from: { type: 'string', description: 'ISO string of start date' },
+                to: { type: 'string', description: 'ISO string of end date' },
                 name: { type: 'string', description: 'Name of the events to get' },
                 metadata: { type: 'object', description: 'Metadata of events to get' },
             },
@@ -30,8 +30,8 @@ const getEventsTimelineTool: AIPlugin_TTool<'getEventsTimeline'> = {
         parameters: {
             type: 'object',
             properties: {
-                from: { type: 'string', description: 'ISO string of start date including hours' },
-                to: { type: 'string', description: 'ISO string of end date including hours' },
+                from: { type: 'string', description: 'ISO string of start date' },
+                to: { type: 'string', description: 'ISO string of end date' },
                 name: { type: 'string', description: 'Name of the events to get' },
                 metadata: { type: 'object', description: 'Metadata of events to get' },
             },
diff --git a/dashboard/server/ai/functions/AI_Visits.ts b/dashboard/server/ai/functions/AI_Visits.ts
index 0c228f6..78ab4fb 100644
--- a/dashboard/server/ai/functions/AI_Visits.ts
+++ b/dashboard/server/ai/functions/AI_Visits.ts
@@ -4,6 +4,11 @@ import { Types } from "mongoose";
 import { AIPlugin, AIPlugin_TTool } from "../Plugin";
 import dayjs from 'dayjs';
 
+import { zodFunction } from "openai/helpers/zod";
+import { z } from 'zod';
+import { Slice } from "@services/DateService";
+
+
 const getVisitsCountsTool: AIPlugin_TTool<'getVisitsCount'> = {
     type: 'function',
     function: {
@@ -12,8 +17,8 @@ const getVisitsCountsTool: AIPlugin_TTool<'getVisitsCount'> = {
         parameters: {
             type: 'object',
             properties: {
-                from: { type: 'string', description: 'ISO string of start date including hours' },
-                to: { type: 'string', description: 'ISO string of end date including hours' },
+                from: { type: 'string', description: 'ISO string of start date' },
+                to: { type: 'string', description: 'ISO string of end date' },
                 website: { type: 'string', description: 'The website of the visits' },
                 page: { type: 'string', description: 'The page of the visit' }
             },
@@ -30,10 +35,15 @@ const getVisitsTimelineTool: AIPlugin_TTool<'getVisitsTimeline'> = {
         parameters: {
             type: 'object',
             properties: {
-                from: { type: 'string', description: 'ISO string of start date including hours' },
-                to: { type: 'string', description: 'ISO string of end date including hours' },
+                from: { type: 'string', description: 'ISO string of start date' },
+                to: { type: 'string', description: 'ISO string of end date' },
                 website: { type: 'string', description: 'The website of the visits' },
-                page: { type: 'string', description: 'The page of the visit' }
+                page: { type: 'string', description: 'The page of the visit' },
+                slice: {
+                    type: 'string',
+                    description: 'The slice for the visit data',
+                    enum: ['hour', 'day', 'month', 'year']
+                }
             },
             required: ['from', 'to']
         }
@@ -47,6 +57,7 @@ export class AiVisits extends AIPlugin<['getVisitsCount', 'getVisitsTimeline']>
         super({
             'getVisitsCount': {
                 handler: async (data: { project_id: string, from?: string, to?: string, website?: string, page?: string }) => {
+
                     const query: any = {
                         project_id: data.project_id,
                         created_at: {
@@ -54,31 +65,46 @@ export class AiVisits extends AIPlugin<['getVisitsCount', 'getVisitsTimeline']>
                             $lt: data.to ? new Date(data.to).getTime() : new Date().getTime(),
                         }
                     }
+
                     if (data.website) query.website = data.website;
+
                     if (data.page) query.page = data.page;
+
                     const result = await VisitModel.countDocuments(query);
+
                     return { count: result };
+
                 },
                 tool: getVisitsCountsTool
             },
             'getVisitsTimeline': {
-                handler: async (data: { project_id: string, from: string, to: string, website?: string, page?: string }) => {
+                handler: async (data: { project_id: string, from: string, to: string, time_offset: number, website?: string, page?: string, slice?: string }) => {
 
-                    const query: AdvancedTimelineAggregationOptions & { customMatch: Record<string, any> } = {
-                        projectId: new Types.ObjectId(data.project_id) as any,
+                    const timelineData = await executeTimelineAggregation({
+                        projectId: new Types.ObjectId(data.project_id),
                         model: VisitModel,
-                        from: dayjs(data.from).startOf('day').toISOString(),
-                        to: dayjs(data.to).startOf('day').toISOString(),
-                        slice: 'day',
-                        customMatch: {}
-                    }
+                        from: data.from,
+                        to: data.to,
+                        slice: (data.slice || 'day') as Slice,
+                        timeOffset: data.time_offset
+                    });
+                    return { data: timelineData };
 
-                    if (data.website) query.customMatch.website = data.website;
-                    if (data.page) query.customMatch.page = data.page;
+                    // const query: AdvancedTimelineAggregationOptions & { customMatch: Record<string, any> } = {
+                    //     projectId: new Types.ObjectId(data.project_id) as any,
+                    //     model: VisitModel,
+                    //     from: dayjs(data.from).startOf('day').toISOString(),
+                    //     to: dayjs(data.to).startOf('day').toISOString(),
+                    //     slice: 'day',
+                    //     customMatch: {}
+                    // }
 
-                    const timelineData = await executeAdvancedTimelineAggregation(query);
-                    const timelineFilledMerged = fillAndMergeTimelineAggregationV2(timelineData, 'day', data.from, data.to);
-                    return { data: timelineFilledMerged };
+                    // if (data.website) query.customMatch.website = data.website;
+                    // if (data.page) query.customMatch.page = data.page;
+
+                    // const timelineData = await executeAdvancedTimelineAggregation(query);
+                    // const timelineFilledMerged = fillAndMergeTimelineAggregationV2(timelineData, 'day', data.from, data.to);
+                    // return { data: timelineFilledMerged };
                 },
                 tool: getVisitsTimelineTool
             }
diff --git a/dashboard/server/api/ai/[chat_id]/get_messages.ts b/dashboard/server/api/ai/[chat_id]/get_messages.ts
index 4aa9133..be08fa2 100644
--- a/dashboard/server/api/ai/[chat_id]/get_messages.ts
+++ b/dashboard/server/api/ai/[chat_id]/get_messages.ts
@@ -7,6 +7,8 @@ export default defineEventHandler(async event => {
     const data = await getRequestData(event);
     if (!data) return;
 
+    const isAdmin = data.user.user.roles.includes('ADMIN');
+
     const { project_id } = data;
 
     if (!event.context.params) return;
@@ -16,13 +18,13 @@ export default defineEventHandler(async event => {
     if (!chat) return;
 
     return (chat.messages as OpenAI.Chat.Completions.ChatCompletionMessageParam[])
-        .filter(e => e.role === 'assistant' || e.role === 'user')
+        .filter(e => isAdmin ? true : (e.role === 'assistant' || e.role === 'user'))
         .map(e => {
             const charts = getChartsInMessage(e);
             const content = e.content;
-            return { role: e.role, content, charts }
+            return { ...e, charts }
         })
-        .filter(e=>{
-            return e.charts.length > 0 || e.content
+        .filter(e => {
+            return isAdmin ? true : (e.charts.length > 0 || e.content);
         })
 });
\ No newline at end of file
diff --git a/dashboard/server/api/ai/send_message.post.ts b/dashboard/server/api/ai/send_message.post.ts
index d89dd04..2383011 100644
--- a/dashboard/server/api/ai/send_message.post.ts
+++ b/dashboard/server/api/ai/send_message.post.ts
@@ -9,7 +9,7 @@ export default defineEventHandler(async event => {
 
     const { pid } = data;
 
-    const { text, chat_id } = await readBody(event);
+    const { text, chat_id, timeOffset } = await readBody(event);
     if (!text) return setResponseStatus(event, 400, 'text parameter missing');
 
     const chatsRemaining = await getAiChatRemainings(pid);
@@ -21,7 +21,7 @@ export default defineEventHandler(async event => {
 
     let targetChatId = '';
 
-    await sendMessageOnChat(text, pid, chat_id, {
+    await sendMessageOnChat(text, pid, timeOffset, chat_id, {
         onChatId: async chat_id => {
             if (!responseSent) {
                 event.node.res.setHeader('Content-Type', 'application/json');
diff --git a/dashboard/server/services/AiService.ts b/dashboard/server/services/AiService.ts
index 65fc4a1..7203383 100644
--- a/dashboard/server/services/AiService.ts
+++ b/dashboard/server/services/AiService.ts
@@ -65,7 +65,7 @@ export function getChartsInMessage(message: OpenAI.Chat.Completions.ChatCompleti
     if (message.role != 'assistant') return [];
     if (!message.tool_calls) return [];
     if (message.tool_calls.length == 0) return [];
-    return message.tool_calls.filter(e => e.function.name === 'createComposableChart').map(e => e.function.arguments);
+    return message.tool_calls.filter((e: any) => e.function.name === 'createComposableChart').map((e: any) => e.function.arguments);
 }
 
 
@@ -87,7 +87,7 @@ type ElaborateResponseCallbacks = {
     onChatId?: (chat_id: string) => any
 }
 
-async function elaborateResponse(messages: OpenAI.Chat.Completions.ChatCompletionMessageParam[], pid: string, chat_id: string, callbacks?: ElaborateResponseCallbacks) {
+async function elaborateResponse(messages: OpenAI.Chat.Completions.ChatCompletionMessageParam[], pid: string, time_offset: number, chat_id: string, callbacks?: ElaborateResponseCallbacks) {
 
     const responseStream = await openai.beta.chat.completions.stream({ model: OPENAI_MODEL, messages, n: 1, tools });
 
@@ -124,35 +124,27 @@ async function elaborateResponse(messages: OpenAI.Chat.Completions.ChatCompletio
             const functionCall: FunctionCall = functionCalls.at(-1) as FunctionCall;
             await callbacks?.onFunctionCall?.(functionCall.name);
             const args = JSON.parse(functionCall.argsRaw.join(''));
-            const functionResult = await functions[functionCall.name]({ project_id: pid, ...args });
+
+
+            const functionResult = await functions[functionCall.name]({ project_id: pid, time_offset, ...args });
             functionCall.result = functionResult;
             await callbacks?.onFunctionResult?.(functionCall.name, functionResult);
 
-
-            
-            addMessageToChat({
-                tool_call_id: functionCall.tool_call_id,
-                role: 'tool',
-                content: JSON.stringify(functionResult)
-            }, chat_id);
-
-            
-            addMessageToChat({
+            await addMessageToChat({
                 role: 'assistant',
                 content: delta.content,
                 refusal: delta.refusal,
                 tool_calls: [
                     {
-                        id: functionCall.tool_call_id,
-                        type: 'function',
+                        id: functionCall.tool_call_id, type: 'function',
                         function: {
-                            name: functionCall.name,
-                            arguments: functionCall.argsRaw.join('')
+                            name: functionCall.name, arguments: functionCall.argsRaw.join('')
                         }
                     }
                 ]
             }, chat_id);
 
+            await addMessageToChat({ tool_call_id: functionCall.tool_call_id, role: 'tool', content: JSON.stringify(functionCall.result) }, chat_id);
 
             functionCall.collecting = false;
             lastFinishReason = finishReason;
@@ -166,13 +158,13 @@ async function elaborateResponse(messages: OpenAI.Chat.Completions.ChatCompletio
         return { tool_call_id: e.tool_call_id, role: "tool", content: JSON.stringify(e.result) }
     });
 
-    if (lastFinishReason == 'tool_calls') return await elaborateResponse([...responseStream.messages, ...toolResponseMesages], pid, chat_id, callbacks);
+    if (lastFinishReason == 'tool_calls') return await elaborateResponse([...responseStream.messages, ...toolResponseMesages], pid, time_offset, chat_id, callbacks);
 
     return responseStream;
 }
 
 
-export async function sendMessageOnChat(text: string, pid: string, initial_chat_id?: string, callbacks?: ElaborateResponseCallbacks) {
+export async function sendMessageOnChat(text: string, pid: string, time_offset: number, initial_chat_id?: string, callbacks?: ElaborateResponseCallbacks) {
 
 
     const messages: OpenAI.Chat.Completions.ChatCompletionMessageParam[] = []
@@ -183,6 +175,7 @@ export async function sendMessageOnChat(text: string, pid: string, initial_chat_
 
     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 = {
@@ -200,7 +193,7 @@ export async function sendMessageOnChat(text: string, pid: string, initial_chat_
     await addMessageToChat(userMessage, chat_id);
 
     try {
-        const streamResponse = await elaborateResponse(messages, pid, chat_id, callbacks);
+        const streamResponse = await elaborateResponse(messages, pid, time_offset, chat_id, callbacks);
         const finalContent = await streamResponse.finalContent();
         await addMessageToChat({ role: 'assistant', refusal: null, content: finalContent }, chat_id);
         return { content: finalContent, charts: [] };
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
index d0aaeda..3929c9d 100644
--- a/pnpm-lock.yaml
+++ b/pnpm-lock.yaml
@@ -87,7 +87,7 @@ importers:
         version: 0.0.11(magicast@0.3.5)(rollup@4.27.2)(vue@3.5.13(typescript@5.6.3))
       openai:
         specifier: ^4.61.0
-        version: 4.72.0
+        version: 4.72.0(zod@3.24.1)
       pdfkit:
         specifier: ^0.15.0
         version: 0.15.1
@@ -118,6 +118,9 @@ importers:
       winston:
         specifier: ^3.14.2
         version: 3.17.0
+      zod:
+        specifier: ^3.24.1
+        version: 3.24.1
     devDependencies:
       '@nuxt/ui':
         specifier: ^2.15.2
@@ -4883,6 +4886,9 @@ packages:
     resolution: {integrity: sha512-zK7YHHz4ZXpW89AHXUPbQVGKI7uvkd3hzusTdotCg1UxyaVtg0zFJSTfW/Dq5f7OBBVnq6cZIaC8Ti4hb6dtCA==}
     engines: {node: '>= 14'}
 
+  zod@3.24.1:
+    resolution: {integrity: sha512-muH7gBL9sI1nciMZV67X5fTKKBLtwpZ5VBp1vsOQzj1MhrBZ4wlVCm3gedKZWLp0Oyel8sIGfeiz54Su+OVT+A==}
+
 snapshots:
 
   '@alloc/quick-lru@5.2.0': {}
@@ -8731,7 +8737,7 @@ snapshots:
       is-docker: 2.2.1
       is-wsl: 2.2.0
 
-  openai@4.72.0:
+  openai@4.72.0(zod@3.24.1):
     dependencies:
       '@types/node': 18.19.64
       '@types/node-fetch': 2.6.12
@@ -8740,6 +8746,8 @@ snapshots:
       form-data-encoder: 1.7.2
       formdata-node: 4.4.1
       node-fetch: 2.7.0
+    optionalDependencies:
+      zod: 3.24.1
     transitivePeerDependencies:
       - encoding
 
@@ -10185,3 +10193,5 @@ snapshots:
       archiver-utils: 5.0.2
       compress-commons: 6.0.2
       readable-stream: 4.5.2
+
+  zod@3.24.1: {}