diff --git a/src/backend/langflow/interface/chains/base.py b/src/backend/langflow/interface/chains/base.py index 63840cf13..ff7e1ee33 100644 --- a/src/backend/langflow/interface/chains/base.py +++ b/src/backend/langflow/interface/chains/base.py @@ -20,7 +20,10 @@ class ChainCreator(LangChainTypeCreator): return ChainFrontendNode #! We need to find a better solution for this - from_method_nodes = {"ConversationalRetrievalChain": "from_llm"} + from_method_nodes = { + "ConversationalRetrievalChain": "from_llm", + "LLMCheckerChain": "from_llm", + } @property def type_to_loader_dict(self) -> Dict: diff --git a/src/backend/langflow/template/frontend_node/chains.py b/src/backend/langflow/template/frontend_node/chains.py index 6b36c28e7..1a1753df3 100644 --- a/src/backend/langflow/template/frontend_node/chains.py +++ b/src/backend/langflow/template/frontend_node/chains.py @@ -33,6 +33,14 @@ class ChainFrontendNode(FrontendNode): field.show = True field.advanced = True + # We should think of a way to deal with this later + # if field.field_type == "PromptTemplate": + # field.field_type = "str" + # field.multiline = True + # field.show = True + # field.advanced = False + # field.value = field.value.template + # Separated for possible future changes if field.name == "prompt" and field.value is None: field.required = True