+j/6dZddlZddlZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddl m Zdd l mZejd Zd e jd efd Z dde jdeejd e jfdZGdde jZGdde jZdS)z&[Experimental] Auth Tokens API client.N)AnyDictListOptional) urlencode) _api_module)_common)_tokens_converters)typeszgoogle_genai.tokenssetupreturncg}|D]U\}t|tr#|r!fd|D}ng}||Vd|S)zReturn field_maskscg|] }d| S).).0kkks W/home/wildlama/visual-decline/.venv/lib/python3.11/site-packages/google/genai/tokens.py z$_get_field_masks..$s#...r!{{b{{...,)items isinstancedictkeysextendjoin)r fieldsvfieldrs @r_get_field_masksr#s &kkmmda!Tq....QVVXX...eece MM% &  r request_dictconfigc|d}|r|dr|d|d<t|d}t|tr*|d|dr(t|tjr|j|js||d<ndt|tr|d!t|tjr|j|ddn|dr|d}t j }|r&g}|D] }||vrd|}| |!ng}|r|dzd |zn||d<n`|ddnI|dg}d |}|r||d<n|dd|ds|dd|S)z"Converts bidiGenerateContentSetup.bidiGenerateContentSetupr lock_additional_fieldsN fieldMaskzgenerationConfig.r) getr#rrr CreateAuthTokenConfigr(popGenerationConfig model_dumprappendr) r$r% bidi_setup field_maskadditional_fields_listgeneration_config_listfield_mask_listr"field_mask_strs r"_convert_bidi_setup_to_token_setupr6,s  :;;*8*JNN7++8*0:~~g/F/FL+,",/I"JKKJ 64  )* JJ/ 0 0 < 344 = 65677 =  ) 5- 6#-l;64  * JJ/ 0 0 865677 9  ) 1{D)))) k "*4@4D4D 55 %577BBDDIIKK + ( (E , , ,///E   ' ' ' ' (  *s SXXo66 6 6;{D))))!!+r22JXXj))N*"0l;{D)))   4 5 57/666 rcteZdZdZejddddeejdej fdZ dS)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%rcLtj|}|jjrt dt j|j|}|d}|rd|}nd}|d}|r|dt|}| dd|rt||}d}||j |j j |j j }tj|}tj|}|jd |||}|jsint'j|j} tj| | } |j| | 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.auth_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.auth_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.auth_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.auth_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'], ) ) r:=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,r6r% http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloads AuthToken_from_responser._verify_response selfr%parameter_modelr$request_url_dictpath query_paramsrLrC response_dict return_values rcreatez Tokens.createws@5O    I  'J   l&))&11 ''(899##H--L1 0 0y.. 0 0dXt$$$N7 fMMl04L#  " .  " / ;$+8l*<88L6|DDL''lLH'mJBBHM1J1JM?11'A'A'C'C2L %%l333 r __name__ __module__ __qualname____doc__r experimental_warningrr CreateAuthTokenConfigOrDictrSr^rrrr8r8qs  7+ FJOOO ABO OOO OOOrr8cteZdZdZejddddeejdej fdZ dS) AsyncTokenszu[Experimental] Async Auth Tokens API client. This class provides asynchronous methods for creating auth tokens. r9Nr:r%rcZKtj|}|jjrt dt j|j|}|d}|rd|}nd}|d}|r|dt|}| ddt||}d}||j |j j |j j }tj|}tj|}|jd ||| d{V}|jsint'j|j} tj| | } |j| | S) 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.' ), ), ) ) r:r<r=r>r?r@r%NrA)rLrB)r rErFrGrHrIrJr*rKrr,r6r%rLr rMrN async_requestrPrQrRrSrTr.rUrVs rr^zAsyncTokens.createsD5O     I  'J   l&))&11 ''(899##H--L1 0 0y.. 0 0dXt$$$6lFKKL04L#  " .  " / ;$+8l*<88L6|DDL%33 ! 4H 'mJBBHM1J1JM?11'A'A'C'C2L %%l333 rr_rrrrgrg s  7+ FJSSS ABS SSS SSSrrg)N)rcrQloggingtypingrrrr urllib.parserr r r rIr getLoggerlogger StringDictstrr#rer6 BaseModuler8rgrrrrss -, ,,,,,,,,,,,,""""""555555  0 1 1 G. 3     ;?BB$B U6 7B BBBBJYYYYY[ #YYYx]]]]]+(]]]]]r