&`iV hddlZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z ddl mZddlmZddlmZmZddlmZdd lmZdd lmZdd lmZmZmZd Zd Zej e!Z"de de#de fdZ$dZ%dZ&dZ'dede#de#ddfdZ(dddde de fde#de#d e)d!e)de*f d"Z+d#e de#fd$Z,dS)%N)md5sha256)Path)AnyCallable)UUID)get_cli_profile)HttpResponseErrorResourceNotFoundError)AzureCliCredential)ResourceManagementClient)DeploymentMode)generate_rsa_key_pairgenerate_ssh_key_namegenerate_ssh_key_pathsz$b24988ac-6180-42a0-ab88-20f7382dd24cclient function_namereturnc t||t|d|d}|*td|jh||S)a2Retrieve a callable function from Azure SDK client object. Newer versions of the various client SDKs renamed function names to have a begin_ prefix. This function supports both the old and new versions of the SDK by first trying the old name and falling back to the prefixed new name. begin_Nz6'{obj}' object has no {func} or begin_{func} attribute)objfunc)getattrAttributeErrorformat__name__)rrrs y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/autoscaler/_private/_azure/config.pyget_azure_sdk_functionrsr  wv/G /G/GNN  D | D K K_%M L     KcBt|}t|}|SN)_configure_key_pair_configure_resource_group)configs rbootstrap_azurer&3s!  ( (F &v . .F Mr c |dd}| t}tt |}||dd<t d|d|dvs Jd|dd}d|dvs Jdd|ddi}d|dvr|dd|d<t d |t|jd }||| ttj }| d }t|d5}tj|} dddn #1swxYwYt d|d|dd} | dt!} | |ddd| dt(} nt+| } | |dd<t d| d|d| } d| d} t/| }d|d|d|}|dd}|||%d<})Yd}%~%nd}%~%wwxYw|(s |&|d;?}*tW|*d@r|*,t dA||)rt *dB|t[|'||st $dC|nY#tP$rt *dD|Yn2tJ$r&}t $dE||Yd}~nd}~wwxYwd/t\j/| dF|idF| idF| idF|idF| idF|idGdHi}+t|j0d },|,|dI|+J1j&j2}-|-dKdF|ddK<|-dLdF|ddL<|-dMdF|ddM<|S)NNprovidersubscription_idzUsing subscription id: %sresource_groupz1Provider config must include resource_group fieldlocationz+Provider config must include location fieldtagsz$Creating/Updating resource group: %screate_or_updaterr)resource_group_name parameterszazure-config-template.jsonrzUsing cluster name: %s cluster_name unique_idutf-8zUsing unique id: %sz{}-{}zray-z-raz/subscriptions/z/resourceGroups/z3/providers/Microsoft.Authorization/roleAssignments/ subnet_maskz 10.{}.0.0/16zUsing subnet mask: %slist_by_resource_groupz substringof('z@', name) and resourceType eq 'Microsoft.Network/virtualNetworks'r get_by_idz 2024-10-01zMicrosoft.Networkc30K|]}|jdk |VdS)virtualNetworksN) resource_type).0rts r z,_configure_resource_group..s='+<<<<<<<r cg|]}d|v| S)previewr=vs r z-_configure_resource_group..s*###)STBTBTABTBTBTr z)Using API version: %s for virtualNetworkszYCould not determine supported API versions for virtualNetworks, using fallback version %szUFailed to query Microsoft.Network provider: %s. Using fallback API version 2024-10-01subnetsc32K|]}|ddk|VdS)typez!Microsoft.Network/virtualNetworksNrB)r=rss rr?z,_configure_resource_group..s>  f:!DDDDDDD  r resources propertiesmsi_namemsi_resource_groupz-msiz.Using msi_name: %s from msi_resource_group: %sz??O)++??AA./A/C/C_UUO,;F:() KK+_=== F:....: /..J'(89N fZ((((4 )((&,Z8 9F  ### +F3v KK6GGG0.>PNvNNNN>>(L ))*FGGM mS ! !*[9[))*************** KK(&*@AAAz"&&{33I fZ()9:AA'JJKKK$$&&~ ~6  NN &/F:{# KK%y111~ 6 BBJ1*111-.BCC K/ K K> K K4H K K $((77K I$++FN1c,B,BCC  KK'555((8PJ  BI B B B  N >Qa #*",K   ( '1556IJJI!%'6  "" " &8&E ##1>###$F'9'F%$**>#?#?#C  ?AQo$    NNgA          4!122=iHK  ";/         $,\:>>yII + #L188 9MNNN fZ((W-AVJEW-W +//0DnUUj!%%j2I2I2I2IJJH KK8(DV! g  Vo V V?Q V VKS V V  1&0 L(<FFL$+L,$K$K$O$O%% ! !% ) ) )$( ! ! !    NNC          !  KK[%    3%    "8",N" " " 5",K   ).%$ (  +    %   ,0 ) LLU$      ( ( ( NNH$    $(  (-& % 33 ; , :v..&OO%%% O( (LLF, :'/( NNB,)    X(   Q(  ". "K0%z2#X.%,.@$A%+;';<'.0D&E     J.*:L  .(!    G !(w 7F:u 'w 7F:u#*8#4W#=F:x MsE>>FF=B(P&& Q"0(QQ"%AV,, W): W)W$$W) Y'Z2 Z2 Z--Z28B]%^$5 ^$>^^$cr|dd}d}d|dv}d|dv}||kr"|rd\}}nd\}}t|d|d |rZ|rWt|dd}t|dd}g} |s| d ||s| d || r5td d | td||t|d5} | }dddn #1swxYwYn|dd} |dd} tdd| | |} t| \}}tj|rtj|rktd||t|d5} | }dddn #1swxYwYntd||tjtj|dt%\}}t|dd5} | |dddn #1swxYwYt|d5} | |dddn #1swxYwYtj|sJd||t)||dd<|ddd|dD](}|ddi}||d<||d <)|S)!z Configure SSH keypair. Use user specified custom paths, otherwise, generate Ray-specific keypair in this format: "ray-autoscaler_azure_{region}_{resource_group}_{ssh_user}_{index}" authssh_userNssh_private_keyssh_public_key)rr)rrz is specified but zu is missing. Both SSH key paths must be specified together, or omit both from your config to use auto-generated keys.zssh_private_key: zssh_public_key: zSSH key files from config do not exist: {}. Please create the keys or remove the custom paths from your config to use auto-generated keys.z, z/Using specified SSH keys from config: {} and {}r1r(r+r*azurez/Using existing Ray-specific SSH keys: {} and {}z5Generating new Ray-specific SSH key pair at {} and {}T)exist_okwc.tj||dS)Ni)osrj)pathflagss rz%_configure_key_pair..s274+F+Fr )openerz)Private key file {} not found for user {}available_node_types node_configazure_arm_parameters adminUsername publicKey) ValueErrorr expanduseris_fileappendrjoinrdrerjreadrrrrexistsmakedirsdirnamerwriterqpopvalues setdefault)r%r public_keyuser_specified_private_keyuser_specified_public_key missing_key specified_keyprivate_key_pathpublic_key_path missing_keysfregionr*key_name private_key node_typers rr#r#Zs f~j)HJ"3fVn!D 0F6N B"%>>> % M)L &K)L &K 6 6  6 6 6   "D &?D v/@ ABBMMOOvf~.>?@@KKMM '')) H    F4D F F G G G&&(( F    D? D D E E E  ..4fTYY|5L5L.M.M   = D D /     /3 ' ' "1J " " " " " " " " " " " " " " " #J/ +,<=) T6>8  -C8,L,L)) 7>>* + + $0O0O $ KKAHH$o    os++ &qVVXX  & & & & & & & & & & & & & & & KKGNN$o    K(899D I I I I&;&=&= #J  FF % $$$  % % % % % % % % % % % % % % % os++ $q ### $ $ $ $ $ $ $ $ $ $ $ $ $ $ $w~~     6 = = h     ),,<(=(=F6N$% 6N'...23::<<77 (7BB "B  19_-,6[)) MsH<FF!$F!'JJ J L;;L?L?M88M<?M<rr* principal_idc |sdSt|jd}t|jd} t||d}tdt |||n4#t $r'}td||Yd}~dSd}~wwxYw|D]@}t|dipi}td t|d d | d | d | d |kr ||j d} t| dr| t d|j|#t$r%td|j|Y t $r-}td|j||Yd}~:d}~wwxYwdS)aDelete all role assignments in the resource group for the given principal. Uses the generic ResourceManagementClient so we avoid depending on azure-mgmt-authorization. All role assignments associated with the provided principal ID are removed. Nr8r.rPz9resourceType eq 'Microsoft.Authorization/roleAssignments'zMFound %d role assignments in resource group %s while cleaning up principal %sz>Failed to enumerate role assignments for resource group %s: %srLzEInspecting role assignment %s with principalId=%s roleDefinitionId=%snamez rOroleDefinitionIdrQrRrUz4Deleted existing role assignment %s for principal %sz:Role assignment %s not found while processing principal %sz8Failed to delete role assignment %s for principal %s: %s)rrKrvrdrrwr r~rrbrxrrUrerr r) rr*rrr assignmentsr assignmentrLrs rrrs '(8PJ4( JK     [           L      "## Zr::@b  S J 4 4 NN= ) ) NN- . .    >>- ( (L 8 8  //&M(Jz6** "!!! KKF     %    LLL          NNJ          ;##s7AA66 B'B""B'/AF+G+5 G+>"G&&G+ ) max_attempts delay_secondsr.rSrrrc @td|dzD]} ||dtd|||nB#t$rYdSt$r(}td||||Yd}~nd}~wwxYw||krt j|dS)zPoll until a role assignment disappears after deletion. Returns True if the assignment is confirmed deleted, False otherwise. Logs detailed progress to aid troubleshooting when transient errors occur. r6rQz?Role assignment %s still present after deletion (attempt %d/%d)Tz@Attempt %d/%d to verify removal of role assignment %s failed: %sNF)rangerdrr rtimesleep)rrSrrrattemptrs rrrsL1,--&&    \ : : : LLQ$     %   444    LLR$           \ ! ! J} % % % 5s A B BA;;Brc dd|D}ttt|dS)zGReplicates ARM template guid() function for creating deterministic IDs.c34K|]}t|VdSr")rqrCs rr?z%_generate_arm_guid..Cs(22a3q66222222r r4)rrqrrrnro)r concatenateds rrrrr@s\7722622222L tC ++G4455??AABB C CCr )-rkloggingrrsrhashlibrrpathlibrtypingrruuidrazure.common.credentialsr azure.core.exceptionsr r azure.identityr azure.mgmt.resourcer $azure.mgmt.resource.resources.modelsrray.autoscaler._private.utilrrrCONTRIBUTOR_ROLE_DEFINITION_IDrp getLoggerrrdrqrr&r$r#rintboolrrrrBr rrss   444444JJJJJJJJ------888888??????"H  8 $ $3sx( ^^^B iiiXN-NNN NNNNl &&&!#s(+&&&  &  & &&&&RDDDDDDDDr