diff --git a/src/backend/langflow/database/models/token.py b/src/backend/langflow/services/database/models/token.py similarity index 100% rename from src/backend/langflow/database/models/token.py rename to src/backend/langflow/services/database/models/token.py diff --git a/src/backend/langflow/database/models/user.py b/src/backend/langflow/services/database/models/user.py similarity index 93% rename from src/backend/langflow/database/models/user.py rename to src/backend/langflow/services/database/models/user.py index 94ceb4e15..f9a3c80f8 100644 --- a/src/backend/langflow/database/models/user.py +++ b/src/backend/langflow/services/database/models/user.py @@ -1,14 +1,15 @@ -from sqlmodel import Field -from uuid import UUID, uuid4 -from pydantic import BaseModel -from typing import Optional, List -from sqlalchemy.orm import Session -from datetime import timezone, datetime -from sqlalchemy.exc import IntegrityError -from fastapi import HTTPException, Depends - +from fastapi import Depends, HTTPException +from langflow.services.database.models.base import SQLModel, SQLModelSerializable from langflow.services.utils import get_session -from langflow.services.database.models.base import SQLModelSerializable, SQLModel +from pydantic import BaseModel +from sqlalchemy.exc import IntegrityError +from sqlalchemy.orm import Session +from sqlmodel import Field + + +from datetime import datetime, timezone +from typing import List, Optional +from uuid import UUID, uuid4 class User(SQLModelSerializable, table=True):