tj-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__i/var/www/html/CA-Chatbot/venv/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|}|r~|d\}}t |t r1|dkrtdtt|j }nt |tr|j dr|rtd|d\}} t | trq| j drW||jd|j |jd | j gn?td t |tr)|j drtd t |tr5||jd |j nt |tr5||jd|j n^t |t&r:||j|j|j ntd |~|j||| S)zNConverts a list of LangChain messages into a PaLM API MessagePrompt structure.rNr8z+System message must be first input message.examplez1Message 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.ryrr,c(jjdi|SNr%)clientchatryrws r'_chat_with_retryz)chat_with_retry.._chat_with_retryssz(((((r&ryrr,rr%rv)rwryretry_decoratorrs` r'chat_with_retryrsO-//O)))))_)   % %f % %%r&c\Kt}|dfd }|di|d{VS)z0Use tenacity to retry the async completion call.ryrr,c8Kjjdi|d{VSr|)r} chat_asyncrs r'_achat_with_retryz+achat_with_retry.._achat_with_retrys6+SZ*44V444444444r&Nrr%r)rwryrrs` 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)3rxa`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_namespace s;:::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) rrrKrL configureget_secret_value ImportErrorr ValueError)rrrrVs 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$rJrDr*r+ run_manager"Optional[CallbackManagerForLLMRun]ryrc t|}t|f|j||j|j|j|jd|}t||SN)modelpromptrrrcandidate_count)r\rrrrrrrBrrJr*rryrr3s r' _generatezChatGooglePalm._generate+sf*(33-<  . /(** F . .  . . #8T222r&'Optional[AsyncCallbackManagerForLLMRun]c Kt|}t||j||j|j|j|jd{V}t||Sr)r\rrrrrrrBrs r' _ageneratezChatGooglePalm._agenerateAsy*(333C /(** F4 4 4 . . . . . . #8T222r&Dict[str, Any]cD|j|j|j|j|jdS)zGet the identifying parameters.rrrrrrrs r'_identifying_paramsz"ChatGooglePalm._identifying_paramsVs-/+ZZ    r&cdS)Nzgoogle-palm-chatr%rs r' _llm_typezChatGooglePalm._llm_typeas!!r&)r,r)r,r)r,r)rrr,r)NN) rJrDr*r+rrryrr,r) rJrDr*r+rrryrr,r)r,r)r,r))r!r"r#r$__annotations__rrrrrrpropertyr classmethodrrrrrrrrr%r&r'rxrxs"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])rwrxryrr,r)6r$ __future__rrptypingrrrrrr r langchain_core.callbacksr r *langchain_core.language_models.chat_modelsr langchain_core.messagesrrrrrlangchain_core.outputsrrlangchain_core.utilsrrrpydanticrrtenacityrrrrrrKrLrV getLoggerr!ro Exceptionr r2rBr\rvrrrxr%r&r'rs,,""""""KKKKKKKKKKKKKKKKKKEDDDDDWVVVVVVVVV))))))))(''''''  8 $ $,,,,,),,,    %/%/%/%/PKKKK\,&&&& - - - -E"E"E"E"E"]IE"E"E"E"E"r&