o ưi3@sddlZddlZddlZddlmZddlmZmZmZddlZddl m Z ddl m Z ddl mZmZmZmZmZdZd Zd Zd ZGd d d ZdS)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/tokenc@seZdZdddZdefddZdefddZdeefd d Zde ee ffd d Z dd dZ ddeede eeffddZ de eeffddZdedefddZdefddZdS) AuthenticatorreturnNcCsVtdtjd|_tj|jtdd|_tj|jtdd|_|dS)zJInitialize the GitHub Copilot authenticator with configurable token paths.ZGITHUB_COPILOT_TOKEN_DIRz ~/.config/litellm/github_copilotZ GITHUB_COPILOT_ACCESS_TOKEN_FILEz access-tokenZGITHUB_COPILOT_API_KEY_FILEz api-key.jsonN) osgetenvpath expanduser token_dirjoinaccess_token_file api_key_file_ensure_token_dirselfr`/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/github_copilot/authenticator.py__init__s   zAuthenticator.__init__c CsTz*t|jd}|}|r|WdWSWdn1s$wYWnty8tdYnwtdD]f}td|ddz5| }zt|jd }| |Wdn1sfwYWntyzt d Ynw|WSt t tfy}ztd |dd t|WYd}~q=d}~wwt d d d)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 rz/3wz!Error saving access token to filezFailed attempt z: z+Failed to get access token after 3 attemptsmessage status_code)openrreadstripIOErrorrwarningrangedebug_loginwriteerrorr r r str)rf access_tokenattempterrrget_access_token,sH        zAuthenticator.get_access_tokenc Csz:t|jd*}t|}|ddtkr&|dWdWSt dt ddd 1s4wYWn4t yHt d Yn'tj t fyf}zt d t|WYd}~n d}~wt ynYnwz0|}t|jd }t||Wdn1swY|d}|r|WStd dd t y}ztdt|tdt|dd d}~wty}z tdt|dd d}~ww)z Get the API key, refreshing if necessary. Returns: str: The GitHub Copilot API key. Raises: GetAPIKeyError: If unable to obtain an API key. rZ 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 timestamprr)r r(JSONDecodeErrorKeyErrorr/_refresh_api_keydumpr r.r )rr0 api_key_infor3r5rrr get_api_keySs\         zAuthenticator.get_api_keyc Csz,t|jd}t|}|di}|d}|WdWS1s%wYWdSttjtfyM}zt dt |WYd}~dSd}~ww)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%rr6r7r8r(r;r<rr)r/)rr0r?rAZ api_endpointr3rrr get_api_bases   (zAuthenticator.get_api_basec Cs|}||}d}t|D]g}z%t}|jt|d}||}d|vr-|WSt d|Wqt j y[}zt d|dd|dt |WYd }~qd }~wtyv}zt d t |WYd }~qd }~wwtd d d )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)headersr5z 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")r4_get_github_headersr*rr8GITHUB_API_KEY_URLraise_for_statusr6rr)httpxHTTPStatusErrorr.r/ Exceptionr ) rr1rD max_retriesr2 sync_clientresponseZ response_jsonr3rrrr=s6    zAuthenticator._refresh_api_keycCs&tj|jstj|jdddSdS)z"Ensure the token directory exists.T)exist_okN)rrexistsrmakedirsrrrrrszAuthenticator._ensure_token_dirr1cCs6dddddd}|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-typer)rr1rDrrrrFs z!Authenticator._get_github_headersc s8z8t}|jt|tddd}||gd}tfdd|Ds6t dt dd d WSt j yZ}zt d t |t d t |d d d }~wtjy{}zt dt |t dt |d d d }~wty}zt dt |t d t |d d d }~ww)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_idscoperDr6) device_code user_codeverification_uric3s|]}|vVqdSNr).0field resp_jsonrr sz1Authenticator._get_device_code..z"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_URLrFGITHUB_CLIENT_IDrHr6allrr.r rIrJr/r;rK)rrMrespZrequired_fieldsr3rr]r_get_device_codesN    zAuthenticator._get_device_coderWc Cst}d}t|D]}zJ|jt|t|ddd}||}d|vr2t d|dWSd|vrL| ddkrLt d |d d |d nt d |Wnct jyw}ztdt|tdt|ddd}~wtjy}ztdt|tdt|ddd}~wty}ztdt|tdt|ddd}~wwtdq tddd)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)rTrWZ grant_typerVr1zAuthentication successful!r.Zauthorization_pendingzAuthorization pending (attempt rrE)zUnexpected response: z%HTTP error polling for access token: zFailed to get access token: r`r"Nraz(Failed to decode access token response: z+Unexpected error polling for access token: z2Timed out waiting for user to authorize the device)rr*rbGITHUB_ACCESS_TOKEN_URLrFrdrHr6rinfor8r+r)rIrJr.r/r r;rKtimesleep)rrWrMZ max_attemptsr2rfr^r3rrr_poll_for_access_tokensj         z$Authenticator._poll_for_access_tokencCsD|}|d}|d}|d}td|d|ddd||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. rWrXrYz Please visit z and enter code z to authenticate.T)flush)rgprintro)rZdevice_code_inforWrXrYrrrr,Us  zAuthenticator._login)rNrZ)__name__ __module__ __qualname__rr/r4r@rrCrrr=rrFrgror,rrrrrs '4 ( 0Fr)r6rrmrtypingrrrrIZlitellm._loggingrZ&litellm.llms.custom_httpx.http_handlerrZ common_utilsr r r r r rdrcrkrGrrrrrs