U h@slddlmZmZmZmZmZddlZddlmZddl m Z ddl m Z m Z ddlmZGddde ZdS) )AnyDictListMappingOptionalN)CallbackManagerForLLMRun)LLM)get_from_dict_or_envpre_init)enforce_stop_tokensc@szeZdZUdZdZeeed<dZeed<dZ ee ed<dZ ee ed<dZ ee ed<dZee ed <dZeeeed <dZee ed <dZee ed <dZee ed <dZeed<dZee ed<dZeeed<dZeeed<GdddZeeedddZeeeefdddZ eeeefdddZ!eedddZ"d"eeeeee#eedd d!Z$dS)#WriteraXWriter large language models. To use, you should have the environment variable ``WRITER_API_KEY`` and ``WRITER_ORG_ID`` set with your API key and organization ID respectively. Example: .. code-block:: python from langchain_community.llms import Writer writer = Writer(model_id="palmyra-base") N writer_org_idzpalmyra-instructmodel_id min_tokens max_tokens temperaturetop_pstoppresence_penaltyrepetition_penaltybest_ofFlogprobsnwriter_api_keybase_urlc@seZdZdZdS)z Writer.ConfigZforbidN)__name__ __module__ __qualname__extrarrC/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/llms/writer.pyConfigBsr!)valuesreturncCs,t|dd}||d<t|dd}||d<|S)z?Validate that api key and organization id exist in environment.rZWRITER_API_KEYr Z WRITER_ORG_ID)r )clsr"rr rrr validate_environmentEs zWriter.validate_environment)r#c Cs.|j|j|j|j|j|j|j|j|j|j d S)z2Get the default parameters for calling Writer API.) Z minTokensZ maxTokensrZtopPrZpresencePenaltyZrepetitionPenaltyZbestOfrr) rrrrrrrrrrselfrrr _default_paramsSszWriter._default_paramscCs|j|jd|jS)zGet the identifying parameters.)rr )rr r(r&rrr _identifying_paramscs zWriter._identifying_paramscCsdS)zReturn type of llm.writerrr&rrr _llm_typekszWriter._llm_type)promptr run_managerkwargsr#c Ksr|jdk r|j}nd|jd|jd}|j|}tj||jdddd|i|d}|j}|dk rnt||}|S) akCall out to Writer's completions 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 = Writer("Tell me a joke.") Nz3https://enterprise-api.writer.com/llm/organization/z/model/z /completionszapplication/json) Authorizationz Content-TypeAcceptr,)urlheadersjson) rr rr(requestspostrtextr ) r'r,rr-r.rparamsresponser6rrr _callps"     z Writer._call)NN)%rrr__doc__r rstr__annotations__rrintrrfloatrrrrrrrboolrrrr!r rr%propertyrrr(r)r+rr9rrrr r sB      r )typingrrrrrr4Zlangchain_core.callbacksrZ#langchain_core.language_models.llmsrZlangchain_core.utilsr r Zlangchain_community.llms.utilsr r rrrr s