Refactor ProcessResponse schema in schemas.py

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-18 10:50:43 -03:00
commit 826a6d07e0

View file

@ -3,12 +3,11 @@ 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_validator
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_validator
class BuildStatus(Enum): class BuildStatus(Enum):
@ -59,6 +58,7 @@ class ProcessResponse(BaseModel):
"""Process response schema.""" """Process response schema."""
result: Any result: Any
status: Optional[str] = None
task: Optional[TaskResponse] = None task: Optional[TaskResponse] = None
session_id: Optional[str] = None session_id: Optional[str] = None
backend: Optional[str] = None backend: Optional[str] = None