h3ddlZddlZddlZddlmZddlmZmZmZddlZddl m Z ddl m Z ddl mZmZmZmZmZdZd Zd Zd ZGd d Zy)N)datetime)AnyDictOptional)verbose_logger)_get_httpx_client)APIKeyExpiredErrorGetAccessTokenErrorGetAPIKeyErrorGetDeviceCodeErrorRefreshAPIKeyErrorzIv1.b507a08c87ecfe98z$https://github.com/login/device/codez+https://github.com/login/oauth/access_tokenz0https://api.github.com/copilot_internal/v2/tokenceZdZddZdefdZdefdZdeefdZde ee ffdZ ddZ dd eede eeffd Z de eeffd Zd 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 e/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/llms/github_copilot/authenticator.py__init__zAuthenticator.__init__s & GG  A B "$ NN II8. I" GGLL NNBII&C^T   c  t|jd5}|jj}|r |cdddS dddtdD]c}t jd|dzd |j} t|jd5}|j|ddd|cStd d #1swYxYw#t$rt j dYwxYw#1swYKxYw#t$rt jd YnwxYw#tttf$r2}t j d |dzd t|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 r z/3wz!Error saving access token to filezFailed attempt z: z+Failed to get access token after 3 attemptsmessage status_code)openrreadstripIOErrorrwarningrangedebug_loginwriteerrorr r rstr)r f access_tokenattemptes r!get_access_tokenzAuthenticator.get_access_token,sf d,,c2 (a vvx~~/ ' ( ( (QxG  #DWq[MQS!T U #{{} Nd44c:.a -.$# "A  / ( (   " "F  ..N"(()LMN'(;=OP &&1 RAx'PQ sC"C  CC8D+ DC;1D9D+ CCC87C8;D DD(%D+'D((D++E1?'E,,E1c t|jd5}tj|}|j ddt j jkDr|j dcdddStjdtdd #1swYnxYwnp#t$rtjd YnPtjtf$r+}tjd t|Yd}~nd}~wt$rYnwxYw |j}t|jd 5}tj ||dddn #1swYnxYw|j d}|r|St#d d #t$r?}tj$dt|t#dt|d d}~wt&$r}t#dt|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: r'zAPI key response missing tokenzError saving API key to file: zFailed to save API key: izFailed to refresh API key: )r,rjsonloadgetrnow timestamprr0r r/JSONDecodeErrorKeyErrorr6_refresh_api_keydumpr r5r)r r7 api_key_infor:r>s r! get_api_keyzAuthenticator.get_api_keySs d''- #yy| ##L!4x||~7O7O7QQ'++G4   #**+HI, 1$'    R  " "#P Q$$h/ Q  " "%Fs1vh#O P P!    002Ld''- + ,* + + + $$W-E $< #   #A#a&!J K 23q6(; "  5c!fX>  sB,AB3 B,="BB($B,,D D$!D  DD&FE# F#E,(F F G>:G G> G99G>ch t|jd5}tj|}|j di}|j d}|cdddS#1swYyxYw#t tj tf$r+}tjdt|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: ) r,rr?r@rAr/rDrErr0r6)r r7rHrK api_endpointr:s r! get_api_basezAuthenticator.get_api_bases d''- $#yy| (,,["= (}}U3 #  $ $ $ --x8   " "%KCPQF8#T U s4A(:A A(A%!A(%A((B1!B,,B1c >|j}|j|}d}t|D]d} t}|j t |}|j |j}d|vr|cStjd|ftd d #tj$r4}tjd|dzd|dt|Yd }~d }~wt$r+}tjd t|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. r&)headersr>z API key response missing token: z'HTTP error refreshing API key (attempt r /z): Nz%Unexpected error refreshing API key: z/Failed to refresh API key after maximum retriesr(r))r;_get_github_headersr1rrAGITHUB_API_KEY_URLraise_for_statusr?rr0httpxHTTPStatusErrorr5r6 Exceptionr) r r8rP max_retriesr9 sync_clientresponse response_jsonr:s r!rFzAuthenticator._refresh_api_keys",,. **<8 [)G W/1 &??+=w?O))+ (  m+(("**:=/J**!E  (( $$=gai[+VYZ]^_Z`Yab W$$'LSQRVH%UVV Ws*AB#<B##D6*C%% D1!DDctjj|js"tj|jdyy)z"Ensure the token directory exists.T)exist_okN)rrexistsrmakedirsrs r!rzAuthenticator._ensure_token_dirs-ww~~dnn- KK 6.r#r8c<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 r8rPs r!rRz!Authenticator._get_github_headerssG)-%610   )/ ~'>GO $  (&8GN #r#c t}|jt|jtdd}|j |j gd}tfd|Ds%tjdtdd S#tj$r?}tjd t|td t|d d }~wt j$r?}tjd t|tdt|d d }~wt$r?}tjdt|td t|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. z read:user) client_idscoperPr?) device_code user_codeverification_uric3&K|]}|v ywNrc).0field resp_jsons r! z1Authenticator._get_device_code..sGeu )Gsz"Response missing required fields: z Response missing required fieldsr)z HTTP error getting device code: zFailed to get device code: NError decoding JSON response: z'Failed to decode device code response: z&Unexpected error getting device code: )rpostGITHUB_DEVICE_CODE_URLrRGITHUB_CLIENT_IDrTr?allrr5r rUrVr6rDrW)r rYresprequired_fieldsr:ros @r!_get_device_codezAuthenticator._get_device_codess$ +-K##&002#3kJ$D  ! ! # INOGGG$$'I)%UV(> #  $$   #CCF8!L M$5c!fX> ##   #A#a&!J K$A#a&J    #I#a&!R S$5c!fX>  s0BBF,:C&&F<:D66 F:E<<Frhc t}d}t|D]} |jt|j t |dd}|j |j}d|vrtjd|dcSd|vr4|jddk(r tjd |d zd |d ntjd |t)j*dt#dd#tj$r?}tjdt!|t#dt!|dd}~wtj$$r?}tjdt!|t#dt!|dd}~wt&$r?}tjdt!|t#dt!|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. z,urn:ietf:params:oauth:grant-type:device_code)rerh grant_typergr8zAuthentication successful!r5authorization_pendingzAuthorization pending (attempt r rQ)zUnexpected response: z%HTTP error polling for access token: zFailed to get access token: rqr)Nrrz(Failed to decode access token response: z+Unexpected error polling for access token: z2Timed out waiting for user to authorize the device)rr1rsGITHUB_ACCESS_TOKEN_URLrRrurTr?rinforAr2r0rUrVr5r6r rDrWtimesleep)r rhrY max_attemptsr9rwror:s r!_poll_for_access_tokenz$Authenticator._poll_for_access_tokens()  \*G, "''+ 446%5'2&T(%%' IIK !Y."''(DE$^44y(! g.2II"((9'!)Al^STU#**-B9++NO, JJqM_+b"H  /(( $$'LSQRVH%UV):3q6(C #'' $$'Ec!fX%NO)Fs1vhO # $$A#a&J*:3q6(C #  s7A+C> AC>>G&:E  G&!:F G&':G!!G&c|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. rhrirjz Please visit z and enter code z to authenticate.T)flush)ryprintr)r device_code_inforhrirjs r!r3zAuthenticator._loginUsi 002&}5 $[1 +,>? ,--=i[HY Z  **;77r#)rNrl)__name__ __module__ __qualname__r"r6r;rIrrNrrrFrrRryrr3rcr#r!rrs! % #% N2S2hhsm"& $sCx.& P7  cSVh4.$sCx..`D #D #D L88r#r)r?rrrtypingrrrrUlitellm._loggingr&litellm.llms.custom_httpx.http_handlerr common_utilsr r r r rrurtrrSrrcr#r!rsP && +D*?GGS8S8r#