3fi-zdZddlmZddlZddlmZmZmZmZm 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)ejTe+Z,Gd d e-Z. ddZ/ ddZ0 ddZ1ddZ2ddZ3ddZ4GddeeZ5y)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_exponentialceZdZdZy)ChatGooglePalmErrorz!Error with the `Google PaLM` API.N)__name__ __module__ __qualname____doc__i/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_community/chat_models/google_palm.pyr r (s+r&r cT||S|D]}|j|}|dk7s|d|} |S)z0Truncates text at the earliest stop token found.N)find)textstop stop_tokenstop_token_idxs r'_truncate_at_stop_tokensr/,sF  | ) :. R (D) Kr&c |js tdg}|jD]}|jd}|td|t|jdd|}|td||dk(r'|j t |t | ||d k(r'|j t |t| |j t |t||  t| S) 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)r2)r+messagehuman)roler2) generations) candidatesr getr/appendrrrrr)responser,r8 candidater1r2s r'_response_to_resultr>;s    !"RSS(*K(( x( >%(J9+&VW W*9==B+GN ?%(J9+&VW W T>   GYw5OP w     (9     'VWE +8 + ..r&c.ddlm}d}g}g}tt|}|rU|j d\}}t |t r,|dk7r tdtt|j}nt |tr|jjdr|r td|j d\}} t | trz| jjdr_|j|j j#d|j|j j#d | jgn+td t |tr&|jjdr td t |tr7|j%|j j#d |jnt |tr7|j%|j j#d|jn\t |t&rA|j%|j j#|j(|jn td |rU|j j+||| S)zNConverts a list of LangChain messages into a PaLM API MessagePrompt structure.rNr3z+System message must be first input message.examplez1Message examples must come before other messages.r6)r1r2r4zNHuman 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$P Q%foo&@&@&S&ST U%foo&@&@&O&OP Q&fgoo>  r&ChatGooglePalmc >t}|dfd }|di|S)z*Use tenacity to retry the completion call.c<jjdi|SNr%)clientchatkwargsllms r'_chat_with_retryz)chat_with_retry.._chat_with_retrysszz(((r&rxrreturnrr%rp)ryrxretry_decoratorrzs` r'chat_with_retryrs--/O))  %f %%r&cZKt}|dfd }|di|d{S7w)z0Use tenacity to retry the async completion call.cXKjjdi|d{S7wrt)ru chat_asyncrws r'_achat_with_retryz+achat_with_retry.._achat_with_retrys)+SZZ**4V4444s *(*Nr{r%r})ryrxr~rs` r'achat_with_retryrs7-/O55#,V, ,, ,s !+)+c&eZdZUdZded<dZded< dZded <dZd ed < dZd ed < dZ d ed< dZ ded< e ddZ e ddZe ddZeddZ d ddZ d d dZe d!dZe d"dZy)#rqa`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() rruzmodels/chat-bison-001rJ model_nameNzOptional[SecretStr]google_api_keyzOptional[float] temperaturetop_pz Optional[int]top_krZintnc ddiS)NrGOOGLE_API_KEYr%selfs r' lc_secretszChatGooglePalm.lc_secretss "233r&cy)NTr%rs r'is_lc_serializablez!ChatGooglePalm.is_lc_serializablesr&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&ctt|dd} ddlm}|j|j ||d<|d%d|dcxkrd kstd td |d %d|d cxkrd kstd td |d |d dkr td|S#t $r tdwxYw)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`rurrZz+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) rrrDrE configureget_secret_value ImportErrorr ValueError)rvaluesrrQs r'validate_environmentz#ChatGooglePalm.validate_environments/ )9;K L   / EOON$C$C$E F!x - ,Q&:O5TST5TJK K6UJK K '? &qF7O/Hq/HDE E0IDE E '? &6'?a+?56 6 # %K  s 'B22Cc t|}t|f|j||j|j|j |j d|}t||SN)modelpromptrrrcandidate_count)rWrrrrrrr>rrCr, run_managerrxrr<s r' _generatezChatGooglePalm._generate+sa*(3-<  . //((**** FF .  . #8T22r&c Kt|}t||j||j|j|j |j d{}t||S7wr)rWrrrrrrr>rs r' _ageneratezChatGooglePalm._agenerateAs`*(33C //((**** FF4 . #8T22. sAA(A&A(cv|j|j|j|j|jdS)zGet the identifying parameters.rrrrrrrs r'_identifying_paramsz"ChatGooglePalm._identifying_paramsVs3//++ZZZZ   r&cy)Nzgoogle-palm-chatr%rs r' _llm_typezChatGooglePalm._llm_typeas!r&)r|zDict[str, str])r|bool)r|z List[str])rrr|r)NN) rCList[BaseMessage]r,Optional[List[str]]rz"Optional[CallbackManagerForLLMRun]rxrr|r) rCrr,rrz'Optional[AsyncCallbackManagerForLLMRun]rxrr|r)r|zDict[str, Any])r|rJ)r!r"r#r$__annotations__rrrrrrpropertyr classmethodrrrrrrrrr%r&r'rqrqsU" K-J-*.N'.#'K'!E?!XE= AsJI44;;>%):> 3#3"38 3  3  32%)?C 3#3"3= 3  3  3*  ""r&)r+rJr,rr|rJ)r<zgenai.types.ChatResponser,rr|r)rPrr|zgenai.types.MessagePromptDict)r|zCallable[[Any], Any])ryrqrxrr|r)6r$ __future__rrjtypingrrrrrr r langchain_core.callbacksr r *langchain_core.language_models.chat_modelsr langchain_core.messagesrrrrrlangchain_core.outputsrrlangchain_core.utilsrrrpydanticrrtenacityrrrrrrDrErQ getLoggerr!ri Exceptionr r/r>rWrprrrqr%r&r'rs,"KKKEWV)'   8 $,),       %/&%/ %/%/PK%K"K\,& -E"]IE"r&