Check rel type before checking properties (#9)

Minor change, but lazy evaluation of the type before checking whether the properties are there saves me a lot of unnecessary exceptions.
langchain-ai · Oct 22, 2024 · ebd1933 · ebd1933
1 parent 6fdfdbe
commit ebd1933
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")
                 ):