From 9c58bfc5844bfc7f4c806ab566fe1b3a1bd9156b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 11 Dec 2023 22:12:59 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(utils.py):=20refactor=20im?= =?UTF-8?q?port=20statement=20to=20improve=20code=20readability=20and=20ma?= =?UTF-8?q?intainability=20=F0=9F=94=A7=20chore(utils.py):=20refactor=20as?= =?UTF-8?q?signment=20of=20langchain.llm=5Fcache=20to=20use=20a=20setter?= =?UTF-8?q?=20function=20for=20better=20encapsulation?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/interface/utils.py b/src/backend/langflow/interface/utils.py index 6f7ec9329..ef29911f5 100644 --- a/src/backend/langflow/interface/utils.py +++ b/src/backend/langflow/interface/utils.py @@ -72,7 +72,7 @@ def setup_llm_caching(): def set_langchain_cache(settings): - import langchain + from langchain.globals import set_llm_cache from langflow.interface.importing.utils import import_class if cache_type := os.getenv("LANGFLOW_LANGCHAIN_CACHE"): @@ -80,7 +80,7 @@ def set_langchain_cache(settings): cache_class = import_class(f"langchain.cache.{cache_type or settings.LANGCHAIN_CACHE}") logger.debug(f"Setting up LLM caching with {cache_class.__name__}") - langchain.llm_cache = cache_class() + set_llm_cache(cache_class()) logger.info(f"LLM caching setup with {cache_class.__name__}") except ImportError: logger.warning(f"Could not import {cache_type}. ")