7|h"ddlmZddlmZmZmZmZmZmZm Z ddl Z ddl m Z ddl mZmZmZGddeeZGdd eeZGd d eZGd d eZGddeZGddZGddZy))Enum)AnyDictListLiteralMappingOptionalUnionN)Document) BaseModel SecretStrmodel_validatorceZdZdZdZdZdZy) ArceeRoutez1Routes available for the Arcee API as enumerator.zmodels/generatezmodels/retrievezmodels/status/{id_or_name}N)__name__ __module__ __qualname____doc__generateretrievemodel_training_statusb/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/utilities/arcee.pyrr s; H H8rrceZdZdZdZdZy)DALMFilterTypez:Filter types available for a DALM retrieval as enumerator. fuzzy_search strict_searchN)rrrrrrrrrrrsD!L#MrrcpeZdZUdZeed<eed<eed<dZeed<e de d e d e fd Z y ) 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_metadatabefore)modevaluesreturnc2|jddv|d<|S)zDdocument and name are reserved arcee keys. Anything else is metadatar!)documentname_is_meta)get)clsr's rset_metazDALMFilter.set_meta6s"$ZZ 5=QQz rN)rrrrstr__annotations__rr$boolr classmethodrrr/rrrr r sP(O JL$(#ds$rr c0eZdZUdZeed<eed<eed<y)ArceeDocumentSourcezSource of an Arcee document.r*r+idN)rrrrr0r1rrrr5r5>s&M I Grr5c:eZdZUdZeed<eed<eed<eed<y) ArceeDocumentzArcee document.indexr6scoresourceN)rrrrr0r1floatr5rrrr8r8Fs J G L rr8c*eZdZdZededefdZy)ArceeDocumentAdapterzAdapter for Arcee documentsarcee_documentr(c t|jj|jj|jj|j |j|j dS)z=Adapts an `ArceeDocument` to a langchain's `Document` object.)r+ source_idr9r6r:) page_contentmetadata)r r;r*r+r6r9r:)r.r?s radaptzArceeDocumentAdapter.adaptRs`'..77'--22+2255'--$'''--  rN)rrrrr3r8r rDrrrr>r>Os%%  =  X    rr>c *eZdZdZdeeefdededeeee fdef dZ dd Z dd e d d ee efdeeee fdeedeedef dZddeedefdZd ee efdefdZdedeee fdeee ffdZdede defdZdede deefdZy ) ArceeWrapperzMWrapper for Arcee API. For more details, see: https://www.arcee.ai/ arcee_api_key arcee_api_urlarcee_api_version model_kwargs model_namect|tr t|}n|}||_||_||_||_ tjjj|}|jd|}|jd|_ |jd|_y#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) isinstancer0r rGrJrHrIrrr#format _make_requestr-rN Exception ValueError) selfrGrHrIrJrKarcee_api_key_routeresponsees r__init__zArceeWrapper.__init__is" mS )&}5N*N(6(*!2 44::AAZAXE))%7H$LL4DM)1h)?D & DZLPSTUSVW  sA(B%% C.CCr(NcT|jdk7rtd|jdy)Ntraining_completezModel z4 is not ready. Please wait for training to complete.)rrSrN)rUs rvalidate_model_training_statusz+ArceeWrapper.validate_model_training_statuss8  % %)< <(88  =rmethod)postr-rWbodyparamsheadersc|j|}|j|}tt|}|||||}|jdvrt d|j |jS)aMake a request to the Arcee API Args: method: The HTTP method to use route: The route to call body: The body of the request params: The query params of the request headers: The headers of the request )rb)rW)jsonrarb)z"Failed to make request. Response: )_make_request_headers_make_request_urlgetattrrequests status_coderStextrd) rUr^rWr`rarburlreq_typerXs rrRzArceeWrapper._make_requestsw ,,W,=$$5$18V,Cd67K   z 1@PQ Q}}rc|xsi}t|jts!tdt |j|jj }|dd}|j ||S)Nz'arcee_api_key must be a SecretStr. Got zapplication/json)zX-Tokenz Content-Type)rPrGr TypeErrortypeget_secret_valueupdate)rUrbapi_keyinternal_headerss rrgz"ArceeWrapper._make_request_headerssv-R$,,i89$t?Q?Q:R9ST $$557.  '(rc>|jd|jd|S)N/)rHrI)rUrWs rrhzArceeWrapper._make_request_urls'$$%Qt'='=&>awGGrpromptkwargsc |jxsi}i||}|jdgDcgc] }tdi|}}t|j||jdd||jScc}w)z;Make the request body for generate/retrieve models endpointfilterssize)rNqueryr|r{r6r)rJr-r dictrN)rUrxry _model_kwargs_paramsfr{s r_make_request_body_for_modelsz*ArceeWrapper._make_request_body_for_modelss|))/R -]-f-,3KK 2,FGq:??GG]]VQ'}}   HsA3c |jdtjj|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_rxr^rWr`rlr)rRrrr#r)rUrxryrXs rrzArceeWrapper.generatesX%%%%++333& rr~c |jdtjj|jdd|i|}|dDcgc] }t j tdi|"c}Scc}w)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_rxrresultsr)rRrrr#rr>rDr8)rUr~ryrXdocs rrzArceeWrapper.retrieves%%%%++333&  *  ! & &}';s'; <   s%A.)r(N)NNN)N)rrrrr r0r r rrrZr]rrrrrRrgrhrrrr rrrrrFrFcsq "S)^,"" " tCH~. "  "H-1!%"&  &Z_%wsCx()    $  4 Xd^ t HuZ_'=H#H  %,S#X%6 c       0    h  rrF)enumrtypingrrrrrr r rjlangchain_core.retrieversr pydanticr r rr0rrr r5r8r>rFrrrrsv EEE.::9d9$S$$D )  I   (] ] r