Check rel type before checking properties

langchain-ai · Oct 9, 2024 · 0bbb6b0 · 0bbb6b0
1 parent 6fdfdbe
commit 0bbb6b0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/libs/experimental/langchain_experimental/graph_transformers/llm.py b/libs/experimental/langchain_experimental/graph_transformers/llm.py
@@ -753,7 +753,8 @@ def process_response(
             for rel in parsed_json:
                 # Check if mandatory properties are there
                 if (
-                    not rel.get("head")
+                    not isinstance(rel, dict)
+                    or not rel.get("head")
                     or not rel.get("tail")
                     or not rel.get("relation")
                 ):