o Zh @sRddlZddlZddlmZmZddlmZddlmZm Z GdddeeZ dS)N)AnyList) Embeddings) BaseModel ConfigDictcseZdZUdZdZeed<  ddededeffdd Ze d d Z d e ed e e e fd dZ ded e e fddZZS)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.bertmodelcpuhardware_targetkwargsc stjd i|zddlm}ddlm}Wnty(}ztd|d}~wwztjt j d<tjt j d<|j |dWnt yN}zt d |d}~wwz"t |tr^|||_WdSt ||ri||_WdS|||_WdSt y}zt d |d}~ww) 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__rb/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/johnsnowlabs.pyrs<        zJohnSnowLabsEmbeddings.__init__Zforbid)extratextsreturncCsB|jj|dd}d}|jD]}d|vr|}q dd||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 Psz:JohnSnowLabsEmbeddings.embed_documents..)r Zpredictcolumnsr$)rr"dfZemb_colcrrr embed_documentsAs  z&JohnSnowLabsEmbeddings.embed_documentstextcCs||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