uPj_>NUddlZddlZddlZddlZddlmZddlmZmZddl Z ddl m Z m Z m Z ddlmZddlmZddlmZmZmZmZmZmZmZmZmZmZd Zeed <d Zeed <d Z eed<dZ!eed<dZ"eed<ee#e fZ$eed<e e$Z%eed<de dzde#dzfdZ&GddZ'y)N)Mapping)Final TypeAlias) JsonValue TypeAdapterValidationError)verbose_logger)_get_httpx_client) CHATGPT_API_BASECHATGPT_AUTH_BASECHATGPT_CLIENT_IDCHATGPT_DEVICE_CODE_URLCHATGPT_DEVICE_TOKEN_URLCHATGPT_DEVICE_VERIFY_URLCHATGPT_OAUTH_TOKEN_URLGetAccessTokenErrorGetDeviceCodeErrorRefreshAccessTokenError<TOKEN_EXPIRY_SKEW_SECONDSiDEVICE_CODE_TIMEOUT_SECONDSi,DEVICE_CODE_COOLDOWN_SECONDSDEVICE_CODE_POLL_SLEEP_SECONDSzhttps://api.openai.com/authOPENAI_AUTH_CLAIM_KEY JsonObject_JSON_OBJECT_ADAPTERvaluereturnc*t|tr|SdS)N) isinstancestr)rs g/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/chatgpt/authenticator.py _optional_strr%'suc*544ceZdZd dZdefdZdefdZdedzfdZd dZde dzfdZ d e ddfd Z d e d ede fd Z dededzfdZdede fdZdedzdedzfdZdeeeffdZdeeeffdZdeeefdeeeffdZdeeefdeeeffdZdedeeeffdZdeeefde fdZd e dzdefdZd dZdededzfdZy)! Authenticatorr Nctjdtjjd|_tjj |jtjdd|_|jy)NCHATGPT_TOKEN_DIRz~/.config/litellm/chatgptCHATGPT_AUTH_FILEz auth.json)osgetenvpath expanduser token_dirjoin auth_file_ensure_token_dirselfs r$__init__zAuthenticator.__init__,s]  GG  : ; dnnbii@SU`6ab  r&cjtjdxstjdxstS)Nr OPENAI_CHATGPT_API_BASE)r,r-r r4s r$ get_api_basezAuthenticator.get_api_base4s'yy+,h :S0ThXhhr&c|j}|rct|jd}|r|j||s|St|jd}|r |j |}|dS|j|}|dkDr|j|}|r|S|j}|dS#t $r }t jd|Yd}~dd}~wwxYw)N access_token refresh_tokenz3ChatGPT refresh token failed, re-login required: %sr) _read_auth_filer%get_is_token_expired_refresh_tokensrr warning#_get_device_code_cooldown_remaining_wait_for_access_token_login_device_code) r5 auth_datar;r< refreshedexccooldown_remainingtokentokenss r$get_access_tokenzAuthenticator.get_access_token7s//1 "/ n0M"NLD$:$:9l$S###01O#PMg'+';';M'JI$^44%)$L$LY$W  !667IJE //1n%%/g"**+`beffgs B66 C?CCc|j}|syt|jd}|r|S|jd}|jd}|jt|xs|}|r|j i|d|i|S)N account_idid_tokenr;)r=r%r>_extract_account_id_write_auth_file)r5rErMrNr;deriveds r$get_account_idzAuthenticator.get_account_idNs//1 ))-- *EF  #-- 3'mmN; 11-@XL2YZ   ! !"FY"F g"F Gr&ctjj|js"tj|jdyy)NT)exist_ok)r,r.existsr0makedirsr4s r$r3zAuthenticator._ensure_token_dir\s-ww~~dnn- KK 6.r&c8 t|jd5}tjt j |cdddS#1swYyxYw#t $rYytjtf$r }tjd|Yd}~yd}~wwxYw)NrzInvalid ChatGPT auth file: %s) openr2rvalidate_pythonjsonloadOSErrorJSONDecodeErrorrr rA)r5frGs r$r=zAuthenticator._read_auth_file`s dnnc* Ja+;;DIIaLI J J J $$o6   " "#BC H s:A(A  A AAA B!B9BBdatac t|jd5}tj||dddy#1swYyxYw#t$r }t j d|Yd}~yd}~wwxYw)Nwz%Failed to write ChatGPT auth file: %s)rYr2r[dumpr]r error)r5r`r_rGs r$rPzAuthenticator._write_auth_filejs` Odnnc* #a $" # # # O  !H# N N Os0A8AAAA A- A((A-rEr;cF|jd}t|ttfr't jt|t z k\S|j |}|r|ji|d|i|yt jt|t z k\S)N expires_atT)r>r"intfloattimer_get_expires_atrP)r5rEr;stored_expires_atderived_expires_ats r$r?zAuthenticator._is_token_expiredqs#,==#> '#u 699;%(9":=V"VV V$($8$8$F   ! !"QY"Q >P"Q R  %yy{e$67:SSSSr&rIc|j|}|jd}t|ttfr t|Sy)Nexp)_decode_jwt_claimsr>r"rgrh)r5rIclaimsrns r$rjzAuthenticator._get_expires_at|s://6ZZ& cC< (s8Or&c6 |jd}t|dkriS|ddt|d dzzz}tj|}tj t j|jdS#t$ricYSwxYw)N.r =zutf-8) splitlenbase64urlsafe_b64decoderrZr[loadsdecode Exception)r5rIparts payload_b64 payload_bytess r$roz Authenticator._decode_jwt_claimss  ;;s+E5zA~ !&qCCaM>A3E,F!FK#)#;#;K#HM'77 =CWCWX_C`8ab b I s B A&B BBc|sy|j|}|jt}t|tr%|jd}t|t r|r|Sy)Nchatgpt_account_id)ror>rr"dictr#)r5rIrp auth_claimsrMs r$rOz!Authenticator._extract_account_idsU//6#ZZ(=> k4 ( +0D EJ*c*z!!r&cz|j|j}|dkDr|j|}|rd|iS|j}|j t dt d|ddd|j|}|j|}|j|}|j||S) Nrr;z1Sign in with ChatGPT using device code: 1) Visit z 2) Enter code: user_codezB Device codes are a common phishing target. Never share this code.T)flush) rBr=rC_request_device_code_record_device_code_requestprintr_poll_for_authorization_code_exchange_code_for_tokens_build_auth_recordrP)r5rHrI device_code auth_coderJrEs r$rDz Authenticator._login_device_codes$($L$LTMaMaMc$d  !667IJE&..!668  ((*  123)+678P P   <<[I 66yA226:  i( r&cl t}|jtdti}|j t j |j}t|jd}t|jdxs|jd}|jd }|r|std |d||t!|xsd d S#tj$r)}td||jjd}~wt$r}td|dd}~wwxYw) N client_idr[zFailed to request device code: message status_codedevice_auth_idrusercodeintervalz%Device code response missing fields: 5)rrr)r postrrraise_for_statusrrZr[httpxHTTPStatusErrorrresponserr|r%r>r#)r5clientrespr`rGrrrs r$rz"Authenticator._request_device_codes5 -/F ++'!#45&D  ! ! #.>>tyy{KD!.dhh7G.H I(+)>)V$((:BVW ((:.Y$?vF  -"HO,  '$$ $9#?LL44  $9#?  s$ACD3.$D D3D..D3rct}t|jdd}tj}tj|z tkr2 |j t |d|dd}|jdk(r}tj|j}t|jd}t|jd }t|jd } |r |r| r||| d S|jd vr$tjt|t|jtjt|ttj|z tkr2t'dd#t j"$rz} | j$r| j$jnd} | d vr)tjt|tYd} ~ t'd | | j$jd} ~ wt($r} t'd | dd} ~ wwxYw)Nrrrr)rrrauthorization_codecode_challenge code_verifier)rrr)iizPolling failed: rrz*Timed out waiting for device authorizationi)r rgr>rirrrrrrZr[r%sleepmaxrrrrrrr|) r5rrr start_timerr`rrrrGrs r$rz*Authenticator._poll_for_authorization_codes)+kooj#>? IIK iikJ&)DD$ {{,*56F*G%0%=###s*/?? LD)6txx@T7U)V&%2488) r5rr redirect_uribodyrr`rGr;r<rNs r$rz'Authenticator._exchange_code_for_tokensso -/F%6$77K"LL"#789 /0!)O">tyy{KD,DHH^,DE ,TXXo-FG '(<==%B4&I  )*   '$$ %1#7LL44  %1#7  s$A8C$$D<7$D D<'D77D<r<c t}|jttd|dd}|j t j |j}t|jd}t|jd }|r|std |d|t|jdxs||d }|j!|} |j#| |S#tj$r)}td||jjd}~wt$r}td|dd}~wwxYw) Nr<zopenid profile email)r grant_typer<scoperzRefresh token failed: rrr;rNz!Refresh response missing fields: r)r rrrrrrZr[rrrrrr|r%r>rrP) r5r<rrr`rGr;rNrFrEs r$r@zAuthenticator._refresh_tokens-sM -/F ++'!2"1%23 &D  ! ! #.>>tyy{KD,DHH^,DE '(<=8);D6B  )*488O+DEV   229=  i(5$$ )06LL44  )06  s$AC--E$D$$ E0EErJc|jd}|jd}|r|j|nd}|j|xs|}||jd|||dS)Nr;rNr<)r;r<rNrfrM)r>rjrO)r5rJr;rNrfrMs r$rz Authenticator._build_auth_recordWsj$jj8  **Z0BND00>TX  44X5MN (#ZZ8 $$   r&c|sy|jd}t|tttfsy t|}tj|z }t|z }td|S#t t f$rYywxYw)Ndevice_code_requested_at) r>r"rgrhr# TypeError ValueErrorrirr)r5rE requested_atrequested_secondselapsed remainings r$rBz1Authenticator._get_device_code_cooldown_remainingds~'mm,FG ,eS(9: ',\': '887'A 3 "" :&  s A))A;:A;c~|jxsi}|ji|dtjiy)Nr)r=rPri)r5rEs r$rz)Authenticator._record_device_code_requestrs7//17R  TT,F TUr&timeout_secondsc tj|z}tj|kr|j}|r0t|jd}|r|j ||s|St t td|tjz }|dkr ytj|tj|kry)Nr;rr) rir=r%r>r?minrrr)r5rdeadlinerEr; sleep_fors r$rCz$Authenticator._wait_for_access_tokenvs))+7iikH$,,.I,Y]]>-JK (>(>y,(W'':CXPTPYPYP[E[<\]IA~ JJy !iikH$r&)r N)__name__ __module__ __qualname__r6r#r9rKrRr3rr=rPboolr?rgrjrorOrrDrrrr@rrhrBrrCr&r$r(r(+s!ici&#&. d 7d!2OZODO T: TS TT TSS4Z    t d DcN,  d38n  D/ S#X/ 4PSUXPX>/ b) 4S>) d3PS8n) V(S(T#s(^(T  c3h  J   #Z$=N #SX #V e d r&r()(rxr[r,ricollections.abcrtypingrrrpydanticrrrlitellm._loggingr &litellm.llms.custom_httpx.http_handlerr common_utilsr r rrrrrrrrr__annotations__rrrrr#rrr%r(rr&r$rs ## <<+D   $&5%%,U,&,e,())<u<Y/ I/)*5e55T)5cDj5WWr&