refac: access_control field

This commit is contained in:
Timothy Jaeryang Baek
2024-11-14 18:57:25 -08:00
parent dae764fa5a
commit 150d0adea2
4 changed files with 35 additions and 3 deletions

View File

@@ -5,7 +5,7 @@ from typing import Optional
from open_webui.apps.webui.internal.db import Base, JSONField, get_db
from open_webui.env import SRC_LOG_LEVELS
from pydantic import BaseModel, ConfigDict
from sqlalchemy import BigInteger, Column, Text
from sqlalchemy import BigInteger, Column, Text, JSON
log = logging.getLogger(__name__)
log.setLevel(SRC_LOG_LEVELS["MODELS"])
@@ -67,6 +67,12 @@ class Model(Base):
Holds a JSON encoded blob of metadata, see `ModelMeta`.
"""
access_control = Column(JSON, nullable=True) # Controls data access levels.
# NULL for public access (open to all users with "user" role).
# {} for individual access (private to the owner).
# {"group_ids": ["group_id1", "group_id2"]} for access restricted to specific groups.
# {"user_ids": ["user_id1", "user_id2"]} for access restricted to specific users.
updated_at = Column(BigInteger)
created_at = Column(BigInteger)
@@ -80,6 +86,8 @@ class ModelModel(BaseModel):
params: ModelParams
meta: ModelMeta
access_control = Optional[dict] = None
updated_at: int # timestamp in epoch
created_at: int # timestamp in epoch