NgzddlmZmZmZmZmZddlmZddlm Z ddl m Z m Z m Z ddlmZmZGdde ZdS) )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(dS))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}||d|d|d|d|d <n#t $rt d wxYw|S) 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/ai-engine/env/lib/python3.11/site-packages/langchain_community/llms/nlpcloud.pyvalidate_environmentzNLPCloud.validate_environment:s&; );=O P P& & !"  OOO'|$)*;;==5MF^ /  F8      A    s A A..Bc |j|j|j|j|j|j|j|j|j|j |j d S)z4Get the default parameters for calling NLPCloud API. rrrrrrrrr r"r#r3selfs r0_default_paramszNLPCloud._default_paramsPsP +/#3 -#'#;ZZ"&"9$($=   cJid|jid|jid|ji|jS)zGet the identifying parameters.rrr)rrrr6r4s r0_identifying_paramszNLPCloud._identifying_paramsasH T_- dh ty! "   r7cdS)zReturn type of llm.r+r4s r0 _llm_typezNLPCloud._llm_typeks zr7promptstop run_managerkwargsc |r"t|dkrtd|rt|dkr |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*  CIIMM.  c$ii1nn7LLL3D(3F3)4;)&VV|VvVV())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$- $$$$:ItCy5E5DE3OOOO ####:Is* !#!!!;,0hy)000:L$4X* c!2   X  WS#X%6   X 3X%):> * * *tCy! *67 *  * * * * * * *r7rN)typingrrrrrlangchain_core.callbacksr#langchain_core.language_models.llmsr langchain_core.utilsr r r pydanticr rrr;r7r0rZs55555555555555======333333VVVVVVVVVV********G*G*G*G*G*sG*G*G*G*G*r7