Update schemas.py with CustomComponentCode field

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-21 16:08:56 -03:00
commit 11eeaa6bfc

View file

@ -2,13 +2,14 @@ from enum import Enum
from pathlib import Path 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 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 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): class BuildStatus(Enum):
"""Status of the build.""" """Status of the build."""
@ -156,6 +157,7 @@ class StreamData(BaseModel):
class CustomComponentCode(BaseModel): class CustomComponentCode(BaseModel):
code: str code: str
field: Optional[str] = None
class CustomComponentResponseError(BaseModel): class CustomComponentResponseError(BaseModel):