From 8fd231086e03af69f100c24e621b11ac6ddb35fc Mon Sep 17 00:00:00 2001 From: liugz18 <57401541+liugz18@users.noreply.github.com> Date: Wed, 5 Jun 2024 01:07:38 +0800 Subject: [PATCH] experimental[patch]: Fix graph_transformers llms #21482 (#22417) Fix AttributeError on calling LLMGraphTransformer.convert_to_graph_documents #21482 since raw_schema is always a str @baskaryan --- .../langchain_experimental/graph_transformers/llm.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/libs/experimental/langchain_experimental/graph_transformers/llm.py b/libs/experimental/langchain_experimental/graph_transformers/llm.py index 29af044237f..e29cfb01b02 100644 --- a/libs/experimental/langchain_experimental/graph_transformers/llm.py +++ b/libs/experimental/langchain_experimental/graph_transformers/llm.py @@ -609,7 +609,9 @@ class LLMGraphTransformer: else: nodes_set = set() relationships = [] - parsed_json = self.json_repair.loads(raw_schema.content) + if not isinstance(raw_schema, str): + raw_schema = raw_schema.content + parsed_json = self.json_repair.loads(raw_schema) for rel in parsed_json: # Nodes need to be deduplicated using a set nodes_set.add((rel["head"], rel["head_type"]))