From 7a93a708ee2ff0786a7c7e3220d1c30b96fc78f8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=8Dtalo=20Johnny?= Date: Mon, 12 Aug 2024 19:09:08 -0300 Subject: [PATCH] fix: message sorting in playground (#3293) --- .../services/database/models/vertex_builds/model.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/services/database/models/vertex_builds/model.py b/src/backend/base/langflow/services/database/models/vertex_builds/model.py index 78d582fe8..e45a659a3 100644 --- a/src/backend/base/langflow/services/database/models/vertex_builds/model.py +++ b/src/backend/base/langflow/services/database/models/vertex_builds/model.py @@ -2,7 +2,7 @@ from datetime import datetime, timezone from typing import TYPE_CHECKING, Optional from uuid import UUID, uuid4 -from pydantic import field_validator, BaseModel +from pydantic import field_serializer, field_validator, BaseModel from sqlmodel import JSON, Column, Field, Relationship, SQLModel @@ -32,6 +32,13 @@ class VertexBuildBase(SQLModel): value = UUID(value) return value + @field_serializer("timestamp") + @classmethod + def serialize_timestamp(cls, value): + if value.tzinfo is None: + value = value.replace(tzinfo=timezone.utc) + return value + class VertexBuildTable(VertexBuildBase, table=True): # type: ignore __tablename__ = "vertex_build"