o ;i: @sUddlZddlZddlZddlZddlZddlZddlmZm Z m Z ddl m Z m Z mZmZmZddlZddlmZddlmZddlmZddlmZddlmZmZdd lmZd d l m!Z!d d l"m#Z#d d l$m%Z%m&Z&d dl'm(Z(d dl)m*Z*d dl+m,Z,m-Z-m+Z+m.Z.d dl/m0Z0m1Z1e+2dZ3e4e5d<e3dZ6e4e5d<de7dee8e9e9fde9de:e9e9ffddZ;edZee e9e>fe e8e9e>ffZ?ededZ@ededZAGd d!d!ZBe&eBZCdS)"N)AsyncGenerator CollectionMapping)AnyClassVarOptionalTypeVarUnion) empty_pb2)Message)asynccontextmanager) synchronizer)api_pb2modal_api_grpc) __version__)print_server_warnings) async_utils) TaskContextsynchronize_api)_AuthTokenManager)ConnectionManager) _check_config _is_remoteconfiglogger) AuthError ClientClosedheartbeat_intervalHEARTBEAT_INTERVALg?HEARTBEAT_TIMEOUT client_type credentialsversionreturnc Cst}|jdkrdtd}}n|j|j}}ddd|||jfD}dtjj tjj tjj f}|t ||t jtt j|d}|r\|tjkr\|\} } || | d |S) NDarwinmacOSr-css|] }|ddVqdS)r'_N)replace).0sr,@/home/ubuntu/.local/lib/python3.10/site-packages/modal/client.py 'sz _get_metadata..z%d.%d.%d)zx-modal-client-versionzx-modal-client-typezx-modal-python-versionz x-modal-nodezx-modal-platform)zx-modal-token-idzx-modal-token-secret)platformunamesystemmac_verreleasejoinmachinesys version_infomajorminormicrostrurllibparsequotenoderCLIENT_TYPE_CLIENTupdate) r!r"r#r0r1r3 platform_strpython_versionmetadatatoken_id token_secretr,r,r- _get_metadata s(  rG ReturnType RequestType)bound ResponseTypec @s|eZdZUdZeeded<dZeeej ed<e ed<dZ eej ed<dZ eejed<dZeeed<dZeed <dZeeed <d ed <efd ed d deeeefdefddZdefddZedefddZedejfddZd edejfddZddZdGdefddZ d d!Z!d"d#Z"d$d%Z#e$e%d ede&d&fd'd(Z'e$dHdId)d*Z(e$d+ed,eddfd-d.Z)e$d edeeefddfd/d0Z*e$d1edfd2d3Z+d4ede,eeeffd5d6Z-d7efd8d9Z.d:d;Z/d ede0j1j2fdd?e0j1j6e7e8fd@e9dAee:dBee;de9f dCdDZdS)J_ClientN_client_from_env_client_from_env_lock_cancellation_context _cancellation_context_event_loop_stub_auth_token_managerF _snapshotted_connection_managerzapi_pb2.ClientType.ValueTyper! server_urlr"r#cCs:||_||_||_||_d|_d|_d|_d|_d|_dS)zjmdmd:hidden The Modal client object is not intended to be instantiated directly by users. FN) rUr! _credentialsr#_closedrQrRrS _owner_pid)selfrUr!r"r#r,r,r-__init__Ps  z_Client.__init__r$cCs|jSN)rWrYr,r,r- is_closeddsz_Client.is_closedcCstj|jj}|dvS)z3Returns True if the server URL points to localhost.>::1 127.0.0.1 172.21.0.1 localhost)r<r=urlparserUhostname)rYrcr,r,r- _is_localhostgsz_Client._is_localhostcCs|jsJ|jS)a\mdmd:hidden The default stub. Stubs can safely be used across forks / client snapshots. This is useful if you want to make requests to the default Modal server in us-east, for example control plane requests. This is equivalent to client.get_stub(default_server_url), but it's cached, so it's a bit faster. )rQr\r,r,r-stubms z _Client.stubcstj||IdHS)aUmdmd:hidden Get a stub for a specific server URL. Stubs can safely be used across forks / client snapshots. This is useful if you want to make requests to a regional Modal server, for example low-latency function calls in us-west. This function is O(n) where n is the number of RPCs in ModalClient. N)rModalClientModal_createrYrUr,r,r-get_stubzs z_Client.get_stubcsd|_|jdus Jt|j|j|j}tdd|_t |_ |j IdHt ||d|_ ||jIdH|_t|j|_t|_dS)NFg?)grace)clientrD)rWrQrGr!rVr#rrOasyncioget_running_looprP __aenter__rrTrirUrrerRosgetpidrX)rYrDr,r,r-_opens   z _Client._openprep_for_restorecsftdt|dd|_t|dr|jdddIdH|jr'|j|r,d|_ | ddS)NClient (z ): closingTrO) rdebugidrWhasattrrO __aexit__rTcloserSset_env_client)rYrrr,r,r-_closes  z_Client._closecs<tdt|d|jtIdH}t|jdS)zaConnect to server and retrieve version information; raise appropriate error for various failures.rsz ): StartingN) rrtrure ClientHellor Emptyrserver_warnings)rYrespr,r,r-hellosz _Client.hellocs|IdH|Sr[)rqr\r,r,r-rnsz_Client.__aenter__cs|IdHdSr[)rz)rYexc_typeexctbr,r,r-rwsz_Client.__aexit__)rLNcCsVtd||tjdd}z|IdH|VW|IdHdS|IdHw)zdmdmd:hidden Create a connection with no credentials; to be used for token creation. z.Client: Starting client without authenticationN)r")rrtrr@rqrz)clsrUrkr,r,r- anonymouss "z_Client.anonymousc st|r |}nt}|jdurt|_|j4IdHb|jr-|jWdIdHS|d}|d}trG|s<|rAtdt j }d}n|rS|rSt j }||f}nt d|d}t |||}|IdHt|||_|WdIdHS1IdHswYdS)zqmdmd:hidden Singleton that is instantiated from the Modal config and reused on subsequent calls. NrErFzzModal tokens provided by MODAL_TOKEN_ID and MODAL_TOKEN_SECRET (or through the config file) are ignored inside containers.zToken missing. Could not authenticate client. If you have token credentials, see modal.com/docs/reference/modal.config for setup help. If you are a new user, register an account at modal.com, then run `modal token new`.rU)rrrNrlLockrMrwarningswarnrCLIENT_TYPE_CONTAINERr@rrLrqr on_shutdownrz) r_override_configcrErFr!r"rUrkr,r,r-from_envsB    0z_Client.from_envrErFcsJttd}tj}||f}t|||}|IdHt||S)aB Constructor based on token credentials; useful for managing Modal on behalf of third-party users. **Usage:** ```python notest client = modal.Client.from_credentials("my_token_id", "my_token_secret") modal.Sandbox.create("echo", "hi", client=client, app=app) ``` rUN) rrrr@rLrqrrrz)rrErFrUr!r"rkr,r,r-from_credentialss  z_Client.from_credentialsc sX||tj|4IdH}|IdHWdIdHdS1IdHs%wYdS)zzmdmd:hidden Check whether can the client can connect to this server with these credentials; raise if not. N)rr@r)rrUr"rkr,r,r-verifys.z_Client.verifyrkcCs ||_dS)z mdmd:hiddenN)rM)rrkr,r,r-ry s z_Client.set_env_clientinput_plane_regioncs0|jsJd|jIdH}d|fd|fgS)Nz3Client must have an instance of auth token manager.zx-modal-input-plane-regionzx-modal-auth-token)rR get_token)rYrtokenr,r,r-get_input_plane_metadatas z _Client.get_input_plane_metadatareadable_methodcs|r|tt|t}||jkr>z|j|}| ||IdHWStj y=|rsD          .     *