7|h8ddlZddlmZmZmZmZddlmZddlm Z m Z m Z m Z ejeZGdde eZy)N)AnyDictListOptional) Embeddings) BaseModel ConfigDictFieldmodel_validatorcveZdZUdZdZeeed< dZeeed< dZ eeed< dZ eeed< dZ eeed< e dd Z eeed < e dd Zeeed < e dd Zeed <d Zeed<eddZedededefdZdeedeeefdZdedeefdZy)ClarifaiEmbeddingsaClarifai embedding models. To use, you should have the ``clarifai`` python package installed, and the environment variable ``CLARIFAI_PAT`` set with your personal access token or pass it as a named parameter to the constructor. Example: .. code-block:: python from langchain_community.embeddings import ClarifaiEmbeddings clarifai = ClarifaiEmbeddings(user_id=USER_ID, app_id=APP_ID, model_id=MODEL_ID) (or) Example_URL = "https://clarifai.com/clarifai/main/models/BAAI-bge-base-en-v15" clarifai = ClarifaiEmbeddings(model_url=EXAMPLE_URL) N model_urlmodel_idmodel_version_idapp_iduser_idT)defaultexcludepattokenmodelzhttps://api.clarifai.comapi_baseforbid)extraprotected_namespacesbefore)modevaluesreturnc  ddlm}|jd}|jd}|jd}|jd}|jd}|jd }|jd } |jd } ||||t | | | || |d<|S#t$r tdwxYw)zuValidate that we have all required info to access Clarifai platform and python package exists in environment.r)ModelzXCould not import clarifai python package. Please install it with `pip install clarifai`.rrrrrrrr)id)urlrr model_versionrrrbase_urlr)clarifai.client.modelr" ImportErrorgetdict) clsrr"rrrrrrrrs f/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/clarifai.pyvalidate_environmentz'ClarifaiEmbeddings.validate_environment0s   3 **Y'H%::j)!::&89JJ{+ ::j)jj 7#"23  w 3 A  s B//Ctextsc lddlm}|j|jj}d}g} t dt ||D]}||||z}t|D cgc]!\}} |jt|| #} }} |jj| } |j| jD cgc].} t| jjdj 0c}  |Scc} }wcc} w#t"$r#} t$j'd| Yd} ~ |Sd} ~ wwxYw)zCall out to Clarifai's embedding models. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. r)Inputs )input_idraw_textPredict failed, exception: N)clarifai.client.inputr0from_auth_helperr auth_helperrangelen enumerateget_text_inputstrpredictextendoutputslistdata embeddingsvector Exceptionloggererror)selfr.r0 input_obj batch_sizerBibatchr#inp input_batchpredict_responseoutputes r,embed_documentsz"ClarifaiEmbeddings.embed_documentsSs6 1++DJJ,B,BC    <1c%j*5 a!j.1$-U#3C,,c"g,L $(::#5#5k#B !!'7&>&>"V[[33A6==> "   < LL6qc: ; ; N=U=U79RWW''*112J!} < LL6qc: ; ;!} rgs9,,0BB   8 $AJAra