From b10708194b6a65fda11359da9cce74d34c5862e7 Mon Sep 17 00:00:00 2001 From: Cristhian Zanforlin Lousa Date: Fri, 4 Oct 2024 16:48:40 -0300 Subject: [PATCH] fix: remove truncate to save data on DB (#4025) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 🔧 (model.py): Remove unused imports and functions related to truncating long strings in database models to clean up the code and improve readability. --- .../services/database/models/transactions/model.py | 8 +------- .../services/database/models/vertex_builds/model.py | 10 ---------- 2 files changed, 1 insertion(+), 17 deletions(-) diff --git a/src/backend/base/langflow/services/database/models/transactions/model.py b/src/backend/base/langflow/services/database/models/transactions/model.py index 210be01ea..effcf36cb 100644 --- a/src/backend/base/langflow/services/database/models/transactions/model.py +++ b/src/backend/base/langflow/services/database/models/transactions/model.py @@ -2,14 +2,12 @@ from datetime import datetime, timezone from typing import TYPE_CHECKING from uuid import UUID, uuid4 -from pydantic import field_serializer, field_validator +from pydantic import field_validator from sqlmodel import JSON, Column, Field, Relationship, SQLModel if TYPE_CHECKING: from langflow.services.database.models.flow.model import Flow -from langflow.utils.util_strings import truncate_long_strings - class TransactionBase(SQLModel): timestamp: datetime = Field(default_factory=lambda: datetime.now(timezone.utc)) @@ -34,10 +32,6 @@ class TransactionBase(SQLModel): value = UUID(value) return value - @field_serializer("outputs") - def serialize_outputs(self, data) -> dict: - return truncate_long_strings(data) - class TransactionTable(TransactionBase, table=True): # type: ignore __tablename__ = "transaction" 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 c48f3f93c..ab535dbc2 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 @@ -8,8 +8,6 @@ from sqlmodel import JSON, Column, Field, Relationship, SQLModel if TYPE_CHECKING: from langflow.services.database.models.flow.model import Flow -from langflow.utils.util_strings import truncate_long_strings - class VertexBuildBase(SQLModel): timestamp: datetime = Field(default_factory=lambda: datetime.now(timezone.utc)) @@ -40,14 +38,6 @@ class VertexBuildBase(SQLModel): value = value.replace(tzinfo=timezone.utc) return value - @field_serializer("data") - def serialize_data(self, data: dict) -> dict: - return truncate_long_strings(data) - - @field_serializer("artifacts") - def serialize_artifacts(self, data) -> dict: - return truncate_long_strings(data) - class VertexBuildTable(VertexBuildBase, table=True): # type: ignore __tablename__ = "vertex_build"