diff --git a/libs/neo4j/tests/unit_tests/chains/test_graph_qa.py b/libs/neo4j/tests/unit_tests/chains/test_graph_qa.py index 8d538a8..4baa34d 100644 --- a/libs/neo4j/tests/unit_tests/chains/test_graph_qa.py +++ b/libs/neo4j/tests/unit_tests/chains/test_graph_qa.py @@ -76,8 +76,15 @@ def test_graph_cypher_qa_chain_prompt_selection_1() -> None: cypher_prompt=cypher_prompt, allow_dangerous_requests=True, ) +<<<<<<< HEAD assert chain.qa_chain.prompt == qa_prompt # type: ignore[union-attr] assert chain.cypher_generation_chain.prompt == cypher_prompt +======= + assert hasattr(chain.qa_chain, "first") + assert chain.qa_chain.first == qa_prompt + assert hasattr(chain.cypher_generation_chain, "first") + assert chain.cypher_generation_chain.first == cypher_prompt +>>>>>>> 63f27b1 (Added hasattr assertions to tests/unit_tests/chains/test_graph_qa.py) def test_graph_cypher_qa_chain_prompt_selection_2() -> None: