tj}<dZddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZdd lmZmZmZmZmZmZmZmZmZm Z m!Z!dd l"m#Z#ed Z$ej%d Z&ej%d Z'ej%dZ(ej%dZ)ej%dZ*ej%dZ+ej%dZ,ej%dZ-ej%dZ.ej%dZ/ej%dZ0ej%dZ1ej%dZ2edGddZ3d2d!Z4d3d#Z5d3d$Z6d3d%Z7d3d&Z8d3d'Z9d3d(Z:d3d)Z;d3d*Zd5d1Z?dS)6uPrimary intent classification — decide the channel BEFORE any retrieval. Five intents (exactly one per turn): 1. document_lookup — answer depends on uploaded PDF/statement contents → structured KB / Chroma 2. document_metadata — filenames, page counts, doc count, upload/index status → inventory only 3. general_knowledge — concepts, geography, science, finance theory → base LLM (no RAG) 4. calculation — pure math or hypothetical numeric scenarios → calculator / LLM math 5. conversational — greetings, smalltalk, capability, personal memory → templates Only (1) and (2) touch the knowledge base. (2) never summarizes document contents. ) annotationsN) dataclass)Literal)is_pdf_overview_question) is_pure_math) _BYE_WORDS_GREETING_WORDS _HELP_WORDS_HOW_ARE_YOU_RE _OFFTOPIC_RE _THANKS_WORDS_detect_intent_likely_greeting_normalize_smalltalk) classify_finance_intentis_analyse_overview_askis_analytical_self_containedis_biggest_expenses_questionis_payment_list_questionis_period_coverage_questionis_profit_loss_questionis_statement_field_questionis_statement_identity_questionis_statement_period_question!is_unrelated_party_spend_question)is_kb_meta_question)document_lookupdocument_metadatageneral_knowledge calculationconversationalu(?i)^(what|huh|eh|and|so|ok|okay|um+|hmm*|complete|done|yes|yep|yeah|sure|cool|nice|great|alright|right|got\s+it|fine|k|kk|tell\s+me|[?!.…]+|what\?+|huh\?+)\s*$z(?i)^\s*(?:hi[,!]?\s+|hello[,!]?\s+)?(?:my\s+name\s+is|i(?:'m|\s+am|\s*m)\s+(?:called\s+)?)\s*([A-Za-z][A-Za-z'\-]{1,30})(?:\s+[A-Za-z][A-Za-z'\-]{1,30})?\s*[!.?]*\s*$a(?i)^\s*(?:what(?:'s|\s+is)\s+my\s+name|what\s+do\s+you\s+think\s+(?:is\s+)?my\s+name|what\s+do\s+you\s+think\s+my\s+name\s+is|(?:can\s+you\s+)?(?:guess|infer)\s+my\s+name|who\s+am\s+i|do\s+you\s+know\s+my\s+name|remind\s+me\s+(?:of\s+)?my\s+name)\s*[?.!]?\s*$z(?i)^\s*(?:what\s+was\s+my\s+first\s+question|what\s+did\s+i\s+(?:ask|say)\s+first|remind\s+me\s+(?:of\s+)?my\s+first\s+question)\s*[?.!]?\s*$z(?i)^\s*what(?:'s|\s+is)\s+(?:the\s+)?date\s+(?:today|todat|todqy|now|currently)\s*[?.!]?\s*$|^\s*what(?:'s|\s+is)\s+today'?s\s+date\s*[?.!]?\s*$|^\s*(?:today'?s\s+date|date\s+today)\s*[?.!]*\s*$ah(?i)\b(knowledge\s*base|how\s+many\s+(?:documents?|files?|pdfs?|pages?|chunks?)|(?:page|pages)\s+count|number\s+of\s+(?:pages?|documents?|files?)|what(?:'s|\s+is)\s+the\s+filename|(?:file|document|pdf)\s+name|which\s+(?:files?|documents?|pdfs?)\s+(?:are|do|did)|what\s+(?:documents?|files?|pdfs?)\s+(?:do\s+you\s+have|are\s+(?:uploaded|ready|indexed|loaded))|upload\s+status|indexing\s+status|is\s+(?:it|the\s+(?:file|pdf|document))\s+(?:ready|uploaded|indexed)|my\s+(?:documents?|files?|uploads?|pdfs?)\b|uploaded\s+(?:documents?|files?|pdfs?)|which\s+(?:statement|pdf|document|file|upload)\b|what\s+(?:statement|pdf|document|file)\s+(?:is\s+)?(?:this|that|loaded|ready|uploaded)|what\s+(?:statement|pdf|document)\s+do\s+(?:you|i)\s+have|what\s+file\s+is\s+(?:this|loaded)|tell\s+me\s+about\s+(?:the\s+)?knowledge\s*base|list\s+(?:my\s+)?(?:documents?|files?|uploads?))\ba*(?i)\b(my\s+(?:spend|spending|statement|payment|payments|account|balance|transaction|transactions|invoice|outflow|inflow|merchant)|i\s+(?:spent|spend|paid|pay|received|receive|owe)|how\s+much\s+(?:did\s+i|have\s+i|was\s+spent|spent)|how\s+many\s+(?:payments?|transactions?|times|credits?|deposits?|debits?)|this\s+(?:statement|pdf|document|file)|passbook|upi\s+statement|spent\s+on|paid\s+to|sent\s+to|money\s+(?:in|out)|branch\s*(?:code|number|no\.?)?|account\s*(?:number|no\.?|branch)|ifsc|micr|cust(?:omer)?\s*id|zepto|blinkit|swiggy|zomato|myntra|nykaa|instamart|meesho|grocer(?:y|ies)|merchant|category\s+spend|biggest\s+expenses?|largest\s+outflows?|\d{1,2}(?:st|nd|rd|th)?\s+(?:jan|feb|mar|apr|may|jun|jul|aug|sep|oct|nov|dec)[a-z]*|(?:jan|feb|mar|apr|may|jun|jul|aug|sep|oct|nov|dec)[a-z]*\s+\d{1,2})\bz(?i)\b(what\s+if|suppose\s+(?:i|we)|hypothetic|if\s+i\s+(?:cut|reduce|increase|spent|spend|saved|save)|(?:would|could)\s+(?:my|the)\s+(?:net|cashflow|balance|spend)|\d+\s*%\s+(?:less|more|of)|percent\s+(?:less|more))\ba(?i)\b(capital\s+of|who\s+invented|who\s+discovered|photosynthesis|gravity|solar\s+system|planet\b|continent|ocean|mountain|president\s+of|population\s+of|distance\s+between|fun\s+fact|tell\s+me\s+a\s+joke|weather|football|cricket|movie|song|joke|recipe|horoscope|ebitda|gaap|ifrs|gst|vat|depreciation|amorti[sz]ation|accrual|working\s+capital|current\s+ratio|debt[\s-]?to[\s-]?equity|roi|roe|npv|irr)\ba(?i)^\s*(what\s+is|what\s+are|what'?s|define|definition\s+of|explain|meaning\s+of|how\s+to|how\s+does|how\s+do(?:\s+i|\s+you)?|how\s+can\s+(?:i|you|one|we)\b|difference\s+between|why\s+is|why\s+are|who\s+(?:is|was|invented|discovered|wrote|painted)|where\s+is|when\s+was|when\s+did|tell\s+me\s+about(?!\s+(?:my|the\s+statement|the\s+pdf|the\s+document|this\s+pdf|the\s+knowledge)))\bz(?i)^\s*how\s+(?:to|do\s+i|can\s+i|does\s+one)\s+(?!much\b|many\b|often\b|(?:i\s+)?(?:spend|save|pay|reduce|cut|budget)|(?:did|do)\s+i\s)z(?i)\b(indexing|embeddings?|vector\s*(?:search|store|database)?|semantic\s+search|chunking|retrieval|rag\b|knowledge\s+base\s+(?:index|search|work))\bzH(?i)\b(stock|market|fund|portfolio|s\s*&\s*p|nasdaq|benchmark\s+index)\bT)frozenc8eZdZUded<ded<ded<ded<dS) PrimaryRoute PrimaryIntentintentstrreasonfinance_intent chat_intentN)__name__ __module__ __qualname____annotations__2/var/www/html/CA-Chatbot/app/saas/intent_router.pyr$r$s?KKKr0r$questionr'return str | Nonec t|pd}|sdS|d}|dvrdS|S)N>okyoualexfinegoodhereokay)_SELF_INTRO_REmatchstripgrouplowertitle)r2mnames r1 is_self_introrGsthn"335566A t 771::    D zz||LLLt ::<<r0boolc|pd}|rt|sdSt|sdSt| S)uCDocument/search indexing concepts — not financial market indexes.r6F)rA_CONCEPT_LEAD_REsearch_KB_TECH_CONCEPT_RE_FINANCE_INDEX_REr2qs r1is_kb_tech_concept_questionrPsl R  A $++A..u  % %a ( (u ''** **r0cxtt|pdSNr6)rH_ASK_MY_NAME_REr@rAr2s r1is_ask_my_namerUs. %%x~2&<&<&>&>?? @ @@r0cxtt|pdSrR)rH_ASK_FIRST_QUESTION_REr@rArTs r1is_ask_first_questionrXs/ &,,hn"-C-C-E-EFF G GGr0cxtt|pdSrR)rH_TODAY_DATE_REr@rArTs r1is_today_date_questionr[s. $$hn"%;%;%=%=>> ? ??r0cv|pd}t|st|st|rdStjd|rdSt |st |rEtjd|rdStjd|rtjd|sdSdSdS)u=Inventory / status of indexed docs — not a content summary.r6Fzr(?i)\b(?:my\s+name|account\s+holder|who\s+am\s+i|whose\s+statement|how\s+much|total\s+spent|biggest\s+expenses?)\bzQ(?i)\b(analy[sz]e|summar(?:y|ize)|overview)\b.*\b(statement|pdf|document|spend)\bz%(?i)\b(spent|spend|paid|how\s+much)\bzj(?i)\b(how\s+many\s+(?:pages?|documents?|files?)|page\s+count|filename|upload\s+status|knowledge\s*base)\bT)rrrrerKr_DOC_METADATA_RErNs r1is_document_metadata_questionr_sBA%a((,H,K,KOjklOmOmu y ;  u1 !1!8!8!;!;  9 `    5 9=q A A ") y K K  5t 5r0c|pd}t|rdSt|rt|sdSt|rdSdS)Nr6TF)rAr_HYPOTHETICAL_RErK_DOC_LOOKUP_RErNs r1is_hypothetical_or_calcrcsz R  AAtq!!.*?*?*B*Btq!!t 5r0c|pd}|sdSt|st|rdSt |st |rdSt |st|rdSt|st|rdSt|st|st|rdStjd|rdSt|}|dkrdSt |rdSt"|st%j|rdSt&|rdSdS)Nr6Fz(?i)\b(my\s+(?:statement|spend|spending|payments?)|how\s+much\s+(?:did|do)\s+i|total\s+spent|uploaded\s+(?:pdf|statement)|(?:the\s+)?statement\s+(?:period|year|date|balance|summary|holder)|account\s+holder|\bifsc\b|branch\s+code)\bT)rAr_rbrKrrrrrrrrrr]lensplit_HOWTO_GENERAL_RE_GENERAL_KNOWLEDGE_REr rJ)r2rOwordss r1is_general_knowledge_questionrk s R  A u$Q''>+@+@+C+Cu"1%%)G)J)Ju#A&&*Ea*H*Huq!!%=a%@%@uq!!%A!%D%DH`abHcHcu y     u  NNE rzzu""t##A&&,*=a*@*@tq!!t 5r0c8|pd}t|}|sdSt|s-t|st |st |rdSddlm}m}||s ||rdSt |st |rdS|tvs|tvs|tvs |tvrdStj |st!|rdSt#|}|dvS)Nr6Tr)is_capability_questionis_identity_question>byehelpthanksgreeting)rArrGrUrXr[app.saas.personalityrmrn_VAGUE_PING_REr@r r rr r rr)r2rOnormrmrnchats r1is_conversational_questionrw1sZ R  A  " "D tQ>!,,0Ea0H0HLbcdLeLetQQQQQQQQA"8"8";";tD!!^%9%9!%<%<t $-"7"74:;M;MQUYdQdQdtT""&6t&<&<t !  D 8 88r0)r)r)c|pd}|pt|}t|}t|}t |st |rt dd|dSt|rt dd||St|rt ddd d St|rt dd d d St|rt ddd dSt|st|st|st|szt|skt!|s\t#|sMt%|s>t'|s/t)|s t*|s|dvr|rt dd|d kr|nd d S|d kr!t|st dd|d St.|st.|rt ddd dSt3jd|s:|dr7t7|dkrt ddd dSt.|s|dkrt ddd dSt dd||S)z+Single pre-retrieval decision for the turn.r6r umath / hypothetical — no RAGmathr!z!greeting / smalltalk / capabilityrz(KB inventory / filename / pages / status document_qadatazunrelated third-party spendgeneralofftopicrubase LLM — no document RAGconcept>r|rzrz-answer depends on uploaded statement contentszdocument content questionz vague pingrpz7(?i)^\s*(what|who|where|when|why|how|which|tell\s+me)\b?r7zopen general questionzclarify / helpzdefault conversational)rArrrrrcr$rwr_rrkrrrrrrrrrrrbrKrtr@r]endswithrfrg)r2r)rOfinrvrus r1classify_primary_intentrFsW R  A  63A66C !  D  " "DAZ1!44ZM+KSRXYYY"!$$^,.QSVX\]]]%Q''   6      )++   )      %Q''g/1OQZ\efff %Q'' &q ) )  *! , ,  #1 % %  $A & &  #1 % %   ( * *  $A & &  'q ) )  ( * *    # #  1 1 1c 1  ;)##CC      m$A!$D$D-/JCQWXXXD!!O^%9%9!%<%<O,lIvNNN yKQOO` 3` NNQ../1H)U^___D!!STV^^,.> 6RRR (*BC N NNr0r&r%c|dkS)uNChroma / statement extract only for content lookup — not metadata inventory.rr/r&s r1primary_uses_retrievalrs & &&r0c|dkS)Nrr/rs r1primary_uses_kb_inventoryrs ( ((r0)r2r'r3r4)r2r'r3rH)r2r'r)r4r3r$)r&r%r3rH)@__doc__ __future__rr] dataclassesrtypingr app.pdf.calcrapp.rag.calculatorr app.rag.chainrr r r r r rrrapp.saas.finance_intentrrrrrrrrrrrapp.saas.knowledge_staterr%compilertr?rSrWrZr^rbrarirJrhrLrMr$rGrPrUrXr[r_rcrkrwrrrr/r0r1rsG  #""""" !!!!!!111111++++++                                                988888 % "*  $C :2: , *2: #    2:   BJ!bj BJO  $++++AAAAHHHH@@@@8    ####L9999*LPKOKOKOKOKOKO\'''' ))))))r0