o i^@sUdZddlZddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z ddlZddlZddlZddlmZddlmZddlmZddlmZdd lmZdd lmZejd kridd lmZnejZejr|dd lmZ ddl!m"Z#n*eZ ej$e%d<eZ#ej$e%d<zdd lmZ ddl!m"Z#Wn e&ydZ dZ#YnwdZ'e(dZ)dej*dej+fddZ,de e-ej.fde e-fddZ/de e-ej.fde e-fddZ0 dTde e-ej1fde ej2dej2fddZ3 dTde ej*de4e5fddZ6   dUde ej*d!e e7e-efd"e8de7e-e ed#efefffd$d%Z9d&ej:dej:fd'd(Z;d&e e7e-ed.ed+ed/e-d0e-def d1d2Z?d&ej:d3ed#efdej:fd4d5Z@d&ej:d6ed#efdefd7d8ZAd&ej:d6ed#efdefd9d:ZBd;ejCdZEd;ejCd| ||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.) rr4callableinspectiscoroutinefunctionr UnsupportedFunctionError__name__itemsr# ValueErrorupdate)rr=r> function_mapr8r; tool_name_rrrget_function_maps8      rLargscCsdd|DS)z:Converts float values in dict with no decimal to integers.cSi|] \}}|t|qSr,convert_number_values_for_function_call_args.0keyvaluerrr zEconvert_number_values_for_dict_function_call_args..)rFrMrrr1convert_number_values_for_dict_function_call_argssrXcCsRt|tr |r t|St|trdd|DSt|tr'dd|DS|S)z2Converts float values with no decimal to integers.cSrNrrOrQrrrrUrVz@convert_number_values_for_function_call_args..cSsg|]}t|qSrrO)rRrTrrr sz@convert_number_values_for_function_call_args..)rfloat is_integerintrrFlistrWrrrrPs  rP annotationcCs.z t|o t|tjWStyYdSw)NF)rBisclass issubclasspydantic BaseModel TypeError)r^rrris_annotation_pydantic_models  rdrT param_name func_namecst|tr3t|r3z|d i|WStjy2}ztddd|d|d| d}~wwt|trOt|tkrOt |dfdd|DSt|trmt|tkrmt |\}fd d | DSt|t t fvrt |D]/}t |rt|tust||st|trt|rz t ||WStjyYqywqytddd|d |d t|tr|tur|St||stddd|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 rcsg|] }t|qSrconvert_if_exist_pydantic_model)rRitem)rf item_typererrrYs z3convert_if_exist_pydantic_model..cs i|] \}}|t|qSrrg)rRkv)rfre value_typerrrUsz3convert_if_exist_pydantic_model..z- cannot be converted to parameter annotation .r)rrrdraValidationErrorr UnknownFunctionCallArgumentErrorr]rrrFrrrhr\rZ)rTr^rerferKargr)rfrjrermrrhs      rhfunctioncCsLt|}|j}i}|jD]\}}||vr#t|||j||||<q|SN)rB signaturerE parametersrFrhr^)rMrsrurfconverted_argsreparamrrrconvert_argument_from_function s ryfunction_to_invokec CsXt||}z|di|WSty+}ztd|jd|d|d|d}~wwNzFailed to invoke function z with converted arguments z, from model returned function call argument z because of error rry Exceptionr FunctionInvocationErrorrErMrzrwrqrrrinvoke_function_from_dict_args1s  rc s`t||}z |di|IdHWSty/}ztd|jd|d|d|d}~wwr{r|rrrr$invoke_function_from_dict_args_async?s   rresponserIc Csg}|jdurtt|jdjtjrt|jdjjdurt|jdjjD]R}|js'q!|jj}|durs|jjdurs||}t |jj}zt|t sLdt ||i}Wnt ye}z dt |i}WYd}~nd}~wwtjj||d} || q!|S)6Returns the function response parts from the response.Nrresulterrornamer) candidatesrcontentrContentparts function_callrrMrXrrr}r!Partfrom_function_responser7) rrIfunc_response_partspartrffuncrM func_responserqfunc_response_partrrrget_function_response_partsMs:    rc sLg}|jdurt|jdjtjr|jdjjdur|jdjjD]}|js(q"|jj}|dur|jjdur||}t |jj}z=t|t r`| tj ||dIdH}|j r[d|i}n!d|i}nt|rpdt||IdHi}n dtt||IdHi}Wnty} z dt| i}WYd} ~ nd} ~ wwtjj||d} || q"|S)rNr)rrMrrr)rrrrrrrrrMrXr call_tool FunctionCallisErrorrBrCrasyncio to_threadrr}r!rrr7) rrIrrrfrrMmcp_tool_responserrqrrrr!get_function_response_parts_asyncpsR        rcCs|sdSt|}|r(|jr(|jjdur(t|jjdkr(td|jjddS|jr1|jjdur3dS|jjrV|jjdurVd|jjvrVt|jjdkrVtd|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.)rautomatic_function_callingrr\loggerwarningdisablemodel_fields_setrr8rrrshould_disable_afcsH  rcCsD|stS t|r tdt|}|jr|jjdurtSt|jjS)NzUautomatic function calling is not enabled, but SDK is trying to get max remote calls.)_DEFAULT_MAX_REMOTE_CALLS_AFCrrGrrrr\rrrrget_max_remote_calls_afcs  rcCs$|sdSt|}|js dS|jj S)NT)rrignore_call_historyrrrrshould_append_afc_historys  rcCsx|sdSt|}|jddid}|j|_|jr:t|jr:|jdur)tjid|_|jjdur3i|j_t |jj|S)zVReturns a parsed config with an appended MCP header if MCP tools or sessions are used.Nr4rH)headers) r model_copyr4r has_mcp_tool_usage http_optionsr HttpOptionsrset_mcp_usage_header)rr8config_model_copyrrrparse_config_for_mcp_usages  rcsi}t|}|s d|fS|jddid}|jrhg|_|jD]I}tdurat|trat||IdH}|j|j|jD]"}|jr_|jD]}|j r^| |j rYt d|j d|||j <qEq=q|j |q||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. Nr4rr?r@) rrr4rrr list_toolsextendr6rr#rGr7)rr= parsed_configparsed_config_copyr;mcp_to_genai_tool_adapter genai_toolfunction_declarationrrrparse_config_for_mcp_sessions s8       rcontentschunkcCsH|dur"|jdur"|jdj}t|}t|tr"|dur"|||S)zFAppends the contents of the chunk to the contents list and returns it.Nr)rrt t_contentsrr]r7)rr chunk_contentrrrappend_chunk_contents8s    rfileuser_http_optionsuser_mime_typec Csfd}|}t|tjr9|durtdt|dr d|jvr td|}|dtj ||}||tj n-t |}|rFtj |sMt|dtj |}|dur^t|\}}|durftd|rt|trutjdi|}|}d |_d d d ||d |_ntjd d d d ||d d}t|ttjfr|jduri|_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)rioIOBaserGhasattrrtellseekosSEEK_ENDSEEK_SETfspathpathisfileFileNotFoundErrorgetsize mimetypes guess_typerrrrrr!PathLikebasename) rrr size_bytes mime_typeoffsetfs_pathrKrrrrprepare_resumable_uploadEsb          rrt)NNF)NN)U__doc__rrBrloggingsystypingrrrrrrrrrrarr r r rr r _adaptersr version_infor_UnionGenericAlias TYPE_CHECKINGmcprr mcp.typesrrType__annotations__ ImportErrorr getLoggerrGenerateContentConfigOrDictrrr!BatchJobSourceOrDictr&r(r$r+r3r]r\r<rboolrL StringDictrXobjectrPrdrhryrrGenerateContentResponserrrrrrrtuplerContentListUnionContentListUnionDictrrrHttpOptionsOrDictrrrrrrs~$                     $     >       # 2 8      ,