+jMeUdZddlZddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z ddlZddlZddlZddlmZddlmZddlmZddlmZdd lmZdd lmZejd krdd lmZnejZejr dd lmZ ddl!m"Z#n5eZ ej$e%d<eZ#ej$e%d< dd lmZ ddl!m"Z#n#e&$rdZ dZ#YnwxYwdZ'ej(dZ)dej*dej+fdZ,de e-ej.fde e-fdZ/de e-ej.fde e-fdZ0 dAde e-ej1fde ej2dej2fdZ3 dBde ej*de4de5e6fdZ7 dCde ej*de e8e-efde4de8e-e edefefffd Z9d!ej:dej:fd"Z;d!e e8e-ed&ed$ed'e-d(e-def d)Z?d!ej:d*edefdej:fd+Z@d!ej:d,edefdefd-ZAd!ej:d,edefdefd.ZBd/ejCd0e8e-e edefeffde5ejDfd1ZEd/ejCd0e8e-e edefeffde5ejDfd2ZF dAde ej*de4fd3ZG dAde ej*de6fd4ZHde ej+ddfd5ZI dAde ej*de4fd6ZJ dAde ej*de ej+fd7ZK dBde ej*de4deLe ej+e8e-efffd8ZMd9e ejNejOfd:ejCde ejNejOffd;ZP dDde e-deLejTe6e-ffd?ZU dAde ej*de4fd@ZVdS)EzNExtra utils depending on types that are shared between sync and async modules.N)AnyCallableDictOptionalUnionget_args get_origin)_common) _mcp_utils) _transformers)errors)types)McpToGenAiToolAdapter) ) UnionType) ClientSession)ToolMcpClientSessionMcpToolrzgoogle_genai.modelsconfigreturncRt|trtjdi|S|S)N) isinstancedictrGenerateContentConfig)rs ]/home/wildlama/visual-decline/.venv/lib/python3.11/site-packages/google/genai/_extra_utils.py%_create_generate_content_config_modelr <s2  & 0 0 0 00 MsrccPt|tr|dr|St|tr-|dr|dr|ddndSt|t jr|jr|jr |jdndSdS)z9Extracts the first GCS URI from the source, if available.zgs://gcs_urirN)rstr startswithrgetrBatchJobSourcer$r"s r _get_gcs_urir*EsS3cnnW553 J#t3!3!33 #I 83y>!  D8#u+,,33 [ 23;q>>d2 r!ct|tr|dr|St|tr|dr|dSt|t jr|jr|jSdS)z8Extracts the BigQuery URI from the source, if available.zbq:// bigquery_uriN)rr%r&rr'rr(r,r)s r_get_bigquery_urir-RsScnnW55 J#t!8!8 ~ #u+,,1A   r!c`|tj}d}|jstj}d||_|jskt |}t|}|r(|dr|ddd|_n#|r!|ptj}|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+  ;133K:[::F @!#&&N+C00@.11(;;@%crc*111fkk @D7#B#D#Dk*??+??fk -r!Fis_agent_platformcL|sgSt|}g}|r|js|St|jD]n\}}t|tjs t |ddr||t |ddr|s||o|S)z9Checks if the config contains any AFC incompatible tools.function_declarationsN mcp_servers)r tools enumeraterrrgetattrappend)rr: config_modelincompatible_tools_indexesindextools r"find_afc_incompatible_tool_indexesrFys  I6v>>,*, &\/& %%|122//keT dEJ ' 't,d33/ ''...t]D))/2C/ ''... ##r!mcp_to_genai_tool_adaptersis_caller_method_async.ci}|s|St|}|jr^|jD]V}t|rEtj|r'|s%t jd|jd|jd|||j<W|rm|st jd|D]-\}}| |rtd|d.| ||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 r>callableinspectiscoroutinefunctionrUnsupportedFunctionError__name__itemsr' ValueErrorupdate)rrGrH function_maprBrE tool_name_s rget_function_maprWs^OQ,  6v>>, +"++ $+  &t , , 5K /G$-GGGGG   '+ T]# 4 !  + ?  388::  1   ) $ $  CI C C C    2333 r!argsc>d|DS)z:Converts float values in dict with no decimal to integers.c4i|]\}}|t|Sr,convert_number_values_for_function_call_args.0keyvalues r zEconvert_number_values_for_dict_function_call_args..s7    #u 7 > >   r!)rQrXs r1convert_number_values_for_dict_function_call_argsrcs+      r!ct|tr#|rt|St|trd|DSt|t r d|DS|S)z2Converts float values with no decimal to integers.c4i|]\}}|t|Srr[r]s rraz@convert_number_values_for_function_call_args..s7    C 9% @ @   r!c,g|]}t|Srr[)r^r`s r z@convert_number_values_for_function_call_args..s.   @E4U;;   r!)rfloat is_integerintrrQlistrbs rr\r\se!2!2 t99d  **,,   d  IM    +r! annotationc tj|ot|tjS#t $rYdSwxYwNF)rMisclass issubclasspydantic BaseModel TypeError)rls ris_annotation_pydantic_modelrtsV ?: & & :H&,,  55s -0 >>r` param_name func_namec t|trRt|rC |d i|S#tj$r(}t jddd|d|d| d}~wwxYwt|tr=t|tkr%t|dfd|DSt|trLt|tkr4t|\}fd| DSt|ttfvrt|D]}t|rt|tus4t||s$t|tr9t|r* t||cS#tj$rYwxYwt jddd|d |d t|tr |tur|St||s!t jddd|d|d |S) NzFailed to parse parameter z for function z> from function call part because function call argument value z- is not compatible with parameter annotation z, due to error rc4g|]}t|Srconvert_if_exist_pydantic_model)r^itemrv item_typerus rrgz3convert_if_exist_pydantic_model..s7     (iYOO   r!c <i|]\}}|t|Srry)r^kvrvru value_types rraz3convert_if_exist_pydantic_model..s>    Aq *1j*i P P   r!z- cannot be converted to parameter annotation .r)rrrtrqValidationErrorr UnknownFunctionCallArgumentErrorrkr rrQrrrzrjrh) r`rlrurverVargr|rs `` @@rrzrzsgt !=j!I!I  Z %   #  3 -z - - - - - - - -*+ - -  tJ!7!74!?!?$$Q'I         tJ!7!74!?!?Z((MAz      KKMM     y111 ##   C== '__44 s # #5%%5*Fs*K*K5 0S*i     '    (   1 Z             s e 3 3 L E: & &  1 CZ C Cy C CDI C C5? C C C   ,s'2A)#A$$A)'F;;G  G functionctj|}|j}i}|jD])\}}||vr t |||j||||<*|SN)rM signaturerP parametersrQrzrl)rXrrrvconverted_argsruparams rconvert_argument_from_functionrs)))).$/5577j%T#B z      $$nZ r!function_to_invokec t||} |di|S#t$r*}tjd|jd|d|d|d}~wwxYwNzFailed to invoke function z with converted arguments z, from model returned function call argument z because of error rr ExceptionrFunctionInvocationErrorrPrXrrrs rinvoke_function_from_dict_argsr-s2$8JKK.   / / / //   ( 6%7%@ 6 6 . 6 6 6 623 6 6  s A%A  Ac Kt||} |di|d{VS#t$r*}tjd|jd|d|d|d}~wwxYwrrrs r$invoke_function_from_dict_args_asyncr;s2$8JKK.##55n55 5 5 5 5 5 55   ( 6%7%@ 6 6 . 6 6 6 623 6 6  s " A%AAresponserTcdg}|j%t|jdjtjr|jdjj|jdjjD]}|js |jj}||jj||}t|jj} t|tsdt||i}n(#t$r}dt|i}Yd}~nd}~wwxYwtj||} || |S)6Returns the function response parts from the response.Nrresulterrornamer) candidatesrcontentrContentparts function_callrrXrcrrrr%Partfrom_function_responserA) rrTfunc_response_partspartrvfuncrX func_responserfunc_response_parts rget_function_response_partsrIsZ % X(+3U] C C&  a ( . :#A&.477  $)i  4#5#:#FI&@   #   ,D"788 8tDDM , , ,"CFF+------ ,"Z>>]?   ""#5666 s)'C C6C11C6cvKg}|jt|jdjtjr|jdjji|jdjjD]P}|js |jj}|6||}|jj |jjni}t|} t|tr@| tj ||d{V}|j rd|i} nUd|i} nPtj|rdt!||d{Vi} n#dt#jt&||d{Vi} n(#t($r} dt+| i} Yd} ~ nd} ~ wwxYwtj|| } || R|S)rNr)rrXrrr)rrrrrrrrrXrcr call_tool FunctionCallisErrorrMrNrasyncio to_threadrrr%rrrA) rrTrrrvrraw_argsrXmcp_tool_responserrrs r!get_function_response_parts_asyncrlsB % X(+3U] C C&  a ( . :#A&.4%7%7  $)i  I&!&2   # #  AJJ , 3 4 4 &*nn" ===''!!!!!! !( <&(9:mm'):;mm*400  DT4 P PPPPPPPMM  12D$!!M  , , ,"CFF+------ ,"Z>>]?   ""#5666 s3B%E E>#E99E>c|sdSt|}|r[|jrT|jj Gt|jjdkr*td|jjddS|jr |jjdS|jjr`|jj Sd|jjvrEt|jjdkr(td|jjd |jjS) z6Returns whether automatic function calling is enabled.FNrz6max_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_callingrrjloggerwarningdisablemodel_fields_setrrBs rshould_disable_afcrsR  56v>>,   1   1 F   l5J K Kq P P NN >  3 H > > > 4  1  0 8 @ 5-5  1 F   !  0 A B B l5J K Ka O O NN D  3 H D D D     0 88r!c|stS t|rtdt|}|jr |jjtSt |jjS)NzUautomatic function calling is not enabled, but SDK is trying to get max remote calls.)_DEFAULT_MAX_REMOTE_CALLS_AFCrrRr rrrjrs rget_max_remote_calls_afcrs| ) ((J    7v>>,  1)  0 E M (( \ 4 I J JJr!c|r|jr |jjsdS|j}|r|jnd}|jjj}|r|st ddSdS)z9Raises an error if the config is not compatible with AFC.NFavRunning in streaming mode with stream_function_call_arguments enabled, this feature is not compatible with automatic function calling (AFC). Please set config.automatic_function_calling.disable to True to disable AFC or leave config.tool_config. function_calling_config.stream_function_call_arguments to be empty or set to False to disable streaming function call arguments.) tool_configfunction_calling_configrrstream_function_call_argumentsrR)r afc_configdisable_afc_configstream_function_calls r'raise_error_for_afc_incompatible_configrs      7  F0*-7Bz))U 0O"4  I  r!cT|sdSt|}|jsdS|jj S)NT)r rignore_call_historyrs rshould_append_afc_historyrs=  46v>>,  0 4  4 H HHr!cb|sdSt|}|ddi}|j|_|jrptj|jrW|jt ji|_|jj i|j_tj |jj|S)zVReturns a parsed config with an appended MCP header if MCP tools or sessions are used.Nr>rS)headers) r model_copyr>r has_mcp_tool_usage http_optionsr HttpOptionsrset_mcp_usage_header)rrBconfig_model_copys rparse_config_for_mcp_usagers  46v>>,#--gt_-EE(.LJ9,:LMML%-','8'D'D'D$%-5/1$,#$5$B$JKKK r!cXKi}t|}|sd|fS|ddi}|jrg|_|jD]}tt |trt ||d{V|}|j|j|jD]V}|jrM|jD]E}|j r<| |j rtd|j d|||j <FW|j |||fS)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>r)r:rJrK) rrr>rrr list_toolsextendr<rr'rRrA) rr:rG parsed_configparsed_config_copyrEmcp_to_genai_tool_adapter genai_toolfunction_declarations rparse_config_for_mcp_sessionsr$sBD,V44- , + ++%///GG.!#..  %*T;K*L*L %$9 ))))))))=N% % % !  ''(A(GHHH39  J  - (2(H  $%*-112F2KLL"&27&&& .++?+DE   ''---- 7 77r!contentschunkc|Y|jR|jdj}tj|}t |t r||||S)zFAppends the contents of the chunk to the contents list and returns it.Nr)rrt t_contentsrrkrA)rr chunk_contents rappend_chunk_contentsrQs_  5+7$Q'/M|H%%H(D!!%m&?oom$$$ /r!fileuser_http_optionsuser_mime_typec d}|}t|tjr|tdt |drd|jvrtd|}|dtj ||z }||tj ntj |}|rtj |st|dtj |}|t!j|\}}|td|r?t|t$rt'jdi|}|}d |_d d d ||d |_nt'jd d d d ||d }t|t.tjfr5|ji|_tj ||jd<|||fS)a-Prepares the HTTP options, file bytes size and mime type for a resumable upload. This function inspects a file (from a path or an in-memory object) to determine its size and MIME type. It then constructs the necessary HTTP headers and options required to initiate a resumable upload session. NzfUnknown mime type: Could not determine the mimetype for your file please set the `mime_type` argumentmodebz'The file must be opened in binary mode.rz is not a valid file path.ziUnknown mime type: Could not determine the mimetype for your file please set the `mime_type` argumentzapplication/json resumablestart)z Content-TypezX-Goog-Upload-ProtocolzX-Goog-Upload-Commandz#X-Goog-Upload-Header-Content-Lengthz!X-Goog-Upload-Header-Content-Type) api_versionrzX-Goog-Upload-File-Namer)rioIOBaserRhasattrrtellseekosSEEK_ENDSEEK_SETfspathpathisfileFileNotFoundErrorgetsize mimetypes guess_typerrrrrr%PathLikebasename) rrr size_bytes mime_typeoffsetfs_pathrVrs rprepare_resumable_uploadr^sD*)bi    8  tVD DI  BCCC YY[[FIIav%JIIfbk""""iooG C"'..11C AAA B BB))J)'22li  ;   #T**A+@@.?@@$L!L*"-!(2<09^ L$.&1%,6@?4=      LsBK())M#! 68g6F6Ft6L6LL23 z9 ,,r!cz|sdSt|}|jsdS|jD]}t|ddrdSdS)zBChecks whether the configuration contains any MCP server requests.Fr=NT)r r>r@)rrBrEs rhas_agent_platform_mcp_serversrse  U8@@L   U" }d + +tt 5r!rrn)NNF)NN)W__doc__rrMrloggingsystypingrrrrrrr rrrqrr r r rrr _adaptersr version_infor_UnionGenericAlias TYPE_CHECKINGmcprr mcp.typesrrType__annotations__ ImportErrorr getLoggerrGenerateContentConfigOrDictrr r%BatchJobSourceOrDictr*r-r(r1r9boolrkrjrFrrW StringDictrcobjectr\rtrzrrrGenerateContentResponserrrrrrrrtuplerContentListUnionContentListUnionDictrrrHttpOptionsOrDictrrrrr!rr s| UTT  MMMMMMMMMMMMMMMMMM  ,,,,,,w')  333333'''''''"%FK%%%'6;555555))))))) GGG!#  0 1 1  -    sE.. /  c]      sE.. /  c]    48 sE(( ) U/ 0 6;?#$$ U6 7$$ #Y$$$$4;? #( !! U6 7! ( S' '(!! ! !  #uXc3h')>>? ?@ !!!!H    S&[!4<7 8 4V d6lF 23$ S T    ; ;;-0;=@;;;;;|  (0c(: "    2:382D         2:382D      + sE(38"46K"KLLM  %*    F1+1sE(38"46K"KLLM1 %*1111j;?4949 U6 749 49494949p;?KK U6 7KKKKK(HU=X4Y 4;?II U6 7I IIII;? U6 7 e)*,;?#*8*8 U6 7*8*8  U ()# #$&*8*8*8*8Z E*E,FFG   (  5 !5#= =>    <@$(F-F- R[%ry0 1F- 78F-SMF-    F-F-F-F-T;?   U6 7        s$ B11 B=<B=