Update CustomComponentCode schema

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-28 23:00:29 -03:00
commit 9c09c724a9

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):
@ -158,6 +157,7 @@ class StreamData(BaseModel):
class CustomComponentCode(BaseModel): class CustomComponentCode(BaseModel):
code: str code: str
field: Optional[str] = None field: Optional[str] = None
template: Optional[dict] = None
class CustomComponentResponseError(BaseModel): class CustomComponentResponseError(BaseModel):