Ng `ddlZddlZddlmZmZddlmZddlmZm Z GddeeZ dS)N)AnyList) Embeddings) BaseModel ConfigDictceZdZUdZdZeed< ddededeffd Ze d Z d e ed e e e fd Z d ed e e fdZxZS)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 tjd i| ddlm}ddlm}n"#t $r}t d|d}~wwxYw tjtj d<tjtj d<| |n"#t$r}td |d}~wwxYw t|tr|||_dSt||r ||_dS|||_dS#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 rrrexc __class__s g/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/embeddings/johnsnowlabs.pyrzJohnSnowLabsEmbeddings.__init__s ""6"""  ( ( ( ( ( ( 5 5 5 5 5 5 5   E   G+.>BJ' (25.BJ. / IIoI 6 6 6 6 G G G<==3 F G >%%% 4 XXe__ E;// 4"  __U33  > > >344# = >sQ ) AAA AB B4B//B48/D)DD D=(D88D=forbid)extratextsreturnc|j|d}d}|jD]}d|vr|} d||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. document) output_levelN embeddingc6g|]}|Sr)tolist).0vecs r' z:JohnSnowLabsEmbeddings.embed_documents..Ps === ===)r predictcolumnsr1)r$r*dfemb_colcs r'embed_documentsz&JohnSnowLabsEmbeddings.embed_documentsAsiZ  J  ? ?  Aa==7 (:(:(<(<====r5textc:||gdS)zCompute query embeddings using a JohnSnowLabs transformer model. Args: text: The text to embed. Returns: Embeddings for the text. r)r;)r$r<s r' embed_queryz"JohnSnowLabsEmbeddings.embed_queryRs##TF++A..r5)r r )__name__ __module__ __qualname____doc__r r__annotations__r!rr model_configrfloatr;r> __classcell__)r&s@r'r r s  'E3&&&+$#>#>#>#> #>#>#>#>#>#>J:L>T#Y>4U 3D>>>>" / /U / / / / / / / /r5r ) rrtypingrrlangchain_core.embeddingsrpydanticrrr rr5r'rJs 000000********R/R/R/R/R/Y R/R/R/R/R/r5