/jfddlmZddlmZmZmZmZddlmZm Z m Z m Z edZ GddZ dS)) annotations) HUB_API_ROOT HUB_WEB_ROOTPREFIXrequest_with_credentials)IS_COLABLOGGERSETTINGSemojisz/settings?tab=api+keyscHeZdZdZdxZxZZdddZddd ZddZ ddZ dZ dS)AuthauManages authentication processes including API key handling, cookie-based authentication, and header generation. The class supports different methods of authentication: 1. Directly using an API key. 2. Authenticating using browser cookies (specifically in Google Colab). 3. Prompting the user to enter an API key. Attributes: id_token (str | bool): Token used for identity verification, initialized as False. api_key (str | bool): API key for authentication, initialized as False. model_key (bool): Placeholder for model key, initialized as False. Methods: authenticate: Attempt to authenticate with the server using either id_token or API key. auth_with_cookies: Attempt to fetch authentication via cookies and set id_token. get_auth_header: Get the authentication header for making API requests. request_api_key: Prompt the user to input their API key. Examples: Initialize Auth with an API key >>> auth = Auth(api_key="your_api_key_here") Initialize Auth without API key (will prompt for input) >>> auth = Auth() Fapi_keystrverboseboolcV|ddd}|ptjdd|_|jrR|jtjdkr |rt jt ddS|}n0tr| }n| }|r=tj d|ji|rt jt ddSdS|r&t jt d td dSdS) aGInitialize Auth class and authenticate user. Handles API key validation, Google Colab authentication, and new key requests. Updates SETTINGS upon successful authentication. Args: api_key (str): API key or combined key_id format. verbose (bool): Enable verbose logging. _rrruAuthenticated ✅Nu!New authentication successful ✅zGet API key from z" and then run 'yolo login API_KEY') splitr getrr infor authenticaterauth_with_cookiesrequest_api_keyupdate API_KEY_URL)selfrrsuccesss Y/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/hub/auth.py__init__z Auth.__init__(sT--Q''*=(,y""="=  < -|x|I6666>K6 < < <===++--  -,,..GG**,,G  e OY 5 6 6 6 J vHHHIIIII J J  e K6ccKccc d d d d d e e max_attemptsintreturncjddl}t|D]|}tjtd|dzd||dt d}|ddd|_|rd S}tttd ) zPrompt the user to input their API key. Args: max_attempts (int): Maximum number of authentication attempts. Returns: (bool): True if authentication is successful, False otherwise. rNzLogin. Attempt rz of zEnter API key from  rTuFailed to authenticate ❌) getpassranger rrrrrrConnectionErrorr )rr$r)attempts input_keys r rzAuth.request_api_keyTs l++  H K6RR(Q,RRLRR S S S(Lk(L(L(LMMI$??32215DL  "" tt f%J%J%JKKLLLr"c|ddl} |x}rX|td|}|ddst ddSt d #t $r.dx|_|_tj td YdSwxYw) zAttempt to authenticate with the server using either id_token or API key. Returns: (bool): True if authentication is successful, False otherwise. rNz/v1/auth)headersrFzUnable to authenticate.Tz#User has not authenticated locally.zInvalid API key) requestsget_auth_headerpostrjsonrr+id_tokenrr warningr)rr0headerrs r rzAuth.authenticategs  --///v MM\";";";VMLLvvxx||Iu55E)*CDDDt!"GHH H   +0 0DMDL Nf555 6 6 655 sA,B4B4B;:B;cHtsdS ttd}|ddrE|didd|_|dSt d#t $r d|_YdSwxYw) zAttempt to fetch authentication via cookies and set id_token. User must be logged in to HUB and running in a supported browser. Returns: (bool): True if authentication is successful, False otherwise. Fz /v1/auth/autordataidTokenNTz/Unable to fetch browser authentication details.)rrrrr4rr+)rauthns r rzAuth.auth_with_cookies{s 5 , -K-K-KLLEyyE**  % &" 5 5 9 9)T J J !!###t!"STT T   !DM55 sA0B =B B! B!cL|jr dd|jiS|jr d|jiSdS)zGet the authentication header for making API requests. Returns: (dict | None): The authentication header if id_token or API key is set, None otherwise. authorizationzBearer z x-api-keyN)r4r)rs r r1zAuth.get_auth_headersE = /#%>t}%>%>? ? \ /. . / /r"N)rF)rrrr)r#)r$r%r&r)r&r) __name__ __module__ __qualname____doc__r4r model_keyr!rrrr1r"r r r s4&+*H*w*e*e*e*e*eXMMMMM&(* / / / / /r"r N) __future__rultralytics.hub.utilsrrrrultralytics.utilsrr r r rr rCr"r rGs#"""""^^^^^^^^^^^^@@@@@@@@@@@@555 N/N/N/N/N/N/N/N/N/N/r"