h :ddlZddlZddlmZejdZej dejdejfdZ ej dejd d d ejdejd e fd Z ej dejdddejdddejdddejdddejde ddejdejfdZej d ejd!d"#ejdejfd$Zy)%N)UsersManagementClientcy)z)Manage users on your LiteLLM proxy serverNrc/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/client/cli/commands/users.pyusersr s rlistctxc t|jd|jd}|j}t|tr d|vr|d}|st j dyddlm}dd l m }|d }|jd d |jdd|jdd|jdd|D]}|jt|jddt|jddt|jdddj|jdgxsg|}|j!|y)zList all usersbase_urlapi_keyr rr zNo users found.Nr)Table)ConsoleUsers)titlezUser IDcyan)styleEmailgreenRolemagentaTeamsyellowuser_id user_email user_rolez, teams)robj list_users isinstancedictclickecho rich.tabler rich.consoler add_columnadd_rowstrgetjoinprint)r clientr rrtableuserconsoles rr"r" s:#CGGJ,?QZI[ \F    E%7e#3g  $% $  E Yf- WG, V9- WH- B' ( r* + b) * IIdhhw+1r 2  iG MM%rr,z--idrzID of the user to retrieve)helpct|jd|jd}|j|}tj|y)z%Get information about a specific userr rr)rdataN)rr!get_userrich print_json)r rr/results rr7r7&s? #CGGJ,?QZI[ \F __W_ -FOO rcreatez--emailTz User email)requiredr3z--role internal_userz User role)defaultr3z--aliasz User aliasz--teamzTeam IDs (can specify multiple))multipler3z --max-budgetzMax budget for user)typer>r3ct|jd|jd}||d}|r||d<|rt||d<|||d<|j|}t j | y) zCreate a new userr rr)rr user_aliasr N max_budgetr5)rr!r create_userr8r9) r emailrolealiasteamrCr/ user_datar:s rrDrD/s|#CGGJ,?QZI[ \FI "' , !$Z '", ,    *FOO rdeleteuser_ids)nargsct|jd|jd}|jt|}t j |y)z#Delete one or more users by user_idr rrr5N)rr! delete_userr r8r9)r rKr/r:s rrOrOFsC #CGGJ,?QZI[ \F   X /FOO r)r%r8rrgroupr command pass_contextContextr"optionr+r7floatrDargumentrOrrrrWs %   vEMM4u fi&BC!%--!#!D! x i$\: hkB iL9 h,MN n5$=RS!U]]!TO:C;! x "%!U]]!&!r