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

refactor to use DefaultModel as base class

Browse source
This commit is contained in:
italojohnny 2024-06-10 16:06:40 -03:00
commit 118da0d482
1 changed files with 3 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/backend/base/langflow/services/monitor/schema.py
View file

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