Skip to content

Commit 598f75f

Browse files
rapsealkclaude
andcommitted
fix: use JWTAlgorithm enum instead of raw string for JWTConfig.algorithm
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent aa7f5ca commit 598f75f

3 files changed

Lines changed: 11 additions & 6 deletions

File tree

src/ai/backend/common/configs/jwt.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,12 @@
1111
from pydantic import AliasChoices, Field
1212

1313
from 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+
)
1520
from 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
)

tests/unit/common/jwt/test_signer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
import jwt as pyjwt
99
import pytest
1010

11-
from ai.backend.common.jwt.config import JWTConfig
11+
from ai.backend.common.jwt.config import JWTAlgorithm, JWTConfig
1212
from ai.backend.common.jwt.signer import JWTSigner
1313
from ai.backend.common.jwt.types import JWTUserContext
1414
from ai.backend.common.types import AccessKey
@@ -18,7 +18,7 @@
1818
def 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

tests/unit/common/jwt/test_validator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
import jwt as pyjwt
88
import pytest
99

10-
from ai.backend.common.jwt.config import JWTConfig
10+
from ai.backend.common.jwt.config import JWTAlgorithm, JWTConfig
1111
from ai.backend.common.jwt.exceptions import (
1212
JWTDecodeError,
1313
JWTExpiredError,
@@ -24,7 +24,7 @@
2424
def 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

0 commit comments

Comments
 (0)