uPj&5UddlZddlZddlZddlmZddlmZmZddlZddlm Z ddl m Z ddl m Z mZmZmZmZdZeed <d Zeed <d Zeed <dZeed<GddZy)N)datetime)AnyFinal)verbose_logger)_get_httpx_client)APIKeyExpiredErrorGetAccessTokenErrorGetAPIKeyErrorGetDeviceCodeErrorRefreshAPIKeyErrorzIv1.b507a08c87ecfe98DEFAULT_GITHUB_CLIENT_IDz$https://github.com/login/device/codeDEFAULT_GITHUB_DEVICE_CODE_URLz+https://github.com/login/oauth/access_tokenDEFAULT_GITHUB_ACCESS_TOKEN_URLz0https://api.github.com/copilot_internal/v2/tokenDEFAULT_GITHUB_API_KEY_URLceZdZddZdefdZdefdZdedzfdZdeee ffdZ ddZ dd edzdeeeffd Z deeeffd Z d edefd ZdefdZy) AuthenticatorreturnNctjdtjjd|_tjj |jtjdd|_tjj |jtjdd|_|jy)zJInitialize the GitHub Copilot authenticator with configurable token paths.GITHUB_COPILOT_TOKEN_DIRz ~/.config/litellm/github_copilot GITHUB_COPILOT_ACCESS_TOKEN_FILEz access-tokenGITHUB_COPILOT_API_KEY_FILEz api-key.jsonN) osgetenvpath expanduser token_dirjoinaccess_token_file api_key_file_ensure_token_dirselfs n/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/github_copilot/authenticator.py__init__zAuthenticator.__init__s & GG  A B "$ NN II8. I" GGLLC`bp9qr  c t|jd5}|jj}|r |cdddS dddtdD]`}t jd|dz |j} t|jd5}|j|ddd|cStd d #1swYxYw#t$rt j dYwxYw#1swYKxYw#t$rt jdYnwxYw#tttf$r%}t j d |dz|Yd}~d}~wwxYw) z Login to Copilot with retry 3 times. Returns: str: The GitHub access token. Raises: GetAccessTokenError: If unable to obtain an access token after retries. rNz4No existing access token found or error reading filez%Access token acquisition attempt %s/3rwz!Error saving access token to filezFailed attempt %s: %sz+Failed to get access token after 3 attemptsmessage status_code)openrreadstripOSErrorrwarningrangedebug_loginwriteerrorr r r )r#f access_tokenattemptes r$get_access_tokenzAuthenticator.get_access_token*sY [d,,c2 (a vvx~~/ ' ( ( (Qx G  !H'TU+ V #{{} Nd44c:.a -.$# "A  + ( ( [  " "#Y Z [..N"(()LMN'(;=OP &&'>! QO sC"C CC5D(DC8.D6D(C CC54C58D =DD%"D($D%%D((E!<EE!c t|jd5}tj|}|j ddt j jkDr|j dcdddStjdtdd #1swYnxYwne#t$rtjd YnEtjtf$r }tjd |Yd}~nd}~wt$rYnwxYw |j}t|jd 5}tj||dddn #1swYnxYw|j d}|r|St!d d #t$r+}tj"d|t!d|d d}~wt$$r}t!d|d d}~wwxYw)z Get the API key, refreshing if necessary. Returns: str: The GitHub Copilot API key. Raises: GetAPIKeyError: If unable to obtain an API key. r( expires_atrtokenNzAPI key expired, refreshingzAPI key expiredr+r,z+No API key file found or error opening filez#Error reading API key from file: %sr*zAPI key response missing tokenz Error saving API key to file: %szFailed to save API key: izFailed to refresh API key: )r/r jsonloadgetrnow timestamprr3r r2JSONDecodeErrorKeyError_refresh_api_keydumpr r8r )r#r9 api_key_infor<r@s r$ get_api_keyzAuthenticator.get_api_keyOs d''- #yy| ##L!4x||~7O7O7QQ'++G4   #**+HI, 1$'    R  " "#P Q$$h/ M  " "#H! L L!    002Ld''- + ,* + + +'++G4E $< #   !CQ G 21#6 "  5aS9  sB,AB3 B,="BB($B,,D D$C?? D D&F8E FE!F9 F G&F55 GGGcR t|jd5}tj|}|j di}|j d}|cdddS#1swYyxYw#t tj tf$r }tjd|Yd}~yd}~wwxYw)z Get the API endpoint from the api-key.json file. Returns: Optional[str]: The GitHub Copilot API endpoint, or None if not found. r( endpointsapiNz(Error reading API endpoint from file: %s) r/r rArBrCr2rFrGrr3)r#r9rJrM api_endpointr<s r$ get_api_basezAuthenticator.get_api_bases d''- $&*iil #/#3#3K#D &/mmE&: #  $ $ $ --x8   " "#Mq Q s4A(:A A(A%!A(%A((B&B!!B&c2|j}|j|}tjdt}d}t |D]^} t }|j||}|j|j}d|vr|cStjd|`t!d d #tj$r%} tjd|dz|| Yd} ~ d} ~ wt$r } tjd | Yd} ~ d} ~ wwxYw) z Refresh the API key using the access token. Returns: Dict[str, Any]: The API key information including token and expiration. Raises: RefreshAPIKeyError: If unable to refresh the API key. GITHUB_COPILOT_API_KEY_URLr))headersr@z"API key response missing token: %sz1HTTP error refreshing API key (attempt %s/%s): %srNz'Unexpected error refreshing API key: %sz/Failed to refresh API key after maximum retriesr+r,)r=_get_github_headersrrrr4rrCraise_for_statusrArr3httpxHTTPStatusErrorr8 Exceptionr ) r#r:rS api_key_url max_retriesr; sync_clientresponse response_jsonr<s r$rHzAuthenticator._refresh_api_keys#335 11,?YY'CE_`  [) SG S/1 &??;?H))+ (  m+(("**+OQ^_ S"!E  (( w$$%XZadeZegrtuvv S$$%NPQRR Ss+ AB7B77D C** D6DDctjj|js"tj|jdyy)z"Ensure the token directory exists.T)exist_okN)rrexistsrmakedirsr"s r$r!zAuthenticator._ensure_token_dirs-ww~~dnn- KK 6.r&r:c<dddddd}|rd||d<d |vrd|d <|S) z Generate standard GitHub headers for API requests. Args: access_token: Optional access token to include in the headers. Returns: Dict[str, str]: Headers for GitHub API requests. zapplication/jsonz vscode/1.85.1zcopilot/1.155.0zGithubCopilot/1.155.0zgzip,deflate,br)acceptzeditor-versionzeditor-plugin-versionz user-agentzaccept-encodingztoken authorizationz content-type)r#r:rSs r$rTz!Authenticator._get_github_headerssG)-%610   )/ ~'>GO $  (&8GN #r&c t}tjdt}tjdt}|j ||j |dd}|j|jgd}tfd|Ds#tjdtd d S#tj$r+}tjd |td |d d}~wtj$r+}tjd|td|d d}~wt $r+}tjd|td |d d}~wwxYw)z Get a device code for GitHub authentication. Returns: Dict[str, str]: Device code information. Raises: GetDeviceCodeError: If unable to get a device code. GITHUB_COPILOT_DEVICE_CODE_URLGITHUB_COPILOT_CLIENT_IDz read:user) client_idscoperSrA) device_code user_codeverification_uric3&K|]}|v ywNre).0field resp_jsons r$ z1Authenticator._get_device_code..sGeu )Gsz$Response missing required fields: %sz Response missing required fieldsr,z"HTTP error getting device code: %szFailed to get device code: N Error decoding JSON response: %sz'Failed to decode device code response: z(Unexpected error getting device code: %s)rrrrrpostrTrUrAallrr8r rVrWrFrX)r#r[device_code_urlriresprequired_fieldsr<rss @r$_get_device_codezAuthenticator._get_device_codesk& !2!4K%'YY/OQo%pO!yy)CE]^I%**002#,{C+D  ! ! ##yy{I%UOGGG$$%KYW(> #  $$   !Eq I$5aS9 ##   !CQ G$A!E    !KQ O$5aS9  s0B?CE/&C<<E/&D88 E/&E**E/rlc t}d}tjdt}tjdt}t |D]} |j ||j||dd}|j|j}d|vrtjd|dcSd |vr/|jd d k(rtjd |d z|ntjd |t+j,dt%dd#tj $r+} tj"d| t%d| dd} ~ wtj&$r+} tj"d| t%d| dd} ~ wt($r+} tj"d| t%d| dd} ~ wwxYw)a Poll for an access token after user authentication. Args: device_code: The device code to use for polling. Returns: str: The access token. Raises: GetAccessTokenError: If unable to get an access token. GITHUB_COPILOT_ACCESS_TOKEN_URLrhz,urn:ietf:params:oauth:grant-type:device_code)rirl grant_typerkr:zAuthentication successful!r8authorization_pendingz%Authorization pending (attempt %s/%s)rzUnexpected response: %sz'HTTP error polling for access token: %szFailed to get access token: rur,Nrvz(Failed to decode access token response: z-Unexpected error polling for access token: %sz2Timed out waiting for user to authorize the device)rrrrrr4rwrTrUrArinforCr5r3rVrWr8r rFrXtimesleep) r#rlr[ max_attemptsaccess_token_urlrir;rzrsr<s r$_poll_for_access_tokenz$Authenticator._poll_for_access_token s/0  "$)),MOn"o99%?AYZ \*( G% "''$ 446%.'2&T(%%' IIK !Y."''(DE$^44 )immG.DH_._"(()PRY\]R]_kl"**+DiP( JJqMQ( T"H  +(( $$%NPQR):1#> #'' $$%GK)FqcJ # $$%TVWX):1#> # s8A#D#6A D##G6&EG2&F G$&G  Gc|j}|d}|d}|d}td|d|dd|j|S) a Login to GitHub Copilot using device code flow. Returns: str: The GitHub access token. Raises: GetDeviceCodeError: If unable to get a device code. GetAccessTokenError: If unable to get an access token. rlrmrnz Please visit z and enter code z to authenticate.T)flush)r|printr)r#device_code_inforlrmrns r$r6zAuthenticator._loginLsi#'"7"7"9-m< +K8 "23E"F ,--=i[HY Z  **;77r&)rNrp)__name__ __module__ __qualname__r%strr=rKrPdictrrHr!rTr|rr6rer&r$rrs !# ## J2S2hcDj"# $sCx.# J7 d d3PS8n40$sCx.0d@ #@ #@ D88r&r)rArrrtypingrrrVlitellm._loggingr&litellm.llms.custom_httpx.http_handlerr common_utilsr r r r r r__annotations__rrrrrer&r$rsa  +D#9%8(NN)VV$VEVI8I8r&