From a0f48b1d123213cb69a4644d1512d1d6a00efbb1 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 23 Jun 2023 10:44:23 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(test=5Fgraph.py):=20add=20as?= =?UTF-8?q?sertion=20to=20check=20if=20root=20node=20is=20not=20None=20The?= =?UTF-8?q?=20test=5Fbuild=5Fparams=20function=20was=20failing=20when=20th?= =?UTF-8?q?e=20root=20node=20was=20None.=20This=20commit=20adds=20an=20ass?= =?UTF-8?q?ertion=20to=20check=20if=20the=20root=20node=20is=20not=20None?= =?UTF-8?q?=20before=20proceeding=20with=20the=20other=20assertions.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_graph.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_graph.py b/tests/test_graph.py index 02bb180f2..21febb435 100644 --- a/tests/test_graph.py +++ b/tests/test_graph.py @@ -238,6 +238,7 @@ def test_build_params(basic_graph): root = get_root_node(basic_graph) # Root node is a TimeTravelGuideChain # which requires an llm and memory + assert root is not None assert isinstance(root.params, dict) assert "llm" in root.params assert "memory" in root.params