a bg!@s4ddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z m Z ddlmZddlmZddlmZmZmZdd lmZmZmZdd lmZmZmZmZmZdd l m!Z!e"e#Z$d d dddZ%ddddddZ&ddddddZ'e ddddGdddeZ(e ddddGd ddee(Z)dS)!) 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_tokensintzCallable[[Any], Any]) max_retriesreturncCsVddl}t|drt|jjntt}d}d}tdt|td||d|t t t j dS) Nrerror T) multiplierminmax)reraisestopwaitr before_sleep) coherehasattrrrZ CohereError ExceptionrrrrloggerloggingWARNING)rr'Zretry_conditionsZ min_secondsZ max_secondsr-m/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/llms/cohere.py_create_retry_decorators  r/Coherer)llmkwargsrc s0tj}|dddfdd }|fi|S)*Use tenacity to retry the completion call.rr2rcsjjfi|SN)clientgenerater2r1r-r._completion_with_retry7sz5completion_with_retry.._completion_with_retryr/rr1r2Zretry_decoratorr:r-r9r.completion_with_retry3s r=c s0tj}|dddfdd }|fi|S)r3rr4csjjfi|IdHSr5) async_clientr7r8r9r-r.r:Bsz6acompletion_with_retry.._completion_with_retryr;r<r-r9r.acompletion_with_retry>s r?z0.0.30z1.0zlangchain_cohere.BaseCohere)ZsinceZremovalZalternative_importc@seZdZUdZdZded<dZded<eddZded<d Z d ed <dZ d ed <dZ ded<eddZ ded<dZ ded<edddddZdS) BaseCoherezBase class for Cohere models.Nrr6r>)default Optional[str]modelg?float temperaturezOptional[SecretStr]cohere_api_keyOptional[List[str]]r$Fbool streamingZ langchainstr user_agentr)valuesrcCs|z ddl}Wnty&tdYnR0tt|dd|d<|d}|j|d|d|d<|j|d|d|d <|S) z?Validate that api key and python package exists in environment.rNzTCould not import cohere python package. Please install it with `pip install cohere`.rFCOHERE_API_KEYrK)Zapi_key client_namer6r>)r' ImportErrorr rZClientZget_secret_valueZ AsyncClient)clsrLr'rNr-r-r.validate_environmentbs&        zBaseCohere.validate_environment)__name__ __module__ __qualname____doc__r6__annotations__r>rrCrErFr$rIrKrrQr-r-r-r.r@Is       r@z0.1.14zlangchain_cohere.Coherec@s eZdZUdZdZded<dZded<dZded<d Zd ed <d Z d ed <d Z ded<dZ ded<e ddZ eddddZeddddZeddddZeddddZd d!d"d#d$d%Zd!d dd&d'd(Zd0dd d)d!dd*d+d,Zd1dd d-d!dd*d.d/Zd S)2r0aCohere 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") r max_tokensrkrpgrDfrequency_penaltypresence_penaltyNrBtruncaterrZforbid)extrazDict[str, Any])rcCs"|j|j|j|j|j|j|jdS)z2Get the default parameters for calling Cohere API.rXrErYrZr[r\r]r_selfr-r-r._default_paramsszCohere._default_paramszDict[str, str]cCsddiS)NrFrMr-r`r-r-r. lc_secretsszCohere.lc_secretscCsid|ji|jS)zGet the identifying parameters.rC)rCrbr`r-r-r._identifying_paramsszCohere._identifying_paramsrJcCsdS)zReturn type of llm.r'r-r`r-r-r. _llm_typeszCohere._llm_typerGrdict)r$r2rcKsL|j}|jdur"|dur"tdn|jdur8|j|d<n||d<i||S)Nz2`stop` found in both the input and default params.stop_sequences)rbr$ ValueError)rar$r2paramsr-r-r._invocation_paramss   zCohere._invocation_params)responser$rcCs|jdj}|rt||}|S)Nr)Z generationstextr)rarkr$rlr-r-r._process_responses  zCohere._process_responsez"Optional[CallbackManagerForLLMRun])promptr$ run_managerr2rcKs@|j|fi|}t|f|j|d|}|d}|||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.") rCrnrg)rjr=rCgetrmrarnr$ror2rirk_stopr-r-r._calls z Cohere._callz'Optional[AsyncCallbackManagerForLLMRun]csF|j|fi|}t|f|j|d|IdH}|d}|||S)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.") rpNrg)rjr?rCrqrmrrr-r-r._acalls  z Cohere._acall)NN)NN)rRrSrTrUrXrVrYrZr[r\r]rrZ model_configpropertyrbrcrdrerjrmrtrur-r-r-r.r0|s6           )* __future__rr+typingrrrrrZlangchain_core._api.deprecationrZlangchain_core.callbacksr r Z#langchain_core.language_models.llmsr Z langchain_core.load.serializabler Zlangchain_core.utilsr rrZpydanticrrrtenacityrrrrrZlangchain_community.llms.utilsr getLoggerrRr*r/r=r?r@r0r-r-r-r.s(        0