From ba081e81b68f7a1f76a8664c96a21fa0836e25e2 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 25 Aug 2023 12:05:20 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=80=20chore(user.py):=20add=20type=20h?= =?UTF-8?q?inting=20for=20ApiKey=20import=20to=20improve=20code=20readabil?= =?UTF-8?q?ity=20=F0=9F=94=80=20chore(user.py):=20add=20relationship=20bet?= =?UTF-8?q?ween=20User=20and=20ApiKey=20models=20to=20establish=20a=20one-?= =?UTF-8?q?to-many=20relationship?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../langflow/services/database/models/user/user.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/services/database/models/user/user.py b/src/backend/langflow/services/database/models/user/user.py index 3a4308b42..b6c27c2dc 100644 --- a/src/backend/langflow/services/database/models/user/user.py +++ b/src/backend/langflow/services/database/models/user/user.py @@ -1,11 +1,14 @@ from langflow.services.database.models.base import SQLModel, SQLModelSerializable -from sqlmodel import Field +from sqlmodel import Field, Relationship from datetime import datetime -from typing import Optional +from typing import Optional, TYPE_CHECKING from uuid import UUID, uuid4 +if TYPE_CHECKING: + from langflow.services.database.models.api_key import ApiKey + class User(SQLModelSerializable, table=True): id: UUID = Field(default_factory=uuid4, primary_key=True, unique=True) @@ -16,6 +19,7 @@ class User(SQLModelSerializable, table=True): create_at: datetime = Field(default_factory=datetime.utcnow) updated_at: datetime = Field(default_factory=datetime.utcnow) last_login_at: Optional[datetime] = Field() + api_keys: list["ApiKey"] = Relationship(back_populates="user") class UserCreate(SQLModel):