diff --git a/src/backend/base/langflow/services/database/models/api_key/model.py b/src/backend/base/langflow/services/database/models/api_key/model.py index 226794cbe..e3780be67 100644 --- a/src/backend/base/langflow/services/database/models/api_key/model.py +++ b/src/backend/base/langflow/services/database/models/api_key/model.py @@ -3,7 +3,7 @@ from typing import TYPE_CHECKING, Optional from uuid import UUID, uuid4 from pydantic import validator -from sqlmodel import Field, Relationship, SQLModel +from sqlmodel import Field, Relationship, SQLModel, Column, func, DateTime if TYPE_CHECKING: from langflow.services.database.models.user import User @@ -11,7 +11,7 @@ if TYPE_CHECKING: class ApiKeyBase(SQLModel): name: Optional[str] = Field(index=True, nullable=True, default=None) - created_at: datetime = Field(default_factory=datetime.utcnow) + created_at: datetime = Field(sa_column=Column(DateTime(timezone=True), server_default=func.now())) last_used_at: Optional[datetime] = Field(default=None, nullable=True) total_uses: int = Field(default=0) is_active: bool = Field(default=True) diff --git a/src/backend/base/langflow/services/database/models/variable/model.py b/src/backend/base/langflow/services/database/models/variable/model.py index 1c314d756..b473b08d3 100644 --- a/src/backend/base/langflow/services/database/models/variable/model.py +++ b/src/backend/base/langflow/services/database/models/variable/model.py @@ -2,8 +2,7 @@ from datetime import datetime, timezone from typing import TYPE_CHECKING, Optional from uuid import UUID, uuid4 -from sqlmodel import Field, Relationship, SQLModel - +from sqlmodel import Column, DateTime, Field, Relationship, SQLModel, func if TYPE_CHECKING: from langflow.services.database.models.user.model import User @@ -26,8 +25,14 @@ class Variable(VariableBase, table=True): description="Unique ID for the variable", ) # name is unique per user - created_at: datetime = Field(default_factory=utc_now, description="Creation time of the variable") - updated_at: Optional[datetime] = Field(None, description="Last update time of the variable") + created_at: datetime = Field( + sa_column=Column(DateTime(timezone=True), server_default=func.now()), + description="Creation time of the variable", + ) + updated_at: Optional[datetime] = Field( + sa_column=Column(DateTime(timezone=True)), + description="Last update time of the variable", + ) # foreign key to user table user_id: UUID = Field(description="User ID associated with this variable", foreign_key="user.id") user: "User" = Relationship(back_populates="variables")