:wdz:ddlmZddlZddlZdZdZdZdZdS)DocumentNcZtjdd|dd}|S)Nz\s+ z...)resubreplace)res_textress X/Users/tsaris/Documents/GitHub/genai-id-lab4-prep/lab-4/rag_app/kendra/kendra_results.py clean_resultrs* &X & & . .uR 8 8C JcF|d|}|dd}|d}|d}|dr9|dddd kr!|ddd d d}n|d d}t|}d |zdz|zdz}|||||ddS)N ResultItems DocumentTitleText DocumentURITypeAdditionalAttributesrKey AnswerTextValueTextWithHighlightsValueDocumentExcerptzDocument Title: z Document Excerpt:  )sourcetitleexcerpttype page_contentmetadata)r) respcountr doc_titledoc_urir_typer doc_excerpt combined_texts r get_top_n_resultsr, s ]E"A/"6*I G vYF !0a(>&?&B5&I\&Y&Y+,Q/89RSTZ[&'/x((K&25MMP[[^bbM(wQZgr}C6D6D E EErc|||td|kr|}ntd}fdtd|D}d|DS)N)IndexId QueryTextrc0g|]}t|S)r,).0iresponses r z kendra_query..s$ F F Fq h * * F F FrrcHg|]}t|d|d S)r"r#r!r)r2ds r r5z kendra_query..s/ _ _ _UVHAn$5!J- P P P _ _ _r)querystriplenrange)kclientkquerykcount kindex_idr_countdocsr4s @r kendra_queryrBs}}Y&,,..}IIH 8M "##f,,h}-.. F F F FE!W4E4E F F FD _ _Z^ _ _ __rc2tjd|}|S)Nkendra) region_name)boto3client)r?kregionr<s r kendra_clientrIsl8999G Nr)langchain.docstore.documentrrFrrr,rBrIr1rr rKsu000000  E E E```r