D,j>ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl Z ddlmZddlmZdZdZdZdZdd Zd Zd ZdZGddZdZedkr edSdS)N)parse_qsurlparse)urls) InputOutputc tjddd|id}||}|diddS#t$rYdSwxYw) a  Checks if the user is on a free tier for OpenRouter. Args: api_key: The OpenRouter API key to check. Returns: A boolean indicating if the user is on a free tier (True) or paid tier (False). Returns True if the check fails. z%https://openrouter.ai/api/v1/auth/key AuthorizationzBearer )headerstimeoutdata is_free_tierT)requestsgetraise_for_statusjson Exception)api_keyresponser s _/home/rurouni/.local/share/uv/tools/aider-chat/lib/python3.11/site-packages/aider/onboarding.pycheck_openrouter_tierrs < 3$&9&9&9:    !!###}}xx##''=== ttsA-A00 A>=A>ctjd}|rt|}|rdSdSgd}|D]*\}}tj|}|r|cS+dS)z Attempts to select a default model based on available API keys. Checks OpenRouter tier status to select appropriate model. Returns: The name of the selected model, or None if no suitable default is found. OPENROUTER_API_KEYz$openrouter/deepseek/deepseek-r1:freez$openrouter/anthropic/claude-sonnet-4))ANTHROPIC_API_KEYsonnet)DEEPSEEK_API_KEYdeepseek)OPENAI_API_KEYzgpt-4o)GEMINI_API_KEYzgemini/gemini-2.5-pro-exp-03-25)VERTEXAI_PROJECTz"vertex_ai/gemini-2.5-pro-exp-03-25N)osenvironrr)openrouter_keyr model_key_pairsenv_key model_name api_key_values rtry_to_select_default_modelr',sZ^^$899N:,^<<  :9999O / w//       4cP|d|ddry|ddt||}|r&|tjd<|d d S|d |d d S)a Offers OpenRouter OAuth flow to the user if no API keys are found. Args: io: The InputOutput object for user interaction. analytics: The Analytics object for tracking events. Returns: True if authentication was successful, False otherwise. z6OpenRouter provides free and paid access to many LLMs.z-Login to OpenRouter or create a free account?y)defaultoauth_flow_initiated openrouterproviderroauth_flow_successToauth_flow_failurez8OpenRouter authentication did not complete successfully.F) tool_output confirm_askeventstart_openrouter_oauth_flowr r! tool_error)io analyticsr"s roffer_openrouter_oauthr9OsNNKLLL ~~7R .FFF4RCC  0>BJ+ , OO0 1 1 14 ,--- PQQQ 5r(cX|jr|jSt}|r2|d|d|d||Sd}||t ||t}|r|S|t jddS)a Selects a default model based on available API keys if no model is specified. Offers OAuth flow for OpenRouter if no keys are found. Args: args: The command line arguments object. io: The InputOutput object for user interaction. analytics: The Analytics object for tracking events. Returns: The name of the selected model, or None if no suitable default is found. zUsing z% model with API key from environment.auto_model_selection)modelz9No LLM model was specified and no API keys were provided.z%Open documentation URL for more info?N)r<r' tool_warningr4r9 offer_urlrmodels_and_keys)argsr7r8r< no_model_msgs rselect_default_modelrBts zz ' ) )E  MMMMNNN.e<<< NLOOL!!!2y))) ( ) )E  LL%'NOOOOOr($!!ct||dzD]F} tjd|fd5|cdddcS#1swxYwY7#t$rYCwxYwdS)N localhost)range socketserver TCPServerOSError) start_portend_portports rfind_available_portrOsj(Q,// 'd(;TBB                         H  4s1A ? A A A A A  AAc@tjd}tj}||dt j| d d}||fS)N@utf-8=) secrets token_urlsafehashlibsha256updateencodebase64urlsafe_b64encodedigestrstripdecode) code_verifierhashercode_challenges rgenerate_pkce_codesrbs)"--M ^  F MM-&&w//000-fmmoo>>EEdKKRRSZ[[N . ((r(c& tjdddi||ddd}||}|d}|s4|d |d |jdS|S#tjj$r|d YdStjj $r\}|d |j j d |j j |d |j jYd}~dSd}~wtjj $r#}|d |Yd}~dSd}~wt$r#}|d|Yd}~dSd}~wwxYw)Nz&https://openrouter.ai/api/v1/auth/keysz Content-Typezapplication/jsonS256)coder_code_challenge_method)r rr keyz.Error: 'key' not found in OpenRouter response.z Response: zInitiates the OpenRouter OAuth PKCE flow using a local server.z7Could not find an available port between 8484 and 8584.zBPlease ensure a port in this range is free, or configure manually.Nzhttp://localhost:/callback/aiderc$eZdZfdZdZdS)9start_openrouter_oauth_flow..OAuthCallbackHandlerct|j}|jdkrt|j}d|vr}|dd|d|dd||jd dS|d|d tj |dS|d|d tj ||jd dS) Nrurerz Content-typez text/htmls}

