Add successors field to VertexBuildResponse

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-22 18:25:38 -03:00
commit d79c8283a6

View file

@ -4,12 +4,13 @@ from pathlib import Path
from typing import Any, Dict, List, Optional, Union from typing import Any, Dict, List, Optional, Union
from uuid import UUID from uuid import UUID
from pydantic import BaseModel, Field, field_serializer, field_validator
from langflow.api.utils import serialize_field from langflow.api.utils import serialize_field
from langflow.services.database.models.api_key.model import ApiKeyRead from langflow.services.database.models.api_key.model import ApiKeyRead
from langflow.services.database.models.base import orjson_dumps from langflow.services.database.models.base import orjson_dumps
from langflow.services.database.models.flow import FlowCreate, FlowRead from langflow.services.database.models.flow import FlowCreate, FlowRead
from langflow.services.database.models.user import UserRead from langflow.services.database.models.user import UserRead
from pydantic import BaseModel, Field, field_serializer, field_validator
class BuildStatus(Enum): class BuildStatus(Enum):
@ -235,6 +236,7 @@ class ResultData(BaseModel):
class VertexBuildResponse(BaseModel): class VertexBuildResponse(BaseModel):
id: Optional[str] = None id: Optional[str] = None
successors: Optional[List[str]] = Field(default_factory=list)
valid: bool valid: bool
params: Optional[str] params: Optional[str]
"""JSON string of the params.""" """JSON string of the params."""