a bgh@sddlmZmZmZmZmZddlZddlmZddl m Z ddl m Z m Z mZddlmZmZmZGdddeZGd d d e ZdS) )AnyDictListOptionalcastN)CallbackManagerForLLMRun)LLM)convert_to_secret_strget_from_dict_or_envpre_init) BaseModel ConfigDict SecretStrc@sZeZdZUdZdZeed<dZeed<dZ eed<dZ eed<dZ eed<dZ eed <d S) AI21PenaltyDataz!Parameters for AI21 penalty data.rscaleTapplyToWhitespacesapplyToPunctuationsapplyToNumbersapplyToStopwords applyToEmojisN) __name__ __module__ __qualname____doc__rint__annotations__rboolrrrrrrk/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/llms/ai21.pyr s      rc@sXeZdZUdZdZeed<dZeed<dZ e ed<dZ e ed <d Z eed <e Ze ed <e Ze ed <e Ze ed<dZe ed<dZeeeefed<dZeeed<dZeeeed<dZeeed<eddZeeedddZeeeefdddZ eeeefdddZ!eedd d!Z"d%eeeeee#eed"d#d$Z$dS)&AI21agAI21 large language models. To use, you should have the environment variable ``AI21_API_KEY`` set with your API key or pass it as a named parameter to the constructor. Example: .. code-block:: python from langchain_community.llms import AI21 ai21 = AI21(ai21_api_key="my-api-key", model="j2-jumbo-instruct") zj2-jumbo-instructmodelgffffff? temperature maxTokensr minTokensg?topPpresencePenalty countPenaltyfrequencyPenalty numResultsN logitBias ai21_api_keystopbase_urlZforbid)extra)valuesreturncCstt|dd}||d<|S)z,Validate that api key exists in environment.r,Z AI21_API_KEY)r r )clsr0r,rrrvalidate_environmentKs  zAI21.validate_environment)r1c Cs6|j|j|j|j|j|j|j|j|j d S)z0Get the default parameters for calling AI21 API.) r!r#r$r%r&r'r(r*r+) r!r#r$r%r&dictr'r(r*r+selfrrr_default_paramsTszAI21._default_paramscCsid|ji|jS)zGet the identifying parameters.r )r r7r5rrr_identifying_paramscszAI21._identifying_paramscCsdS)zReturn type of llm.Zai21rr5rrr _llm_typehszAI21._llm_type)promptr- run_managerkwargsr1c Ks|jdur|durtdn|jdur.|j}n |dur:g}|jdurL|j}n|jdvr\d}nd}i|j|}tt|j|_tj |d|jddd |j i||d |d }|j d kr| d }td|j d|| } | ddddS)adCall out to AI21's complete endpoint. Args: prompt: The prompt to pass into the model. stop: Optional list of stop words to use when generating. Returns: The string generated by the model. Example: .. code-block:: python response = ai21("Tell me a joke.") Nz2`stop` found in both the input and default params.)zj1-grande-instructz+https://api.ai21.com/studio/v1/experimentalzhttps://api.ai21.com/studio/v1/z /complete AuthorizationzBearer )r:Z stopSequences)urlheadersjsonerrorz,AI21 /complete call failed with status code z . Details: Z completionsrdatatext)r- ValueErrorr.r r7rrr,requestspostZget_secret_value status_coderAget) r6r:r-r;r<r.paramsresponseZoptional_detailZ response_jsonrrr_callms6       z AI21._call)NN)%rrrrr strrr!floatr#rr$r%rr&r'r(r*r+rrr,rr-rr.r Z model_configr r3propertyrr7r8r9rrMrrrrrsD        r)typingrrrrrrGZlangchain_core.callbacksrZ#langchain_core.language_models.llmsrZlangchain_core.utilsr r r Zpydanticr r rrrrrrrs