U h1@sddlZddlZddlmZmZmZmZmZmZm Z m Z ddl m Z ddl mZmZddlmZddlmZddlmZddlmZdd lmZmZmZdd lmZmZmZm Z dd l!m"Z"m#Z#Gd d d eZ$e ddddGdddee$Z%dS)N)Any AsyncIteratorCallableDictIteratorListMappingOptional) deprecated)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)BaseLanguageModel)LLM)GenerationChunk) PromptValue)Field SecretStrroot_validator)check_package_versionget_from_dict_or_envget_pydantic_field_namespre_init)build_extra_kwargsconvert_to_secret_strc@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<eddeedddZ e!eeddd Z"e#e$e efd!d"d#Z%e#e$e efd!d$d%Z&d)e e'e e'e d&d'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_kwargsT)prevaluesreturncCs(|di}t|}t||||d<|S)Nr/)getrr)clsr2extraZall_required_field_namesr7F/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/llms/anthropic.py build_extraEs z_AnthropicCommon.build_extracCstt|dd|d<t|dddd|d<zddl}td d d |j|d|d|d |d d|d<|j|d|d|d |d d|d<|j|d<|j|d<|dj |d<Wnt k rt dYnX|S)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)r5r2r:r7r7r8validate_environmentNs@          z%_AnthropicCommon.validate_environmentr3cCsT|j|jd}|jdk r"|j|d<|jdk r6|j|d<|jdk rJ|j|d<||jS)z5Get the default parameters for calling Anthropic API.)r!rNr"r#r$)r!rr"r#r$r/)selfdr7r7r8_default_paramsws      z _AnthropicCommon._default_paramscCs i|jS)zGet the identifying parameters.)rCrAr7r7r8_identifying_paramssz$_AnthropicCommon._identifying_params)stopr3cCs2|jr |jstd|dkr g}||jg|S)N-Please ensure the anthropic package is loaded)r+r, NameErrorextend)rArFr7r7r8_get_anthropic_stops  z$_AnthropicCommon._get_anthropic_stop)N))__name__ __module__ __qualname__rr__annotations__rrrstrr!intr"r floatr#r$r%boolr&r(r)r*rr+r,r-rdictr/rrr9rr?propertyrrCrErrJr7r7r7r8r!s2     (rz0.0.28z1.0z langchain_anthropic.AnthropicLLM)ZsinceZremovalZalternative_importc@s eZdZdZGdddZeeedddZee ddd Z e e d d d Z de e e e e eee dddZee d ddZde e e e e eee dddZde e e e e eeeedddZde e e e e eeeedddZe edddZd 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) c@seZdZdZdZdS)zAnthropic.ConfigTN)rKrLrMZallow_population_by_field_nameZarbitrary_types_allowedr7r7r7r8ConfigsrUr1cCstd|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)r5r2r7r7r8 raise_warningszAnthropic.raise_warningr@cCsdS)zReturn type of llm.z anthropic-llmr7rDr7r7r8 _llm_typeszAnthropic._llm_type)promptr3cCs\|jr |jstd||jr$|Std|j|\}}|dkrD|S|jd||jdS)NrGz ^\n*Human: z Sure, here you go: )r+r,rH startswithresubn)rArZZcorrected_promptZn_subsr7r7r8 _wrap_prompts  zAnthropic._wrap_promptN)rZrF run_managerkwargsr3c Ksp|jr6d}|jf|||d|D]}||j7}q"|S||}|j|}|jjjf|||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) rZrFrarZstop_sequences) r%_streamtextrJrCr completionscreater` completion rArZrFrarbrkchunkparamsresponser7r7r8_calls(     zAnthropic._callcCs||S)N)r`Z to_string)rArZr7r7r8convert_promptszAnthropic.convert_promptc s|jr@d}|jf|||d|2z3dHW}||j7}q"6|S||}|j|}|jjjf|||d|IdH}|j S)z;Call out to Anthropic's completion endpoint asynchronously.rcrdNre) r%_astreamrhrJrCrrirjr`rkrlr7r7r8_acalls(    zAnthropic._acallcksf||}|j|}|jjjf|||dd|D]*}t|jd}|rZ|j|j |d|Vq6dS)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 TrZrfstreamrhrmN) rJrCrrirjr`rrkon_llm_new_tokenrhrArZrFrarbrntokenrmr7r7r8rgs     zAnthropic._streamcKs|||}|j|}|jjjf|||dd|IdH2z83dHW}t|jd}|rn|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 TrtNrvrw) rJrCrrirjr`rrkrxrhryr7r7r8rrAs    zAnthropic._astream)rhr3cCs|jstd||S)zCalculate number of tokens.rG)r-rH)rArhr7r7r8get_num_tokenseszAnthropic.get_num_tokens)NN)NN)NN)NN)rKrLrM__doc__rUrrrXrTrOrYr`r rr rrprrqr rsrrrgrrrrPr{r7r7r7r8r=sZ   )    %  $r=)&r^rVtypingrrrrrrrr Zlangchain_core._api.deprecationr Zlangchain_core.callbacksr r Zlangchain_core.language_modelsr Z#langchain_core.language_models.llmsrZlangchain_core.outputsrZlangchain_core.prompt_valuesrZlangchain_core.pydantic_v1rrrZlangchain_core.utilsrrrrZlangchain_core.utils.utilsrrrr=r7r7r7r8s$(     w