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

fix: remove truncate to save data on DB (#4025)

Browse source 
🔧 (model.py): Remove unused imports and functions related to truncating long strings in database models to clean up the code and improve readability.
This commit is contained in:
Cristhian Zanforlin Lousa 2024-10-04 16:48:40 -03:00 • committed by GitHub
commit b10708194b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 1 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/base/langflow/services/database/models/transactions/model.py
View file

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

10
src/backend/base/langflow/services/database/models/vertex_builds/model.py
View file

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