o ߭j5@sddlmZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl Z ddl mmZddlmmZddlmZddlmZddlmZddlmZdd lmZdd lmZddl Z!ddl"Z!z dd l#m$Z$e$j%Z%Wn e&yd Z%Ynwd Z'dZ(ej)ddZ*Gdddej+Z+Gdddej,Z,ej-GdddZ.ddddddddNd$d%Z/e.Z0e0/dOd(d)Z1dPd+d,Z2dQd.d/Z3dRd1d2Z4dSd4d5Z5dTd7d8Z6dUd:d;Z7dVd=d>Z8dWd@dAZ9dXdCdDZ:dYdFdGZ;dZdIdJZ(sz-patch_colab_gce_credentials..)r_default_get_gce_credentialsosenviron)get_gcerrrpatch_colab_gce_credentials$s  rcs:eZdZfddZddZddddddddZZS)FileServiceClientcsd|_tj|i|dSN)_discovery_apisuper__init__selfrr __class__rrr1szFileServiceClient.__init__cCsh|jj}|dur tdtjjtddtd|d}| \}}| d}tj j ||d|_ dS)NzaInvalid operation: Uploading to the File API requires an API key. Please provide a valid API key.cSs||fSrr)respcontentrrrr>sz8FileServiceClient._setup_discovery_api..z?version=v1beta&key=)httppostprocurizutf-8) developerKey)_client_optionsapi_key ValueErrorgoogleapiclientr$ HttpRequesthttplib2HttpGENAI_API_DISCOVERY_URLexecutedecode discoverybuild_from_documentr)rr)requestresponser# discovery_docrrr_setup_discovery_api5s    z&FileServiceClient._setup_discovery_apiNT) mime_typename display_name resumablepath str | pathlib.Path | os.PathLiker8 str | Noner9r:r;boolreturn protos.Filec Cs~|jdur |i}|dur||d<|dur||d<tjj|||d}|jjd|i|d}|} |d| ddiS)Nr9 displayName)filenamemimetyper;file)body media_body) rr7r+r$MediaFileUploadmediauploadr0get_file) rr<r8r9r:r;rErIr4resultrrr create_fileHs zFileServiceClient.create_file) r<r=r8r>r9r>r:r>r;r?r@rA)__name__ __module__ __qualname__rr7rM __classcell__rrr rr0s rc@seZdZddZdS)FileServiceAsyncClientcs td)NzPThe `create_file` method is currently not supported for the asynchronous client.)NotImplementedErrorrrrrrMdsz"FileServiceAsyncClient.create_fileN)rNrOrPrMrrrrrRcs rRc@seZdZUejedZded<dZded<dZ ded <dZ d ed <ejedZ ded <ddddddd d"ddZ ddZ ddZd#d d!ZdS)$_ClientManager)default_factoryzdict[str, Any] client_configrSequence[tuple[str, str]]default_metadataNzglm.DiscussServiceClient | Nonediscuss_clientz$glm.DiscussServiceAsyncClient | Nonediscuss_async_clientclientsr)r transportr client_inforXr)r>r(ga_credentials.Credentials | dict | Noner]r 8client_options_lib.ClientOptions | dict[str, Any] | Noner^&gapic_v1.client_info.ClientInfo | Noner@Nonec Cst|tr t|}|durt}ttj|}t|dd}|r)|dur(tdn |dur2t d}||_ t dt }|rP|j rL|j d|7_ n ||_ ntjj|d}||||d} d d | D} | |_||_i|_dS) aInitializes default client configurations using specified parameters or environment variables. If no API key has been provided (either directly, or on `client_options`) and the `GOOGLE_API_KEY` environment variable is set, it will be used as the API key. Note: Not all arguments are detailed below. Refer to the `*ServiceClient` classes in `google.ai.generativelanguage` for details on the other arguments. Args: transport: A string, one of: [`rest`, `grpc`, `grpc_asyncio`]. api_key: The API-Key to use when creating the default clients (each service uses a separate client). This is a shortcut for `client_options={"api_key": api_key}`. If omitted, and the `GOOGLE_API_KEY` environment variable is set, it will be used. default_metadata: Default (key, value) metadata pairs to send with every request. when using `transport="rest"` these are sent as HTTP headers. Nr)z`Invalid configuration: Please set either `api_key` or `client_options['api_key']`, but not both.GOOGLE_API_KEY/ ) user_agent)rr]r r^cSsi|] \}}|dur||qSrr).0keyvaluerrr sz,_ClientManager.configure..) isinstancedictclient_options_lib from_dict ClientOptionsrgetattrr*rgetenvr) USER_AGENT __version__rfr r^ ClientInfoitemsrVrXr[) rr)rr]r r^rXhad_api_key_valuerfrVrrr configuress< !     z_ClientManager.configurec s&|dkrt}n&|dkrt}n|dr$|dd}tt|d}n tt|d}js3tzt |d ij}Wdn1sJwYWnt j yb}zd|_ |d}~wwj sh|Sd d }fd d }|jD]\}}|||sqwt||}||}t|||qw|S)NrE file_async_async_rServiceAsyncClient ServiceClient)a No API_KEY or ADC found. Please either: - Set the `GOOGLE_API_KEY` environment variable. - Manually pass the key with `genai.configure(api_key=my_api_key)`. - Or set up Application Default Credentials, see https://ai.google.dev/gemini-api/docs/oauth for more information.cSsJ|drdS|dkr dSt|tjsdSt|trdSt|tr#dSdS)NrzFrMT) startswithrktypes FunctionType classmethod staticmethod)r9frrrkeeps    z(_ClientManager.make_client..keepcsddfdd }|S)Nr)metadatacs(t|tj}|i|d|iS)Nr)listrX)rrr)rrrrcallszN_ClientManager.make_client..add_default_metadata_wrapper..callr)rrr)rradd_default_metadata_wrappersz@_ClientManager.make_client..add_default_metadata_wrapperr)rrRendswithsplitrpglmtitlerVrwr ga_exceptionsDefaultCredentialsErrorrrX__dict__rusetattr) rr9clsclienterrrirrrr make_clients>     z_ClientManager.make_clientcCsD|}|dkr |S|j|}|dur ||}||j|<|SN operations)lowerget_default_operations_clientr[getr)rr9rrrrget_default_clients   z!_ClientManager.get_default_clientoperations_v1.OperationsClientcCs6|jdd}|dur|d}|jj}||jd<|S)NrModel)r[rr _transportoperations_client)rr model_clientrrrrs   z,_ClientManager.get_default_operations_client)r)r>rr_r]r>r r`r^rarXrWr@rbr@r)rNrOrP dataclassesfieldrlrV__annotations__rXrYrZr[rwrrrrrrrrTjs    M< rTrr\r)r>rr_r]r .client_options_lib.ClientOptions | dict | Noner^rarXrWcCstj||||||dS)aCaptures default client configuration. If no API key has been provided (either directly, or on `client_options`) and the `GOOGLE_API_KEY` environment variable is set, it will be used as the API key. Note: Not all arguments are detailed below. Refer to the `*ServiceClient` classes in `google.ai.generativelanguage` for details on the other arguments. Args: transport: A string, one of: [`rest`, `grpc`, `grpc_asyncio`]. api_key: The API-Key to use when creating the default clients (each service uses a separate client). This is a shortcut for `client_options={"api_key": api_key}`. If omitted, and the `GOOGLE_API_KEY` environment variable is set, it will be used. default_metadata: Default (key, value) metadata pairs to send with every request. when using `transport="rest"` these are sent as HTTP headers. r\)_client_managerrwr\rrrrws rwr@glm.DiscussServiceClientcC tdS)Ndiscussrrrrrrget_default_discuss_client> rglm.DiscussServiceAsyncClientcCr)N discuss_asyncrrrrr get_default_discuss_async_clientBrrglm.FilesServiceClientcCr)NrErrrrrget_default_file_clientFrrglm.FilesServiceAsyncClientcCr)Nrxrrrrrget_default_file_async_clientJrrglm.GenerativeServiceClientcCr)N generativerrrrrget_default_generative_clientNrr glm.GenerativeServiceAsyncClientcCr)Ngenerative_asyncrrrrr#get_default_generative_async_clientRrrglm.TextServiceClientcCr)Ntextrrrrrget_default_text_clientVrrrcCrrrrrrrrZrrglm.ModelServiceAsyncClientcCr)Nmodelrrrrrget_default_model_client^rrglm.RetrieverClientcCr)N retrieverrrrrrget_default_retriever_clientbrrglm.RetrieverAsyncClientcCr)Nretriever_asyncrrrrr"get_default_retriever_async_clientfrrglm.PermissionServiceClientcCr)N permissionrrrrrget_default_permission_clientjrr glm.PermissionServiceAsyncClientcCr)Npermission_asyncrrrrr#get_default_permission_async_clientnrr) r)r>rr_r]r>r rr^rarXrW)r@r)r@r)r@r)r@r)r@r)r@r)r@rr)r@r)r@r)r@r)r@r)r@r)> __future__rr contextlibrpathlibr~typingrrcollections.abcrr-google.ai.generativelanguageaigenerativelanguagergoogle.generativeai.protos generativeaiprotos google.authrga_credentialsrrgooglergoogle.api_corer rmr r googleapiclient.httpr+googleapiclient.discoverygoogle.generativeair rs ImportErrorrrr/contextmanagerrrrR dataclassrTrwrrrrrrrrrrrrrrrrrrsp             3(*