jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🔧 chore(utils.py): refactor import statement to improve code readability and maintainability

Browse source 
🔧 chore(utils.py): refactor assignment of langchain.llm_cache to use a setter function for better encapsulation
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-11 22:12:59 -03:00
commit 9c58bfc584
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/interface/utils.py
View file

@ -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}. ")