Ng <ddlmZmZmZmZmZmZmZddlm Z ddl m Z m Z ddl mZmZmZddlmZmZmZmZmZmZddlmZmZmZddlmZddlmZdd l m!Z!d ed e"d e"d e"fdZ#ddddeed e"d e"d e"fdZ$e dddGddee!Z%dS))Any AsyncIteratorDictIteratorListOptionalcast) deprecated)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun) BaseChatModelagenerate_from_streamgenerate_from_stream) AIMessageAIMessageChunk BaseMessage ChatMessage HumanMessage SystemMessage)ChatGenerationChatGenerationChunk ChatResult) PromptValue) ConfigDict)_AnthropicCommonmessage human_prompt ai_promptreturncltt|j}t|tr d|jd|}ndt|tr|d|}nGt|tr|d|}n*t|tr|}ntd||S)Nz z:  zGot unknown type ) r strcontent isinstancerrole capitalizerrr ValueError)rrrr# message_texts e/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/chat_models/anthropic.py_convert_one_message_to_textr*s 3((G';'' 8Dgl5577DD7DD G\ * *8&2222 GY ' '8#//g// G] + +8 6W66777 z Human:z Assistant:)rrmessagesc|}t|dts#|tddfd|D}|S)aFormat a list of messages into a full prompt for the Anthropic model Args: messages (List[BaseMessage]): List of BaseMessage to combine. human_prompt (str, optional): Human prompt tag. Defaults to " Human:". ai_prompt (str, optional): AI prompt tag. Defaults to " Assistant:". Returns: str: Combined string with necessary human_prompt and ai_prompt tags. r#c3:K|]}t|VdSN)r*).0rrrs r) z7convert_messages_to_prompt_anthropic..BsC  %WlIFFr+)copyr$rappendjoinrstrip)r,rrtexts `` r)$convert_messages_to_prompt_anthropicr:/s}}H hrlI . ./ "---... 77  D ;;==r+z0.0.28z1.0z!langchain_anthropic.ChatAnthropic)sinceremovalalternative_importc TeZdZdZeddZedeeeffdZ edefdZ e de fdZ e deefdZd eedefd Zd edefd Z dd eedeeedeededeef dZ dd eedeeedeededeef dZ dd eedeeedeededef dZ dd eedeeedeededef dZdede fdZ!d S) ChatAnthropica`Anthropic` chat 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.chat_models import ChatAnthropic model = ChatAnthropic(model="", anthropic_api_key="my-api-key") T)populate_by_namearbitrary_types_allowedrc ddiS)Nanthropic_api_keyANTHROPIC_API_KEYselfs r) lc_secretszChatAnthropic.lc_secretsds#%899r+cdS)zReturn type of chat model.zanthropic-chatrErFs r) _llm_typezChatAnthropic._llm_typehs  r+cdS)z9Return whether this model can be serialized by Langchain.TrEclss r)is_lc_serializablez ChatAnthropic.is_lc_serializablems tr+c gdS)z*Get the namespace of the langchain object.) langchain chat_models anthropicrErLs r)get_lc_namespacezChatAnthropic.get_lc_namespacers9888r+r,cfi}|jr |j|d<|jr |j|d<tdd|i|S)aFormat a list of messages into a full prompt for the Anthropic model Args: messages (List[BaseMessage]): List of BaseMessage to combine. Returns: str: Combined string with necessary HUMAN_PROMPT and AI_PROMPT tags. rrr,rE) HUMAN_PROMPT AI_PROMPTr:)rGr, prompt_paramss r)_convert_messages_to_promptz)ChatAnthropic._convert_messages_to_promptwsP   >,0,=M. ) > 8)-M+ &3WWXWWWWr+promptcP||Sr2)rX to_messages)rGrYs r)convert_promptzChatAnthropic.convert_prompts"//0B0B0D0DEEEr+Nstop run_managerkwargsc+ K||}d|i|j|}|r||d<|jjjdi|ddi}|D]D}|j} t t| } |r|| | | VEdS NrYstop_sequencesstreamTr0r)chunkrE) rX_default_paramsclient completionscreate completionrron_llm_new_token rGr,r]r^r_rYparams stream_respdatadeltares r)_streamzChatAnthropic._streams11(;;"*F!Ud6J!Uf!U  ,'+F# $4dk-4KKvKKdKKK   DOE'u0M0M0MNNNE A,,U%,@@@KKKK   r+c FK||}d|i|j|}|r||d<|jjjdi|ddid{V}|23d{V}|j} t t| } |r|| | d{V| WVQ6dSra) rXrf async_clientrhrirjrrrkrls r)_astreamzChatAnthropic._astreams11(;;"*F!Ud6J!Uf!U  ,'+F# $@D-9@WW6WWRVWWWWWWWWW %       $OE'u0M0M0MNNNE G!2252FFFFFFFFFKKKKK &++sB c <|jr |j|f||d|}t|S||}d|i|j|}|r||d<|jjjdi|}|j} t| } tt| gSN)r]r^rYrbr0rd) generationsrE) streamingrqrrXrfrgrhrirjrrr rGr,r]r^r_ stream_iterrYrmresponserjrs r) _generatezChatAnthropic._generates > 5&$,#@FK( 44 411    f" "" "   ,'+F# $14;*1;;F;;( J///~g'F'F'F&GHHHHr+cXK|jr&|j|f||d|}t|d{VS||}d|i|j|}|r||d<|jjjdi|d{V}|j} t| } tt| gSrv) rxrtrrXrfrsrhrirjrrrrys r) _ageneratezChatAnthropic._agenerates > <'$-#@FK/{;;;;;;;; ;11    f" "" "   ,'+F# $=*6=GGGGGGGGGG( J///~g'F'F'F&GHHHHr+r9cX|jstd||S)zCalculate number of tokens.z-Please ensure the anthropic package is loaded) count_tokens NameError)rGr9s r)get_num_tokenszChatAnthropic.get_num_tokenss1  MKLL L  &&&r+)NN)"__name__ __module__ __qualname____doc__r model_configpropertyrr"rHrJ classmethodboolrNrrSrrXrr\rr rrrrqr rrtrr|r~intrrEr+r)r?r?Ks  : $L :DcN:::X: 3   X 4[9c999[9 XD4E X# X X X XF[FSFFFF %):> {#tCy!67    % & .%)?C {#tCy!;<    * + .%):> II{#ItCy!I67 I  I  IIII<%)?C II{#ItCy!I;< I  I  IIII6'3'3''''''r+r?N)&typingrrrrrrr langchain_core._api.deprecationr langchain_core.callbacksr r *langchain_core.language_models.chat_modelsr rrlangchain_core.messagesrrrrrrlangchain_core.outputsrrrlangchain_core.prompt_valuesrpydanticr"langchain_community.llms.anthropicrr"r*r:r?rEr+r)rsAKKKKKKKKKKKKKKKKKK666666 SRRRRRRRRR444444??????  ,%% ;   8   : Z'Z'Z'Z'Z'M#3Z'Z'  Z'Z'Z'r+