o pkjlq@s8UdZddlZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z m Z mZmZmZmZddlZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZejd krsdd lmZnej Zej!rddl"m#Z$ddl%m&Z'ne Z$ej e(d<e Z'ej e(d<edddZ)dZ*e+dZ,dej-dej.fddZ/dee0ej1fde e0fddZ2dee0ej1fde e0fddZ3 dddee0ej4fde ej5dej5fd d!Z6  "dede ej-d#e7de8e9fd$d%Z:de ej-d&e8e9ddfd'd(Z;   "dfde ej-d)e edej>fd/d0Z?d.eed;e d+e fdej>fdd>e d+e fde fd?d@ZEd.ej>d>e d+e fde fdAdBZFdCejGdDe| ||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)rrNrO function_maprArD tool_name_rrr#get_function_maps8      r]argscCsdd|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..)rWr^rrr#1convert_number_values_for_dict_function_call_argssricCsRt|tr |r t|St|trdd|DSt|tr'dd|DS|S)z2Converts float values with no decimal to integers.cSr_rr`rbrrr#rfrgz@convert_number_values_for_function_call_args..cSsg|]}t|qSrr`)rcrerrr# sz@convert_number_values_for_function_call_args..)r float is_integerintr!rWlistrhrrr#ras  ra annotationcCs.z t|o t|tjWStyYdSwNF)rSisclass issubclasspydantic BaseModel TypeError)rorrr#is_annotation_pydantic_models  rvre 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)rcitem)rx item_typerwrr#rjs z3convert_if_exist_pydantic_model..cs i|] \}}|t|qSrry)rckv)rxrw value_typerr#rf sz3convert_if_exist_pydantic_model..z- cannot be converted to parameter annotation .r)r r!rvrsValidationErrorr UnknownFunctionCallArgumentErrorrnr r rWrrrzrmrk)rerorwrxer\argr)rxr|rwrr#rzs      rzfunctioncCsLt|}|j}i}|jD]\}}||vr#t|||j||||<q|SN)rS signaturerV parametersrWrzro)r^rrrxconverted_argsrwparamrrr#convert_argument_from_function6s rfunction_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 rr ExceptionrFunctionInvocationErrorrVr^rrrrrr#invoke_function_from_dict_argsGs  rc s`t||}z |di|IdHWSty/}ztd|jd|d|d|d}~wwrrrrrr#$invoke_function_from_dict_args_asyncUs   rresponserZc 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) candidatesr contentrContentparts function_callrr^rirrrr'Partfrom_function_responser@) rrZfunc_response_partspartrxfuncr^ func_responserfunc_response_partrrr#get_function_response_partscs:    rc sfg}|jdurt|jdjtjr|jdjjdur|jdjjD]}|js(q"|jj}|dur||}|jjdur>|jjni}t |}zFt|t rm| tj ||dIdH}t |dt |dd} | rhd|i} n!d|i} nt|r}dt||IdHi} n dtt||IdHi} Wnty} z dt| i} WYd} ~ nd} ~ wwtjj|| d } || q"|S) rNr)rr^is_errorisErrorFrrr)rr rrrrrrr^rir call_tool FunctionCallr?rSrTrasyncio to_threadrrr'rrr@) rrZrrrxrraw_argsr^mcp_tool_responserrrrrrr#!get_function_response_parts_asyncs`          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.)r$automatic_function_callingrrmrIrJdisablemodel_fields_setrrArrr#should_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_AFCrrXr$rrrmrrrr#get_max_remote_calls_afcs  rcCsL|r |jr |jjs dS|j}|r|jnd}|jjj}|r"|s$tddSdS)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_argumentsrX)r afc_configdisable_afc_configstream_function_callrrr#'raise_error_for_afc_incompatible_config s rcCs$|sdSt|}|js dS|jj S)NT)r$rignore_call_historyrrrr#should_append_afc_history#s  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.Nr=rY)headers) r$ model_copyr=r has_mcp_tool_usage http_optionsr HttpOptionsrset_mcp_usage_header)rrAconfig_model_copyrrr#parse_config_for_mcp_usage.s  rc s&i}t|}|s d|fS|jddid}|jrg|_ts*|j|j||fSzddlm}WntyAt ddi}Ynw|jD]I}t ||rt || IdH|d}|j|j|jD]$}|j r|j D]} | jdur|| jrtd | jd ||| j<qjqbqE|j|qE||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=rrr DummySessionr)r:rPrQ)rrr=r _is_mcp_loadedextendmcpr ImportErrortyper r list_toolsr;rr)rXr@) rr:rN parsed_configparsed_config_copy_McpClientSessionrDmcp_to_genai_tool_adapter genai_toolfunction_declarationrrr#parse_config_for_mcp_sessionsCsH          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_contentsr rnr@)rr chunk_contentrrr#append_chunk_contentszs    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)r ioIOBaserXhasattrrtellseekosSEEK_ENDSEEK_SETfspathpathisfileFileNotFoundErrorgetsize mimetypes guess_typer!rrrrr'PathLikebasename) rrr size_bytes mime_typeoffsetfs_pathr\rrrr#prepare_resumable_uploadsb          rcCs<|sdSt|}|js dS|jD] }t|ddrdSqdS)zBChecks whether the configuration contains any MCP server requests.Fr<NT)r$r=r?)rrArDrrr#has_agent_platform_mcp_serverss  r config_clsusagecCsdtjd|}|s|}nt|tr|di|}n|}t|ds%|St|dd}|dur9t}t|d|t tj|}|j pDi}dD]>}||vrd|||vr|||vrdtj||vru|| dtj|||<qG||d|7<qG|||<qG||_ |S)z(Returns the usage header for the config.zgoogle-genai-sdk/+rN)z user-agentzx-goog-api-client r) public_version __version__r r!rr?rrsetattrtypingcastrreplace)rrr usage_headerrArexisting_headers header_keyrrr#get_usage_headers:          r rrp)NNF)NN)\__doc__rrSrloggingrrsysrrrrrrrrr r rsrr r rrrrrr _adaptersr version_infor_UnionGenericAlias TYPE_CHECKINGrrr mcp.typesrr__annotations__rr getLoggerrIGenerateContentConfigOrDictr"r$r'BatchJobSourceOrDictr,r.r*r1r9boolrnrmrErMr!r] StringDictriobjectrarvrzrrrGenerateContentResponserrrrrrrrtuplerContentListUnionContentListUnionDictrrrHttpOptionsOrDictrrrr rrrr#s,                      $     >       # : 8       7  J