From 5170323eee8cd5079ca24237b48c977d4c8d8eac Mon Sep 17 00:00:00 2001 From: cheng-tan Date: Thu, 15 Feb 2024 13:51:38 -0500 Subject: [PATCH] bug fix: logging test may fail if some config fails (#1695) --- test/agentchat/test_agent_logging.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/agentchat/test_agent_logging.py b/test/agentchat/test_agent_logging.py index 1cd2629f3bf8..d85ff034dc9e 100644 --- a/test/agentchat/test_agent_logging.py +++ b/test/agentchat/test_agent_logging.py @@ -88,7 +88,7 @@ def test_two_agents_logging(db_connection): cur.execute(CHAT_COMPLETIONS_QUERY) rows = cur.fetchall() - assert len(rows) == 3 + assert len(rows) >= 3 # some config may fail session_id = rows[0]["session_id"] for idx, row in enumerate(rows):