100% coverage for GraphCypherQAChain

libs/neo4j/tests/integration_tests/chains/test_graph_database.py

@@ -77,6 +77,7 @@ def test_cypher_generating_run() -> None:
     chain = GraphCypherQAChain.from_llm(
         llm=llm,
         graph=graph,
+        validate_cypher=True,
         allow_dangerous_requests=True,
     )
     output = chain.run("Who starred in Pulp Fiction?")

libs/neo4j/tests/unit_tests/chains/test_graph_qa.py

@@ -304,6 +304,19 @@ def test_graph_cypher_qa_chain() -> None:
     assert True
 
 
+def test_cypher_generation_failure() -> None:
+    """Test the chain doesn't fail if the Cypher query fails to be generated."""
+    llm = FakeLLM(queries={"query": ""}, sequential_responses=True)
+    chain = GraphCypherQAChain.from_llm(
+        llm=llm,
+        graph=FakeGraphStore(),
+        allow_dangerous_requests=True,
+        return_direct=True,
+    )
+    response = chain.run("Test question")
+    assert response == []
+
+
 def test_no_backticks() -> None:
     """Test if there are no backticks, so the original text should be returned."""
     query = "MATCH (n) RETURN n"