Ng8-ddZddlmZddlZddlmZmZmZmZm Z m Z m Z ddl m Z mZddlmZddlmZmZmZmZmZddlmZmZdd lmZmZmZdd lmZm Z dd l!m"Z"m#Z#m$Z$m%Z%m&Z&erddl'm(Z)ej*e+Z,Gd d e-Z.d%dZ/d&dZ0d'dZ1d(dZ2d)d"Z3d)d#Z4Gd$deeZ5dS)*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_exponentialceZdZdZdS)ChatGooglePalmErrorz!Error with the `Google PaLM` API.N)__name__ __module__ __qualname____doc__g/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/chat_models/google_palm.pyr r (s++++r&r textstrstopOptional[List[str]]returncb||S|D]'}||}|dkr |d|}(|S)z0Truncates text at the earliest stop token found.N)find)r(r* stop_tokenstop_token_idxs r'_truncate_at_stop_tokensr2,sM  | )) :.. R  (D Kr&responsegenai.types.ChatResponserc v|jstdg}|jD] }|d}|td|t|dd|}|td||dkr3|t |t |  |d kr3|t |t|  |t |t||   t| S)z9Converts a PaLM API response into a LangChain ChatResult.z.ChatResponse must have at least one candidate.authorNz"ChatResponse must have an author: contentz"ChatResponse must have a content: ai)r7)r(messagehuman)roler7) generations) candidatesr getr2appendrrrrr)r3r*r= candidater6r7s r'_response_to_resultrB;s  T!"RSSS(*K( x(( >%&V9&V&VWW W*9==B+G+GNN ?%&V9&V&VWW W T>>   GYw5O5O5OPPP    w      (999        'VWEEE     + . . ..r&input_messagesList[BaseMessage]genai.types.MessagePromptDictcddlm}d}g}g}tt|}|rE|d\}}t |t r1|dkrtdtt|j }nt |tr|j r|rtd|d\}} t | tr^| j rW||jd|j |jd| j gn,td t |tr|j rtd t |tr5||jd|j nt |tr5||jd|j n^t |t$r:||j|j|j ntd |E|j||| S) zNConverts a list of LangChain messages into a PaLM API MessagePrompt structure.rNr8z+System message must be first input message.z1Message examples must come before other messages.r;)r6r7r9zNHuman example message must be immediately followed by an AI example response.zKAI example message must be immediately preceded by a Human example message.z$P Q Q%fo&@&STT U%fo&@&OPP Q&fgo>>    r&llmChatGooglePalmkwargsrc Lt}|dfd }|di|S)z*Use tenacity to retry the completion call.rxrr,c(jjdi|SNr%)clientchatrxrvs r'_chat_with_retryz)chat_with_retry.._chat_with_retryssz(((((r&rxrr,rr%ru)rvrxretry_decoratorrs` r'chat_with_retryrsO-//O)))))_)   % %f % %%r&c\Kt}|dfd }|di|d{VS)z0Use tenacity to retry the async completion call.rxrr,c8Kjjdi|d{VSr{)r| chat_asyncr~s r'_achat_with_retryz+achat_with_retry.._achat_with_retrys6+SZ*44V444444444r&Nrr%r)rvrxrrs` r'achat_with_retryrse-//O55555_5#",,V,, , , , , , ,,r&c(eZdZUdZded<dZded< dZded <dZd ed < dZd ed < dZ d ed< dZ ded< e d*dZ e d+dZe d,dZed-dZ d.d/d$Z d.d0d&Ze d1d(Ze d2d)ZdS)3rwa`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() rr|zmodels/chat-bison-001r) model_nameNzOptional[SecretStr]google_api_keyzOptional[float] temperaturetop_pz Optional[int]top_kr_intnr,Dict[str, str]c ddiS)NrGOOGLE_API_KEYr%selfs r' lc_secretszChatGooglePalm.lc_secretss "233r&boolcdS)NTr%rs r'is_lc_serializablez!ChatGooglePalm.is_lc_serializablestr& List[str]c gdS)z*Get the namespace of the langchain object.) langchain chat_models google_palmr%)clss r'get_lc_namespacezChatGooglePalm.get_lc_namespaces;:::r&valuesrctt|dd} ddlm}|j|n#t $rtdwxYw||d<|d$d|dcxkrd ksntd |d $d|d cxkrd ksntd |d |d dkrtd|S)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`r|rr_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) rrrJrK configureget_secret_value ImportErrorr ValueError)rrrrUs r'validate_environmentz#ChatGooglePalm.validate_environmentsC/ )9;K L L    / / / / / / EON$C$C$E$E F F F F F   %K   !x - ,Q&:O5T5T5T5TST5T5T5T5TJKK K '? &qF7O/H/H/H/Hq/H/H/H/HDEE E '? &6'?a+?+?566 6 s )A A$rIrDr*r+ run_manager"Optional[CallbackManagerForLLMRun]rxrc t|}t|f|j||j|j|j|jd|}t||SN)modelpromptrrrcandidate_count)r[rrrrrrrBrrIr*rrxrr3s r' _generatezChatGooglePalm._generate%sf*(33-<  . /(** F . .  . . #8T222r&'Optional[AsyncCallbackManagerForLLMRun]c Kt|}t||j||j|j|j|jd{V}t||Sr)r[rrrrrrrBrs r' _ageneratezChatGooglePalm._agenerate;sy*(333C /(** F4 4 4 . . . . . . #8T222r&Dict[str, Any]cD|j|j|j|j|jdS)zGet the identifying parameters.rrrrrrrs r'_identifying_paramsz"ChatGooglePalm._identifying_paramsPs-/+ZZ    r&cdS)Nzgoogle-palm-chatr%rs r' _llm_typezChatGooglePalm._llm_type[s!!r&)r,r)r,r)r,r)rrr,r)NN) rIrDr*r+rrrxrr,r) rIrDr*r+rrrxrr,r)r,r)r,r))r!r"r#r$__annotations__rrrrrrpropertyr classmethodrrrrrrrrr%r&r'rwrws"KKK-J----*.N....#'K''''!E!!!!XE AJJJJI444X4[;;;[;X>%):> 333332%)?C 33333*   X """X"""r&)r(r)r*r+r,r))r3r4r*r+r,r)rCrDr,rE)r,r\)rvrwrxrr,r)6r$ __future__rrotypingrrrrrr r langchain_core.callbacksr r *langchain_core.language_models.chat_modelsr langchain_core.messagesrrrrrlangchain_core.outputsrrlangchain_core.utilsrrrpydanticrrtenacityrrrrrrJrKrU getLoggerr!rn Exceptionr r2rBr[rurrrwr%r&r'rs,,""""""KKKKKKKKKKKKKKKKKKEDDDDDWVVVVVVVVV))))))))(''''''  8 $ $,,,,,),,,    %/%/%/%/PEEEEP,&&&& - - - -E"E"E"E"E"]IE"E"E"E"E"r&