o ZhY1@sddlZddlZddlmZmZmZmZmZmZm Z m Z ddl m Z ddl mZmZddlmZddlmZddlmZddlmZdd lmZmZmZmZdd lmZmZdd l m!Z!m"Z"m#Z#m$Z$Gd d d eZ%e ddddGdddee%Z&dS)N)Any AsyncIteratorCallableDictIteratorListMappingOptional) deprecated)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)BaseLanguageModel)LLM)GenerationChunk) PromptValue)check_package_versionget_from_dict_or_envget_pydantic_field_namespre_init)_build_model_kwargsconvert_to_secret_str) ConfigDictField SecretStrmodel_validatorc@seZdZUdZeed<dZeed<edddZe ed< edd dZ e ed < dZ e eed < dZe e ed < dZe eed < dZeed< dZe eed< dZe ed< dZe e ed<dZe eed<dZe e ed<dZe e ed<dZe ee ge fed<eedZee efed<edde dedefddZ!e"dedefd d!Z#e$de%e effd"d#Z&e$de%e effd$d%Z'd)d&e e(e de(e fd'd(Z)dS)*_AnthropicCommonNclient async_clientzclaude-2Z model_name)defaultaliasmodelZ max_tokensmax_tokens_to_sample temperaturetop_ktop_pF streamingdefault_request_timeout max_retriesanthropic_api_urlanthropic_api_key HUMAN_PROMPT AI_PROMPT count_tokens)default_factory model_kwargsbefore)modevaluesreturncCst|}t||}|SN)rr)clsr3Zall_required_field_namesr7Y/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/llms/anthropic.py build_extraEs z_AnthropicCommon.build_extracCstt|dd|d<t|dddd|d<zHddl}td d d |j|d|d|d |d d|d<|j|d|d|d |d d|d<|j|d<|j|d<|dj |d<W|St ygt dw)z?Validate that api key and python package exists in environment.r+ZANTHROPIC_API_KEYr*ZANTHROPIC_API_URLzhttps://api.anthropic.com)rrN anthropicz0.3)Z gte_versionr'r))base_urlZapi_keytimeoutr)rrr,r-r.z]Could not import anthropic python package. Please it install it with `pip install anthropic`.) rrr:r AnthropicZget_secret_valueZAsyncAnthropicr,r-r. ImportError)r6r3r:r7r7r8validate_environmentLsB         z%_AnthropicCommon.validate_environmentcCsX|j|jd}|jdur|j|d<|jdur|j|d<|jdur%|j|d<i||jS)z5Get the default parameters for calling Anthropic API.)r"r Nr#r$r%)r"r r#r$r%r0)selfdr7r7r8_default_paramsus      z _AnthropicCommon._default_paramscCsii|jS)zGet the identifying parameters.)rBr@r7r7r8_identifying_paramssz$_AnthropicCommon._identifying_paramsstopcCs2|jr|js td|durg}||jg|S)N-Please ensure the anthropic package is loaded)r,r- NameErrorextend)r@rEr7r7r8_get_anthropic_stops z$_AnthropicCommon._get_anthropic_stopr5)*__name__ __module__ __qualname__rr__annotations__rrr strr"intr#r floatr$r%r&boolr'r)r*r+rr,r-r.rdictr0rr classmethodr9rr?propertyrrBrDrrIr7r7r7r8r!sF    ($rz0.0.28z1.0z langchain_anthropic.AnthropicLLM)ZsinceZremovalZalternative_importc @sBeZdZdZedddZededefddZe de fdd Z d e de fd d Z dd e de ee de edede f ddZd ede fddZ dd e de ee de edede f ddZ dd e de ee de ededeef ddZ dd e de ee de ededeef ddZde defddZd S)r=aAnthropic large language models. To use, you should have the ``anthropic`` python package installed, and the environment variable ``ANTHROPIC_API_KEY`` set with your API key, or pass it as a named parameter to the constructor. Example: .. code-block:: python import anthropic from langchain_community.llms import Anthropic model = Anthropic(model="", anthropic_api_key="my-api-key") # Simplest invocation, automatically wrapped with HUMAN_PROMPT # and AI_PROMPT. response = model.invoke("What are the biggest risks facing humanity?") # Or if you want to use the chat mode, build a few-shot-prompt, or # put words in the Assistant's mouth, use HUMAN_PROMPT and AI_PROMPT: raw_prompt = "What are the biggest risks facing humanity?" prompt = f"{anthropic.HUMAN_PROMPT} {prompt}{anthropic.AI_PROMPT}" response = model.invoke(prompt) T)Zpopulate_by_nameZarbitrary_types_allowedr3r4cCstd|S)z,Raise warning that this class is deprecated.zpThis Anthropic LLM is deprecated. Please use `from langchain_community.chat_models import ChatAnthropic` instead)warningswarn)r6r3r7r7r8 raise_warningszAnthropic.raise_warningcCsdS)zReturn type of llm.z anthropic-llmr7rCr7r7r8 _llm_typeszAnthropic._llm_typepromptcCs\|jr|js td||jr|Std|j|\}}|dkr"|S|jd||jdS)NrFz ^\n*Human: z Sure, here you go: )r,r-rG startswithresubn)r@rYZcorrected_promptZn_subsr7r7r8 _wrap_prompts  zAnthropic._wrap_promptNrE run_managerkwargsc Kst|jrd}|jd|||d|D]}||j7}q|S||}i|j|}|jjjd|||d|}|j S)aCall out to Anthropic's completion 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 prompt = "What are the biggest risks facing humanity?" prompt = f"\n\nHuman: {prompt}\n\nAssistant:" response = model.invoke(prompt) rYrEr`rYstop_sequencesNr7) r&_streamtextrIrBr completionscreater_ completion r@rYrEr`rarjchunkparamsresponser7r7r8_calls$   zAnthropic._callcCs||Sr5)r_Z to_string)r@rYr7r7r8convert_promptszAnthropic.convert_promptc s|jr!d}|jd|||d|2z 3dHW}||j7}q6|S||}i|j|}|jjjd|||d|IdH}|j S)z;Call out to Anthropic's completion endpoint asynchronously.rbrcNrdr7) r&_astreamrgrIrBrrhrir_rjrkr7r7r8_acalls(    zAnthropic._acallcksl||}i|j|}|jjjd|||dd|D]}t|jd}|r0|j|j |d|VqdS)a\Call Anthropic completion_stream and return the resulting generator. Args: prompt: The prompt to pass into the model. stop: Optional list of stop words to use when generating. Returns: A generator representing the stream of tokens from Anthropic. Example: .. code-block:: python prompt = "Write a poem about a stream." prompt = f"\n\nHuman: {prompt}\n\nAssistant:" generator = anthropic.stream(prompt) for token in generator: yield token TrYrestreamrgrlNr7) rIrBrrhrir_rrjon_llm_new_tokenrgr@rYrEr`rarmtokenrlr7r7r8rfs    zAnthropic._streamcKs||}i|j|}|jjjd|||dd|IdH2z3dHW}t|jd}|r:|j|j |dIdH|Vq!6dS)a[Call Anthropic completion_stream and return the resulting generator. Args: prompt: The prompt to pass into the model. stop: Optional list of stop words to use when generating. Returns: A generator representing the stream of tokens from Anthropic. Example: .. code-block:: python prompt = "Write a poem about a stream." prompt = f"\n\nHuman: {prompt}\n\nAssistant:" generator = anthropic.stream(prompt) for token in generator: yield token TrsNrurvr7) rIrBrrhrir_rrjrwrgrxr7r7r8rq@s   zAnthropic._astreamrgcCs|jstd||S)zCalculate number of tokens.rF)r.rG)r@rgr7r7r8get_num_tokensds zAnthropic.get_num_tokens)NN)rJrKrL__doc__rZ model_configrrrWrTrNrXr_r rr rrorrpr rrrrrfrrqrOrzr7r7r7r8r=s   )    %  $r=)'r]rUtypingrrrrrrrr Zlangchain_core._api.deprecationr Zlangchain_core.callbacksr r Zlangchain_core.language_modelsr Z#langchain_core.language_models.llmsrZlangchain_core.outputsrZlangchain_core.prompt_valuesrZlangchain_core.utilsrrrrZlangchain_core.utils.utilsrrZpydanticrrrrrr=r7r7r7r8s&(     u