o %g0 @s(dZddlZddlZddlZddlZddlmZmZmZmZm Z m Z m Z ddl Z ddl mZddl mZddl mZejdkrFdd lmZnejZd Zed Zd ejd ejfddZ d/ded eejd ejfddZ d/d eejd eeeffddZdeeefd eeeffddZ de eee!fe"e!e!fd e eee!fe"e!e!ffddZ#ded e$fddZ%dedededed ef d d!Z&deeefd"ed efd#d$Z'd%ej(d&eeefd e"ej)fd'd(Z* d/d eejd e$fd)d*Z+ d/d eejd e,fd+d,Z- d/d eejd e$fd-d.Z.dS)0zNExtra utils depending on types that are shared between sync and async modules.N)AnyCallableDictOptionalUnionget_args get_origin)_common)errors)types) ) UnionTyperzgoogle_genai.modelsconfigreturncCst|tr tjdi|S|S)N) isinstancedictr GenerateContentConfigrrr[/mnt/skqttb/ctump_chatbot/chatbot/lib/python3.10/site-packages/google/genai/_extra_utils.py%_create_generate_content_config_model(s rsrccCstj|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_namerrrformat_destination1s"  r%cCsbi}|s|St|}|jr/|jD]}t|r.t|r)td|jd|jd|||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.)rtoolscallableinspectiscoroutinefunctionr UnsupportedFunctionError__name__)r function_map config_modeltoolrrrget_function_mapOs     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 hzEconvert_number_values_for_dict_function_call_args..)itemsr0rrr1convert_number_values_for_dict_function_call_argsdsr<cCsRt|tr |r t|St|trdd|DSt|tr'dd|DS|S)z2Converts float values with no decimal to integers.cSr1rr2r4rrrr8ur9z@convert_number_values_for_function_call_args..cSsg|]}t|qSrr2)r5r7rrr zsz@convert_number_values_for_function_call_args..)rfloat is_integerintrr:listr;rrrr3ns  r3 annotationcCs.z t|o t|tjWStyYdSw)NF)r(isclass issubclasspydantic BaseModel TypeError)rBrrris_annotation_pydantic_models  rHr7 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)r5item)rJ item_typerIrrr=s z3convert_if_exist_pydantic_model..cs i|] \}}|t|qSrrK)r5kv)rJrI value_typerrr8sz3convert_if_exist_pydantic_model..z- cannot be converted to parameter annotation .r)rrrHrEValidationErrorr UnknownFunctionCallArgumentErrorrArrr:rrrLr@r>)r7rBrIrJe_argr)rJrNrIrQrrLs      rLfunction_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+ parametersr:rLrB Exceptionr FunctionInvocationError)r0rXrYrJconverted_argsrIparamrUrrrinvoke_function_from_dict_argss4 r_responser,c Csg}|jdurot|jdjtjro|jdjjduro|jdjjD]M}|js'q!|jj}|durn|jjdurn||}t |jj}z dt ||i}Wnt y`}z dt |i}WYd}~nd}~wwtj j||d} || q!|S)z6Returns the function response parts from the response.Nrresulterror)namer`) candidatesrcontentr Contentparts function_callrcr0r<r_r[strPartfrom_function_responseappend) r`r,func_response_partspartrJfuncr0 func_responserUfunc_response_partrrrget_function_response_partss6   rrcCs|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_callingrsr@loggerwarningdisablemodel_fields_setrr-rrrshould_disable_afcsH  rzcCsD|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_AFCrzr#rrtrsr@ryrrrget_max_remote_calls_afc;s  r|cCs$|sdSt|}|js dS|jj S)NT)rrtignore_call_historyryrrrshould_append_afc_historyOs  r~)N)/__doc__r(loggingsystypingrrrrrrrrEr r r version_infor_UnionGenericAliasr{ getLoggerruGenerateContentConfigOrDictrrriCreateBatchJobConfigOrDictrr%rr/r<objectrAr3boolrHrLr_GenerateContentResponserjrrrzr@r|r~rrrrs$                 >    # 8