From 11eeaa6bfc3d1118a35e1c826f67f28af303d904 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 21 Nov 2023 16:08:56 -0300 Subject: [PATCH] Update schemas.py with CustomComponentCode field --- src/backend/langflow/api/v1/schemas.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/api/v1/schemas.py b/src/backend/langflow/api/v1/schemas.py index 99a8b08cb..9c7a40983 100644 --- a/src/backend/langflow/api/v1/schemas.py +++ b/src/backend/langflow/api/v1/schemas.py @@ -2,13 +2,14 @@ from enum import Enum from pathlib import Path from typing import Any, Dict, List, Optional, Union from uuid import UUID -from langflow.services.database.models.api_key.api_key import ApiKeyRead -from langflow.services.database.models.flow import FlowCreate, FlowRead -from langflow.services.database.models.user import UserRead -from langflow.services.database.models.base import orjson_dumps from pydantic import BaseModel, Field, field_validator +from langflow.services.database.models.api_key.api_key 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 + class BuildStatus(Enum): """Status of the build.""" @@ -156,6 +157,7 @@ class StreamData(BaseModel): class CustomComponentCode(BaseModel): code: str + field: Optional[str] = None class CustomComponentResponseError(BaseModel):