From 37ede63d2db98fd22dc1c2148ccb17db9341e827 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 28 Mar 2024 17:01:13 -0300 Subject: [PATCH] Add import statement for get_settings_service() in base.py --- src/backend/base/langflow/interface/llms/base.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/interface/llms/base.py b/src/backend/base/langflow/interface/llms/base.py index 1eacade6b..ba611da8d 100644 --- a/src/backend/base/langflow/interface/llms/base.py +++ b/src/backend/base/langflow/interface/llms/base.py @@ -4,6 +4,7 @@ from loguru import logger from langflow.interface.base import LangChainTypeCreator from langflow.interface.custom_lists import llm_type_to_cls_dict +from langflow.services.deps import get_settings_service from langflow.template.frontend_node.llms import LLMFrontendNode from langflow.utils.util import build_template_from_class @@ -33,10 +34,11 @@ class LLMCreator(LangChainTypeCreator): return None def to_list(self) -> List[str]: + settings_service = get_settings_service() return [ llm.__name__ for llm in self.type_to_loader_dict.values() - # if llm.__name__ in settings_service.settings.LLMS or settings_service.settings.DEV + if llm.__name__ in settings_service.settings.LLMS or settings_service.settings.DEV ]