dh8-SrSSKJr SSKrSSKJrJrJrJrJ r J r J r SSK J r Jr SSKJr SSKJrJrJrJrJr SSKJrJr SS KJrJrJr SS KJrJ r SS K!J"r"J#r#J$r$J%r%J&r& \(aSSK'J(r) \RT"\+5r,"S S \-5r.SSjr/SSjr0SSjr1SSjr2SSjr3SSjr4"SS\\5r5g)z&Wrapper around Google's PaLM Chat API.) annotationsN) TYPE_CHECKINGAnyCallableDictListOptionalcast)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun) BaseChatModel) AIMessage BaseMessage ChatMessage HumanMessage SystemMessage)ChatGeneration ChatResult)convert_to_secret_strget_from_dict_or_envpre_init) BaseModel SecretStr)before_sleep_logretryretry_if_exception_typestop_after_attemptwait_exponentialc\rSrSrSrSrg)ChatGooglePalmError(z!Error with the `Google PaLM` API.N)__name__ __module__ __qualname____firstlineno____doc____static_attributes__r"c/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/chat_models/google_palm.pyr r (s+r)r c^UcU$UH!nURU5nUS:wdMUSUnM# U$)z0Truncates text at the earliest stop token found.N)find)textstop stop_tokenstop_token_idxs r*_truncate_at_stop_tokensr2,sA  |  :. R (D Kr)c UR(d [S5e/nURHnURS5nUc[SU35e[URSS5U5nUc[SU35eUS:Xa"UR [ U[ US9S 95 MzUS :Xa"UR [ U[US9S 95 MUR [ U[XES 9S 95 M [US 9$) z9Converts a PaLM API response into a LangChain ChatResult.z.ChatResponse must have at least one candidate.authorz"ChatResponse must have an author: contentz"ChatResponse must have a content: ai)r5)r.messagehuman)roler5) generations) candidatesr getr2appendrrrrr)responser/r; candidater4r5s r*_response_to_resultrA;s   !"RSS(*K(( x( >%(J9+&VW W*9==B+GN ?%(J9+&VW W T>   GYw5OP w     (9     'VE +)8 + ..r)c2SSKJn Sn/n/n[[U55nU(GaUUR S5upg[ U[ 5(a-US:wa [S5e[[UR5nGO[ U[5(aUR(aU(a [S5eUR S5up[ U [5(ajU R(aYURURR!SURS9URR!SU RS9/5 GO,[S 5e[ U[5(aUR(a [S 5e[ U[5(a4UR#URR!SURS95 O[ U[5(a4UR#URR!SURS95 O^[ U[$5(a>UR#URR!UR&URS95 O [S 5eU(aGMUURR)UUUS 9$) zNConverts a list of LangChain messages into a PaLM API MessagePrompt structure.rNr6z+System message must be first input message.z1Message examples must come before other messages.r9)r4r5r7zNHuman example message must be immediately followed by an AI example response.zKAI example message must be immediately preceded by a Human example message.zrr:MessagePromptDict) input_messagesgenairCrDrE remainingindex input_message_next_input_messages r*_messages_to_prompt_dictrYcsG(G.0H.0HY~./I (}}Q/ m] 3 3z)*WXX3 5 56G  | 4 49N9N)G%.MM!$4 !A,i88=O=W=W //#*M4I4I0 //#'1C1K1K0  *, y 1 1m6K6K%#  y 1 1 OO ''t]=R=R'S  | 4 4 OO ''w @U@U'V  { 3 3 OO ''(--}7L7L(  &N c )j ;; ( ( ) r)c SSKnSnSnSnSn[S[U5[XUS9[ UR R R5[ UR R R5-[ UR R R5-[[[R5S 9$) zKReturns a tenacity retry decorator, preconfigured to handle PaLM exceptionsrN< T) multiplierminmax)reraiser/waitr before_sleep)google.api_core.exceptionsrrrrapi_core exceptionsResourceExhaustedServiceUnavailableGoogleAPIErrorrloggerloggingWARNING)googler_ min_seconds max_seconds max_retriess r*_create_retry_decoratorrrs%JKKK   , + V #FOO$>$>$P$P Q%foo&@&@&S&ST U%foo&@&@&O&OP Q&fgoo>  r)ChatGooglePalmc B^[5nUSU4Sjj5nU"S0UD6$)z*Use tenacity to retry the completion call.c<>TRR"S0UD6$Nr")clientchatkwargsllms r*_chat_with_retry)chat_with_retry.._chat_with_retryszz(((r)rzrreturnrr"rr)r{rzretry_decoratorr|s` r*chat_with_retryrs--/O))  %f %%r)c^^# [5nUSU4Sjj5nU"S0UD6IShvN $N7f)z0Use tenacity to retry the async completion call.cX># TRR"S0UD6IShvN $N7frv)rw chat_asyncrys r*_achat_with_retry+achat_with_retry.._achat_with_retrys&ZZ**4V4444s *(*Nr~r"r)r{rzrrs` r*achat_with_retryrs6-/O55#,V, ,, ,s #-+-cB\rSrSr%SrS\S'SrS\S'SrS \S 'SrS \S 'Sr S \S 'Sr S\S'Sr S\S'\ SSj5r \SSj5r\SSj5r\SSj5rS S!SjjrS S"Sjjr\ S#Sj5r\ S$Sj5rSrg)%rsa`Google PaLM` Chat models API. To use you must have the google.generativeai Python package installed and either: 1. The ``GOOGLE_API_KEY`` environment variable set with your API key, or 2. Pass your API key using the google_api_key kwarg to the ChatGoogle constructor. Example: .. code-block:: python from langchain_community.chat_models import ChatGooglePalm chat = ChatGooglePalm() rrwzmodels/chat-bison-001rL model_nameNzOptional[SecretStr]google_api_keyzOptional[float] temperaturetop_pz Optional[int]top_kr\intnc SS0$)NrGOOGLE_API_KEYr"selfs r* lc_secretsChatGooglePalm.lc_secretss "233r)cg)NTr"rs r*is_lc_serializable!ChatGooglePalm.is_lc_serializablesr)c /SQ$)z*Get the namespace of the langchain object.) langchain chat_models google_palmr")clss r*get_lc_namespaceChatGooglePalm.get_lc_namespaces ;:r)cv[[USS55nSSKJn UR"UR 5S9 X1S'USbSUSs=::aS ::d O [S 5eUS bSUS s=::aS ::d O [S 5eUS bUS S::a [S5eU$![ a [S5ef=f)zGValidate api key, python package exists, temperature, top_p, and top_k.rrrN)api_keyzmCould not import google.generativeai python package. Please install it with `pip install google-generativeai`rwrr\z+temperature must be in the range [0.0, 1.0]rz%top_p must be in the range [0.0, 1.0]rztop_k must be positive) rrrFrG configureget_secret_value ImportErrorr ValueError)rvaluesrrSs r*validate_environment#ChatGooglePalm.validate_environments/ )9;K L   / OON$C$C$E F!x - ,Q&:O5TST5TJK K '? &qF7O/Hq/HDE E '? &6'?a+?56 6 # %K  s $B""B8c [U5n[U4URUURURUR UR S.UD6n[Xb5$N)modelpromptrrrcandidate_count)rYrrrrrrrArrEr/ run_managerrzrr?s r* _generateChatGooglePalm._generate%s_*(3-<  . //((**** FF .  . #822r)c # [U5n[UURUURURUR UR S9IShvN n[Xb5$N7fr)rYrrrrrrrArs r* _agenerateChatGooglePalm._agenerate;s]*(33C //((**** FF4 . #822. sAA$A"A$cvURURURURURS.$)zGet the identifying parameters.rrrrrrrs r*_identifying_params"ChatGooglePalm._identifying_paramsPs3//++ZZZZ   r)cg)Nzgoogle-palm-chatr"rs r* _llm_typeChatGooglePalm._llm_type[s!r)r")rzDict[str, str])rbool)rz List[str])rrrr)NN) rEList[BaseMessage]r/Optional[List[str]]rz"Optional[CallbackManagerForLLMRun]rzrrr) rErr/rrz'Optional[AsyncCallbackManagerForLLMRun]rzrrr)rzDict[str, Any])rrL)r#r$r%r&r'__annotations__rrrrrrpropertyr classmethodrrrrrrrrr(r"r)r*rsrssU" K-J-*.N'.#'K'!E?!XE= AsJI44;;>%):> 3#3"38 3  3  32%)?C 3#3"3= 3  3  3*  ""r))r.rLr/rrrL)r?zgenai.types.ChatResponser/rrr)rRrrzgenai.types.MessagePromptDict)rzCallable[[Any], Any])r{rsrzrrr)6r' __future__rrltypingrrrrrr r langchain_core.callbacksr r *langchain_core.language_models.chat_modelsr langchain_core.messagesrrrrrlangchain_core.outputsrrlangchain_core.utilsrrrpydanticrrtenacityrrrrrrFrGrS getLoggerr#rk Exceptionr r2rArYrrrrrsr"r)r*rs,"KKKEWV)'   8 $,),       %/&%/ %/%/PE%E"EP,& -E"]IE"r)