[ThOUdZddlZddlZddlZddlZddlmZmZmZmZm Z m Z m Z ddl Z ddl mZddl mZddl mZddl mZdd l mZdd lmZej,d k\rdd lmZn ej0Zej2r dd lmZddlmZn/eZej@e!d<eZej@e!d< dd lmZddlmZdZ#ejHdZ%dejLdejNfdZ(de e)ejTfdee)fdZ+de e)ejTfdee)fdZ, d8de e)ejZfdeej\dej\fdZ/ d9deejLdee0e)efde1de0e)e edefefffdZ2dejfdejffdZ4de e0e)e5fe6e5e5fde e0e)e5fe6e5e5ffd Z7d!ede1fd"Z8d#ed!ed$e)d%e)def d&Z9dejfd'edefdejffd(Z:dejfd)edefdefd*Z;dejfd)edefdefd+Z! 8D8#u++, [[3;;q>2d2 r!ct|tr|jdr|St|tr|j dr|dSt|t j r|jr |jSy)z8Extracts the BigQuery URI from the source, if available.zbq:// bigquery_uriN)rr%r&rr'rr(r,r)s r_get_bigquery_urir-OseScnnW5 J#t!8 ~ #u++,1A1A    r!c^|tj}d}|jstj}d||_|j s^t |}t|}|r |jdr|ddd|_|S|r$|xstj}|d||_|S)zBFormats the destination uri based on the source uri for Vertex AI.Ngenai_batch_job_z.jsonliz/dest_dest_) rCreateBatchJobConfig display_namer timestamped_unique_namedestr*r-endswith)r"r unique_namegcs_source_uribigquery_source_uris rformat_destinationr9\s  ^  ' ' )F+   113K,[M:F !#&N+C0.11(;%cr*+51fk - D7#B#B#Dk*+6+?fk -r!mcp_to_genai_tool_adaptersis_caller_method_async.ci}|s|St|}|jrs|jD]d}t|stj|r2|s0t j d|jd|jd|||j<f|r`|st j d|jD]%\}}|j|std|d|j||S)z'Returns a function map from the config.z Function zh is a coroutine function, which is not supported for automatic function calling. Please manually invoke z to get the function response.z3MCP tools are not supported in synchronous methods.Tool $ is already defined for the request.) r toolscallableinspectiscoroutinefunctionrUnsupportedFunctionError__name__itemsr' ValueErrorupdate)rr:r; function_map config_modeltool tool_name_s rget_function_maprMvsOQ,  6v>,""+ $  & &t ,5K//$--)'EG  '+ T]]#+ !  + + ? 388: 1   ) $I;B C   23 r!argsch|jDcic]\}}|t|c}}Scc}}w)z:Converts float values in dict with no decimal to integers.)rE,convert_number_values_for_function_call_argsrNkeyvalues r1convert_number_values_for_dict_function_call_argsrTs9    #u 7 >>  s.c>t|tr|jr t|St|tr-|j Dcic]\}}|t |c}}St|tr|Dcgc] }t |c}S|Scc}}wcc}w)z2Converts float values with no decimal to integers.)rfloat is_integerintrrErPlistrQs rrPrPse!2 t9d**,  C 9% @@ dIM @E4U;  +   s B=B annotationc tj|xrt|tjS#t $rYywxYw)NF)rAisclass issubclasspydantic BaseModel TypeError)rZs ris_annotation_pydantic_modelrasC ??: & :H&&,  s 03 ??rS param_name func_namec :t|trt|r |d i|St|trt|\}}|jD cic]\} } | t| |||c} } St|ttfvrt|D]W} t| rt| tus)t|| st|ts from function call part because function call argument value z- is not compatible with parameter annotation z, due to error rz- cannot be converted to parameter annotation .r)rrrar^ValidationErrorr UnknownFunctionCallArgumentErrorrYr rconvert_if_exist_pydantic_modelrErrrXrV) rSrZrbrce item_typeitemrL value_typekvargs rrhrhsot!=j!I  %  tJ!74!?$Q'I   (iYO tJ!74!?Z(MAzKKM  Aq *1j*i PP   y11 #  C=Z_4 s #%*Fs*K 0S*i     1 1 $ZL1 ; >]?  ""#56)7*   ,"CF+- ,s,E E  EE cKg}|jt|jdjtjrp|jdjj L|jdjj D]%}|j s|j j}|*|j jA||}t|j j} t|trD|jtj||d{}|jrd|i}n?d|i}n:tj|rdt!||d{i}ndt#||i}tj(j+||} |j-| (|S77M#t$$r} dt'| i}Yd} ~ Vd} ~ wwxYww)rNr)rrNrrr)rrrrrrrrrNrTr call_tool FunctionCallisErrorrArBrrr|r%rrr) rrHrrrcrrNmcp_tool_responserrirs r!get_function_response_parts_asyncrPs % X((+33U]] C   a ( ( . . :##A&..44 7   $$))i  4#5#5#:#:#FI&@    # #  , 3 4&*nn"" ='! !((&(9:m'):;m**40 DT4 PPM 8tDM #ZZ>>]?  ""#56A 7B +!Q  ,"CF+- ,s`B2G5G $G19F:*F6+?F:*F8+F:?7G6F:8F:: G GGGGc|syt|}|rs|jrg|jj Pt|jjdkr.tj d|jjdy|jr|jj y|jj r~|jj gd|jjvrOt|jjdkDr-tj d|jjd|jj S) z6Returns whether automatic function calling is enabled.Frz6max_remote_calls in automatic_function_calling_config zu is less than or equal to 0. Disabling automatic function calling. Please set max_remote_calls to a positive integer.Tmaximum_remote_callsz`automatic_function_calling.disable` is set to `True`. And `automatic_function_calling.maximum_remote_calls` is a positive number a9. Disabling automatic function calling. If you want to enable automatic function calling, please set `automatic_function_calling.disable` to `False` or leave it unset, and set `automatic_function_calling.maximum_remote_calls` to a positive integer or leave `automatic_function_calling.maximum_remote_calls` unset.)r automatic_function_callingrrXloggerwarningdisablemodel_fields_setrrIs rshould_disable_afcrsF  6v>,  1 1  1 1 F F   l55JJ Kq P NN  3 3 H H IJ> >   1 1  0 0 8 8 @ --55  1 1 F F   !  0 0 A A B l55JJ Ka O NN   3 3 H H IJD D   0 0 8 88r!c|stS t|r tdt|}|jr|jj tSt |jj S)NzUautomatic function calling is not enabled, but SDK is trying to get max remote calls.)_DEFAULT_MAX_REMOTE_CALLS_AFCrrFr rrrXrs rget_max_remote_calls_afcrsn  ((J   7v>,  1 1  0 0 E E M (( \ 4 4 I I JJr!cf|syt|}|jsy|jj S)NT)r rignore_call_historyrs rshould_append_afc_historyrs5  6v>,  0 0   4 4 H H HHr!c|syt|}|jddi}|j|_|jrtj|jrw|j t ji|_|j ji|j _tj|j j|S)zVReturns a parsed config with an appended MCP header if MCP tools or sessions are used.Nr?rG)headers) r model_copyr?r has_mcp_tool_usage http_optionsr HttpOptionsrset_mcp_usage_header)rrIconfig_model_copys rparse_config_for_mcp_usagers  6v>,#--gt_-E(..J99,:L:LM%%-','8'8'D$%%--5/1$$,##$5$B$B$J$JK r!cKi}t|}|sd|fS|jddi}|jrg|_|jD]}tt |trt ||j d{}|jj|j|jD]p}|js|jD]R}|js|j|jrtd|jd|||j<Tr|jj|||fS7έw)zReturns a parsed config with MCP sessions converted to GenAI tools. Also returns a map of MCP tools to GenAI tool adapters to be used for AFC. Nr?rr=r>) rrr?rrr list_toolsextendfunction_declarationsrr'rFr)rr: parsed_configparsed_config_copyrJmcp_to_genai_tool_adapter genai_toolfunction_declarations rparse_config_for_mcp_sessionsrshBD,V4-  + ++%///G!##.  %*T;K*L$9 ))% !   ''(A(G(GH399 J  - -(2(H(H $%**-112F2K2KL"27789&& .++?+D+DE     ''-)., 7 77'*s A:E <E =AE E "A*E contentschunkc|a|jT|jdj}tj|}t |t r||j |yyyyy)z7Appends the contents of the chunk to the contents list.Nr)rrt t_contentsrrYr)rr chunk_contents rappend_chunk_contentsrse  5++7$$Q'//M||H%H(D!m&?oom$'@!8r!rr)NNF)K__doc__rAloggingsystypingrrrrrrr r^r r r rrr _adaptersr version_infor_UnionGenericAlias TYPE_CHECKINGmcprr mcp.typesrrType__annotations__ ImportErrorr getLoggerrGenerateContentConfigOrDictrr r%BatchJobSourceOrDictr*r-r(r1r9rboolrM StringDictrTobjectrYrPrarhrwrrGenerateContentResponserrrrrXrrrtuplerContentListUnionContentListUnionDictrrr!rrs U MMM ,w'') 3'"%FKK%'6;;5) !#   0 1  - -     sE... /  c]   sE... /  c] 48 sE((( ) U// 0 6;? #( ! U66 7! ( S' '(!! ! !  #uXc3h')>>? ?@ !H     S&[!4<7 8 4V d6lF 23$ S T ; ;;-0;=@;;|   (0c(: "     2:382D       2:382D   ++ sE(38"46K"KLLM  %** F,++,sE(38"46K"KLLM, %**,`;?49 U66 749 49p;?K U66 7KK*;?I U66 7I I;? U66 7 e))*,;?)8 U66 7)8  U ( ()# #$&)8X %E**E,F,FFG %  ( ( %  %M Gs, M MM