bug fix: logging test may fail if some config fails (#1695)

cheng-tan · web-flow · commit 5ca69da8e5b2 · 2024-02-15T18:51:38.000Z
diff --git 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):
