tj08UdZddlmZddlZddlmZmZddlmZddl m Z e dZ edZ d e d <iZd e d <dCdZdDdZejdZejdZejdZejdZejdZejdZejdZeGddZdEd ZdFd!ZdGd#ZdDd$ZdHd&ZdDd'ZdId)Z dFd*Z!dFd+Z"dId,Z#dJd.Z$dKd1Z%dLd2Z&dMd5Z'd6d7dNd;Z(dOd?Z)dPd@Z*dQdBZ+dS)RaConversational intent + history-aware question resolution for SaaS chat. Distinguishes new document questions from follow-ups about prior answers and from verify / re-check challenges. Keeps a short in-process memory so follow-ups work even when audit persistence is off. ) annotationsN) defaultdictdeque) dataclass)Literal) new_question follow_upverify explain_calcc"tdS)N)maxlen)r1/var/www/html/CA-Chatbot/app/saas/conversation.pyrsUVrz!dict[str, deque[tuple[str, str]]]_SESSION_TURNSzdict[str, dict[str, str]]_SESSION_PROFILE session_id str | NonenamestrreturnNonecr|r|sdS|t|id<dSN display_name)stripr setdefaultrrs rset_session_display_namer!s? TBF**,, B//???rc|sdSt|pid}|r|ndSr)rgetrr s rget_session_display_namer$!sL t   , , 2 7 7 G GD )4::<<(?i)\b(how\s+did\s+you\s+(?:get|calculate|compute|count|arrive|derive)|how\s+(?:is|was)\s+that\s+(?:calculated|computed|counted)|where\s+does\s+(?:that|this|the)\s+(?:number|total|figure|amount)\s+come\s+from|show\s+(?:your|the)\s+(?:work|math|calculation)|explain\s+(?:your|the)\s+(?:calculation|math|total|figure))\ba(?i)\b(\bit\b|\bthat\b|\bthis\b|\bthose\b|\bthese\b|\bthem\b|previous\s+answer|your\s+answer|you\s+said|you\s+mentioned|as\s+you\s+(?:said|mentioned)|the\s+(?:above|earlier|last)\b|why\s+(?:does|did|is|are|only|not)|how\s+come|what\s+about|and\s+(?:august|july|june|may|april|march|february|january|september|october|november|december)\b|only\s+(?:contain|contains|include|includes|cover|covers|has|have|\d+)|not\s+(?:august|july|june|may|april|march|february|january|september|october|november|december)|in\s+a\s+paragraph|as\s+prose|in\s+words|paragraph\s+form|where\s+(?:are|is)\s+the\s+rest|missing|incomplete|why\s+(?:only|just)\s+\d+|rest\s+of\s+(?:the\s+)?(?:payments?|transactions?|list)|same\s+(?:as|for)|compared\s+to|vs\.?|versus)\ba^(?i)\b(why\s+(?:only|just)\s+\d+|where\s+(?:are|is)\s+the\s+rest|(?:list|extraction|index(?:ing)?)\s+is\s+incomplete|incomplete\s+(?:list|extraction|index)|missing\s+(?:payments?|transactions?|lines?)|not\s+(?:showing|listing)\s+(?:all|every)|only\s+\d+\s+(?:of\s+\d+\s+)?(?:payments?|transactions?)|should\s+be\s+\d+|header\s+(?:says|shows)\s+\d+)\bz3(?i)^(why|how|what\s+about|and|also|ok\s+but|but)\ba((?i)^(hi|hii|hello|hey|yo|sup|wass?u+p+|whass?u+p+|(?:what'?s|whats)\s+(?:up|good)|thanks?|thank\s+you|bye|goodbye|how\s+are\s+(?:you|u)(?:\s+doing)?|how(?:'s|\s+is)\s+it\s+going|how\s+have\s+you\s+been|how\s+you\s+doing|help(?:\s+me)?|how\s+can\s+you\s+help(?:\s+me)?|what\s+can\s+you\s+(?:do|help(?:\s+me)?(?:\s+with)?)|who\s+are\s+you|what\s+are\s+you|tell\s+me\s+about\s+(?:yourself|you|the\s+bot|this\s+bot)|my\s+name\s+is\b.|i\s*(?:'m|am)\s+[A-Za-z].|what(?:'s|\s+is)\s+my\s+name|what\s+was\s+my\s+first\s+question|complete|done|ok|okay)[?.!\s]*$z(?i)\b(spent|spend|total|payment|paid|received|income|merchant|categor|grocer|july|august|date|month|period|transaction|outflow|inflow)\bcXeZdZUded<ded<ded<dZded<dZded <d Zded <dS) ResolvedTurnroriginal rewritten ConvIntent conv_intentNrprior_question prior_answer history_block)__name__ __module__ __qualname____annotations__r+r,r.rrrr&r&scMMMNNN!%N%%%%#L####Mrr&questionanswerc|r|sdSt|||pdfdSNr-)rappendr)rr3r4s r remember_turnr8sT X:%%x~~'7'7&,B9M9M9O9O&PQQQQQrcz|r8t|dt|ddSdSN)rpoprrs r clear_turnsr=sF/:t,,,Z.....//rlist[tuple[str, str]]c\|sgStt|pgSr:)listrr#r<s r get_turnsrAs0  "":..4" 5 55rctt|}|sdS|ddpd}|pdS)Nrr-)rAr)rturnsqs rget_first_user_questionrEsD j ! !E t q! !!##A 9r evidence_jsonc|r|sdS|ddt|id<dt|id<dS)zXStore last financial-analysis evidence for short follow-ups (reduce it / that merchant).Ni@ last_analysis1prior_was_analysisrr)rrFs rremember_analysis_contextrLs_ ]CPQVRVQVCW B//@HK B//0DEEErch|sdSt|pidS)NrHrr#r<s rget_analysis_contextrOs6 t   , , 2 7 7 H HHrboolcp|sdSt|piddkS)NFrJrIrNr<s rrJrJs= u   , , 2 7 78L M MQT TTrcv|sdSt|}|r|dddSdS)NrJ)rr#r;)rprofs rclear_analysis_flagrTsO     + +D - %t,,,,,--rcJ|sdSdt|id<dS)zARemember that Alex already introduced themselves in this session.NrI introducedrKr<s rmark_session_introducedrWs0 @C B// ===rcp|sdSt|piddkS)NFrVrIrNr<s rsession_was_introducedrYs; u   , , 2 7 7 E E LLrmetricsct|r|sdSdd|ddD}t|pidd}|r|dz|zn|}|ddd}d|t|id<dS) zITrack metric cards already shown so later turns can avoid repeating them.N|c3*K|]\}}|d|VdS)=Nr).0abs r z)remember_shown_metrics..s077$!Qa::!::777777r shown_metricsr-z||)joinrr#splitr)rrZkeyprevmergedpartss rremember_shown_metricsrls W ((77'#2#,777 7 7C   , , 2 7 7 L LD$( 1dTkCcF LL  rss #ECG99UCSCS B//@@@rlabelvaluec~|sdSt|pidpd}|d||vS)NFrdr-r^rN)rrmrnblobs rrecently_shown_metricrqsS u   , , 2 7 7 H H NBD  e   %%rcTtt|pdSr6)rP_COVERAGE_CHALLENGE_REsearch)r3s ris_coverage_challengerus# &--hn"== > >>r has_historyr)c|pd}|sdSt|rdSt|rdSt |rdSt |r|rdndS|sdSddlm}||rdS| }t|rwt|dkst|rdStjd |rdStjd |rt|rdSt|d kr6t|rt|rdSdS) Nr-rr r r r)is_analytical_self_containedz'(?i)\b(it|that|this|those|these|them)\bz (?i)\bwhy\br )r_HELP_OR_SMALLTALK_REmatch _VERIFY_RErt_EXPLAIN_CALC_REruapp.saas.finance_intentrxrg _FOLLOW_UP_RElen_SHORT_FOLLOW_REre_ANALYTICAL_HINT_RE)r3rvrDrxwordss rclassify_conversational_intentrs R  A ~""1%%~xq!!~Q>!,=~~+= ~EDDDDD##A&&~ GGIIEA u::  /66q99 ; 9? C C ; 9^Q ' ' ,?,F,Fq,I,I ; 5zzQ+22155:M:T:TUV:W:W{ >r)limitrCrintcp|sdSdg}|| dD]\}}|d|tjdd|}t |dkr|dddz}|d |d |S) Nr-z?Recent conversation (use this to answer follow-ups coherently):zUser: \s+ ii…z Assistant:  )r7rsubrrrstriprf)rCrlinesrDr`snippets r_history_blockrs r N OEufgg..1 \a\\"""&a((..00 w<<#  dsdm**,,u4G ,7,,---- 99U  rprior_qprior_aintentc:|}|pd}|pd}|dkr |pd}d|d|dS|dkst|r |pd}d |d |S|g}|r|d |d |rstjd d|}t |dkr|dddz}|d|d d|S)Nr-r zthe previous figuresz,Re-check the underlying statement data for: u.. The user challenged the previous answer (“uF”). Recompute from the full document; do not defend the prior reply.r zthe previous totaluExplain how the figures for “u” were derived from the full statement (header totals vs indexed transactions). Address incomplete extraction if indexed lines are fewer than the header count. User asked: z%(Referring to the previous question: )rrihierz(Previous answer said: )rrur7rrrrrf)r3rrrrDfocusrkanchors r_rewrite_follow_uprs A}"##%%G}"##%%G 11 O5 O O;< O O O #8#;#;// Ne N NJK N N CEI GWGGGHHH:W--3355 v;;  DSD\((**U2F 8v888999 88E??rc $|pd}t|}dx}}|r |d\}}t|t|}|dkr|}nt ||||}t |||||t |S)zFClassify conversational intent and rewrite the question using history.r-N)rvr)r'r(r*r+r,r.)rrArrPrr&r)rr3r'rCrrconvr(s r resolve_turnrFsB%%''H j ! !EGg % 9 )(U L L LD ~ &x'4HH $U++    rfinance_intentc |dvS)N> incomeinsights comparison date_lookup payment_countperiod_coveragefinancial_advicespending_summarycategory_spendingmerchant_analysisstatement_summarytransaction_searchr)rs ris_analytical_finance_intentr^s    r)rrrrrr)rrrr)rrr3rr4rrr)rrrr)rrrr>)rrrFrrr)rrrrP)rrrZr>rr)rrrmrrnrrrP)r3rrrP)r3rrvrPrr))rCr>rrrr) r3rrrrrrr)rr)rrr3rrr&)rrrrP),__doc__ __future__rr collectionsrr dataclassesrtypingrr)rr2rr!r$compiler|r}rrsrrzrr&r8r=rArErLrOrJrTrWrYrlrqrurrrrrrrrrsj#""""" ********!!!!!!  5@K@W@W4X4XXXXX.00000OOOO ****RZ    2:    $$    2:: # .!bj   RRRR //// 6666 LLLLIIII UUUU ----DDDDMMMM TTTT&&&&????++++\BC      $$$$N0r