o Zh!@sddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z m Z ddlmZddlmZddlmZmZmZdd lmZmZmZdd lmZmZmZmZmZdd l m!Z!e"e#Z$d#ddZ%d$ddZ&d$ddZ'e ddddGdddeZ(e d dd!dGd"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_tokens max_retriesintreturnCallable[[Any], Any]cCsVddl}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.V/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/llms/cohere.py_create_retry_decorators  r0llmCoherekwargsrc *tj}|dfdd }|di|S) *Use tenacity to retry the completion call.r3rrcsjjdi|SNr.)clientgenerater3r1r.r/_completion_with_retry7sz5completion_with_retry.._completion_with_retryNr3rrrr.r0rr1r3Zretry_decoratorr;r.r:r/completion_with_retry3 r?c r4) r5r3rrcsjjdi|IdHSr6) async_clientr8r9r:r.r/r;Bsz6acompletion_with_retry.._completion_with_retryNr<r.r=r>r.r:r/acompletion_with_retry>r@rBz0.0.30z1.0zlangchain_cohere.BaseCohere)ZsinceZremovalZalternative_importc@seZdZUdZdZded<dZded<eddZded< d Z d ed < dZ d ed < dZ ded<eddZ ded< dZ ded< edddZdS) BaseCoherezBase class for Cohere models.Nrr7rA)default Optional[str]modelg?float temperaturezOptional[SecretStr]cohere_api_keyOptional[List[str]]r%Fbool streamingZ langchainstr user_agentvaluesrrcCsxzddl}Wn tytdwtt|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`.rICOHERE_API_KEYrN)Zapi_key client_namer7rA)r( ImportErrorr rZClientZget_secret_valueZ AsyncClient)clsrOr(rQr.r.r/validate_environmentbs(       zBaseCohere.validate_environment)rOrrr)__name__ __module__ __qualname____doc__r7__annotations__rArrFrHrIr%rLrNrrTr.r.r.r/rCIs"       rCz0.1.14zlangchain_cohere.Coherec@seZdZUdZdZded< dZded< dZded< d Zd ed < d Z d ed < d Z ded< dZ ded< e ddZ ed2ddZed3ddZed2ddZed4ddZd5d%d&Zd6d(d)Z d7d8d-d.Z d7d9d0d1Zd S):r2aCohere 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_tokensrkr pgrGfrequency_penaltypresence_penaltyNrEtruncaterrZforbid)extrarDict[str, Any]cCs"|j|j|j|j|j|j|jdS)z2Get the default parameters for calling Cohere API.r[rHr\r]r^r_r`rcselfr.r.r/_default_paramsszCohere._default_paramsDict[str, str]cCsddiS)NrIrPr.rdr.r.r/ lc_secretsszCohere.lc_secretscCsid|ji|jS)zGet the identifying parameters.rF)rFrfrdr.r.r/_identifying_paramsszCohere._identifying_paramsrMcCsdS)zReturn type of llm.r(r.rdr.r.r/ _llm_typeszCohere._llm_typer%rJr3rdictcKsJ|j}|jdur|durtd|jdur|j|d<n||d<i||S)Nz2`stop` found in both the input and default params.stop_sequences)rfr% ValueError)rer%r3paramsr.r.r/_invocation_paramss   zCohere._invocation_paramsresponsecCs|jdj}|r t||}|S)Nr)Z generationstextr)rerpr%rqr.r.r/_process_responses  zCohere._process_responseprompt run_manager"Optional[CallbackManagerForLLMRun]cKs@|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.") rFrsrl)ror?rFgetrrrersr%rtr3rnrp_stopr.r.r/_calls  z Cohere._call'Optional[AsyncCallbackManagerForLLMRun]csH|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.") rvNrl)rorBrFrwrrrxr.r.r/_acalls  z Cohere._acall)rrb)rrg)rrM)r%rJr3rrrk)rprr%rJrrM)NN) rsrMr%rJrtrur3rrrM) rsrMr%rJrtr{r3rrrM)rUrVrWrXr[rYr\r]r^r_r`rrZ model_configpropertyrfrhrirjrorrrzr|r.r.r.r/r2|sF                )rrrr)r1r2r3rrr)* __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 getLoggerrUr+r0r?rBrCr2r.r.r.r/s*           0