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

whiskyboy · Feb 15, 2024 · 5170323 · 5170323
1 parent 52d5641
commit 5170323
Showing 1 changed file with 1 addition and 1 deletion.
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):