diff --git a/api/db/services/user_service.py b/api/db/services/user_service.py index 44cafb87a..04789fd16 100644 --- a/api/db/services/user_service.py +++ b/api/db/services/user_service.py @@ -13,6 +13,7 @@ # See the License for the specific language governing permissions and # limitations under the License. # +import hashlib from datetime import datetime import peewee @@ -24,6 +25,7 @@ from api.db.db_models import User, Tenant from api.db.services.common_service import CommonService from api.utils import get_uuid, current_timestamp, datetime_format from api.db import StatusEnum +from rag.settings import MINIO class UserService(CommonService): @@ -48,6 +50,12 @@ class UserService(CommonService): else: return None + @classmethod + @DB.connection_context() + def user_gateway(cls, tenant_id): + hashobj = hashlib.sha256(tenant_id.encode("utf-8")) + return int(hashobj.hexdigest(), 16)%len(MINIO) + @classmethod @DB.connection_context() def save(cls, **kwargs):