diff --git a/src/backend/langflow/interface/utils.py b/src/backend/langflow/interface/utils.py index 1a37e89b4..8d45aa1b1 100644 --- a/src/backend/langflow/interface/utils.py +++ b/src/backend/langflow/interface/utils.py @@ -63,11 +63,12 @@ def extract_input_variables_from_prompt(prompt: str) -> list[str]: def setup_llm_caching(): """Setup LLM caching.""" - import langchain - from langflow.settings import settings - from langflow.interface.importing.utils import import_class try: + import langchain + from langflow.settings import settings + from langflow.interface.importing.utils import import_class + cache_class = import_class(f"langchain.cache.{settings.cache}") logger.debug(f"Setting up LLM caching with {cache_class.__name__}")