o +i;@sdZddlmZddlZddlZddlmZmZmZm Z ddl m Z ddl Z ddl mZmZddlmZmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZdd lm Z m!Z!m"Z"m#Z#ddl$m%Z%GdddeZ&Gdddee&Z'dS)z;Anthropic LLM wrapper. Chat models are in `chat_models.py`.) annotationsN) AsyncIteratorCallableIteratorMapping)Any)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)BaseLanguageModelLangSmithParams)LLM)GenerationChunk) PromptValue)get_pydantic_field_names)_build_model_kwargsfrom_envsecret_from_env) ConfigDictField SecretStrmodel_validator)Selfc@sleZdZUdZded<dZded<edddZded < ed d dZd ed < dZ ded< dZ ded< dZ ded< dZ ded< dZ ded< dZd ed< ededdddZded< ed ed!d"ddZd#ed$< dZded%<dZded&<dZd'ed(<eed)Zd*ed+<ed,d-edAd1d2Zed3d-dBd5d6ZedCd8d9ZedCd:d;ZdDdEd?d@ZdS)F_AnthropicCommonNrclient async_clientzclaude-sonnet-4-5 model_name)defaultaliasstrmodelimax_tokens_to_sampleint max_tokensz float | None temperaturez int | Nonetop_ktop_pFbool streamingdefault_request_timeout max_retriesbase_urlANTHROPIC_API_URLzhttps://api.anthropic.com)r)rdefault_factoryz str | Noneanthropic_api_urlapi_keyANTHROPIC_API_KEYranthropic_api_key HUMAN_PROMPT AI_PROMPTzCallable[[str], int] | None count_tokens)r-dict[str, Any] model_kwargsbeforemodevaluesdictreturncCst|}t||SN)rr)clsr;all_required_field_namesrAU/var/www/html/psymed-ai/venv/lib/python3.10/site-packages/langchain_anthropic/llms.py build_extraRs z_AnthropicCommon.build_extraafterrcCs`tj|j|j|j|jd|_tj|j|j|j|jd|_ t tdd|_ t tdd|_ |S)z?Validate that api key and python package exists in environment.)r+r/timeoutr*r3Nr4) anthropic Anthropicr.r2get_secret_valuer(r*rAsyncAnthropicrgetattrr3r4selfrArArBvalidate_environmentXsz%_AnthropicCommon.validate_environmentMapping[str, Any]cCsX|j|jd}|jdur|j|d<|jdur|j|d<|jdur%|j|d<i||jS)z5Get the default parameters for calling Anthropic API.)r"rNr#r$r%)r"rr#r$r%r7)rLdrArArB_default_paramsls      z _AnthropicCommon._default_paramscCs i|jS)Get the identifying parameters.)rPrKrArArB_identifying_params{s z$_AnthropicCommon._identifying_paramsstoplist[str] | None list[str]cCs|durg}|Sr>rA)rLrSrArArB_get_anthropic_stopsz$_AnthropicCommon._get_anthropic_stopr;r<r=r)r=r)r=rNr>)rSrTr=rU) __name__ __module__ __qualname__r__annotations__rrrr"r#r$r%r'r(r*rr.rr2r3r4r5r<r7r classmethodrCrMpropertyrPrRrVrArArArBrs\               rcseZdZdZedddZedded7d d Ze d8d dZ e d9ddZ ed:ddZ e d;ddZ dd!d"Z  d?d@d%d&ZdAd(d)Z  d?dBd+d,Z  d?dCd.d/Z  d?dDd1d2ZdEd5d6ZZS)F AnthropicLLMatAnthropic text completion large language model (legacy LLM). To use, you should have the environment variable `ANTHROPIC_API_KEY` set with your API key, or pass it as a named parameter to the constructor. Example: ```python from langchain_anthropic import AnthropicLLM model = AnthropicLLM(model="claude-sonnet-4-5") ``` T)populate_by_namearbitrary_types_allowedr8r9r;r<r=rcCstjddd|S)z,Raise warning that this class is deprecated.zdThis Anthropic LLM is deprecated. Please use `from langchain_anthropic import ChatAnthropic` insteadr)) stacklevel)warningswarn)r?r;rArArB raise_warnings zAnthropicLLM.raise_warningrcCdS)zReturn type of llm.z anthropic-llmrArKrArArB _llm_typezAnthropicLLM._llm_typedict[str, str]cCsddiS)z9Return a mapping of secret keys to environment variables.r2r0rArKrArArB lc_secretsszAnthropicLLM.lc_secretsr&cCre)z2Whether this class can be serialized by langchain.TrA)r?rArArBis_lc_serializablergzAnthropicLLM.is_lc_serializabler6c Cs*|j|j|j|j|j|j|j|j|jd S)rQ rr"r#r$r%r7r'r(r*rkrKrArArBrRsz AnthropicLLM._identifying_paramsNrSrTkwargsr c s>tjdd|i|}|j}|d|d}r||d<|S)z Get standard params for tracing.rSr" ls_max_tokensNrA)super_get_ls_paramsrRget)rLrSrlparamsidentifying_paramsr" __class__rArBroszAnthropicLLM._get_ls_paramspromptlist[dict[str, str]]c Cs g}|jr[|j|vr[||j}t|D]E\}}|sq|jrK|j|vrK||jd\}}|r<|d|d|rJ|d|dq|rY|d|dqntdd|}tdd|}|rw|d|d|sd|pddg}|S) z&Convert prompt to Messages API format.user)rolecontent assistantz ^\n*Human:\s*r1z\n*Assistant:\s*.*$Hello)r3split enumeratestripr4appendresub) rLrumessagesparts_part human_partassistant_partrzrArArB_format_messagess4  zAnthropicLLM._format_messages run_managerCallbackManagerForLLMRun | Nonec Ks|jrd}|jd|||d|D]}||j7}q|S||}i|j|}dd|D}|jjjd| ||r=|ndd|}|j djS) ajCall 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. run_manager: Optional callback manager for LLM run. kwargs: Additional keyword arguments to pass to the model. Returns: The string generated by the model. Example: ```python prompt = "What are the biggest risks facing humanity?" prompt = f"\n\nHuman: {prompt}\n\nAssistant:" response = model.invoke(prompt) ``` r1rurSrcSi|] \}}|dkr||qSr rA.0kvrArArB !z&AnthropicLLM._call..Nrstop_sequencesrrA) r'_streamtextrVrPitemsrrcreaterrz rLrurSrrl completionchunkrqresponserArArB_calls*     zAnthropicLLM._callrcCs|S)z$Convert a `PromptValue` to a string.) to_string)rLrurArArBconvert_prompt*szAnthropicLLM.convert_prompt$AsyncCallbackManagerForLLMRun | Nonec s|jr!d}|jd|||d|2z 3dHW}||j7}q6|S||}i|j|}dd|D}|jjjd| ||rC|ndd|IdH}|j djS) z;Call out to Anthropic's completion endpoint asynchronously.r1rNcSrrrArrArArBrErz'AnthropicLLM._acall..rrrA) r'_astreamrrVrPrrrrrrzrrArArB_acall.s.     zAnthropicLLM._acallIterator[GenerationChunk]c ks||}i|j|}dd|D}|jjjd |||r#|ndd|-}|D]!}|jdkrNt|j drNt |j j d}|rK|j |j |d|Vq-WddS1sZwYdS) 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. run_manager: Optional callback manager for LLM run. kwargs: Additional keyword arguments to pass to the model. Returns: A generator representing the stream of tokens from Anthropic. Example: ```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 ``` cSrrrArrArArBrmrz(AnthropicLLM._stream..Nrcontent_block_deltarrrrA) rVrPrrrstreamrtypehasattrdeltar ron_llm_new_token rLrurSrrlrqreventrrArArBrNs(   "zAnthropicLLM._streamAsyncIterator[GenerationChunk]c Ks||}i|j|}dd|D}|jjjd |||r#|ndd|4IdH8}|2z(3dHW}|jdkrYt|j drYt |j j d}|rV|j |j |dIdH|Vq16WdIdHdS1IdHslwYdS) rcSrrrArrArArBrrz)AnthropicLLM._astream..NrrrrrrA) rVrPrrrrrrrrr rrrrArArBr{s(   .zAnthropicLLM._astreamrr!cCs d}t|)zCalculate number of tokens.zAnthropic's legacy count_tokens method was removed in anthropic 0.39.0 and langchain-anthropic 0.3.0. Please use ChatAnthropic.get_num_tokens_from_messages instead.)NotImplementedError)rLrmsgrArArBget_num_tokenss zAnthropicLLM.get_num_tokensrW)r=r)r=rh)r=r&)r=r6r>)rSrTrlrr=r )rurr=rv)NN) rurrSrTrrrlrr=r)rurr=r) rurrSrTrrrlrr=r) rurrSrTrrrlrr=r) rurrSrTrrrlrr=r)rrr=r!)rXrYrZ__doc__r model_configrr\rdr]rfrirjrRrorrrrrrr __classcell__rArArsrBr^sD      * 1 # 0 -r^)(r __future__rrrbcollections.abcrrrrtypingrrFlangchain_core.callbacksrr langchain_core.language_modelsr r #langchain_core.language_models.llmsr langchain_core.outputsr langchain_core.prompt_valuesrlangchain_core.utilsrlangchain_core.utils.utilsrrrpydanticrrrrtyping_extensionsrrr^rArArArBs$       m