7|hrddlmZmZmZmZmZddlmZddlm Z ddl m Z m Z m Z ddlmZmZGdde Zy) )AnyDictListMappingOptional)CallbackManagerForLLMRun)LLM)convert_to_secret_strget_from_dict_or_envpre_init) ConfigDict SecretStrc eZdZUdZdZeed<dZeed< dZ e ed< dZ eed < d Z e ed < d Zeed < dZe ed< dZe ed< dZe ed< gZeeed< dZe ed< dZeed< dZe ed< dZeed< dZeed< dZeeed<edZede de fdZ!e"de#eeffd Z$e"de#eeffd!Z%e"defd"Z& d(d#ed$eeed%ee'd&edef d'Z(y))NLPCloudaaNLPCloud large language models. To use, you should have the ``nlpcloud`` python package installed, and the environment variable ``NLPCLOUD_API_KEY`` set with your API key. Example: .. code-block:: python from langchain_community.llms import NLPCloud nlpcloud = NLPCloud(model="finetuned-gpt-neox-20b") Nclientzfinetuned-gpt-neox-20b model_nameTgpuenlanggffffff? temperature max_lengthlength_no_input remove_inputremove_end_sequence bad_wordsg?top_p2top_krepetition_penalty num_beamsnum_return_sequencesnlpcloud_api_keyforbid)extravaluesreturnctt|dd|d< ddl}|j|d|dj |d|d|d <|S#t $r t d wxYw) z?Validate that api key and python package exists in environment.r$NLPCLOUD_API_KEYrNrrr)rrrzXCould not import nlpcloud python package. Please install it with `pip install nlpcloud`.)r r nlpcloudClientget_secret_value ImportError)clsr'r+s `/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/llms/nlpcloud.pyvalidate_environmentzNLPCloud.validate_environment:s&; );=O P& !"  '|$)*;;=5MF^ / F8   A  s 6AA(c |j|j|j|j|j|j |j |j|j|j|jd S)z4Get the default parameters for calling NLPCloud API. rrrrrrrrr r"r#r3selfs r0_default_paramszNLPCloud._default_paramsPsg ++//#33 --#'#;#;ZZZZ"&"9"9$($=$=  crid|jid|jid|ji|jS)zGet the identifying parameters.rrr)rrrr6r4s r0_identifying_paramszNLPCloud._identifying_paramsasQ T__- dhh tyy! ""   r7cy)zReturn type of llm.r+r4s r0 _llm_typezNLPCloud._llm_typeksr7promptstop run_managerkwargsc |rt|dkDr td|rt|dk(r|d}nd}i|j|}|jj|fd|i|}|dS)alCall out to NLPCloud's create endpoint. Args: prompt: The prompt to pass into the model. stop: Not supported by this interface (pass in init method) Returns: The string generated by the model. Example: .. code-block:: python response = nlpcloud("Tell me a joke.") r!zXNLPCloud only supports a single stop sequence per generation.Pass in a list of length 1.rN end_sequencegenerated_text)len ValueErrorr6r generation)r5r=r>r?r@rBparamsresponses r0_callzNLPCloud._callps* CIM. c$i1n7LL3D((3F3)4;;))&V|VvV())r7)NN))__name__ __module__ __qualname____doc__rr__annotations__rstrrboolrrfloatrintrrrrrrrr r"r#r$rrr model_configr rr1propertyrr6r9r<rrIr;r7r0rr s FC.J.C%D#.K+JE OT SL$- $$:ItCy5E5DE3OO ##:Is* !#!;,0hy)0L$4* c!2   WS#X%6  3%):> * *tCy! *67 *  * *r7rN)typingrrrrrlangchain_core.callbacksr#langchain_core.language_models.llmsr langchain_core.utilsr r r pydanticr rrr;r7r0rZs(55=3VV*G*sG*r7