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
This commit is contained in:
liugz18 2024-06-05 01:07:38 +08:00 committed by GitHub
parent 6db25b4e31
commit 8fd231086e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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