diff --git a/app/general_interceptors/add_general_interceptor_llm_user.py b/app/general_interceptors/add_general_interceptor_llm_user.py index 5477292..f39329e 100644 --- a/app/general_interceptors/add_general_interceptor_llm_user.py +++ b/app/general_interceptors/add_general_interceptor_llm_user.py @@ -1,4 +1,6 @@ from app.general.general_interceptors import add_general_interceptor, GeneralInterceptor +from app.model.UserModel import UserServiceModel, UserService +from app.utils.redis_utils import remove_redis_cache def add_general_interceptor_llm_user(): @@ -16,6 +18,8 @@ def add_general_interceptor_llm_user(): async def before_update(row, session, user): print("before_update:llm_user", row, session, user) + user_cls: UserServiceModel = await UserService.query_item(session, {"id": row["id"]}) + await remove_redis_cache(f"access_token_username_{user_cls.username}") async def after_update(row, session, user): print("after_update:llm_user", row, session, user) @@ -34,6 +38,8 @@ def add_general_interceptor_llm_user(): async def before_delete(query_config, session, user): print("before_delete:llm_user", query_config, session, user) + user_cls: UserServiceModel = await UserService.query_item(session, {"id": query_config["id"]}) + await remove_redis_cache(f"access_token_username_{user_cls.username}") async def after_delete(query_config, session, user): print("after_delete:llm_user", query_config, session, user)