
    i                         S r SSKrSSKr\R                  R	                  \R                  R                  \R                  R                  \5      5      5        SSKJ	r	  S r
S r\S:X  a  \
" 5         \" 5         gg)z/Test script for the enhanced agentic RAG system    N)	RAGSystemc            	         [        S5        [        S5        [        5       n U R                  5       nUS:X  a  [        S5        g[        SU S35        [        5         / SQn[        US	5       H  u  p4[        S
U SU 35        [        S5         U R	                  U5      n[        U[        5      (       a  [        SU 35        Ox[        SUS   SS  [        US   5      S:  a  SOS 35        [        S[        US   5       S35        [        S US    5       5      n[        SSR                  U5       35         [        5         M     g! [         a  n[        SU 35         SnAN,SnAff = f)z"Test the enhanced agentic behaviorz#Testing Enhanced Agentic RAG System2==================================================r   z0No documents available. Please add PDFs to test.NzSystem ready with z
 documents)-what cultures were in america before columbusz/tell me about columbus and jewish participationz%what is the secret destiny of americazwho are the nagaszexplain esoteric traditions   z	Question z: z(----------------------------------------zAnswer: answeri  z... z	Sources: sourcesz chunksc              3   *   #    U  H	  oS    v   M     g7f)sourceN ).0srcs     test_enhanced_agent.py	<genexpr>&test_enhanced_agent.<locals>.<genexpr>1   s     "N<MSx=<Ms   zSource documents: z, zError: )printr   setup_system	enumeratesearch_and_answer
isinstancestrlensetjoin	Exception)
rag_system	doc_counttest_questionsiquestionresultsource_nameses           r   test_enhanced_agentr%   
   si   	
/0	(O J'')IA~@A	yk
45	GN !3	!Bxj)*h	!11(;F&#&&)*!1$3!7 8#fXFVBWZ]B]ce8fgh	#fY&7"8!9AB  #"NF9<M"NN*499\+B*CDE
 	) 4"  	!GA3-  	!s   4E
A7E


E,E''E,c                  :   [        S5        [        S5        [        5       n U R                  5       nUS:X  a  gU R                  5         U R                  n[        S5        SnUR                  U5      n[        SU 35        [        SU 35        [        5         [        S	5        / S
QnU H%  nUR                  U5      n[        SU SU 35        M'     [        5         [        S5        / SQnU H!  u  pUR                  X5      n[        S5        M#     [        5         g)z"Test specific agentic improvementsz%Testing Specific Agentic Improvementsr   r   Nz1. Spelling Correction Test:z*what culures were in amerca befor columbisz
   Input: z   Corrected: z2. Question Analysis Test:)r   z%explain the secret destiny of americaz%compare different esoteric traditionsz   'z' -> z3. Fuzzy Matching Test:))americaamericas)culturecultures)columbuscolumbis)beforebeforz.2f)r   r   r   initialize_model	rag_agentcorrect_spellinganalyze_questionfuzzy_match_score)r   r   agent	test_text	corrected	questionsqqtype
test_pairsw1w2scores               r   test_agentic_improvementsr>   9   s   	
12	(OJ'')IA~ !  E 

()<I&&y1I	Jyk
"#	N9+
&'	G 

&'I
 &&q)QCuUG$%  
G 

#$J ''/e  
G    __main__)__doc__sysospathappenddirnameabspath__file__mainr   r%   r>   __name__r   r?   r   <module>rK      sb    5 
 	  9: ; -^.` z r?   