github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-18 18:53:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

experimental[patch]: Fix graph_transformers llms #21482 (#22417)

Browse Source 
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
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/experimental/langchain_experimental/graph_transformers/llm.py
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"]))