昺h""ddlmZddlmZmZmZmZmZmZm Z ddl Z ddl m Z m Z mZddlmZGddeeZGdd eeZGd d e ZGd d e ZGdde ZGddZGddZdS))Enum)AnyDictListLiteralMappingOptionalUnionN) BaseModel SecretStrroot_validator)DocumentceZdZdZdZdZdZdS) ArceeRoutez1Routes available for the Arcee API as enumerator.zmodels/generatezmodels/retrievezmodels/status/{id_or_name}N)__name__ __module__ __qualname____doc__generateretrievemodel_training_statusk/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/utilities/arcee.pyrr s&;; H H8rrceZdZdZdZdZdS)DALMFilterTypez:Filter types available for a DALM retrieval as enumerator. fuzzy_search strict_searchN)rrrrrrrrrrrsDD!L#MMMrrcpeZdZUdZeed<eed<eed<dZeed<e de de fd Z d S) DALMFilteraXFilters available for a DALM retrieval and generation. Arguments: field_name: The field to filter on. Can be 'document' or 'name' to filter on your document's raw text or title. Any other field will be presumed to be a metadata field you included when uploading your context data filter_type: Currently 'fuzzy_search' and 'strict_search' are supported. 'fuzzy_search' means a fuzzy search on the provided field is performed. The exact strict doesn't need to exist in the document for this to find a match. Very useful for scanning a document for some keyword terms. 'strict_search' means that the exact string must appear in the provided field. This is NOT an exact eq filter. ie a document with content "the happy dog crossed the street" will match on a strict_search of "dog" but won't match on "the dog". Python equivalent of `return search_string in full_string`. value: The actual value to search for in the context data/metadata field_name filter_typevalueF _is_metadatavaluesreturnc:|ddv|d<|S)zDdocument and name are reserved arcee keys. Anything else is metadatar!)documentname_is_meta)get)clsr%s rset_metazDALMFilter.set_meta6s%$ZZ 55=QQz rN) rrrrstr__annotations__rr$boolr rr-rrrr r s(OOO JJJL$^dtrr c2eZdZUdZeed<eed<eed<dS)ArceeDocumentSourcezSource of an Arcee document.r(r)idN)rrrrr.r/rrrr2r2=s1&&MMM III GGGGGrr2c<eZdZUdZeed<eed<eed<eed<dS) ArceeDocumentzArcee document.indexr3scoresourceN)rrrrr.r/floatr2rrrr5r5Es? JJJ GGG LLL rr5c2eZdZdZededefdZdS)ArceeDocumentAdapterzAdapter for Arcee documentsarcee_documentr&c t|jj|jj|jj|j|j|jdS)z=Adapts an `ArceeDocument` to a langchain's `Document` object.)r) source_idr6r3r7) page_contentmetadata)rr8r(r)r3r6r7)r,r<s radaptzArceeDocumentAdapter.adaptQsT'.7'-2+25'-$''-    rN)rrrr classmethodr5rrArrrr;r;NsF%%  =  X    [    rr;c eZdZdZdeeefdededeeee fdef dZ dd Z dd e d d ee efdeeee fdeedeedef dZddeedefdZd ee efdefdZdedeee fdeee ffdZdede defdZdede deefdZd S) ArceeWrapperzMWrapper for Arcee API. For more details, see: https://www.arcee.ai/ arcee_api_key arcee_api_urlarcee_api_version model_kwargs model_namect|trt|}n|}||_||_||_||_ tjj |}| d|}| d|_ | d|_dS#t$r} td|d| | d} ~ wwxYw)a5Initialize ArceeWrapper. Arguments: arcee_api_key: API key for Arcee API. arcee_api_url: URL for Arcee API. arcee_api_version: Version of Arcee API. model_kwargs: Keyword arguments for Arcee API. model_name: Name of an Arcee model. ) id_or_namer+model_idstatusz2Error while validating model training status for 'z': N) isinstancer.r rErHrFrGrrr#format _make_requestr+rL Exception ValueError) selfrErFrGrHrIarcee_api_key_routeresponsees r__init__zArceeWrapper.__init__hs" mS ) ) +&}55NN*N(6(*!2 4:AAZAXXE))%77H$LL44DM)1h)?)?D & & &   WZWWTUWW  sA/B66 CCCr&NcL|jdkrtd|jddS)Ntraining_completezModel z4 is not ready. Please wait for training to complete.)rrQrL)rSs rvalidate_model_training_statusz+ArceeWrapper.validate_model_training_statussC  %)< < <8888  = z>ArceeWrapper._make_request_body_for_models..s"GGGq:????GGGrfilterssize)rLqueryrr~r3)rHr+dictrL)rSrvrw _model_kwargs_paramsr~s r_make_request_body_for_modelsz*ArceeWrapper._make_request_body_for_modelssy)/R -]-f-GGGKK 2,F,FGGG]VQ''}     rc z|dtjj|jdd|i|}|dS)a+Generate text from Arcee DALM. Args: prompt: Prompt to generate text from. size: The max number of context results to retrieve. Defaults to 3. (Can be less if filters are provided). filters: Filters to apply to the context dataset. r]rvr\rUr^rjr)rPrrr#r)rSrvrwrVs rrzArceeWrapper.generates^%%%+33&  rrc |dtjj|jdd|i|}d|dDS)aIRetrieve {size} contexts with your retriever for a given query Args: query: Query to submit to the model size: The max number of context results to retrieve. Defaults to 3. (Can be less if filters are provided). filters: Filters to apply to the context dataset. r]rvrc Vg|]&}ttdi|'Srz)r;rAr5)r{docs rr}z)ArceeWrapper.retrieve..sD    ! & &}';';s';'; < <   rresultsr)rPrrr#r)rSrrwrVs rrzArceeWrapper.retrievesw%%%+33&     *    r)r&N)NNN)N)rrrrr r.r r rrrXr[rrrrrPrerfrrrrrrrrrDrDbs  "S)^,"" " tCH~. "  """"H-1!%"&  &Z_%wsCx()    $  4  Xd^ t    HuZ_'=H#HHHH  %,S#X%6 c             0    h       rrD)enumrtypingrrrrrr r rhlangchain_core.pydantic_v1r r r langchain_core.retrieversrr.rrr r2r5r;rDrrrrs EEEEEEEEEEEEEEEEEEKKKKKKKKKK......99999d999$$$$$S$$$$B     )        I           (] ] ] ] ] ] ] ] ] ] r