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

Refactor validation methods in schema.py for better readability and maintainability

Browse source
This commit is contained in:
ogabrielluiz 2024-05-24 09:15:40 -03:00
commit 90034c1c61
1 changed files with 4 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -81,9 +81,8 @@ class MessageModel(BaseModel):
from_attributes = True from_attributes = True
populate_by_name = True populate_by_name = True
@validator("files", pre=True) @field_validator("files", mode="before")
@field_validator("artifacts", mode="before") def validate_files(cls, v):
def validate_target_args(cls, v):
if isinstance(v, str): if isinstance(v, str):
return json.loads(v) return json.loads(v)
return v return v
@ -108,12 +107,6 @@ class MessageModel(BaseModel):
class MessageModelResponse(MessageModel): class MessageModelResponse(MessageModel):
index: Optional[int] = Field(default=None) index: Optional[int] = Field(default=None)
@field_validator("artifacts", mode="before")
def serialize_artifacts(v):
if isinstance(v, str):
return json.loads(v)
return v
@field_validator("index", mode="before") @field_validator("index", mode="before")
def validate_id(cls, v): def validate_id(cls, v):
if isinstance(v, float): if isinstance(v, float):
@ -151,9 +144,8 @@ class VertexBuildModel(BaseModel):
return v.model_dump_json() return v.model_dump_json()
return v return v
@validator("logs", pre=True) @field_validator("logs", mode="before")
@field_validator("params", mode="before") def validate_logs(cls, v):
def validate_params(cls, v):
if isinstance(v, str): if isinstance(v, str):
try: try:
return json.loads(v) return json.loads(v)
@ -173,14 +165,6 @@ class VertexBuildModel(BaseModel):
return json.loads(v) return json.loads(v)
return v return v
@field_validator("artifacts", mode="before")
def validate_artifacts(cls, v):
if isinstance(v, str):
return json.loads(v)
elif isinstance(v, BaseModel):
return v.model_dump()
return v
class VertexBuildResponseModel(VertexBuildModel): class VertexBuildResponseModel(VertexBuildModel):
messages: list[MessageModel] = [] messages: list[MessageModel] = []