Success!

Aider has received the authentication code. You can close this browser tab.

i.Locations Not Found) rpathrquery send_response send_header end_headerswfilewritesetrwebsite)self parsed_path query_params auth_code server_errorshutdown_servers rdo_GETz@start_openrouter_oauth_flow..OAuthCallbackHandler.do_GETsc"49--K#444' (9:: \)) ,V 4Q 7I&&s+++$$^[AAA$$&&&J$$M$'')))))&&s+++$$Z>>>$$&&&&& ""3'''  T\:::  """   .....r(cdSN)rformatr@s r log_messagezEstart_openrouter_oauth_flow..OAuthCallbackHandler.log_messages Dr(N)__name__ __module__ __qualname__rr)rrrsrOAuthCallbackHandlerrwsH / / / / / / /@     r(rc tjdf5}dds<|t jd<ddddddS#1swxYwYdS#t$r8}d|Yd}~dSd}~wwxYw)NrGzTemporary server listening on T)log_onlyg?zShutting down temporary server.z)Failed to start or run temporary server: ) rIrJr2ris_sethandle_requesttimesleepr) httpdrrr callback_urlr7rNrserver_startedrs r run_serverz/start_openrouter_oauth_flow..run_server s "'d(;=QRR QV[N NNY]^^^""$$$)0022$((***JsOOO *0022$ @4PPP Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q " " "JqJJL       ! ! ! ! ! ! ! ! ! "s;C BB=0 C =CC CC D -DD T)targetdaemonr )r z8Temporary authentication server failed to start in time.rFzhttps://openrouter.ai/authrd)rrarf?&c3*K|]\}}|d|VdS)=Nr).0kvs r z.start_openrouter_oauth_flow..5s0*V*V$!Qa::!::*V*V*V*V*V*Vr(zG Please open this URL in your browser to connect Aider with OpenRouter:z Waiting up to z, minutes for you to finish in the browser...zUse Control-C to interrupt.F<z OAuth flow interrupted.oauth_flow_failedr-user_interrupt)r/rpzAuthentication failed: z&Authentication with OpenRouter failed.r.zCompleting authentication...oauth_flow_code_receivedrz~/.aider)exist_okzoauth-keys.envarR)encodingzOPENROUTER_API_KEY="z" zDAider will load the OpenRouter key automatically in future sessions.r0z:Successfully obtained key, but failed to save it to file: zBSet OPENROUTER_API_KEY environment variable for this session only.oauth_flow_save_failedcode_exchange_failed)rOr6 threadingEventhttpserverSimpleHTTPRequestHandlerThreadstartwaitrjoinrbitemsr2print webbrowseropenrKeyboardInterruptr=r4rsr r!r{ expandusermakedirsrstr)r7r8r server_threadr_ra auth_url_base auth_paramsauth_urlMINUTES interruptedr config_dirkey_filefrrrrrrNrrrs` @@@@@@@rr5r5s   D  OPPP Z[[[t0M$(!'K  XX#((*V*V+BSBSBUBU*V*V*V"V"VXXHNN]^^^NN (OOOGNN[g[[[\\\NN0111 !!!!      KWr\2222  3444+lK[\\\  q!!!t > >>???+l<XXXt  >???+lCCCtNN1222 OO.OFFF$I}bAAG+2 '( ++J77J K T 2 2 2 2w||J0@AAHhg666 =!;w;;;<<< = = = = = = = = = = = = = = = OOb c c c NN    OO0???+lKabbbtsjH22 H?>H?IA J/.J/A'Q'P  Q PQPAQ S$AR=7S=SceZdZdZdS)DummyAnalyticscdSrr)rr@kwargss rr4zDummyAnalytics.events r(N)rrrr4rr(rrrs#     r(rcZtdtdddddd}t}dtjvrtd t ||}|r+td td |dd d ntdtddS)z0Main function to test the OpenRouter OAuth flow.z&Starting OpenRouter OAuth flow test...TFNBLUERED)prettyyesinput_history_filechat_history_filetool_output_colortool_error_colorrz:Warning: OPENROUTER_API_KEY is already set in environment.z# OAuth flow completed successfully!z"Obtained API Key (first 5 chars): r z...z$ OAuth flow failed or was cancelled.z% OpenRouter OAuth flow test finished.)rrrr r!r5)r7r8rs rmainrs 2333       B  Irz)) JKKK*"i88G7 4555 C72A2;CCCDDDD 5666 233333r(__main__)rCrD)rZrV http.serverrr rTrIrrr urllib.parserrraiderraider.iorrr'r9rBrOrbrsr5rrrrr(rrs   ++++++++ 4   F"""J!P!P!PJ    )))###NiiiZ        444D zDFFFFFr(