From f8f99c68815336f89434d92a95f99a1e410622f7 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 13 Aug 2023 23:40:12 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(utils.py):=20rename=20cache?= =?UTF-8?q?=5Ftype=20variable=20to=20langchain=5Fcache=5Ftype=20for=20clar?= =?UTF-8?q?ity=20=F0=9F=90=9B=20fix(manager.py):=20add=20test=20connection?= =?UTF-8?q?s=20to=20cache=20and=20database=20managers=20for=20initializati?= =?UTF-8?q?on?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/utils.py | 6 ++++-- src/backend/langflow/services/manager.py | 5 +++++ 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/interface/utils.py b/src/backend/langflow/interface/utils.py index 1fddbf80f..822e32c35 100644 --- a/src/backend/langflow/interface/utils.py +++ b/src/backend/langflow/interface/utils.py @@ -77,8 +77,10 @@ def set_langchain_cache(settings): import langchain from langflow.interface.importing.utils import import_class - cache_type = os.getenv("LANGFLOW_LANGCHAIN_CACHE") - cache_class = import_class(f"langchain.cache.{cache_type or settings.CACHE}") + langchain_cache_type = os.getenv("LANGFLOW_LANGCHAIN_CACHE") + cache_class = import_class( + f"langchain.cache.{langchain_cache_type or settings.CACHE}" + ) logger.debug(f"Setting up LLM caching with {cache_class.__name__}") langchain.llm_cache = cache_class() diff --git a/src/backend/langflow/services/manager.py b/src/backend/langflow/services/manager.py index 1606b3a82..1f7e4b2bd 100644 --- a/src/backend/langflow/services/manager.py +++ b/src/backend/langflow/services/manager.py @@ -85,3 +85,8 @@ def initialize_services(): service_manager.register_factory(database_factory.DatabaseManagerFactory()) service_manager.register_factory(cache_factory.CacheManagerFactory()) service_manager.register_factory(chat_factory.ChatManagerFactory()) + + # Test cache connection + service_manager.get(ServiceType.CACHE_MANAGER) + # Test database connection + service_manager.get(ServiceType.DATABASE_MANAGER)