Skip to content

Commit 9916c08

Browse files
committed
fix: code review comments
Signed-off-by: Tomas Weiss <[email protected]>
1 parent 40e7b96 commit 9916c08

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

apps/agentstack-server/src/agentstack_server/api/auth/auth.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ def issue_internal_jwt(
121121
"global": global_permissions.model_dump(mode="json"),
122122
"context": context_permissions.model_dump(mode="json"),
123123
},
124-
"token_version": role_version,
124+
"role_version": role_version,
125125
}
126126
return jwt.encode(header, payload, key=secret_key), expires_at
127127

@@ -137,7 +137,7 @@ def verify_internal_jwt(token: str, configuration: Configuration) -> ParsedToken
137137
"exp": {"essential": True},
138138
"iss": {"essential": True, "value": "agentstack-server"},
139139
"aud": {"essential": True, "value": "agentstack-server"},
140-
"token_version": {"essential": True},
140+
"role_version": {"essential": True},
141141
},
142142
)
143143
context_id = UUID(payload["resource"][0].replace("context:", ""))

apps/agentstack-server/src/agentstack_server/api/dependencies.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -129,8 +129,8 @@ async def authorized_user(
129129
parsed_token = verify_internal_jwt(bearer_auth.credentials, configuration=configuration)
130130
user = await user_service.get_user(parsed_token.user_id)
131131

132-
token_version = parsed_token.role_version
133-
if token_version < user.role_version:
132+
token_role_version = parsed_token.role_version
133+
if token_role_version < user.role_version:
134134
raise HTTPException(
135135
status_code=status.HTTP_401_UNAUTHORIZED,
136136
detail="Token invalidated due to role change",

apps/agentstack-server/src/agentstack_server/service_layer/services/users.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from agentstack_server.configuration import Configuration
1010
from agentstack_server.domain.models.user import User, UserRole
1111
from agentstack_server.domain.repositories.env import EnvStoreEntity
12-
from agentstack_server.exceptions import UsageLimitExceededError
12+
from agentstack_server.exceptions import PlatformError, UsageLimitExceededError
1313
from agentstack_server.service_layer.unit_of_work import IUnitOfWorkFactory
1414
from agentstack_server.utils.utils import utc_now
1515

@@ -65,7 +65,7 @@ async def change_role(self, user_id: UUID, new_role: UserRole) -> User:
6565
user = await uow.users.get(user_id=user_id)
6666

6767
if user.role == new_role:
68-
raise ValueError("User already has this role")
68+
raise PlatformError("User already has this role", status_code=400)
6969

7070
user.role = new_role
7171
user.role_version += 1

0 commit comments

Comments
 (0)