jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-04-10 16:34:48 -03:00
commit 4d05187f6b
2 changed files with 11 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/services/database/models/api_key/model.py
View file

@ -3,7 +3,7 @@ from typing import TYPE_CHECKING, Optional
from uuid import UUID, uuid4 from uuid import UUID, uuid4
from pydantic import validator from pydantic import validator
from sqlmodel import Field, Relationship, SQLModel from sqlmodel import Field, Relationship, SQLModel, Column, func, DateTime
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.services.database.models.user import User from langflow.services.database.models.user import User
@ -11,7 +11,7 @@ if TYPE_CHECKING:
class ApiKeyBase(SQLModel): class ApiKeyBase(SQLModel):
name: Optional[str] = Field(index=True, nullable=True, default=None) 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) last_used_at: Optional[datetime] = Field(default=None, nullable=True)
total_uses: int = Field(default=0) total_uses: int = Field(default=0)
is_active: bool = Field(default=True) is_active: bool = Field(default=True)

13
src/backend/base/langflow/services/database/models/variable/model.py
View file

@ -2,8 +2,7 @@ from datetime import datetime, timezone
from typing import TYPE_CHECKING, Optional from typing import TYPE_CHECKING, Optional
from uuid import UUID, uuid4 from uuid import UUID, uuid4
from sqlmodel import Field, Relationship, SQLModel from sqlmodel import Column, DateTime, Field, Relationship, SQLModel, func
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.services.database.models.user.model import User from langflow.services.database.models.user.model import User
@ -26,8 +25,14 @@ class Variable(VariableBase, table=True):
description="Unique ID for the variable", description="Unique ID for the variable",
) )
# name is unique per user # name is unique per user
created_at: datetime = Field(default_factory=utc_now, description="Creation time of the variable") created_at: datetime = Field(
updated_at: Optional[datetime] = Field(None, description="Last update time of the variable") 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 # foreign key to user table
user_id: UUID = Field(description="User ID associated with this variable", foreign_key="user.id") user_id: UUID = Field(description="User ID associated with this variable", foreign_key="user.id")
user: "User" = Relationship(back_populates="variables") user: "User" = Relationship(back_populates="variables")