github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-22 14:49:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add check for prompt based approach in llm graph transformer (#26519)

Browse Source
This commit is contained in:
Tomaz Bratanic 2024-09-17 07:01:09 +09:00 committed by GitHub
parent 0bd98c99b3
commit 3bcd641bc1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
libs/experimental/langchain_experimental/graph_transformers/llm.py
View File

@ -749,12 +749,20 @@ class LLMGraphTransformer:
if isinstance(parsed_json, dict): if isinstance(parsed_json, dict):
parsed_json = [parsed_json] parsed_json = [parsed_json]
for rel in parsed_json: for rel in parsed_json:
# Check if mandatory properties are there
if (
not rel.get("head")
or not rel.get("tail")
or not rel.get("relation")
):
continue
# Nodes need to be deduplicated using a set # Nodes need to be deduplicated using a set
nodes_set.add((rel["head"], rel["head_type"])) # Use default Node label for nodes if missing
nodes_set.add((rel["tail"], rel["tail_type"])) nodes_set.add((rel["head"], rel.get("head_type", "Node")))
nodes_set.add((rel["tail"], rel.get("tail_type", "Node")))
source_node = Node(id=rel["head"], type=rel["head_type"]) source_node = Node(id=rel["head"], type=rel.get("head_type", "Node"))
target_node = Node(id=rel["tail"], type=rel["tail_type"]) target_node = Node(id=rel["tail"], type=rel.get("tail_type", "Node"))
relationships.append( relationships.append(
Relationship( Relationship(
source=source_node, target=target_node, type=rel["relation"] source=source_node, target=target_node, type=rel["relation"]