dh"SSKJr SSKJrJrJrJrJrJrJ r SSK r SSK J r SSK JrJrJr "SS\\5r"SS \\5r"S S \5r"S S \5r"SS\5r"SS5r"SS5rg))Enum)AnyDictListLiteralMappingOptionalUnionN)Document) BaseModel SecretStrmodel_validatorc$\rSrSrSrSrSrSrSrg) ArceeRoute z1Routes available for the Arcee API as enumerator.zmodels/generatezmodels/retrievezmodels/status/{id_or_name}N) __name__ __module__ __qualname____firstlineno____doc__generateretrievemodel_training_status__static_attributes__r[/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/utilities/arcee.pyrr s; H H8rrc \rSrSrSrSrSrSrg)DALMFilterTypez:Filter types available for a DALM retrieval as enumerator. fuzzy_search strict_searchrN)rrrrrr!r"rrrrrrsD!L#Mrrct\rSrSr%Sr\\S'\\S'\\S'Sr\ \S'\ "SS 9\ S \ S \ 4S j55rS rg) DALMFilteraFilters 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)modevaluesreturnc4URS5S;US'U$)zDdocument and name are reserved arcee keys. Anything else is metadatar&)documentname_is_meta)get)clsr,s rset_metaDALMFilter.set_meta6s"$ZZ 5=QQz rrN)rrrrrstr__annotations__rr)boolr classmethodrrr4rrrrr$r$sP(O JL$(#ds$rr$c8\rSrSr%Sr\\S'\\S'\\S'Srg)ArceeDocumentSource>zSource of an Arcee document.r/r0idrN)rrrrrr6r7rrrrr;r;>s&M I Grr;cB\rSrSr%Sr\\S'\\S'\\S'\\S'Sr g) ArceeDocumentFzArcee document.indexr=scoresourcerN) rrrrrr6r7floatr;rrrrr?r?Fs J G L rr?c4\rSrSrSr\S\S\4Sj5rSr g)ArceeDocumentAdapterOzAdapter for Arcee documentsarcee_documentr-c [URRURRURRUR URUR S.S9$)z=Adapts an `ArceeDocument` to a langchain's `Document` object.)r0 source_idrAr=rB) page_contentmetadata)r rCr/r0r=rArB)r3rHs radaptArceeDocumentAdapter.adaptRs`'..77'--22+2255'--$'''--  rrN) rrrrrr9r?r rMrrrrrFrFOs%%  =  X    rrFc F\rSrSrSrS\\\4S\S\S\\ \\ 4S\4 Sjr SS jr SS \ S S\\\4S\\\\ 4S\\S\\S \4 SjjrSS\\ S \ 4SjjrS\\\4S \4SjrS\S\\\ 4S \\\ 44SjrS\S\ S \4SjrS\S\ S \\4SjrSrg ) ArceeWrapperczEWrapper for Arcee API. For more details, see: https://www.arcee.ai/ arcee_api_key arcee_api_urlarcee_api_version model_kwargs model_namec[U[5(a [U5nOUnX`lX@lX lX0l[RRRUS9nURSU5nURS5Ul URS5Ulg![an [SUSU 35U eSn A ff=f)zInitialize 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_namer2model_idstatusz2Error while validating model training status for 'z': N) isinstancer6r rRrUrSrTrrr(format _make_requestr2rY Exception ValueError) selfrRrSrTrUrVarcee_api_key_routeresponsees r__init__ArceeWrapper.__init__is" mS ) )&}5N*N(6(*!2 44::AAZAXE))%7H$LL4DM)1h)?D & DZLPSTUSVW  sA%B## C-B??Cr-NcVURS:wa[SURS35eg)Ntraining_completezModel z4 is not ready. Please wait for training to complete.)rr^rY)r`s rvalidate_model_training_status+ArceeWrapper.validate_model_training_statuss8  % %)< <(88  =rmethod)postr2rbbodyparamsheaderscURUS9nURUS9n[[U5nU"XcXES9nURS;a[ SUR 35eUR5$)zMake 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 )ro)rb)jsonrnro)z"Failed to make request. Response: )_make_request_headers_make_request_urlgetattrrequests status_coder^textrq) r`rkrbrmrnrourlreq_typercs rr]ArceeWrapper._make_requestss ,,W,=$$5$18V,C6K   z 1@PQ Q}}rcU=(d 0n[UR[5(d![S[ UR535eURR 5nUSS.nUR U5 U$)Nz'arcee_api_key must be a SecretStr. Got zapplication/json)zX-Tokenz Content-Type)r[rRr TypeErrortypeget_secret_valueupdate)r`roapi_keyinternal_headerss rrt"ArceeWrapper._make_request_headerssy-R$,,i889$t?Q?Q:R9ST $$557.  '(rc>URSURSU3$)N/)rSrT)r`rbs rruArceeWrapper._make_request_urls'$$%Qt'='=&>awGGrpromptkwargsc UR=(d 0n0UEUEnURS/5Vs/sHn[S0UD6PM nn[URUURSS5UURS9$s snf)z;Make the request body for generate/retrieve models endpointfilterssize)rYqueryrrr=r)rUr2r$dictrY)r`rr _model_kwargs_paramsfrs r_make_request_body_for_models*ArceeWrapper._make_request_body_for_modelss))/R -]-f-,3KK 2,FG,Fq:??,FG]]VQ'}}   HsA7c URS[RRUR"SSU0UD6S9nUS$)zGenerate 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. rlrrkrbrmryr)r]rrr(r)r`rrrcs rrArceeWrapper.generatesV%%%%++33& rrc URS[RRUR"SSU0UD6S9nUSVs/sH!n[ R [S0UD65PM# sn$s snf)aRetrieve {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. rlrrresultsr)r]rrr(rrFrMr?)r`rrrcdocs rrArceeWrapper.retrieves%%%%++33&  * * ! & &}';s'; <*   s'A-)rRrSrTrYrUr)r-N)NNN)N)rrrrrr r6r r rrrerirrrrr]rtrurrrr rrrrrrPrPcsq "S)^,"" " tCH~. "  "H-1!%"&  &Z_%wsCx()    $  4 Xd^ t HuZ_'=H#H  %,S#X%6 c       0    h  rrP)enumrtypingrrrrrr r rwlangchain_core.retrieversr pydanticr r rr6rrr$r;r?rFrPrrrrsv EEE.::9d9$S$$D )  I   (] ] r