F()j5ddlmZddlmZddlZddlZddlmZejeZ ejejd<ejejejZd deded efd Zd!ded ed eded eded efdZdefdZd"dededefdZd"defdZd#defdZdedefdZd#dedefdZdefdZdefdZd#ded efdZd#dededed efdZd#deded efdZd#dededed efdZ y)$) firestore)settingsN)datetimeGOOGLE_APPLICATION_CREDENTIALS)projectdatabase tenant_id thread_iduser_idc |d|}tjdj|}|j}|jrf|j }|rB|jdr1|jd|k7rt jd|d|gS|jdgSgS#t$r&}t jd|d gcYd }~Sd }~wwxYw) zN Fetches chat history for a given tenant, thread, and user isolation. _ chat_historyr z'Unauthorized access attempt for thread z by user messagesz,Error fetching chat history from Firestore: Texc_infoN) db collectiondocumentgetexiststo_dictloggerwarning Exceptionerror)r r r doc_iddoc_refdocdataes =/var/www/html/strategist-ai/app/services/firestore_service.pyget_chat_historyr"s;a {+--/88@kkm ::;;=D488I.488I3F'3Q!H S\]d\efg 88J+ +  CA3GRV W s*BB1B1/B11 C :CC C rolemessage user_namesummaryc  |d|}tjdj|}||tjj d} tjj } t j| g| ||d} |jjs| | d<|r|| d<|r|| d <|r|| d <|j| d tjd |d|d|d|d |dk(r t|yy#t$r"} tjd| Yd} ~ d} ~ wwxYw#t$r$} tj!d| d Yd} ~ yd} ~ wwxYw)zM Saves a new chat message to Firestore with optional user isolation. r r)r#content timestamp)r updated_atr r created_atz4Could not check existing chat document before save: Nr r%r&TmergezSaved z message to Firestore for z (User: z , Thread: ) assistantz(Error saving chat message to Firestore: r)rrrrutcnow isoformatr ArrayUnionrrrrrsetinfoincrement_ai_response_countr)r r#r$r r%r&r rrnew_msgnow update_dataread_errr s r!save_chat_messager:%s'T;a {+--/88@!*446  oo))+!,,gY7""   ^;;='',/ L) %,K " '0K $ %,K " Kt , fTF"=<=rrAz#Error aggregating AI responses for rCN) rrrwherestreamrrrrr) r rIrJrGquerydocstotalrr s r!get_aggregated_ai_responsesrRjsMM.1::9E $$]3 KKj9E KKh7E||~ :C S[[]&&~q9 9E :  :9+RsKL sBB C (CC c |r|jnd}|r|jnd}tjdjdd|}|j }d}|D]l}|j xsi} | j dgD]A} | j ddk7r| j d } |r| r| |kr3|r| r| |kDr=|d z }Cn|S#t$r#} tjd |d | d d} ~ wwxYw)z Counts assistant messages directly from chat_history. This is the authoritative source for enforcing AI response limits. Nrr ==rrr#r/r)r?z(Error counting AI response messages for rCTr r1rrrMrNrrrrr) r rIrJ start_isoend_iso threads_refrPcountrrmsgr)r s r!get_ai_response_message_countr[s .8J((*d *2($$&mmN399+tYW !!# C;;=&BDxx B/ 776?k1GGK0 i9y3HyY-@     ? {"QCP[_ ` sCC DC<<Dc |r|jnd}tjdjdd|}|j }d}|D]I}|j xsi}|j dxs|j d}|r|r||krE|dz }K|S#t$r#} tjd |d | d d} ~ wwxYw) zl Counts tenant conversation threads from Firestore, the same store used to detect new chat threads. Nrr rTrr+r*r?z(Error counting conversation threads for rCTrrU) r rIrVrXrPrYrrr+r s r!get_conversation_thread_countr]s.8J((*d mmN399+tYW !!# C;;=&BD,/I488L3IJ* Y0F QJE     ? {"QCP[_ ` sBB C 'CC  size_bytesc  tj}|jd}tj dj |}|j dj |}|j |||jddtjd|d|d |d y #t$r%}tjd |d |Yd }~y d }~wwxYw)zV Logs the current storage size into the daily bucket for historical tracking. r<r=r@) storage_bytesrBr*Tr,zLogged storage snapshot for z (z): z bytesz#Error logging storage snapshot for rCN) rr0rDrrrr3r1rr4rr)r r^r7rFrGrHr s r!log_storage_snapshotrasMoo<< +MM.1::9E ((7@@J  '--/     29+RzZLX^_` M :9+RsKLLMsB-B00 C9CCc tjdj|}|jdjdtj j }|r|jdd|}|jdj}|D]#}|jjddcSy#t$r%}tjd |d |Yd }~yd }~wwxYw) zT Retrieves the latest available storage snapshot before or on the end_date. r=r@rB) directionrLr?r`rz*Error getting latest storage snapshot for rCN)rrrorder_byrQuery DESCENDINGrMlimitrNrrrrr)r rJrGrOrPrr s r!get_latest_storage_snapshotrhs MM.1::9E $$]3<rz*Error getting total AI response count for rCN) rrrrrrsync_ai_responses_from_historyrrr)r rGrrYr s r!get_total_ai_responses_countrks MM.1::9E mmo ::KKM%%&:A>EL2)< <  A)BqcRSsA!A/$ A// B8BBc. tjd|dtjdj dd|}|j }d}|D]7}|j jdg}|td|Dz }9tjd j|}|jd |id tjd |d|d|S#t$r%}tjd|d|Yd}~yd}~wwxYw)z Scans all historical threads to calculate and initialize the global counter. This handles "past ones" by calculating the sum once and saving it. z8Initializing/Syncing AI response count from history for ...rr rTrrc3JK|]}|jddk(sdyw)r#r/r?N)r).0ms r! z1sync_ai_responses_from_history..sSQaeeFm{6RqSs##r=r>Tr,zSync complete for rCz responses found.zError syncing AI responses for N) rr4rrrMrNrrsumrr3rr)r rXrP total_countrrrGr s r!rjrjs   NykY\]^mmN399+tYW !!#  TC{{}((R8H 3S(SS SK T MM.1::9E  +[9 F ( 2k]BSTU  6ykA3GHsC#C&& D/DDc$ tjdjdd|}|r|jdd|}|j}g}|D]}|j }|j dxs|j d}|s:|j dr|j dgdnd }|j ||j d|j d d |j d d |j dd|j dd|j d|j d|r |dd ddznd d |jdd|S#t$r&} tjd| dgcYd } ~ Sd } ~ wwxYw)z[ Retrieves all chat threads for a specific tenant, optionally filtered by user_id. rr rTr r session_idrNr% Anonymousr&takeover_activeFintervention_requestedr+r*r(drm) r r r%r&ryrzr+r*last_message_previewc|dxsdS)Nr*rx)xs r!z!get_all_threads..#s1\?#8bT)keyreversez&Error listing threads from Firestore: r) rrrMrNrrappendsortrrr) r r rXrPthreadsrr thread_id_vallast_msgr s r!get_all_threadsrs} mmN399+tYW %++ItWEK!!# C;;=D HH[1KTXXl5KM 7;xx 7Ktxx B/3QUH NN*88I.!XXk;?88Ir2#'88,=u#E*.((3KU*S"hh|4"hh|4MU(;DS(AE(I[]   (  8$ G  =aSAD Q sEE F)F F Fstatusct |d|}tjdj|}|tjj d}|durd|d<|j |dtjd|d |y#t$r$}tjd |d Yd }~yd }~wwxYw) zB Toggles the human takeover status for a specific thread. r r)ryr*TFrzr,zUpdated takeover_active to  for z,Error setting takeover status in Firestore: rN rrrrr0r1r3rr4rrr r rr rrr8r s r!set_takeover_statusr)s;a {+--/88@ &"//+557 T>49K0 1 Kt , 1&vhGH  CA3GRV WBB B7B22B7c4 |d|}tjdj|}|j}|jr |j jddSy#t $r$}tjd|dYd}~yd}~wwxYw) z= Retrieves the current takeover status for a thread. r rryFz.Error getting takeover status from Firestore: TrN) rrrrrrrrr)r r r rrrr s r!get_takeover_statusr@s ;a {+--/88@kkm ::;;=$$%6> >  EaSITX YsA&A** B3BBct |d|}tjdj|}|||tjj d}|r||d<|j |dtjd|d|y#t$r$}tjd |d Yd }~y d }~wwxYw) zJ Toggles the intervention_requested status for a specific thread. r r)r r rzr*r Tr,z"Updated intervention_requested to rz0Error setting intervention status in Firestore: rNFrrs r!set_intervention_statusrOs;a {+--/88@#"&,"//+557  %,K " Kt , 8fXNO  GsKVZ [r)defaultN)rNNN)NN)N)! google.cloudrapp.core.configrloggingosr getLogger__name__rrenvironClientFIRESTORE_PROJECTFIRESTORE_DATABASErstrr"r:r5rRr[r]intrarhrkrjrboolrrrr~rr!rs"$    8 $08/V/V +,Y  & &  ( ( #(+T+TC+T#+T#+Tfi+T{~+TQT+TZS3S03CRU,S8S*MCMSM*3#&C$c0$s$S$L33s. 3 3  ssDSVr