jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Remove base class limitations

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-22 17:24:59 -03:00
commit 71de618519
2 changed files with 2 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/template/frontend_node/base.py
View file

@ -5,10 +5,7 @@ from typing import ClassVar, Dict, List, Optional, Union
from pydantic import BaseModel, Field, field_serializer, model_serializer from pydantic import BaseModel, Field, field_serializer, model_serializer
from langflow.template.field.base import TemplateField from langflow.template.field.base import TemplateField
from langflow.template.frontend_node.constants import ( from langflow.template.frontend_node.constants import FORCE_SHOW_FIELDS
CLASSES_TO_REMOVE,
FORCE_SHOW_FIELDS,
)
from langflow.template.frontend_node.formatter import field_formatters 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
@ -88,7 +85,7 @@ class FrontendNode(BaseModel):
def process_base_classes(self, base_classes: List[str]) -> List[str]: def process_base_classes(self, base_classes: List[str]) -> List[str]:
"""Removes unwanted base classes from the list of base classes.""" """Removes unwanted base classes from the list of base classes."""
return [base_class for base_class in base_classes if base_class not in CLASSES_TO_REMOVE] return list(set(base_classes))
@field_serializer("display_name") @field_serializer("display_name")
def process_display_name(self, display_name: str) -> str: def process_display_name(self, display_name: str) -> str:

10
src/backend/langflow/template/frontend_node/constants.py
View file

@ -63,13 +63,3 @@ You can change this to use other APIs like JinaChat, LocalAI and Prem.
INPUT_KEY_INFO = """The variable to be used as Chat Input when more than one variable is available.""" INPUT_KEY_INFO = """The variable to be used as Chat Input when more than one variable is available."""
OUTPUT_KEY_INFO = """The variable to be used as Chat Output (e.g. answer in a ConversationalRetrievalChain)""" OUTPUT_KEY_INFO = """The variable to be used as Chat Output (e.g. answer in a ConversationalRetrievalChain)"""
CLASSES_TO_REMOVE = [
"RunnableSerializable",
"Serializable",
"BaseModel",
"object",
"Runnable",
"Generic",
]