Refactor frontend_node/base.py

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-24 12:45:41 -03:00
commit 280b72adab

View file

@ -1,15 +1,15 @@
from collections import defaultdict
import re
from collections import defaultdict
from typing import ClassVar, DefaultDict, Dict, List, Optional
from pydantic import BaseModel, Field
from langflow.template.frontend_node.formatter import field_formatters
from langflow.template.field.base import TemplateField
from langflow.template.frontend_node.constants import (
CLASSES_TO_REMOVE,
FORCE_SHOW_FIELDS,
)
from langflow.template.field.base import TemplateField
from langflow.template.frontend_node.formatter import field_formatters
from langflow.template.template.base import Template
from langflow.utils import constants
@ -51,7 +51,9 @@ class FrontendNode(BaseModel):
documentation: str = ""
custom_fields: Optional[DefaultDict[str, List[str]]] = defaultdict(list)
output_types: List[str] = []
full_path: Optional[str] = None
field_formatters: FieldFormatters = Field(default_factory=FieldFormatters)
beta: bool = False
error: Optional[str] = None