7|h XddlZddlZddlmZmZddlmZddlmZm Z GddeeZ y)N)AnyList) Embeddings) BaseModel ConfigDictceZdZUdZdZeed< ddededeffd Ze dZ d e ed e e e fd Z d ed e e fd ZxZS)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.bertmodelhardware_targetkwargsc :t|d i| ddlm}ddlm} tjtjd<tjtjd<|j| t|tr|j||_yt||r||_y|j#||_y#t $r}t d|d}~wwxYw#t$r}td |d}~wwxYw#t$r}td |d}~wwxYw) z"Initialize the johnsnowlabs model.r)nlp) NLUPipelinez`Could not import johnsnowlabs python package. Please install it with `pip install johnsnowlabs`.NPYSPARK_PYTHONPYSPARK_DRIVER_PYTHON)r zFailure starting Spark SessionzFailure loading model)super__init__ johnsnowlabsrnlu.pipe.pipeliner ImportErrorsys executableosenvironstart Exception isinstancestrloadr to_nlu_pipe)selfr r r rrexc __class__s j/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/johnsnowlabs.pyrzJohnSnowLabsEmbeddings.__init__s "6"  ( 5 G+.>>BJJ' (25..BJJ. / IIoI 6  >%% XXe_ E;/"  __U3 + E   G<=3 F G >34# = >sS CAC#4&DD/D C  CC # C=, C88C= D DDforbid)extratextsreturnc|jj|d}d}|jD] }d|vs|} ||jDcgc]}|jc}Scc}w)zCompute doc embeddings using a JohnSnowLabs transformer model. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. document) output_levelN embedding)r predictcolumnstolist)r#r)dfemb_colcvecs r&embed_documentsz&JohnSnowLabsEmbeddings.embed_documentsAskZZ  J  ? Aa )+7 (:(:(<= ===s A'textc,|j|gdS)zCompute query embeddings using a JohnSnowLabs transformer model. Args: text: The text to embed. Returns: Embeddings for the text. r)r6)r#r7s r& embed_queryz"JohnSnowLabsEmbeddings.embed_queryRs##TF+A..)r cpu)__name__ __module__ __qualname____doc__r r__annotations__r rr model_configrfloatr6r9 __classcell__)r%s@r&r r s 'E3&+$#>#>#> #>JL>T#Y>4U 3D>" / /U /r:r ) rrtypingrrlangchain_core.embeddingsrpydanticrrr rr:r&rGs$ 0*R/Y R/r: