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):