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