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