昺hddlmZmZmZmZddlmZddlmZm Z m Z ddl m Z ddl mZdgZGddeeZGdd Zd S) )AnyDictListOptional) Embeddings) BaseModelExtraroot_validator)get_from_dict_or_env)parseGradientEmbeddingsceZdZUdZeed< dZeeed< dZeeed< dZ eed< dZ eeed< dZ e ed < Gd d Z ed dedefdZdeedeeefdZdeedeeefdZdedeefdZdedeefdZdS)r aGradient.ai Embedding models. GradientLLM is a class to interact with Embedding Models on gradient.ai To use, set the environment variable ``GRADIENT_ACCESS_TOKEN`` with your API token and ``GRADIENT_WORKSPACE_ID`` for your gradient workspace, or alternatively provide them as keywords to the constructor of this class. Example: .. code-block:: python from langchain_community.embeddings import GradientEmbeddings GradientEmbeddings( model="bge-large", gradient_workspace_id="12345614fc0_workspace", gradient_access_token="gradientai-access_token", ) modelNgradient_workspace_idgradient_access_tokenzhttps://api.gradient.ai/apigradient_api_urlquery_prompt_for_retrievalclientc eZdZdZejZdS)GradientEmbeddings.Configz'Configuration for this pydantic object.N)__name__ __module__ __qualname____doc__r forbidextrar/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/embeddings/gradient_ai.pyConfigr5s55 rr T) allow_reusevaluesreturnct|dd|d<t|dd|d<t|dd|d< ddl}n#t$rtd wxYwt|jtd krtd ||d|d|d }||d  |d<|S)z?Validate that api key and python package exists in environment.rGRADIENT_ACCESS_TOKENrGRADIENT_WORKSPACE_IDrGRADIENT_API_URLrNzAGradientEmbeddings requires `pip install -U "gradientai>=1.4.0"`.z1.4.0) access_token workspace_idhostr)slugr)r gradientai ImportErrorr __version__Gradientget_embeddings_model)clsr"r,gradients rvalidate_environmentz'GradientEmbeddings.validate_environment:s6+? +-D+ + &'+? +-D+ + &'&: &(:& & !"         S   ' ( (5>> 9 9S && 78 78*+'   $88fWo8NNx s AAtextscrd|D}|j|j}d|DS)zCall out to Gradient's embedding endpoint. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. cg|]}d|iSinputr.0texts r z6GradientEmbeddings.embed_documents..f444d7D/444rinputscg|] }|j Sr embeddingr:es rr<z6GradientEmbeddings.embed_documents..j,,, ,,,r)rembed embeddingsselfr4r?results rembed_documentsz"GradientEmbeddings.embed_documents]sF54e444""&"11<,,V,,,,rcKd|D}|j|d{Vj}d|DS)zAsync call out to Gradient's embedding endpoint. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. cg|]}d|iSr7rr9s rr<z7GradientEmbeddings.aembed_documents..ur=rr>Ncg|] }|j SrrArCs rr<z7GradientEmbeddings.aembed_documents..yrEr)raembedrGrHs raembed_documentsz#GradientEmbeddings.aembed_documentsls\54e444 **&*99999999E,,V,,,,rr;cd|jr |jd|n|}||gdS)zCall out to Gradient's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text.  r)rrK)rIr;querys r embed_queryzGradientEmbeddings.embed_query{sJ. t. 7 7 7 7 7  ##UG,,Q//rcxK|jr |jd|n|}||gd{V}|dS)zAsync call out to Gradient's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. rRNr)rrP)rIr;rSrGs r aembed_queryzGradientEmbeddings.aembed_querysd. t. 7 7 7 7 7   00%99999999 !}r)rrrrstr__annotations__rrrrrrrr r rr3rfloatrKrPrTrVrrrr r s& JJJ&+/8C=///*+/8C=/// :c99904 444FC ^%%% $ 4   &% D -T#Y -4U 3D - - - - -DI -$tE{:K - - - -00U 0000 stE{rceZdZdZddZdS) TinyAsyncGradientEmbeddingClientzDeprecated, TinyAsyncGradientEmbeddingClient was removed. This class is just for backwards compatibility with older versions of langchain_community. It might be entirely removed in the future. r#Nc td)Nz8Deprecated,TinyAsyncGradientEmbeddingClient was removed.) ValueError)rIargskwargss r__init__z)TinyAsyncGradientEmbeddingClient.__init__sSTTTr)r#N)rrrrr`rrrr[r[s8UUUUUUrr[N)typingrrrrlangchain_core.embeddingsrlangchain_core.pydantic_v1rr r langchain_core.utilsr packaging.versionr __all__r r[rrrrgs,,,,,,,,,,,,000000GGGGGGGGGG555555######  OOOOOJOOOd U U U U U U U U U Ur