dh8SrSSKJr SSKrSSKrSSKrSSKJrJr SSK J r J r J r J r JrJrJrJr SSKJr SSKrSSKJr SSKJrJr SS KJrJr SS KJrJrJ r J!r!J"r" SS K#J$r$J%r%J&r& SS K'J(r( SS K)J*r*J+r+J,r, SSK-J.r.J/r/ \ (aSSK0J1r1J2r2J3r3J4r4 SSK5J6r6 \Rn"\85r9\"SS55r:SSjr;SSjr<S Sjr=S!Sjr>S"Sjr?\"SSSS9"SS\*\55r@g)#z1Wrapper around Google VertexAI chat-based models.) annotationsN) dataclassfield) TYPE_CHECKINGAnyDictIteratorListOptionalUnioncast)urlparse) deprecated)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun) BaseChatModelgenerate_from_stream) AIMessageAIMessageChunk BaseMessage HumanMessage SystemMessage)ChatGenerationChatGenerationChunk ChatResult)pre_init)_VertexAICommonis_codey_modelis_gemini_model)load_image_from_gcsraise_vertex_import_error) ChatMessage ChatSessionCodeChatSessionInputOutputTextPair)Contentc>\rSrSr%Sr\"\S9rS\S'Sr S\S'S r g) _ChatHistory6z/Represents a context and a history of messages.)default_factoryzList['ChatMessage']historyN Optional[str]context) __name__ __module__ __qualname____firstlineno____doc__rlistr+__annotations__r-__static_attributes__r.`/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/chat_models/vertexai.pyr(r(6s9#(#>G >!G]!r7r(cSSKJn /Sp2[U5HupE[[UR 5nUS:Xa[ U[5(aUnM>[ U[5(a$U"UR SS9nURU5 Mw[ U[5(a$U"UR SS9nURU5 M[S[U5SUS 35e [X2S 9nU$) zParse a sequence of messages into history. Args: history: The list of messages to re-create the history of the chat. Returns: A parsed chat history. Raises: ValueError: If a sequence of message has a SystemMessage not at the first place. r)r"Nbot)contentauthoruserUnexpected message with type  at the position .)r-r+)vertexai.language_modelsr" enumerater strr; isinstancerrappendr ValueErrortyper() r+r"vertex_messagesr-imessager;vertex_message chat_historys r8_parse_chat_historyrM>s5!4W( sGOO, 6j-88G  + +(ON  " "> 2  . .(PN  " "> 2/W >OPQsRST ) IL r7c[U5n[URUR/5$![a"n[ R SU35 SnAgSnAff=f)NzUnable to parse URL: F)rallschemenetloc Exceptionloggerdebug)sresultes r8_is_urlrX^sO!FMM6==122  ,QC01s+. AAAc^^ ^ SSKJnJm Jm S U U U4Sjjn/n[ U5HupVUS:Xa [ U[ 5(a [S5e[ U[5(aSnO3[ U[5(aSnO[S[U5SUS 35eURn[ U[5(aU/nUV s/sH o"U 5PM n n U"XzS 9n URU 5 M U$s sn f) Nr)r&ImagePartc8>[U[5(aTRU5$[U[5(d[ S[ U5S35eUSS:XaTRUS5$USS:XGaUSSnUR S5(a [UTS9nOUR S 5(aa[R"S U5nU(aURS 5nO [ S 5eTR[R"U55nOo[U5(aB[R "U5nUR#5 TRUR$5nOTR'U5nO [ S 5eTR)U5$)Nz0Message's content is expected to be a dict, got !rGtext image_urlurlzgs://)pathprojectz data:image/zdata:image/\w{2,4};base64,(.*)zdInvalid image uri. It should be in the format data:image/;base64,.z,Only text and image_url types are supported!)rDrC from_textrrFrG startswithr researchgroup from_bytesbase64 b64decoderXrequestsgetraise_for_statusr;load_from_file from_image)partraimageencodedresponserZr[rbs r8_convert_to_prompt6_parse_chat_history_gemini.._convert_to_promptlse dC >>$' '$%%B4:,aP  <6 !>>$v,/ / &\[ ( $U+Dw''+wG//!yy)JDQ%mmA.G$Q(()9)9')BC#<<-))+(()9)9:,,T2KL Lu%%r7z%SystemMessages are not yet supported!modelr=r>r?r@)roleparts)rqzUnion[str, Dict]returnr[)"vertexai.preview.generative_modelsr&rZr[rBrDrrFrrrGr;rCrE)r+rbr&rurHrIrJrx raw_contentrqryrKrZr[s ` @@r8_parse_chat_history_geminir}gsHG!&!&FO(  6j-88DE E  + +D  . .D/W >OPQsRST oo k3 ' '&-K6ABkd#D)kB d8~.#)$ CsC2cSSKJn [U5S-S:wa[S[U5S35e/nSn[ U5HupEUS-S:Xa<[ U[ 5(d[S[U5SUS35eURnUS-S :XdMU[ U[5(d[S [U5SUS35eU"X5RS 9nURU5 M U$) Nr)r%z8Expect examples to have an even amount of messages, got r@z;Expected the first message in a part to be from human, got z for the z th message.rcz9Expected the second message in a part to be from AI, got ) input_text output_text) rAr%lenrFrBrDrrGr;rrE)examplesr% example_pairsrrIexamplepairs r8_parse_examplesrs< 8}qAFs8}oUV W  MJ)  q5A:g|44 QG}oYqc>!J q5A:gy11 OG}oYqc>'%??D   &#*$ r7cU(d [S5eUSn[U[5(d[SURS35eU$)zMGet the human message at the end of a list of input messages to a chat model.z:You should provide at least one message to start the chat!z3Last message in the list should be from human, got r@)rFrDrrG)messagesquestions r8 _get_questionrsL UVV|H h - -A(--PQ R   Or7z0.0.12z1.0z&langchain_google_vertexai.ChatVertexAI)sinceremovalalternative_importc\rSrSr%SrSrS\S'SrS\S'\SS j5r \SS j5r \ SS j5r SSS jjr SSS jjrSSSjjrSSjrSrg) ChatVertexAIz+`Vertex AI` Chat large language models API.z chat-bisonrC model_nameNzOptional[List[BaseMessage]]rcg)NTr.)selfs r8is_lc_serializableChatVertexAI.is_lc_serializablesr7c /SQ$)z*Get the namespace of the langchain object.) langchain chat_modelsvertexair.)clss r8get_lc_namespaceChatVertexAI.get_lc_namespaces 87r7c,[US5nURU5 SSKJnJn U(aSSKJn U(aW"USS9US'U$[US5(aWnOWnURUS5US'U$![a [5 N\f=f)z7Validate that the python package exists in environment.rr) ChatModel CodeChatModel)GenerativeModel)rclient) r_try_init_vertexairArrr{r ImportErrorr!rfrom_pretrained)rvalues is_geminirrr model_clss r8validate_environment!ChatVertexAI.validate_environments$F<$89  v& ( I .&:NOF8  f\233) % (88 9MNF8   ( % ' (sA<<BBc UbUO URnU(a UR"U4X#S.UD6n[U5$[U5nUR"SUSS.UD6n 0n SU ;aU R S5U S'UR (aL[XRS9n U R 5n URRU S9n U RXS9nOw[USS 5nURS 5=(d URnU(a[U5U S 'UR "U40U D6n U R"UR"40U D6nUR$Vs/sHn['[)UR*S 9S 9PM nn[-US 9$s snf)aGenerate next turn in the conversation. Args: messages: The history of the conversation as a list of messages. Code chat does not support context. stop: The list of stop words (optional). run_manager: The CallbackManager for LLM run, it's not used at the moment. stream: Whether to use the streaming endpoint. Returns: The ChatResult that contains outputs generated by the model. Raises: ValueError: if the last message in the list is not from human. N)stop run_managerFrstreamcandidate_countrbr+generation_configrrr;rJ generationsr.) streaming_streamrr_prepare_paramspop_is_gemini_modelr}rbr start_chat send_messagerMrmrr _start_chatr; candidatesrrr^r)rrrrrkwargs should_stream stream_iterrparams msg_paramshistory_geminirJchatrtr+rrrs r8 _generateChatVertexAI._generates.#)"4$.. ,,#@FK( 4 4 *%%H4HH  &,2JJ7H,IJ( )  7,,WN$((*G;;)).)AD(((KH)(3B-8Gzz*->H%4X%>z"##G6v6D(()9)9HZHH(( ( 9QVV#< =(  k22  s#Fc # SU;a&URS5 [RS5 UR"SSU0UD6n0nSU;aURS5US'UR(aT[ XR S9nUR5nURRUS9n U RXS9IShvN n Ox[U5n [USS 5n URS S5n U (a[U 5US 'UR"U 40UD6n U R"U R40UD6IShvN n U R Vs/sHn[#[%UR&S 9S 9PM nn[)US 9$NNBs snf7f)aAsynchronously generate next turn in the conversation. Args: messages: The history of the conversation as a list of messages. Code chat does not support context. stop: The list of stop words (optional). run_manager: The CallbackManager for LLM run, it's not used at the moment. Returns: The ChatResult that contains outputs generated by the model. Raises: ValueError: if the last message in the list is not from human. rz8ChatVertexAI does not currently support async streaming.rrrrrNrrrrrr.)rrSwarningrrr}rbrrsend_message_asyncrrMrmrrr;rrrr^r)rrrrrrrrrJrrtrr+rrrs r8 _agenerateChatVertexAI._agenerate2sw* v  JJx NNU V%%:4:6:  &,2JJ7H,IJ( )  7,,WN$((*G;;)).)AD!44W4WWH$X.H)(3B-8Gzz*d3H%4X%>z"##G6v6D!44X5E5ETTTH(( ( 9QVV#< =(  k22XU s7BE8?A8F7E:8F #E<- F:F<Fc+d# UR"S USS.UD6nUR(aN[XRS9nUR 5nUR R US9nURUSUS9n Op[U5n [USS5n URSS5n U (a[U 5US'UR"U 40UD6nUR"U R40UD6n U HBn [[!U R"S9S 9nU(aUR%U R"US 9 Uv MD g7f) NTrrr)rrrrrr)chunkr.)rrr}rbrrrrrrMrmrrsend_message_streamingr;rrr^on_llm_new_token)rrrrrrrrJr responsesrr+rrtrs r8rChatVertexAI._streamds#%%G4GG  7,,WN$((*G;;)).)AD))*I%X.H)(3B-8Gzz*d3H%4X%>z"##G6v6D33H4D4DOOI!H'x}}0UVE,,X]]%,HK "sD.D0c UR(d3URR"SURURS.UD6$URR"SSUR0UD6$)N)r-message_historyrr.)rrrr-r+)rr+rs r8rChatVertexAI._start_chatsa"";;))LR ;;))T'//TVT Tr7r.)rzbool)rzz List[str])rrrzr)NNN) rList[BaseMessage]rOptional[List[str]]r"Optional[CallbackManagerForLLMRun]rzOptional[bool]rrrzr)NN) rrrrrz'Optional[AsyncCallbackManagerForLLMRun]rrrzr) rrrrrrrrrzzIterator[ChatGenerationChunk])r+r(rrrzz#Union[ChatSession, CodeChatSession])r/r0r1r2r3rr5r classmethodrrrrrrrrr6r.r7r8rrs@ 6"J",0H)0884%):>!% 43#43"438 43  43  43 43r%)?C 03#03"03= 03  03  03j%):> #"8    ' :U#U/2U ,Ur7r)r+rrzr()rUrCrzr)r+rrbr,rzzList['Content'])rrrzzList['InputOutputTextPair'])rrrzr)Ar3 __future__rrjloggingrf dataclassesrrtypingrrrr r r r r urllib.parserrllangchain_core._api.deprecationrlangchain_core.callbacksrr*langchain_core.language_models.chat_modelsrrlangchain_core.messagesrrrrrlangchain_core.outputsrrrlangchain_core.utilsr!langchain_community.llms.vertexairrr&langchain_community.utilities.vertexair r!rAr"r#r$r%r{r& getLoggerr/rSr(rMrXr}rrrr.r7r8rs7"  (RRR!6SR)   ;   8 $ "" "@; ;)6;;|<   ? uU?MuU  uUr7