diff --git a/src/backend/langflow/components/chains/LLMChain.py b/src/backend/langflow/components/chains/LLMChain.py index b14eef302..ec88e128a 100644 --- a/src/backend/langflow/components/chains/LLMChain.py +++ b/src/backend/langflow/components/chains/LLMChain.py @@ -1,10 +1,12 @@ -from langflow import CustomComponent +from typing import Callable, Optional, Union + from langchain.chains import LLMChain -from typing import Optional, Union, Callable + +from langflow import CustomComponent from langflow.field_typing import ( - BasePromptTemplate, BaseLanguageModel, BaseMemory, + BasePromptTemplate, Chain, )