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

🐛 fix(utils.py): rename cache_type variable to langchain_cache_type for clarity

Browse source 
🐛 fix(manager.py): add test connections to cache and database managers for initialization
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-13 23:40:12 -03:00
commit f8f99c6881
2 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -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()

5
src/backend/langflow/services/manager.py
View file

@ -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)