From ad3fd44a7f83e3c5e6b25024ce1b49f1a828defe Mon Sep 17 00:00:00 2001 From: Tomaz Bratanic Date: Tue, 7 May 2024 08:59:55 +0200 Subject: [PATCH] experimental: Fix llm graph transformer bug (#21362) --- .../langchain_experimental/graph_transformers/llm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/experimental/langchain_experimental/graph_transformers/llm.py b/libs/experimental/langchain_experimental/graph_transformers/llm.py index 5a752060b02..0c98517ea5c 100644 --- a/libs/experimental/langchain_experimental/graph_transformers/llm.py +++ b/libs/experimental/langchain_experimental/graph_transformers/llm.py @@ -531,8 +531,8 @@ class LLMGraphTransformer: source=source_node, target=target_node, type=rel["relation"] ) ) - # Create nodes list - nodes = [Node(id=el[0], type=el[1]) for el in list(nodes_set)] + # Create nodes list + nodes = [Node(id=el[0], type=el[1]) for el in list(nodes_set)] # Strict mode filtering if self.strict_mode and (self.allowed_nodes or self.allowed_relationships):