diff --git a/src/backend/langflow/interface/types.py b/src/backend/langflow/interface/types.py index 58238958c..f331c3416 100644 --- a/src/backend/langflow/interface/types.py +++ b/src/backend/langflow/interface/types.py @@ -24,6 +24,9 @@ from fastapi import HTTPException import traceback # Used to get the base_classes list +from langchain.chains import ConversationChain # noqa: F401 +from langchain.llms.base import BaseLLM # noqa: F401 +from langchain.tools import Tool # noqa: F401 def get_type_list():