fix: app token's last_used_at can't be updated when last_used_at is null

langgenius · Jan 15, 2025 · 6f20275 · 6f20275
1 parent bc3a570
commit 6f20275
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/api/controllers/service_api/wraps.py b/api/controllers/service_api/wraps.py
@@ -195,7 +195,8 @@ def validate_and_get_api_token(scope: str | None = None):
     with Session(db.engine, expire_on_commit=False) as session:
         update_stmt = (
             update(ApiToken)
-            .where(ApiToken.token == auth_token, ApiToken.last_used_at < cutoff_time, ApiToken.type == scope)
+            .where(ApiToken.token == auth_token,
+                   (ApiToken.last_used_at.is_(None) | (ApiToken.last_used_at < cutoff_time)), ApiToken.type == scope)
             .values(last_used_at=current_time)
             .returning(ApiToken)
         )