tjUdZddlmZddlZddlZddlmZddlmZddl m Z edZ dZ d Zejd ZiZd ed <d ZedGddZd$dZd%d&dZd'dZd(dZd)d*d#ZdS)+uDKnowledge-awareness for SaaS chat — empty KB vs no match vs found.) annotationsN) dataclass)Literal)db)empty no_relevantfounduI don't have a knowledge base yet, so I can't answer questions about your finances or documents. Please upload a knowledge base first: • Upload a PDF statement • Upload a CSV export • Add files in the Knowledge panelzI couldn't find any information related to your question in your uploaded knowledge base. Try rephrasing, ask about totals or a merchant on your statement, or upload another document.a*(?i)\b(knowledge\s*base|my\s+(?:documents?|files?|uploads?|pdfs?)|what\s+(?:documents?|files?)\s+(?:do\s+you\s+have|are\s+uploaded|are\s+ready|are\s+indexed)|uploaded\s+(?:documents?|files?|pdfs?)|which\s+(?:statement|pdf|document|file|upload)|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)|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|upload\s+status|indexing\s+status|is\s+(?:it|the\s+(?:file|pdf|document))\s+(?:ready|uploaded|indexed)|list\s+(?:my\s+)?(?:documents?|files?|uploads?)|tell\s+me\s+about\s+(?:the\s+)?knowledge\s*base)\bz*dict[str, tuple[float, 'KnowledgeStatus']] _STATUS_CACHEgF@T)frozencTeZdZUdZded<ded<ded<ded<ed d Zd S) KnowledgeStatuszBWhether the selected agent has any indexed / structured knowledge.int vector_count ready_docs ready_chunksstatement_jobsreturnboolcZ|jdkp |jdkp|jdkp |jdkS)Nrrrrr)selfs 4/var/www/html/CA-Chatbot/app/saas/knowledge_state.py has_knowledgezKnowledgeStatus.has_knowledge=sE   ! '" ' 1$ '"Q&  N)rr)__name__ __module__ __qualname____doc____annotations__propertyrrrr r 4sgLLOOO    X   rr docs list[dict]rc pi}|pgD]}|ddkrtjddt|dpd}|s|t|dp|dpd}||}|r=|t|dp|dpdkr|||<t |S) zDCollapse duplicate ready uploads of the same filename (keep newest).statusreadyz\s+ filename updated_at created_at)getresubstrstriplowerlistvalues)r"bestdkeystampprevs runique_ready_docsr9GsD ZR   55??g % % fVS#aeeJ&7&7&=2">">"D"D"F"F"L"L"N"NOO  AEE,''D155+>+>D"EExx}} uDHH\$:$:$Zdhh|>T>T$ZXZ [ [[[DI   rorg_id str | Noneagent_idNonec$|r$|r"t|d|ddS|rN|d}ttD]2}||rt|d3dStdS)z5Drop cached KB status after upload / delete / ingest.:N)r popr2 startswithclear)r:r<prefixr6s rinvalidate_knowledge_cacherDWs (V00h00$777  && - -C~~f%% -!!#t,,,rr/c|d|}t|}tj}|r||dz tkr|dS t j||pg}n#t$rg}YnwxYwd|D}t|}t|}td|D} td|D} | dkr| n|rdnd} t| || | } || ft|<| S)zInspect the agent's KB before retrieval or LLM generation. Uses a single documents query (no Chroma open, no PDF re-extract) so chat stays fast on remote Postgres. r?rcDg|]}|ddk|S)r%r&)r,.0r5s r z*assess_agent_knowledge..vs, ; ; ;1h7 : :Q : : :rc3^K|](}t|dpdV)dS) chunk_countrN)rr,rHs r z)assess_agent_knowledge..ys;FF!s155//4155FFFFFFrc3K|]O}t|dpddKdVPdS)r(r)z.pdfrFN)r/r,r1endswithrHs rrMz)assess_agent_knowledge..{snSz!2!2!8b99??AAJJ6RR rr) r r,time monotonic _STATUS_TTL_Ssaas_dblist_kb_documents Exceptionr9lensumr ) r:r<r6hitnowr"r&uniquerrrrr%s rassess_agent_knowledger[esc  h C   C C .  C c!f --1v (::@b  < ; ; ; ;E u % %FVJFFvFFFFFLN$0!#3#3<general>byehelpmaththanksconceptgreetingofftopicFTN)rb>document_lookupdocument_metadata)app.saas.intent_routerrer0intent)rbrcr\rer`primarys rneeds_knowledgerss ?>>>>> R  A N+$E$E ^ ^ ^5t u%%a8N$OOOG >E EEr)r"r#rr#)NN)r:r;r<r;rr=)r:r/r<r/rr )r\r/rr)r))rbr/rcr/r\r/rr)r __future__rr-rP dataclassesrtypingrapp.saasrrSKnowledgeStateEMPTY_KB_MESSAGENO_RELEVANT_MESSAGEcompiler_r rrRr r9rDr[rarsr!rrr|s|JJJ"""""" !!!!!!""""""89+d bj  (=? >>>>  $        $         ####L ' ' ' 'FFFFFFFr