Ng!|ddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z m Z ddlmZddlmZddlmZmZmZdd lmZmZmZdd lmZmZmZmZmZdd l m!Z!ej"e#Z$d dZ%d!dZ&d!dZ'e dddGddeZ(e dddGddee(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]c (ddl}t|drt|jjntt }d}d}t dt|td|||tttj S) Nrerror T) multiplierminmax)reraisestopwaitr before_sleep) coherehasattrrr CohereError ExceptionrrrrloggerloggingWARNING)rr*retry_conditions min_seconds max_secondss [/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/llms/cohere.py_create_retry_decoratorr5sMMM 67 # # 0 8999 $Y / / KK   , ,   M M M%fgo>>    llmCoherekwargsrc Xtj}|dfd }|di|S)*Use tenacity to retry the completion call.r9rrc(jjdi|SN)clientgenerater9r7s r4_completion_with_retryz5completion_with_retry.._completion_with_retry7s"sz",,V,,,r6r9rrrr>r5rr7r9retry_decoratorrBs` r4completion_with_retryrG3sS-co>>O-----_- " ! + +F + ++r6c Xtj}|dfd }|di|S)r;r9rrc8Kjjdi|d{VSr=) async_clientr@rAs r4rBz6acompletion_with_retry.._completion_with_retryBs5.S%.88888888888r6rCr>rDrEs` r4acompletion_with_retryrK>sS-co>>O99999_9 " ! + +F + ++r6z0.0.30z1.0zlangchain_cohere.BaseCohere)sinceremovalalternative_importceZdZUdZdZded<dZded<edZded< d Z d ed < dZ d ed < dZ ded<edZ ded< dZ ded< eddZdS) BaseCoherezBase class for Cohere models.Nrr?rJ)default Optional[str]modelg?float temperaturezOptional[SecretStr]cohere_api_keyOptional[List[str]]r'Fbool streaming langchainstr user_agentvaluesrrcf ddl}tt|dd|d<|d}||d||d<||d||d<n#t $rt d wxYw|S) z?Validate that api key and python package exists in environment.rNrVCOHERE_API_KEYr\)api_key client_namer?rJzTCould not import cohere python package. Please install it with `pip install cohere`.)r*rrClientget_secret_value AsyncClient ImportError)clsr]r*ras r4validate_environmentzBaseCohere.validate_environmentbs  MMM(=$V-=?OPP((F# $!.K%}}/0AACC' -  F8 &,%7%7/0AACC'&8&&F> " "   ?  $ s BB.)r]rrr)__name__ __module__ __qualname____doc__r?__annotations__rJrrSrUrVr'rYr\rrgr>r6r4rPrPIs('FL 5...E....KQ*.N....V $D$$$$eE***I****(!J!!!!< Xr6rPz0.1.14zlangchain_cohere.Coherec(eZdZUdZdZded< dZded< dZded< d Zd ed < d Z d ed < d Z ded< dZ ded< e dZ ed*dZed+dZed*dZed,dZd-d!Zd.d#Z d/d0d'Z d/d1d)Zd S)2r8aCohere 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"pgrTfrequency_penaltypresence_penaltyNrRtruncater!rforbid)extrarDict[str, Any]c\|j|j|j|j|j|j|jdS)z2Get the default parameters for calling Cohere API.rorUrprqrrrsrtryselfs r4_default_paramszCohere._default_paramss9/+!%!7 $ 5    r6Dict[str, str]c ddiS)NrVr_r>rzs r4 lc_secretszCohere.lc_secretss "233r6c&id|ji|jS)zGet the identifying parameters.rS)rSr|rzs r4_identifying_paramszCohere._identifying_paramssA7DJ'@4+?@@r6r[cdS)zReturn type of llm.r*r>rzs r4 _llm_typezCohere._llm_types xr6r'rWr9rdictc z|j}|j|td|j |j|d<n||d<i||S)Nz2`stop` found in both the input and default params.stop_sequences)r|r' ValueError)r{r'r9paramss r4_invocation_paramszCohere._invocation_paramssY% 9 T%5QRR R Y "'+yF# $ $'+F# $#&#F##r6responsecN|jdj}|rt||}|S)Nr) generationstextr)r{rr'rs r4_process_responsezCohere._process_responses0#A&+  3&tT22D r6prompt run_manager"Optional[CallbackManagerForLLMRun]c |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.") rSrr)rrGrSgetrr{rr'rr9rr_stops r4_callz Cohere._callss*)(8888(   6  5;   +,,%%h666r6'Optional[AsyncCallbackManagerForLLMRun]cK|j|fi|}t|f|j|d|d{V}|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.") rNr)rrKrSrrrs r4_acallz Cohere._acalls*)(8888/   6  5;         +,,%%h666r6)rrw)rr})rr[)r'rWr9rrr)rrr'rWrr[)NN) rr[r'rWrrr9rrr[) rr[r'rWrrr9rrr[)rhrirjrkrorlrprqrrrsrtrr model_configpropertyr|rrrrrrrr>r6r4r8r8|s  JA AJJJJ@ AJJJJD"""""L!!!!!5"H""""0K<:L    X  444X4AAAXAX$$$$%):> 77777>%)?C 7777777r6)rrrr)r7r8r9rrr)* __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 getLoggerrhr.r5rGrKrPr8r>r6r4rsA""""""66666666666666666666433333999999VVVVVVVVVV1111111111?>>>>>  8 $ $.,,,,,,,,  E6S--------` (E>WXXXL7L7L7L7L7S*L7L7YXL7L7L7r6