j=,ddlZddlmZmZddlmZmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZddlmZmZmZmZdd lmZd ed efd Zd dd efdZe eeeedfZ GddedZ!GddedZ"GddedZ#GddedZ$GddedZ%GddedZ&GddeedZ'Gdd e'Z(Gd!d"e'Z)Gd#d$edZ*Gd%d&edZ+d'd(d)d*d+d,Z,ed-d.d/0Gd1d2eZ-y)3N)ABCabstractmethod)AnyCallableDictListLiteralOptionalSequenceUnion) deprecated)CallbackManagerForRetrieverRun)Document) BaseRetriever) BaseModelFieldmodel_validator validator) Annotatedexcerptreturnc\|s|Stjdd|jdd}|S)zClean an excerpt from Kendra. Args: excerpt: The excerpt to clean. Returns: The cleaned excerpt. z\s+ z...)resubreplace)rress o/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/retrievers/kendra.py clean_excerptr s0  &&g & . .ub 9C Jitem ResultItemcd}|j}|r |d|dz }t|j}|r |d|dz }|S)zCombine a ResultItem title and excerpt into a single string. Args: item: the ResultItem of a Kendra search. Returns: A combined text of the title and excerpt of the given item. rzDocument Title:  zDocument Excerpt: ) get_titler get_excerpt)r"texttitlers r combined_textr*+s[ D NN E  "5',,D,,./G &wir22 Kr!cLeZdZUdZeed< eed< eeed< eeed<y) Highlightz8Information that highlights the keywords in the excerpt. BeginOffset EndOffset TopAnswerTypeN) __name__ __module__ __qualname____doc__int__annotations__r boolstrr!rr,r,Gs0BLNJ~7 3-selfs rget_value_textz(AdditionalResultAttribute.get_value_textqszz11666r!N) r1r2r3r4r8r6r rArMr9r!rrDrDgs1& H#344 ))%77r!rDceZdZUdZdZeeed< dZee ed< dZ ee eed< dZ eeed< e defdZy) DocumentAttributeValuezValue of a document attribute.N DateValue LongValueStringListValue StringValuerc|jr |jS|jr |jS|jr |jS|jr |jSy)zThe only defined document attribute value or None. According to Amazon Kendra, you can only provide one value for a document attribute. N)rPrQrRrSrKs rvaluezDocumentAttributeValue.valuesS >>>> ! >>>> !   '' '   ## #r!)r1r2r3r4rPr r8r6rQr5rRrrSpropertyDocumentAttributeValueTyperUr9r!rrOrOvsh(#Ix}#3#Ix}#+/OXd3i(/ !%K#% 1r!rOc(eZdZUdZeed< eed<y)DocumentAttributezDocument attribute.rErHN)r1r2r3r4r8r6rOr9r!rrYrYs H# !!%r!rYceZdZUdZeeed< eeed< eeed< gZeee ed< ee ed< e defdZ e defd Z de fd Zdeeeffd Zdefd Zefd edgefdefdZy)r#zBase class of a result item.Id DocumentId DocumentURIDocumentAttributesScoreAttributesrcy)zDocument title.Nr9rKs rr&zResultItem.get_titler!cy)zDDocument excerpt or passage original content as retrieved by Kendra.Nr9rKs rr'zResultItem.get_excerptrar!ciS)zDocument additional metadata dict. This returns any extra metadata except these: * result_id * document_id * source * title * excerpt * document_attributes r9rKs rget_additional_metadataz"ResultItem.get_additional_metadatas  r!c|jxsgDcic]#}|j|jj%c}Scc}w)zDocument attributes dict.)r^rErHrU)rLattrs rget_document_attributes_dictz'ResultItem.get_document_attributes_dicts>8<8O8O8USU8UW8Ut$*****8UWWWs(>c:|j|jdSy)zDocument Score ConfidenceScoreConfidence NOT_AVAILABLE)r_rKs rget_score_attributezResultItem.get_score_attributes#    +''(9: :"r!page_content_formatterc *||}|j}|j|j|j|j|j |j |j|jdt||S)z!Converts this item to a Document.) result_id document_idsourcer)rdocument_attributesscore) page_contentmetadata) rdupdater[r\r]r&r'rgrkr)rLrlrsrts rto_doczResultItem.to_docs.d3 //1!WW#**)++-'+'H'H'J113  \HEEr!N)r1r2r3r4r r8r6r^rrYdictrr&r'rdrrWrgrkr*rrrvr9r!rr#r#s& - #<>&7!89>"d^#%3SSSS  Xd38R3R.SX#S#GTF&. ~s/B&CF Fr!ceZdZUdZeed< eeed< eeed< eeed< gZee e ed< eeed< defd Z defd Z defd Z defd Zy )QueryResultItemzQuery API result item. DocumentTitle FeedbackTokenFormatr0AdditionalAttributesDocumentExcerptrc.|jjSrJ)rzr>rKs rr&zQueryResultItem.get_titles!!&&&r!cv|jsy|jdsy|jdjS)Nrr)r}rMrKs rget_attribute_valuez#QueryResultItem.get_attribute_values9((((+,,Q/>>@ @r!c|jr.|jdjdk(r|j}|S|jr|jj}|Sd}|S)Nr AnswerTextr)r}rErr~r>)rLrs rr'zQueryResultItem.get_excerptsh  % %))!,00L@..0G   ! !**//GGr!c"d|ji}|S)Ntype)r0)rLadditional_metadatas rrdz'QueryResultItem.get_additional_metadata s%tyy1""r!N)r1r2r3r4r=r6r r8r}rrDr&rr'rwrdr9r!rryrys %%C= A SM 3-?FH(4(A#BCHG011''3'ASA S ##r!rycNeZdZUdZeeed< eeed< defdZdefdZy)RetrieveResultItemzRetrieve API result item.rzContentrc"|jxsdSNr)rzrKs rr&zRetrieveResultItem.get_titles!!'R'r!c"|jxsdSr)rrKs rr'zRetrieveResultItem.get_excerpts||!r!r!N) r1r2r3r4r r8r6r&r'r9r!rrrs7#C=  c]"(3("S"r!rc"eZdZUdZeeed<y) QueryResulta`Amazon Kendra Query API` search result. It is composed of: * Relevant suggested answers: either a text excerpt or table excerpt. * Matching FAQs or questions-answer from your FAQ file. * Documents including an excerpt of each document with its title. ResultItemsN)r1r2r3r4rryr6r9r!rrr!so&&r!rc.eZdZUdZeed< eeed<y)RetrieveResultz`Amazon Kendra Retrieve API` search result. It is composed of: * relevant passages or text excerpts given an input query. QueryIdrN)r1r2r3r4r8r6rrr9r!rrr/s  L())r!rg?g?g??)rjLOWMEDIUMHIGH VERY_HIGHz0.3.16z1.0z#langchain_aws.AmazonKendraRetriever)sinceremovalalternative_importceZdZUdZeed<dZeeed<dZeeed<dZ e ed<dZ ee ed<dZ eee ed <eZeegefed <eed <dZee ed <eeeed dfed<edde de fdZedede eefdefdZdedeefdZdeedeefdZ deedeefdZ!dede"deefdZ#y) AmazonKendraRetrievera`Amazon Kendra Index` retriever. Args: index_id: Kendra index id region_name: The aws region e.g., `us-west-2`. Fallsback to AWS_DEFAULT_REGION env variable or region specified in ~/.aws/config. credentials_profile_name: The name of the profile in the ~/.aws/credentials or ~/.aws/config files, which has either access keys or role information specified. If not specified, the default credential profile or, if on an EC2 instance, credentials from IMDS will be used. top_k: No of results to return attribute_filter: Additional filtering of results based on metadata See: https://docs.aws.amazon.com/kendra/latest/APIReference document_relevance_override_configurations: Overrides relevance tuning configurations of fields/attributes set at the index level See: https://docs.aws.amazon.com/kendra/latest/APIReference page_content_formatter: generates the Document page_content allowing access to all result item attributes. By default, it uses the item's title and excerpt. client: boto3 client for Kendra user_context: Provides information about the user context See: https://docs.aws.amazon.com/kendra/latest/APIReference Example: .. code-block:: python retriever = AmazonKendraRetriever( index_id="c0806df7-e76b-4bce-9b5c-d5582f6b1a03" ) index_idN region_namecredentials_profile_nametop_kattribute_filter*document_relevance_override_configurationsrlclient user_contextrr)gelemin_score_confidencerUrc.|dkrtd|d|S)Nrtop_k () cannot be negative.) ValueError)clsrUs rvalidate_top_kz$AmazonKendraRetriever.validate_top_ks# 19wug-BCD D r!before)modevaluesc|jd}||dkrtd|d|jd|S ddl}|jdr|j|d}n|j}i}|jdr|d|d<|jd i||d<|S#t $r t d t $r}td |d}~wwxYw) Nrrrrrr) profile_namerzRCould not import boto3 python package. Please install it with `pip install boto3`.zCould not load credentials to authenticate with AWS client. Please check that credentials in the specified profile name are valid.)kendra)getrboto3Sessionr ImportError Exception)rrrrsession client_paramses r create_clientz#AmazonKendraRetriever.create_clients 7#  wug-BCD D ::h  +M  zz45--V  *   sA,B))C CCqueryc|j|jdd|jd}|j|j|d<|j|j|d<|j |j |d<|j jdi|}tj|}|jr |jS|j jdi|}tj|}|jS)Nri)IndexId QueryTextPageSizeAttributeFilter'DocumentRelevanceOverrideConfigurations UserContextr9) rstriprrrrrretriever parse_objrrr)rLr kendra_kwargsresponser_resultq_results r _kendra_queryz#AmazonKendraRetriever._kendra_querys}}q-    ,/3/D/DM+ ,  : : F?? C D    (+/+<+"c"$"H$3$8J+?$6HZ,@T(^X4>&<<4 < h rs] #   7C-3 ( 3 3   (#3T#Y#<= =  ='*Yg* 7  7Yg>& &>FCw>FB.#j.#b " " )7  Yg      < f