hvddlZddlZddlZddlZddlmZddlmZmZm Z ddl Z de fdZ dee efddfdZ de ee effdZdd Zde e fd Ze j"d e j$d e j&fdZe j"d dZe j"d dZy)N)Path)AnyDictOptionalreturncttj}|dz }|jdt|dz S)z.Get the path to store the authentication tokenz.litellmT)exist_okz token.json)rhomemkdirstr)home_dir config_dirs b/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/client/cli/commands/auth.pyget_token_file_pathr s8yy{HJ&Jd# zL( )) token_datact}t|d5}tj||ddddt j |dy#1swY xYw)zSave token data to filew)indentNi)ropenjsondumposchmod)r token_filefs r save_tokenrsK$&J j# +! *a*+HHZ++s AAct}tjj|sy t |d5}t j |cdddS#1swYyxYw#t jtf$rYywxYw)zLoad token data from fileNr) rrpathexistsrrloadJSONDecodeErrorIOError)rrs r load_tokenr&sj$&J 77>>* % *c " a99Q<    ' *s. A#A A#A A# A##A?>A?ct}tjj|rtj|yy)zClear stored tokenN)rrr!r"remove)rs r clear_tokenr)'s+$&J ww~~j! *"rc.t}|r d|vr|dSy)z&Get the stored API key from token filekeyN)r&)rs rget_stored_api_keyr,-s!Jez)%  rlogin)namectxc ddl}ddl}ddlm}ddlm}|j d}dt|j} |d|d|}tjd |tjd tjd |tj|tjd |d |}d} d} t| | zD]J} |j|d} | jdk(r| j!} | jddk(r| jd}|rt#|dddddt%j$dtjdtjd|dddtjd tjd!|y| jdk(r| dzdk(r8tjd"n"tjd#| jt%j(| Mtjd%y#|j&$r*}| dzdk(rtjd$|Yd}~bd}~wwxYw#t*$rtjd&Yyt,$r"}tjd'|Yd}~yd}~wwxYw)(z/Login to LiteLLM proxy using SSO authenticationrN)LITELLM_CLI_SOURCE_IDENTIFIER) show_commandsbase_urlzsk-z/sso/key/generate?source=z&key=zOpening browser to: z9Please complete the SSO authentication in your browser...z Session ID: zWaiting for authentication...z/sso/cli/poll/i,r )timeoutstatusreadyr+zcli-userunknowncli Authorization)r+user_id user_email user_roleauth_header_name jwt_token timestampu✅ Login successful!z API Key: z...z4You can now use the CLI without specifying --api-keyz= ============================================================z#Still waiting for authentication...zPolling error: HTTP zConnection error (will retry): u/❌ Authentication timed out. Please try again.u& ❌ Authentication cancelled by user.u❌ Authentication failed: )uuidrequestslitellm.constantsr1"litellm.proxy.client.cli.interfacer2objr uuid4clickecho webbrowserrrangeget status_coderrtimeRequestExceptionsleepKeyboardInterrupt Exception)r/rDrEr1r2r3key_idsso_urlpoll_urlr5 poll_intervalattemptresponsedataapi_keyes rr-r-6s?@wwz"H3tzz|$% &FCJ78U7VV[\b[cd )'34 NO \&*+    23Z~fX6 W 56G$ F#<<"<=''3.#==?Dxx)W4"&((5/"&'.+5.7-24C-/-1YY[("JJ'>?!JJ73B<.'DE!JJ']^"JJ}5)O"))S0|q( #HIJJ!5h6J6J5KLM JJ} %O7R  DE,, FR<1$JJ!@DE F  <=  045sWBI3C H78I3:AH7 -I37I0 I+&I3+I00I33J=J=J88J=logoutcBttjdy)z&Logout and clear stored authenticationu:✅ Logged out successfully. Authentication token cleared.N)r)rJrKrrr^r^sM JJKLrwhoamic$t}|stjdytjdtjd|jddtjd|jddtjd |jd d|jd d }t j|z d z }tjd|dd|dkDrtjdyy)z"Show current authentication statusuA❌ Not authenticated. Run 'litellm-proxy login' to authenticate.Nu✅ Authenticatedz User Email: r>Unknownz User ID: r=z User Role: r?rBriz Token age: z.1fz hoursuE⚠️ Warning: Token is more than 24 hours old and may have expired.)r&rJrKrNrP)rrB age_hourss rrarasJ  VW JJ"# JJjnn\9EFGH JJ:>>)Y?@AB JJZ^^KCDEF{A.Iy(D0I JJYsO6232~ Z[r)rN)rrrPrLpathlibrtypingrrrrJr rrr&r)r,command pass_contextContextr-r^rar`rrrks && *S* 4S> d  HT#s(^,  HSMGQu}}QQfHMM H\\r