From d79c8283a679fb81e9bfb13c55ac8a1c31e66a30 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 22 Feb 2024 18:25:38 -0300 Subject: [PATCH] Add successors field to VertexBuildResponse --- src/backend/langflow/api/v1/schemas.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/api/v1/schemas.py b/src/backend/langflow/api/v1/schemas.py index c50787b97..8da328195 100644 --- a/src/backend/langflow/api/v1/schemas.py +++ b/src/backend/langflow/api/v1/schemas.py @@ -4,12 +4,13 @@ from pathlib import Path from typing import Any, Dict, List, Optional, Union from uuid import UUID +from pydantic import BaseModel, Field, field_serializer, field_validator + from langflow.api.utils import serialize_field from langflow.services.database.models.api_key.model import ApiKeyRead from langflow.services.database.models.base import orjson_dumps from langflow.services.database.models.flow import FlowCreate, FlowRead from langflow.services.database.models.user import UserRead -from pydantic import BaseModel, Field, field_serializer, field_validator class BuildStatus(Enum): @@ -235,6 +236,7 @@ class ResultData(BaseModel): class VertexBuildResponse(BaseModel): id: Optional[str] = None + successors: Optional[List[str]] = Field(default_factory=list) valid: bool params: Optional[str] """JSON string of the params."""