Li/bdZddlZddlZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddl m Zdd l mZej d Zd e j$d efd Z dde j$deej*d e j$fdZGdde j.ZGdde j.Zy)z&[Experimental] Auth Tokens API client.N)AnyDictListOptional) urlencode) _api_module)_common)_tokens_converters)typeszgoogle_genai.tokenssetupreturncg}|jD]N\}}t|tr%|r#|jDcgc] }|d| }}n|g}|j |Pdj |Scc}w)zReturn field_masks.,)items isinstancedictkeysextendjoin)r fieldskvkkfields Q/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/google/genai/tokens.py_get_field_masksrsx &kkmda!Tq%&VVX.r!AbT{.e.ce MM%  &  /sA5 request_dictconfigc |jd}|r|jdr|jd|d<t|d}t|tr"|jd|jdr2t|tj r|j |j s||d<n7t|tr|jd&t|tj r|j |jddn|dr|jd}t jjj}|r$g}|D]}||vrd|}|j|ng}|r|dzdj|zn||d<nP|jddn=|jdg}dj|}|r||d<n|jdd|jds|jdd|S)z"Converts bidiGenerateContentSetup.bidiGenerateContentSetupr lock_additional_fieldsN fieldMaskzgenerationConfig.r) getrrrr CreateAuthTokenConfigr#popGenerationConfig model_dumprappendr) rr bidi_setup field_maskadditional_fields_listgeneration_config_listfield_mask_listrfield_mask_strs r"_convert_bidi_setup_to_token_setupr1,s   :;*JNN7+0:~~g/FL+,",/I"JKJ 64 JJ/ 0 < 3465667  ) ) 5--#-l;64 JJ/ 0 865667  ) ) 1{D) k "4@4D4D 5 %557BBDIIK + (E , ,'w/E   ' (  s SXXo6 6;{D)!!+r2JXXj)N"0l;{D)   4 5/6 c~eZdZdZej ddddeejdejfdZ y)Tokenszb[Experimental] Auth Tokens API client. This class provides methods for creating auth tokens. [The SDK's token creation implementation is experimental, and may change in future versions.Nr r rc~tj|}|jjr t dt j |j|}|jd}|rdj|}nd}|jd}|r|dt|}|jdd|r t||}d}|8|j,|jj|jj}tj|}tj |}|jj#d |||}|j$sint'j(|j$} tj*j-| |j/ } |jj1| | S) a [Experimental] Creates an auth token. Args: config (CreateAuthTokenConfig): Optional configuration for the request. The CreateAuthTokenConfig's `live_constrained_parameters` attrubite Can be used to lock the parameters of the live session so they can't be changed client side. This behavior has two basic modes depending on whether `lock_additional_fields` is set: If you do not pass `lock_additional_fields` the entire `live_constrained_parameters` is locked and can't be changed by the token's user. If you set `lock_additional_fields`, then the non-null fields of `live_constrained_parameters` are locked, and any additional fields specified in `lock_additional_fields`. Usage: .. code-block:: python # Case 1: If LiveEphemeralParameters is unset, unlock LiveConnectConfig # when using the token in Live API sessions. Each session connection can # use a different configuration. config = types.CreateAuthTokenConfig( uses=10, expire_time='2025-05-01T00:00:00Z', ) auth_token = client.tokens.create(config=config) .. code-block:: python # Case 2: If LiveEphemeralParameters is set, lock all fields in # LiveConnectConfig when using the token in Live API sessions. For # example, changing `output_audio_transcription` in the Live API # connection will be ignored by the API. auth_token = client.tokens.create( config=types.CreateAuthTokenConfig( uses=10, live_constrained_parameters=types.LiveEphemeralParameters( model='gemini-live-2.5-flash-preview', config=types.LiveConnectConfig( system_instruction='You are an LLM called Gemini.' ), ), ) ) .. code-block:: python # Case 3: If LiveEphemeralParameters is set and lockAdditionalFields is # empty, lock LiveConnectConfig with set fields (e.g. # system_instruction in this example) when using the token in Live API # sessions. auth_token = client.tokens.create( config=types.CreateAuthTokenConfig( uses=10, live_constrained_parameters=types.LiveEphemeralParameters( config=types.LiveConnectConfig( system_instruction='You are an LLM called Gemini.' ), ), lock_additional_fields=[], ) ) .. code-block:: python # Case 4: If LiveEphemeralParameters is set and lockAdditionalFields is # set, lock LiveConnectConfig with set and additional fields (e.g. # system_instruction, temperature in this example) when using the token # in Live API sessions. auth_token = client.tokens.create( config=types.CreateAuthTokenConfig( uses=10, live_constrained_parameters=types.LiveEphemeralParameters( model='gemini-live-2.5-flash-preview', config=types.LiveConnectConfig( system_instruction='You are an LLM called Gemini.' ), ), lock_additional_fields=['temperature'], ) ) r6=This method is only supported in the Gemini Developer client._url auth_tokens_query?r Npostresponsekwargs)r CreateAuthTokenParameters _api_clientvertexai ValueErrortokens_converters#_CreateAuthTokenParameters_to_mldevr% format_maprr'r1r http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloads AuthToken_from_responser)_verify_response selfr parameter_modelrrequest_url_dictpath query_paramsrHr? response_dict return_values rcreatez Tokens.createws@55O    I 'JJ    l&))&1 ''(89##H-LvQy./ 0dXt$7 fMl04L#  " " .  " " / / ;$++88l**<8L66|DL''lLH'mmBHMM1JM??11'A'A'C2L %%l3 r2 __name__ __module__ __qualname____doc__r experimental_warningrr CreateAuthTokenConfigOrDictrOrZr2rr4r4qsW  7+ FJO A ABO O Or2r4c~eZdZdZej ddddeejdejfdZ y) AsyncTokenszu[Experimental] Async Auth Tokens API client. This class provides asynchronous methods for creating auth tokens. r5Nr6r rcKtj|}|jjr t dt j |j|}|jd}|rdj|}nd}|jd}|r|dt|}|jddt||}d}|8|j,|jj|jj}tj|}tj |}|jj#d ||| d{}|j$sint'j(|j$} tj*j-| |j/ } |jj1| | S7}w) aCreates an auth token asynchronously. Support in v1alpha only. Args: config (CreateAuthTokenConfig): Optional configuration for the request. Usage: .. code-block:: python client = genai.Client( api_key=API_KEY, http_options=types.HttpOptions(api_version='v1alpha'), ) auth_token = await client.aio.tokens.create( config=types.CreateAuthTokenConfig( uses=10, live_constrained_parameters=types.LiveEphemeralParameters( model='gemini-live-2.5-flash-preview', config=types.LiveConnectConfig( system_instruction='You are an LLM called Gemini.' ), ), ) ) r6r8r9r:r;r<r Nr=)rHr>)r rArBrCrDrErFr%rGrr'r1r rHr rIrJ async_requestrLrMrNrOrPr)rQrRs rrZzAsyncTokens.createsD55O     I 'JJ    l&))&1 ''(89##H-LvQy./ 0dXt$6lFKL04L#  " " .  " " / / ;$++88l**<8L66|DL%%33 ! 4H 'mmBHMM1JM??11'A'A'C2L %%l3 sEG  G A>G r[rbr2rrdrd sW  7+ FJS A ABS S Sr2rd)N)r_rMloggingtypingrrrr urllib.parserr r r rEr getLoggerlogger StringDictstrrrar1 BaseModuler4rdrbr2rrps - ,,"5   0 1 G.. 3  ;?B$$B U66 7B BJY[ # #Yx]+((]r2