44from datetime import datetime
55from openrouter import errors , models , utils
66from openrouter ._hooks import HookContext
7+ from openrouter .models import (
8+ createauthkeyscodeop as models_createauthkeyscodeop ,
9+ exchangeauthcodeforapikeyop as models_exchangeauthcodeforapikeyop ,
10+ )
711from openrouter .types import OptionalNullable , UNSET
812from openrouter .utils import get_security_from_env
913from openrouter .utils .unmarshal_json_response import unmarshal_json_response
@@ -19,7 +23,7 @@ def exchange_auth_code_for_api_key(
1923 code : str ,
2024 code_verifier : Optional [str ] = None ,
2125 code_challenge_method : OptionalNullable [
22- models .ExchangeAuthCodeForAPIKeyCodeChallengeMethod
26+ models_exchangeauthcodeforapikeyop .ExchangeAuthCodeForAPIKeyCodeChallengeMethod
2327 ] = UNSET ,
2428 retries : OptionalNullable [utils .RetryConfig ] = UNSET ,
2529 server_url : Optional [str ] = None ,
@@ -135,7 +139,7 @@ async def exchange_auth_code_for_api_key_async(
135139 code : str ,
136140 code_verifier : Optional [str ] = None ,
137141 code_challenge_method : OptionalNullable [
138- models .ExchangeAuthCodeForAPIKeyCodeChallengeMethod
142+ models_exchangeauthcodeforapikeyop .ExchangeAuthCodeForAPIKeyCodeChallengeMethod
139143 ] = UNSET ,
140144 retries : OptionalNullable [utils .RetryConfig ] = UNSET ,
141145 server_url : Optional [str ] = None ,
@@ -251,7 +255,7 @@ def create_auth_code(
251255 callback_url : str ,
252256 code_challenge : Optional [str ] = None ,
253257 code_challenge_method : Optional [
254- models .CreateAuthKeysCodeCodeChallengeMethod
258+ models_createauthkeyscodeop .CreateAuthKeysCodeCodeChallengeMethod
255259 ] = None ,
256260 limit : Optional [float ] = None ,
257261 expires_at : OptionalNullable [datetime ] = UNSET ,
@@ -371,7 +375,7 @@ async def create_auth_code_async(
371375 callback_url : str ,
372376 code_challenge : Optional [str ] = None ,
373377 code_challenge_method : Optional [
374- models .CreateAuthKeysCodeCodeChallengeMethod
378+ models_createauthkeyscodeop .CreateAuthKeysCodeCodeChallengeMethod
375379 ] = None ,
376380 limit : Optional [float ] = None ,
377381 expires_at : OptionalNullable [datetime ] = UNSET ,
0 commit comments