File tree Expand file tree Collapse file tree
src/ai/backend/common/configs Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1111from pydantic import AliasChoices , Field
1212
1313from ai .backend .common .config import BaseConfigSchema
14- from ai .backend .common .jwt .config import JWTConfig as CoreJWTConfig
14+ from ai .backend .common .jwt .config import (
15+ JWTAlgorithm as CoreJWTAlgorithm ,
16+ )
17+ from ai .backend .common .jwt .config import (
18+ JWTConfig as CoreJWTConfig ,
19+ )
1520from ai .backend .common .meta import BackendAIConfigMeta , ConfigExample
1621
1722
@@ -58,6 +63,6 @@ class SharedJWTConfig(BaseConfigSchema):
5863 def to_jwt_config (self ) -> CoreJWTConfig :
5964 """Convert to ai.backend.common.jwt.config.JWTConfig."""
6065 return CoreJWTConfig (
61- algorithm = self .algorithm ,
66+ algorithm = CoreJWTAlgorithm ( self .algorithm ) ,
6267 token_expiration_seconds = self .token_expiration_seconds ,
6368 )
Original file line number Diff line number Diff line change 88import jwt as pyjwt
99import pytest
1010
11- from ai .backend .common .jwt .config import JWTConfig
11+ from ai .backend .common .jwt .config import JWTAlgorithm , JWTConfig
1212from ai .backend .common .jwt .signer import JWTSigner
1313from ai .backend .common .jwt .types import JWTUserContext
1414from ai .backend .common .types import AccessKey
1818def jwt_config () -> JWTConfig :
1919 """Create test JWT configuration."""
2020 return JWTConfig (
21- algorithm = " HS256" ,
21+ algorithm = JWTAlgorithm . HS256 ,
2222 token_expiration_seconds = 900 ,
2323 )
2424
Original file line number Diff line number Diff line change 77import jwt as pyjwt
88import pytest
99
10- from ai .backend .common .jwt .config import JWTConfig
10+ from ai .backend .common .jwt .config import JWTAlgorithm , JWTConfig
1111from ai .backend .common .jwt .exceptions import (
1212 JWTDecodeError ,
1313 JWTExpiredError ,
2424def jwt_config () -> JWTConfig :
2525 """Create test JWT configuration."""
2626 return JWTConfig (
27- algorithm = " HS256" ,
27+ algorithm = JWTAlgorithm . HS256 ,
2828 token_expiration_seconds = 900 ,
2929 )
3030
You can’t perform that action at this time.
0 commit comments