o Th$0 @sdZddlZddlZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddl m Zdd l mZed Zd eeefd efd dZ ddeeefdeejd eeeffddZGddde jZGddde jZdS)z&[Experimental] Auth Tokens API client.N)AnyDictListOptional) urlencode) _api_module)_common)_tokens_converters)typeszgoogle_genai.tokenssetupreturncsVg}|D]\}t|tr|rfdd|D}ng}||qd|S)zReturn field_maskscsg|] }d|qS).).0kkkrV/home/air/segue/gemini/backup/venv/lib/python3.10/site-packages/google/genai/tokens.py $sz$_get_field_masks..,)items isinstancedictkeysextendjoin)r fieldsvfieldrrr_get_field_maskss  r request_dictconfigc Cs|d}|r|dr|d|d<t|d}t|tr*|ddur*|dr8t|tjr=|jdur=|js=||d<nxt|trI|ddusTt|tjr[|jdur[|ddnZ|dr|d}t }|rg}|D]}||vr}d|}| |qrng}|r|dd |n||d<n|ddn|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_listrfield_mask_strrrr"_convert_bidi_setup_to_token_setup,s`               r2c@:eZdZdZeddddeejdej fddZ 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"r c Cstj|d}|jjrtdt|j|}|d}|r"d|}nd}|d}|r4|dt |}| dd|rAt ||}d}|durV|j durV|j j durV|j j }t|}t|}|jd |||}|jsnd nt|j} |jjs}t| } tjj| |d } |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.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-2.0-flash-live-001', 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-2.0-flash-live-001', 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(r2r" http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloads_AuthToken_from_mldev AuthToken_from_responser*_verify_response selfr"parameter_modelr!request_url_dictpath query_paramsrHr? response_dict return_valuerrrcreatewsR`          z Tokens.create __name__ __module__ __qualname____doc__r experimental_warningrr CreateAuthTokenConfigOrDictrPr[rrrrr4qr4c@r3) AsyncTokenszu[Experimental] Async Auth Tokens API client. This class provides asynchronous methods for creating auth tokens. r5Nr6r"r c s$tj|d}|jjrtdt|j|}|d}|r#d|}nd}|d}|r5|dt |}| ddt ||}d}|durU|j durU|j j durU|j j }t|}t|}|jjd |||d IdH}|jsqd nt|j} |jjst| } tjj| |d } |j| | S) aWCreates an auth token asynchronously. Args: config (CreateAuthTokenConfig): Optional configuration for the request. Usage: .. code-block:: python auth_token = await client.aio.tokens.create( config=types.CreateAuthTokenConfig( uses=10, live_constrained_parameters=types.LiveEphemeralParameters( model='gemini-2.0-flash-live-001', config=types.LiveConnectConfig( system_instruction='You are an LLM called Gemini.' ), ), ) ) r6r7r8r9r:r;r"Nr<)rHr=r>)r rArBrCrDrErFr&rGrr(r2r"rHr rIrJ async_requestrLrMrNrOrPrQr*rRrSrrrr[sX           zAsyncTokens.creater\rrrrrdrcrd)N)r`rMloggingtypingrrrr urllib.parserr=rr r rEr getLoggerloggerstrr rrbr2 BaseModuler4rdrrrrs,         E"