a bg @sRddlZddlZddlmZmZddlmZddlmZm Z GdddeeZ dS)N)AnyList) Embeddings) BaseModel ConfigDictcsteZdZUdZdZeed<deeedfdd Ze dd Z e ee e e d d d Z ee e d ddZZS)JohnSnowLabsEmbeddingsayJohnSnowLabs embedding models To use, you should have the ``johnsnowlabs`` python package installed. Example: .. code-block:: python from langchain_community.embeddings.johnsnowlabs import JohnSnowLabsEmbeddings embedding = JohnSnowLabsEmbeddings(model='embed_sentence.bert') output = embedding.embed_query("foo bar") embed_sentence.bertmodelcpu)r hardware_targetkwargsc s&tjfi|zddlm}ddlm}Wn.ty\}ztd|WYd}~n d}~00z(tjt j d<tjt j d<|j |dWn.t y}zt d |WYd}~n d}~00z:t |tr|||_nt ||r||_n |||_Wn0t y }zt d |WYd}~n d}~00dS) z"Initialize the johnsnowlabs model.r)nlp) NLUPipelinez`Could not import johnsnowlabs python package. Please install it with `pip install johnsnowlabs`.NZPYSPARK_PYTHONZPYSPARK_DRIVER_PYTHON)r zFailure starting Spark SessionzFailure loading model)super__init__Z johnsnowlabsr Znlu.pipe.pipeliner ImportErrorsys executableosenvironstart Exception isinstancestrloadr Z to_nlu_pipe)selfr r r r rexc __class__y/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/johnsnowlabs.pyrs0      zJohnSnowLabsEmbeddings.__init__Zforbid)extra)textsreturncCsB|jj|dd}d}|jD]}d|vr|}qdd||DS)zCompute doc embeddings using a JohnSnowLabs transformer model. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. Zdocument)Z output_levelNZ embeddingcSsg|] }|qSr)tolist).0Zvecrrr Pz:JohnSnowLabsEmbeddings.embed_documents..)r Zpredictcolumnsr$)rr"ZdfZemb_colcrrr embed_documentsAs  z&JohnSnowLabsEmbeddings.embed_documents)textr#cCs||gdS)zCompute query embeddings using a JohnSnowLabs transformer model. Args: text: The text to embed. Returns: Embeddings for the text. r)r*)rr+rrr embed_queryRs z"JohnSnowLabsEmbeddings.embed_query)rr )__name__ __module__ __qualname____doc__r r__annotations__rrrZ model_configrfloatr*r, __classcell__rrrr r s  %r) rrtypingrrZlangchain_core.embeddingsrZpydanticrrrrrrr s