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"]))