Update SQLModel imports and add sa_column to created_at and updated_at fields in model.py and variable/model.py

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)

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")