diff --git a/src/backend/langflow/interface/agents/base.py b/src/backend/langflow/interface/agents/base.py index 2aa9fc4c2..b272144bc 100644 --- a/src/backend/langflow/interface/agents/base.py +++ b/src/backend/langflow/interface/agents/base.py @@ -32,10 +32,7 @@ class AgentCreator(LangChainTypeCreator): def get_signature(self, name: str) -> Optional[Dict]: try: - if ( - name in get_custom_nodes(self.type_name).keys() - and name in settings.agents - ): + if name in get_custom_nodes(self.type_name).keys(): return get_custom_nodes(self.type_name)[name] elif name in self.from_method_nodes: return build_template_from_method( diff --git a/src/backend/langflow/interface/chains/base.py b/src/backend/langflow/interface/chains/base.py index 9459f49e5..ff7e1ee33 100644 --- a/src/backend/langflow/interface/chains/base.py +++ b/src/backend/langflow/interface/chains/base.py @@ -45,10 +45,7 @@ class ChainCreator(LangChainTypeCreator): def get_signature(self, name: str) -> Optional[Dict]: try: - if ( - name in get_custom_nodes(self.type_name).keys() - and name in settings.chains - ): + if name in get_custom_nodes(self.type_name).keys(): return get_custom_nodes(self.type_name)[name] elif name in self.from_method_nodes.keys(): return build_template_from_method( diff --git a/src/backend/langflow/interface/memories/base.py b/src/backend/langflow/interface/memories/base.py index 9d3ceb30f..0f97a02fe 100644 --- a/src/backend/langflow/interface/memories/base.py +++ b/src/backend/langflow/interface/memories/base.py @@ -32,10 +32,7 @@ class MemoryCreator(LangChainTypeCreator): def get_signature(self, name: str) -> Optional[Dict]: """Get the signature of a memory.""" try: - if ( - name in get_custom_nodes(self.type_name).keys() - and name in settings.memories - ): + if name in get_custom_nodes(self.type_name).keys(): return get_custom_nodes(self.type_name)[name] elif name in self.from_method_nodes: return build_template_from_method( diff --git a/src/backend/langflow/interface/prompts/base.py b/src/backend/langflow/interface/prompts/base.py index ef4f03991..39bd94c5b 100644 --- a/src/backend/langflow/interface/prompts/base.py +++ b/src/backend/langflow/interface/prompts/base.py @@ -40,10 +40,7 @@ class PromptCreator(LangChainTypeCreator): def get_signature(self, name: str) -> Optional[Dict]: try: - if ( - name in get_custom_nodes(self.type_name).keys() - and name in settings.prompts - ): + if name in get_custom_nodes(self.type_name).keys(): return get_custom_nodes(self.type_name)[name] return build_template_from_class(name, self.type_to_loader_dict) except ValueError as exc: