From 826a6d07e0401754a6f53a1647426614f64d574b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 18 Dec 2023 10:50:43 -0300 Subject: [PATCH] Refactor ProcessResponse schema in schemas.py --- src/backend/langflow/api/v1/schemas.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/api/v1/schemas.py b/src/backend/langflow/api/v1/schemas.py index 8c8c4efd7..252df4669 100644 --- a/src/backend/langflow/api/v1/schemas.py +++ b/src/backend/langflow/api/v1/schemas.py @@ -3,12 +3,11 @@ from pathlib import Path from typing import Any, Dict, List, Optional, Union from uuid import UUID -from pydantic import BaseModel, Field, field_validator - 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_validator class BuildStatus(Enum): @@ -59,6 +58,7 @@ class ProcessResponse(BaseModel): """Process response schema.""" result: Any + status: Optional[str] = None task: Optional[TaskResponse] = None session_id: Optional[str] = None backend: Optional[str] = None