o 0Xxi, @sdZddlZddlZddlZddlmZmZmZmZmZm Z m Z ddl Z ddl Z ddl mZddl mZddl mZe jdkrFdd lmZnejZd Z d*d ed eejd ejfddZ d*d eejd eeeffddZdeeeefeeefd eeeefeeeffddZded efddZ dedededed ef ddZ!deeefded efddZ"d ej#d!eeefd eej$fd"d#Z% d*d eejd efd$d%Z& d*d eejd e'fd&d'Z( d*d eejd efd(d)Z)dS)+zNExtra utils depending on types that are shared between sync and async modules.N)AnyCallableDictOptionalUnionget_args get_origin)_common)errors)types) ) UnionTypersrcconfigreturncCstj|djp t}d}|jst}d||_|jsO|dr3| dr3|ddd|_|S|drH|p=t}|d ||_|St d ||S) z4Formats the destination uri based on the source uri.)rNgenai_batch_job_zgs://z.jsonliz/destzbq://_dest_zUnsupported source: ) r _CreateBatchJobParametersrCreateBatchJobConfig display_namer timestamped_unique_namedest startswithendswith ValueError)rr unique_namerX/home/air/biblejyuku/back/venv/lib/python3.10/site-packages/google/genai/_extra_utils.pyformat_destination&s"  r cCs||rt|trtjdi|n|}i}|s|S|jr<|jD]}t|r;t|r6t d|j d|j d|||j <q|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.Nr) isinstancedictr GenerateContentConfigtoolscallableinspectiscoroutinefunctionr UnsupportedFunctionError__name__)r config_model function_maptoolrrrget_function_mapDs*    r-argscCsRt|tr |r t|St|trdd|DSt|tr'dd|DS|S)z2Converts float values with no decimal to integers.cSsi|] \}}|t|qSr,convert_number_values_for_function_call_args).0keyvaluerrr dsz@convert_number_values_for_function_call_args..cSsg|]}t|qSrr/)r1r3rrr isz@convert_number_values_for_function_call_args..)r!float is_integerintr"itemslist)r.rrrr0]s  r0 annotationcCs.z t|o t|tjWStyYdSw)NF)r&isclass issubclasspydantic BaseModel TypeError)r;rrris_annotation_pydantic_modelos  rAr3 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)r1item)rC item_typerBrrr5s z3convert_if_exist_pydantic_model..cs i|] \}}|t|qSrrD)r1kv)rCrB value_typerrr4sz3convert_if_exist_pydantic_model..z- cannot be converted to parameter annotation .r)r!r"rAr>ValidationErrorr UnknownFunctionCallArgumentErrorr:rrr9rrrEr8r6)r3r;rBrCe_argr)rCrGrBrJrrE{s      rEfunction_to_invokec Cst|}|j}i}|jD]\}}||vr#t|||j||||<qz|di|WStyI}zt d|d|d|d|d}~ww)NzFailed to invoke function z with converted arguments z, from model returned function call argument z because of error r) r& signaturer) parametersr9rEr; Exceptionr FunctionInvocationError)r.rQrRrCconverted_argsrBparamrNrrrinvoke_function_from_dict_argss4 rXresponser+c Csg}|jdjjD]C}|jsq |jj}||}t|jj}z dt||i}Wnty>}z dt |i}WYd}~nd}~wwt j j ||d}| |q |S)z6Returns the function response parts from the response.rresulterrorN)namerY) candidatescontentparts function_callr\r0r.rXrTstrr Partfrom_function_responseappend) rYr+func_response_partspartrCfuncr.rN func_responserrrget_function_response_partss$  ricCs|rt|trtjd i|n|}|r1|jr1|jjdur1t|jjdkr1td|jjddS|r<|jr<|jj dur>dS|jj r[|jjdur[t|jjdkr[td|jjd|jj S) z6Returns whether automatic function calling is enabled.Nrz6max_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.TFz`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) r!r"r r#automatic_function_callingmaximum_remote_callsr8loggingwarningdisablerr*rrrshould_disable_afcsL  rpcCsXt|rtd|rt|trtjdi|n|}|r$|jr$|jjdur&tSt |jjS)zBReturns the remaining remote calls for automatic function calling.zUautomatic function calling is not enabled, but SDK is trying to get max remote calls.Nr) rprr!r"r r#rjrk_DEFAULT_MAX_REMOTE_CALLS_AFCr8rorrrget_max_remote_calls_afc!s"  rrcCs:|rt|trtjdi|n|}|r|jsdS|jj S)NTr)r!r"r r#rjignore_call_historyrorrrshould_append_afc_history8s  rt)N)*__doc__r&rltypingrrrrrrrsysr>r r r version_infor_UnionGenericAliasrqraCreateBatchJobConfigOrDictrr GenerateContentConfigOrDictr"objectr-r:r0boolrArErXGenerateContentResponserbrirpr8rrrtrrrrs$          >     9