7|hAddlmZmZmZmZddlmZddlmZddl m Z ddl m Z m Z mZddlmZdgZGdde eZGd d Zy ) )AnyDictListOptional) Embeddingsget_from_dict_or_env)parse) BaseModel ConfigDictmodel_validator)SelfGradientEmbeddingscXeZdZUdZeed< dZeeed< dZeeed< dZ eed< dZ eeed< dZ e ed < e d Zed edede fdZed defdZdeedeeefdZdeedeeefdZdedeefdZdedeefdZy)raGradient.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_tokenhttps://api.gradient.ai/apigradient_api_urlquery_prompt_for_retrievalclientforbid)extrabefore)modevaluesreturncjt|dd|d<t|dd|d<t|ddd|d<|S) z?Validate that api key and python package exists in environment.rGRADIENT_ACCESS_TOKENrGRADIENT_WORKSPACE_IDrGRADIENT_API_URLr)defaultr)clsrs i/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/gradient_ai.pyvalidate_environmentz'GradientEmbeddings.validate_environment:s] +? +-D+ &'+? +-D+ &'&:   1 & !"  afterc> ddl}t|jtdkr td|j |j |j |j}|j|j|_ |S#t$r tdwxYw)NrzAGradientEmbeddings requires `pip install -U "gradientai>=1.4.0"`.z1.4.0) access_token workspace_idhost)slug) gradientai ImportErrorr __version__Gradientrrrget_embeddings_modelrr)selfr-gradients r$ post_initzGradientEmbeddings.post_initNs   '' (5> 9S &&3333&&'  333D  ! S  s BBtextsc|Dcgc]}d|i}}|jj|j}|Dcgc]}|jc}Scc}wcc}w)zCall out to Gradient's embedding endpoint. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. inputinputs)rembed embeddings embeddingr2r5textr9resultes r$embed_documentsz"GradientEmbeddings.embed_documentsdsW/44d7D/44""&"1<<%+, ,, 5-s AAcK|Dcgc]}d|i}}|jj|d{j}|Dcgc]}|jc}Scc}w7-cc}ww)zAsync call out to Gradient's embedding endpoint. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. r7r8N)raembedr;r<r=s r$aembed_documentsz#GradientEmbeddings.aembed_documentsss_/44d7D/44 **&*99EE%+, ,, 59,s+A( A!A(A!A(A# A(#A(r>cj|jr|jd|n|}|j|gdS)zCall out to Gradient's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text.  r)rrA)r2r>querys r$ embed_queryzGradientEmbeddings.embed_querysF..../q 7  ##UG,Q//r&cK|jr|jd|n|}|j|gd{}|dS7 w)zAsync call out to Gradient's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. rFNr)rrD)r2r>rGr;s r$ aembed_queryzGradientEmbeddings.aembed_querysV..../q 7   00%99 !}:s5AA A)__name__ __module__ __qualname____doc__str__annotations__rrrrrrrr model_configr classmethodrr%rr4rfloatrArDrHrJr&r$rr s2& J&+/8C=/*+/8C=/ :c904 4FCL(#$3$$'"4#* -T#Y -4U 3D - -DI -$tE{:K -00U 0 stE{r&ceZdZdZddZy) 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. Nctd)Nz8Deprecated,TinyAsyncGradientEmbeddingClient was removed.) ValueError)r2argskwargss r$__init__z)TinyAsyncGradientEmbeddingClient.__init__sSTTr&)rN)rKrLrMrNr[rTr&r$rVrVs Ur&rVN)typingrrrrlangchain_core.embeddingsrlangchain_core.utilsr packaging.versionr pydanticr r r typing_extensionsr__all__rrVrTr&r$rcsB,,05#;;"  UJUp U Ur&