-fi66ddlZddlZddlZddlmZmZmZmZmZddl m Z ddl Z ddl m Z ddl mZmZ ddlmZddlmZmZej.eZd Zd Zd Zd ZGd dZeZdde de!de fdZ"GddZ#e#Z$y#e$r ddlmZY]wxYw)N)ListDictAnyOptionalTuple)datetime)Session)funcdesc)settings)SmartInventoryChatSessionSmartInventoryChatMessagezsmart_inventory:chat:ii:c 6eZdZdZdZedejfdZde de fdZ de de fdZ de de ee effd Zde d ee efdefd Z dde d e de de eedef dZde deee e ffdZde defdZ dde de de defdZde defdZde defdZy )ChatSessionServicez.Service for managing chat sessions using Rediscd|_y)zInitialize Redis connectionN) _redis_clientselfs I/var/www/html/hubwallet-dev/src/smart_inventory/apps/chat_bot/services.py__init__zChatSessionService.__init__ s !returnc|jHtjtjtj tj d|_|jS)z#Lazy initialization of Redis clientT)hostportdbdecode_responses)rredisRedisr REDIS_HOST REDIS_PORTREDIS_DBrs r redis_clientzChatSessionService.redis_client$sM    %!&(((($$!% "D  !!!r session_idct|S)z Generate Redis key for a session)CHAT_SESSION_PREFIX)rr&s r_get_session_keyz#ChatSessionService._get_session_key0s%&zl33r company_idcJttj}||tjj gdd}|j |}|jj|ttj|tjd||S)zE Create a new chat session and return the session ID N)r&r* created_atmessagescontext_summaryzCreated new chat session: )struuiduuid4rnow isoformatr)r%setexCHAT_SESSION_TTLjsondumpsloggerinfo)rr*r& session_datakeys rcreate_sessionz!ChatSessionService.create_session4s& $$",,.224#  ##J/    JJ| $  0 =>rc|j|}|jj|}|tj d|yt j |S)zi Get session data by session ID Returns None if session doesn't exist or expired NzSession not found or expired: )r)r%getr8r9r6loads)rr&r;datas r get_sessionzChatSessionService.get_sessionKsU ##J/  $$S) < KK8 E Fzz$rr:c|j|}|jj|stj d|y|jj |t tj|y)z5 Update session data and refresh TTL z$Cannot update non-existent session: FT) r)r%existsr8warningr4r5r6r7)rr&r:r;s rupdate_sessionz!ChatSessionService.update_sessionYsl##J/  '', NNA*N O    JJ| $ rNrolecontentraw_datac|j|}|y||tjjd}|dk(r|rd|d<t ||d<|dj ||j ||S)zZ Add a message to the session history role: 'user' or 'assistant' F)rFrG timestamp assistantThas_data data_countr-)rArr2r3lenappendrE)rr&rFrGrHr:messages r add_messagezChatSessionService.add_messageks '' 3  !113  ; 8"&GJ $'MGL !Z ''0"":|<rO)rr&r:historymsgs rget_conversation_historyz+ChatSessionService.get_conversation_historys '' 3  I   - . NN <\J[=\<]^   + C NNF y>   rc|j|}|y|jdg}t|tk\ryt d|D}|t kDryy)zF Check if conversation history needs to be summarized Fr-Tc3RK|]}t|jdd!yw)rGN)rNr>).0rVs r z9ChatSessionService.needs_summarization..s J##cggi45Js%')rAr>rNMAX_HISTORY_MESSAGESsumMAX_CONTEXT_CHARS)rr&r:r- total_charss rneeds_summarizationz&ChatSessionService.needs_summarizationsb'' 3  ##J3 x=0 0JJJ * *rsummary keep_last_ncp|j|}|y||d<|d| d|d<|j||S)zK Set the context summary and keep only the last N messages NFr.r-)rArE)rr&rbrcr:s rset_context_summaryz&ChatSessionService.set_context_summarysS '' 3  *1 &'#/ #;[LM#J Z "":|<>r)N))__name__ __module__ __qualname____doc__rpropertyr r!r%r/r)intr<rrrrAboolrErrQrWrarerirlrrrrsJ8" "ekk " "43434.  c  htCH~.F  DcNt&6:=c==s=&tDz2=>B=234S#X;O6cd,/0 =c =C =(+ =48 =?c?d?rrquestion max_lengthrc0|j}gd}|j}|D]1}|j|s|t|dj}n|r|dj |ddz}t||dz kDr |d|dz dz}|r|SdS)zp Generate a chat name from the first question. Truncates to max_length and adds ellipsis if needed. )zwhat iszwhat arez tell me aboutzshow mezcan youNrr z...zNew Chat)striplower startswithrNupper)rvrw chat_nameprefixes_to_remove lower_nameprefixs rgenerate_chat_namers  IX"J$   (!#f+,/557I  aL&&(9QR=8  9~ Q&ozA~.6 !91z1rceZdZdZdededededededed efd Zdeded e efd Z deded ed ed e e f dZ ddededed e eeeffdZ ddedededededed ee eeefeffdZdeded e eeeffdZy)ChatHistoryServicez1Service for managing chat history in the databaserchat_session_idr*user_idstore_id branch_idfirst_questionrct|}t||||||} |j| |j|j | t j d|| S)z< Create a new chat session in the database. )rr*rrrr~zCreated chat session in DB: )rraddcommitrefreshr8r9) rrrr*rrrrr~sessions rr<z!ChatHistoryService.create_sessionsf'~6 ++!   w  7 2?2CDErc|jtjtj|k(j S)z< Get a chat session by its chat_session_id. )queryrfilterrfirst)rrrs rget_session_by_chat_idz)ChatHistoryService.get_session_by_chat_ids6xx1299 % 5 5 H %' rrFrGcF|j||}|tjd|y|j}t |j ||}|j |tj|_ |j|j||S)z< Add a message to an existing chat session. Nz(Cannot add message - session not found: )r&rFrG) rr8rDr{ridrrutcnow updated_atrr)rrrrFrGr message_rolerPs rrQzChatHistoryService.add_message!s--b/B ? NNEoEVW Xzz| +zz  w&__.  7rnc|j||}|gS|jtjtj|j k(j ttjj|j}tt|}|Dcgc]}|j|jdc}Scc}w)z Get the last N messages (n/2 questions and n/2 answers) for LLM context. Returns list of {role, content} dicts. rT)rrrrr&rorder_byr r,limitalllistreversedrFrG)rrrrrr-rVs rget_last_n_messagesz&ChatHistoryService.get_last_n_messagesCs--b/B ?I8856== % 0 0GJJ > ( *55 6 %(335 *+  XX#++ 6   s1 Cpage page_sizecf|jtjtj|k(tj|k(tj |k(}|j }|dz |z} |jttjj| j|j} g} | D]} |jtj tjjtj | jk(j#} | j%| j| j&| j(| j*| j| j| j | j,| j| d | |fS)z Get paginated list of chat sessions for a user/store/branch. Returns (sessions_with_count, total_count). r ) rrr~r*rrrr,r message_count)rrrrrrcountrr roffsetrrr rrr&scalarrOrr~r*r,)rrrrrrrrtotalrsessionsrhrrs rget_chat_history_listz(ChatHistoryService.get_chat_history_list`sg23:: % - - 8 % . .( : % / /9 <  (i'>> *55 6 &.y)##%   GHHTZZ0I0L0L%MNUU)44 Bfh  MMjj#*#:#:$..%00"??#,,$..%00%00!.    $u}rcN|j||}|y|jtjtj|j k(j tjj}|j |j|j|j|j|j|j|j|j|Dcgc]1}|j |j |j"|jd3c}d Scc}w)z> Get detailed chat session with all messages. N)rrFrGr,) rrr~r*rrrr,rr-)rrrrr&rrr,rrr~r*rrrrrFrG)rrrrr-rVs rget_chat_session_detailz*ChatHistoryService.get_chat_session_details--b/B ?8856== % 0 0GJJ > (,77 8  **&66 **!,,(( **!,,!,,$ &&HH"{{"%..   s'6D"N) )r r)rnrorprqr r/rsrr<rrrrQrrrrrrrrurrrrs;       #>     + ,         + , L       d38n   F1 11 1  1  11 tDcN#S( )1f$ $ $  $sCx. ! $ rr)d)%r6r0loggingtypingrrrrrrr sqlalchemy.ormr sqlalchemyr r src.utils.settingsr ImportErrorutils.settingsmodelsrr getLoggerrnr8r(r5r]r_rchat_session_servicer/rsrrchat_history_servicerurrrs 33 "!(+I   8 $.r?r?l*+ 22#226D D P*+_ ('(sB BB