From 118da0d482cfce379b81f83106f82fb1e51ac264 Mon Sep 17 00:00:00 2001
From: italojohnny <italojohnnydosanjos@gmail.com>
Date: Mon, 10 Jun 2024 16:06:40 -0300
Subject: [PATCH] refactor to use DefaultModel as base class

---
 .../base/langflow/services/monitor/schema.py  | 22 +++----------------
 1 file changed, 3 insertions(+), 19 deletions(-)

diff --git a/src/backend/base/langflow/services/monitor/schema.py b/src/backend/base/langflow/services/monitor/schema.py
index f3261c559..f39e41e21 100644
--- a/src/backend/base/langflow/services/monitor/schema.py
+++ b/src/backend/base/langflow/services/monitor/schema.py
@@ -37,10 +37,6 @@ class TransactionModel(DefaultModel):
     error: Optional[str] = None
     flow_id: Optional[str] = Field(default=None, alias="flow_id")
 
-    class Config:
-        from_attributes = True
-        populate_by_name = True
-
     # validate target_args in case it is a JSON
     @field_validator("outputs", "inputs", mode="before")
     def validate_target_args(cls, v):
@@ -55,7 +51,7 @@ class TransactionModel(DefaultModel):
         return v
 
 
-class TransactionModelResponse(BaseModel):
+class TransactionModelResponse(DefaultModel):
     index: Optional[int] = Field(default=None)
     timestamp: Optional[datetime] = Field(default_factory=datetime.now, alias="timestamp")
     vertex_id: str
@@ -67,10 +63,6 @@ class TransactionModelResponse(BaseModel):
     source: Optional[str] = None
     target: Optional[str] = None
 
-    class Config:
-        from_attributes = True
-        populate_by_name = True
-
     # validate target_args in case it is a JSON
     @field_validator("outputs", "inputs", mode="before")
     def validate_target_args(cls, v):
@@ -88,7 +80,7 @@ class TransactionModelResponse(BaseModel):
         return v
 
 
-class MessageModel(BaseModel):
+class MessageModel(DefaultModel):
     index: Optional[int] = Field(default=None)
     flow_id: Optional[str] = Field(default=None, alias="flow_id")
     timestamp: datetime = Field(default_factory=datetime.now)
@@ -98,10 +90,6 @@ class MessageModel(BaseModel):
     text: str
     files: list[str] = []
 
-    class Config:
-        from_attributes = True
-        populate_by_name = True
-
     @field_validator("files", mode="before")
     def validate_files(cls, v):
         if isinstance(v, str):
@@ -144,7 +132,7 @@ class MessageModelRequest(MessageModel):
     session_id: str = Field(default="")
 
 
-class VertexBuildModel(BaseModel):
+class VertexBuildModel(DefaultModel):
     index: Optional[int] = Field(default=None, alias="index", exclude=True)
     id: Optional[str] = Field(default=None, alias="id")
     flow_id: str
@@ -154,10 +142,6 @@ class VertexBuildModel(BaseModel):
     artifacts: dict
     timestamp: datetime = Field(default_factory=datetime.now)
 
-    class Config:
-        from_attributes = True
-        populate_by_name = True
-
     @field_serializer("data", "artifacts")
     def serialize_dict(v):
         if isinstance(v, dict):