U h\@sddlmZmZmZmZmZddlZddlmZddl m Z ddl m Z m Z ddlmZmZmZGddde ZGd d d e ZdS) )AnyDictListOptionalcastN)CallbackManagerForLLMRun)LLM) BaseModel SecretStr)convert_to_secret_strget_from_dict_or_envpre_initc@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__rboolrrrrrrA/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/llms/ai21.pyr s      rc@s\eZdZUdZdZeed<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<GdddZeeedddZeeeefdddZeeeefdddZ 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_urlc@seZdZdZdS)z AI21.ConfigZforbidN)rrrextrarrrrConfigGsr/)valuesreturncCstt|dd}||d<|S)z,Validate that api key exists in environment.r+Z AI21_API_KEY)r r )clsr0r+rrrvalidate_environmentJs  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_paramsSszAI21._default_paramscCsd|ji|jS)zGet the identifying parameters.r)rr7r5rrr_identifying_paramsbszAI21._identifying_paramscCsdS)zReturn type of llm.Zai21rr5rrr _llm_typegszAI21._llm_type)promptr, run_managerkwargsr1c Ks|jdk r|dk rtdn|jdk r.|j}n |dkr:g}|jdk rL|j}n|jdkr\d}nd}|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-rr7rr r+requestspostZget_secret_value status_coderAget) r6r:r,r;r<r-paramsresponseZoptional_detailZ response_jsonrrr_callls2       z AI21._call)NN)$rrrrrstrrr floatr"rr#r$rr%r&r'r)r*rrr+r r,rr-r/r r3propertyrr7r8r9rrMrrrrrs@        r)typingrrrrrrGZlangchain_core.callbacksrZ#langchain_core.language_models.llmsrZlangchain_core.pydantic_v1r r Zlangchain_core.utilsr r r rrrrrrs