fix: access control

This commit is contained in:
Timothy Jaeryang Baek 2024-11-17 14:03:51 -08:00
parent 17c9a92767
commit 6906a0bd26

View File

@ -1147,7 +1147,7 @@ async def generate_chat_completions(
# Check if user has access to the model # Check if user has access to the model
if user.role == "user": if user.role == "user":
model_info = Models.get_model_by_id(model_id) model_info = Models.get_model_by_id(model_id)
if not ( if model_info and not (
user.id == model_info.user_id user.id == model_info.user_id
or has_access( or has_access(
user.id, type="read", access_control=model_info.access_control user.id, type="read", access_control=model_info.access_control