7|h!ZddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z m Z ddlmZddlmZddlmZmZmZdd lmZmZmZdd lmZmZmZmZmZdd l m!Z!ejDe#Z$dd Z%ddZ&ddZ'e dddGddeZ(e dddGdd ee(Z)y)) annotationsN)AnyCallableDictListOptional) deprecated)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)LLM) Serializable)convert_to_secret_strget_from_dict_or_envpre_init) ConfigDictField SecretStr)before_sleep_logretryretry_if_exception_typestop_after_attemptwait_exponential)enforce_stop_tokensc ddl}t|drt|jjntt }d}d}t dt|td|||tttjS) Nrerror T) multiplierminmax)reraisestopwaitr before_sleep) coherehasattrrr CohereError ExceptionrrrrloggerloggingWARNING) max_retriesr&retry_conditions min_seconds max_secondss ^/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/llms/cohere.py_create_retry_decoratorr2su 67 #  8 89 $Y / KK   ,   M%fgoo>  Coherec Ttj}|dfd }|di|S)*Use tenacity to retry the completion call.c<jjdi|SN)clientgeneratekwargsllms r1_completion_with_retryz5completion_with_retry.._completion_with_retry7s"szz"",V,,r3r=rreturnrr9r2r-r>r=retry_decoratorr?s` r1completion_with_retryrE3s3-coo>O-- " +F ++r3c Ttj}|dfd }|di|S)r6cXKjjdi|d{S7wr8) async_clientr;r<s r1r?z6acompletion_with_retry.._completion_with_retryBs).S%%..88888s *(*r@r9rBrCs` r1acompletion_with_retryrI>s3-coo>O99 " +F ++r3z0.0.30z1.0zlangchain_cohere.BaseCohere)sinceremovalalternative_importceZdZUdZdZded<dZded<edZded< d Z d ed < dZ d ed < dZ ded<edZ ded< dZ ded< eddZy) BaseCoherezBase class for Cohere models.Nrr:rH)default Optional[str]modelg?float temperaturezOptional[SecretStr]cohere_api_keyOptional[List[str]]r#Fbool streaming langchainstr user_agentc ddl}tt|dd|d<|d}|j|dj ||d<|j |dj ||d<|S#t $r t d wxYw) z?Validate that api key and python package exists in environment.rNrTCOHERE_API_KEYrZ)api_key client_namer:rHzTCould not import cohere python package. Please install it with `pip install cohere`.)r&rrClientget_secret_value AsyncClient ImportError)clsvaluesr&r^s r1validate_environmentzBaseCohere.validate_environmentbs  (=$V-=?OP(F# $!.K%}}/0AAC' - F8 &,%7%7/0AAC'&8&F> " % ?  s A44B )rdrrAr)__name__ __module__ __qualname____doc__r:__annotations__rHrrQrSrTr#rWrZrrer9r3r1rNrNIs(FCL# .E=.KQ*.N'.V $D $E*It*(!J!< r3rNz0.1.14zlangchain_cohere.Coherec,eZdZUdZdZded< dZded< dZded< d Zd ed < d Z d ed < d Z ded< dZ ded< e dZ eddZeddZeddZeddZddZd dZ d! d"dZ d! d#dZy )$r4aCohere large language models. To use, you should have the ``cohere`` python package installed, and the environment variable ``COHERE_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 Cohere cohere = Cohere(model="gptd-instruct-tft", cohere_api_key="my-api-key") int max_tokensrkrpgrRfrequency_penaltypresence_penaltyNrPtruncaterr-forbid)extrac|j|j|j|j|j|j |j dS)z2Get the default parameters for calling Cohere API.rnrSrorprqrrrsrwselfs r1_default_paramszCohere._default_paramssE//++!%!7!7 $ 5 5   r3c ddiS)NrTr\r9rxs r1 lc_secretszCohere.lc_secretss "233r3c:id|ji|jS)zGet the identifying parameters.rQ)rQrzrxs r1_identifying_paramszCohere._identifying_paramss$A7DJJ'@4+?+?@@r3cy)zReturn type of llm.r&r9rxs r1 _llm_typezCohere._llm_typesr3c |j}|j | td|j|j|d<n||d<i||S)Nz2`stop` found in both the input and default params.stop_sequences)rzr# ValueError)ryr#r=paramss r1_invocation_paramszCohere._invocation_paramss\%% 99 T%5QR R YY "'+yyF# $'+F# $#&#F##r3cT|jdj}|r t||}|S)Nr) generationstextr)ryresponser#rs r1_process_responsezCohere._process_responses.##A&++ &tT2D r3c |j|fi|}t|f|j|d|}|jd}|j ||S)ahCall out to Cohere's generate 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 = cohere("Tell me a joke.") rQpromptr)rrErQgetrryrr# run_managerr=rr_stops r1_callz Cohere._callsb*)((88(   6 5;  +,%%h66r3cK|j|fi|}t|f|j|d|d{}|jd}|j ||S7'w)atAsync call out to Cohere's generate 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 = await cohere("Tell me a joke.") rNr)rrIrQrrrs r1_acallz Cohere._acallsp*)((88/   6 5;   +,%%h66  s0AA(A)rAzDict[str, Any])rAzDict[str, str])rArY)r#rUr=rrAdict)rrr#rUrArY)NN) rrYr#rUrz"Optional[CallbackManagerForLLMRun]r=rrArY) rrYr#rUrz'Optional[AsyncCallbackManagerForLLMRun]r=rrArY)rfrgrhrirnrjrorprqrrrsr-r model_configpropertyrzr|r~rrrrrr9r3r1r4r4|sK JA AsJ@ AsJD"u"L!e!5"Hm"0K<L    44AA$%):> 77"78 7  7 7>%)?C 77"7= 7  7 7r3)r-rmrAzCallable[[Any], Any])r>r4r=rrAr)* __future__rr+typingrrrrrlangchain_core._api.deprecationr langchain_core.callbacksr r #langchain_core.language_models.llmsr langchain_core.load.serializabler langchain_core.utilsrrrpydanticrrrtenacityrrrrrlangchain_community.llms.utilsr getLoggerrfr*r2rErIrNr4r9r3r1rs"66649VV11?   8 $.,, E6S---` (E>WXL7S*L7YL7r3